Inital work on detailed statuses for runtimes

Author: Knerio

Gemfile

@@ -79,7 +79,7 @@ gem 'good_job', '~> 4.0'
 gem 'rotp'
 
 gem 'grpc', '~> 1.67'
-gem 'tucana', '0.0.47'
+gem 'tucana', '0.0.0', path: '../tucana/build/ruby'
 
 gem 'code0-identities', '~> 0.0.2'
 

Gemfile.lock

@@ -1,3 +1,9 @@
+PATH
+  remote: ../tucana/build/ruby
+  specs:
+    tucana (0.0.0)
+      grpc (~> 1.64)
+
 GEM
   remote: https://rubygems.org/
   specs:
@@ -376,8 +382,6 @@ GEM
     thor (1.4.0)
     timeout (0.4.4)
     tsort (0.2.0)
-    tucana (0.0.47)
-      grpc (~> 1.64)
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
     unicode-display_width (2.6.0)
@@ -430,7 +434,7 @@ DEPENDENCIES
   simplecov (~> 0.22.0)
   simplecov-cobertura (~> 3.0)
   test-prof (~> 1.0)
-  tucana (= 0.0.47)
+  tucana (= 0.0.0)!
   tzinfo-data
 
 RUBY VERSION

app/graphql/types/runtime_status_configuration_endpoint_type.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Types
+  class RuntimeStatusConfigurationEndpointType < Types::BaseObject
+    description 'Detailed information about a runtime status'
+
+    field :endpoint, String, null: false, description: 'The endpoint URL of the runtime'
+
+    id_field ::RuntimeStatusConfiguration
+    timestamps
+  end
+end

app/graphql/types/runtime_status_configuration_type.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Types
+  class RuntimeStatusConfigurationType < Types::BaseUnion
+    description 'Detailed configuration about a runtime status, either: endpoint, ...'
+
+    possible_types Types::RuntimeStatusConfigurationEndpointType
+
+    def self.resolve_type(object, _context)
+      return Types::RuntimeStatusConfigurationEndpointType if object.endpoint.present?
+
+      raise "Unknown RuntimeStatusInformation type: #{object.class.name}"
+    end
+  end
+end

app/graphql/types/runtime_status_enum.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module Types
+  class RuntimeStatusEnum < BaseEnum
+    description 'Represent all available aquila statuses'
+
+    value :CONNECTED, 'No problem with the connection to aquila', value: :connected
+    value :DISCONNECTED, 'The runtime is disconnected, cause unknown', value: :disconnected
+  end
+end

app/graphql/types/runtime_status_status_enum.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Types
+  class RuntimeStatusStatusEnum < Types::BaseEnum
+    description 'The enum status of the detailed status'
+
+    value :NOT_RESPONDING, 'The runtime is not responding to heartbeats', value: 'NOT_RESPONDING'
+    value :NOT_READY, 'The runtime is not ready to compute stuff', value: 'NOT_READY'
+    value :RUNNING, 'The runtime is running and healthy', value: 'RUNNING'
+    value :STOPPED, 'The runtime has been stopped', value: 'STOPPED'
+  end
+end

app/graphql/types/runtime_status_type.rb

@@ -1,10 +1,25 @@
 # frozen_string_literal: true
 
 module Types
-  class RuntimeStatusType < BaseEnum
-    description 'Represent all available types of statuses of a runtime'
+  class RuntimeStatusType < Types::BaseObject
+    description 'A runtime status information entry'
 
-    value :CONNECTED, 'No problem with connection, everything works as expected', value: 'connected'
-    value :DISCONNECTED, 'The runtime is disconnected, cause unknown', value: 'disconnected'
+    field :configurations, Types::RuntimeStatusConfigurationType.connection_type,
+          null: false,
+          description: 'The detailed configuration entries for this runtime status (only for adapters)',
+          method: :runtime_status_configurations
+    field :feature_set, [String], null: false, description: 'The set of features supported by the runtime'
+    field :identifier, String, null: false, description: 'The unique identifier for this runtime status'
+    field :last_heartbeat, Types::TimeType, null: true,
+                                            description: 'The timestamp of the last heartbeat received from the runtime'
+    field :status, Types::RuntimeStatusStatusEnum,
+          null: false,
+          description: 'The current status of the runtime (e.g. running, stopped)'
+    field :type, Types::RuntimeStatusTypeEnum,
+          null: false,
+          description: 'The type of runtime status information (e.g. adapter, execution)', method: :status_type
+
+    id_field RuntimeStatus
+    timestamps
   end
 end

app/graphql/types/runtime_status_type_enum.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module Types
+  class RuntimeStatusTypeEnum < Types::BaseEnum
+    description 'The type of runtime status'
+
+    value :ADAPTER, 'Indicates that the runtime status is related to an adapter.', value: 'adapter'
+    value :EXECUTION, 'Indicates that the runtime status is related to an execution.', value: 'execution'
+  end
+end

app/graphql/types/runtime_type.rb

@@ -16,10 +16,16 @@ class RuntimeType < Types::BaseObject
     field :runtime_function_definitions, Types::RuntimeFunctionDefinitionType.connection_type,
           null: false,
           description: 'Functions of the runtime'
-    field :status, Types::RuntimeStatusType, null: false, description: 'The status of the runtime'
 
     field :token, String, null: true, description: 'Token belonging to the runtime, only present on creation'
 
+    field :status, Types::RuntimeStatusStatusEnum,
+          null: false,
+          description: 'Wheater the last heartbeat was recent enough to consider the runtime as connected'
+    field :statuses, Types::RuntimeStatusType.connection_type, null: false,
+                                                               description: 'Statuses of the runtime',
+                                                               method: :runtime_statuses
+
     expose_abilities %i[
       delete_runtime
       update_runtime
@@ -29,6 +35,17 @@ class RuntimeType < Types::BaseObject
     id_field Runtime
     timestamps
 
+    # If the last heartbeat was within the last 10 minutes, consider the runtime as 'running'
+    def status
+      last_heartbeat = object.last_heartbeat
+
+      if last_heartbeat && last_heartbeat >= 10.minutes.ago
+        :connected
+      else
+        :disconnected
+      end
+    end
+
     def token
       object.token if object.token_previously_changed?
     end

app/grpc/flow_handler.rb

@@ -25,22 +25,12 @@ def self.update_runtime(runtime)
     )
   end
 
-  def self.update_started(runtime_id)
-    runtime = Runtime.find(runtime_id)
-    runtime.connected!
-    runtime.save
-
+  def self.update_started(_runtime_id)
     logger.info(message: 'Runtime connected', runtime_id: runtime.id)
 
     update_runtime(runtime)
   end
 
-  def self.update_died(runtime_id)
-    runtime = Runtime.find(runtime_id)
-    runtime.disconnected!
-    runtime.save
-  end
-
   def self.encoders = { update: ->(grpc_object) { Tucana::Sagittarius::FlowResponse.encode(grpc_object) } }
 
   def self.decoders = { update: ->(string) { Tucana::Sagittarius::FlowResponse.decode(string) } }