grpc-orin3-provider 0.1.0 → 1.0.0

data/lib/grpc/client/orin3/remoteengine/orin3_remote_engine.rb

@@ -0,0 +1,158 @@
+require_relative '../../../../../gen/Design.ORiN3.Common/V1/orin3_common_remote_engine_pb'
+require_relative '../../../../../gen/Message.ORiN3.RemoteEngine/V1/orin3_remote_engine_pb'
+require_relative '../../../../../gen/Message.ORiN3.RemoteEngine/V1/orin3_remote_engine_services_pb'
+require_relative '../message_client_error'
+
+module O3
+  include Design::ORiN3::Common::RemoteEngine::V1::AutoGenerated
+  include Message::ORiN3::RemoteEngine::V1::AutoGenerated
+end
+
+module Grpc
+  module Client
+    module ORiN3
+      class RemoteEngineStatus
+        attr_reader :result_code, :status, :instance_id, :host, :addresses, :version, :os_platform, :os_description, :running_provider_count
+
+        def initialize(result_code, status, instance_id, host, addresses, version, os_platform, os_description, running_provider_count)
+          @result_code = result_code
+          @status = status
+          @instance_id = instance_id
+          @host = host
+          @addresses = addresses
+          @version = version
+          @os_platform = os_platform
+          @os_description = os_description
+          @running_provider_count = running_provider_count
+        end
+      end
+
+      class ProviderEndpoint
+        attr_accessor :index, :ip_address, :port, :uri, :protocol_type
+      
+        def initialize(index, ip_address, port, uri, protocol_type)
+          @index = index
+          @ip_address = ip_address
+          @port = port
+          @uri = uri
+          @protocol_type = protocol_type
+        end
+      end
+      
+      class ProviderInformation
+        attr_accessor :endpoints
+
+        def initialize(endpoints)
+          @endpoints = endpoints
+        end
+      end
+
+      class WakeupProviderResponse
+        attr_accessor :id, :provider_information
+
+        def initialize(id, provider_information)
+          @id = id
+          @provider_information = provider_information
+        end
+      end
+
+      class RemoteEngine
+        def initialize(channel)
+          @channel = channel
+        end
+
+        def get_status
+          begin
+            service = O3::RemoteEngineService::Stub.new(nil, :this_channel_is_insecure, channel_override: @channel)
+            request = O3::GetRemoteEngineStatusRequest.new(common: O3::CommonRequest.new)
+            response = service.get_remote_engine_status(request)
+            if (response.common.result_code != :SUCCEEDED)
+              raise MessageClientError.new(response.common.result_code, response.common.detail)
+            end
+            return RemoteEngineStatus.new(response.common.result_code, response.status,
+              response.instance_id, response.host, response.addresses, response.version,
+              response.os_platform, response.os_description, response.running_provider_count)
+          rescue MessageClientError
+            raise
+          rescue StandardError => e
+            raise MessageClientError.new(e)
+          end
+        end
+
+        def wakeup_provider(id, version, host, port)
+          begin
+            service = O3::RemoteEngineService::Stub.new(nil, :this_channel_is_insecure, channel_override: @channel)
+            endpoint_info = create_provider_core_startup_argument_endpoint_info(0, host, port)
+            telemetry_option = create_telemetry_option(true)
+            argument = create_provider_core_startup_argument(false, [endpoint_info], 0, telemetry_option)
+            request = O3::WakeupProviderRequest.new(
+              common: O3::CommonRequest.new(reserved: 0),
+              id: id, version: version, provider_startup_argument: argument)
+            response = service.wakeup_provider(request)
+            if (response.common.result_code != :SUCCEEDED)
+              raise MessageClientError.new(response.common.result_code, response.common.detail)
+            end
+
+            endpoints = response.provider_information.endpoints.map do |endpoint|
+              ProviderEndpoint.new(
+                endpoint.index,
+                endpoint.ip_address,
+                endpoint.port,
+                endpoint.uri,
+                endpoint.protocol_type
+              )
+            end
+            return WakeupProviderResponse.new(response.id, ProviderInformation.new(endpoints))
+          rescue MessageClientError
+            raise
+          rescue StandardError => e
+            raise MessageClientError.new(e)
+          end
+        end
+
+        def terminate_provider(id)
+          begin
+            service = O3::RemoteEngineService::Stub.new(nil, :this_channel_is_insecure, channel_override: @channel)
+            request = O3::TerminateProviderRequest.new(common: O3::CommonRequest.new, id: id)
+            response = service.terminate_provider(request)
+            if (response.common.result_code != :SUCCEEDED)
+              raise MessageClientError.new(response.common.result_code, response.common.detail)
+            end
+          rescue MessageClientError
+            raise
+          rescue StandardError => e
+            raise MessageClientError.new(e)
+          end
+        end
+
+        private
+        def create_provider_core_startup_argument(thread_safe_mode, endpoint_infos, log_level, telemetry_option)
+          O3::ProviderCoreStartupArgument.new(
+            thread_safe_mode: thread_safe_mode,
+            provider_core_startup_argument_endpoint_infos: endpoint_infos,
+            log_level: log_level,
+            telemetry_option: telemetry_option,
+            extension: {}
+          )
+        end
+      
+        def create_telemetry_option(use_remote_engine_telemetry_setting)
+          O3::TelemetryOption.new(
+            attributes: {},
+            use_remote_engine_telemetry_setting: use_remote_engine_telemetry_setting,
+            telemetry_endpoints: []
+          )
+        end
+
+        def create_provider_core_startup_argument_endpoint_info(protocol_type, host, port)
+          O3::ProviderCoreStartupArgumentEndpointInfo.new(
+            protocol_type: protocol_type,
+            host: host,
+            port: port,
+            reserved: "".b # 空のバイト列
+          )
+        end
+      end
+    end
+  end
+end

data/lib/grpc/orin3/provider/date_time_converter.rb

@@ -0,0 +1,25 @@
+require 'time'
+
+module Grpc
+  module ORiN3
+    module Provider
+      class DateTimeConverter
+        TICKS_PER_SECOND = 10_000_000
+        EPOCH = Time.utc(1, 1, 1, 0, 0, 0)
+
+        def self.to_int64(time)
+          utc_time = time.utc
+          ticks = ((utc_time - EPOCH) * TICKS_PER_SECOND).to_i
+          return ticks & 0x3FFFFFFFFFFFFFFF
+        end
+  
+        def self.from_int64(int64)
+          filter = 0x3FFFFFFFFFFFFFFF
+          datetime_num = int64 & filter
+          seconds_since_windows_epoch = datetime_num.to_f / TICKS_PER_SECOND
+          return EPOCH + seconds_since_windows_epoch
+        end
+      end
+    end
+  end
+end