persistence-providers 0.0.2.1 → 0.0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c76149a5275203cc75b1db6cc9d28abbc7c01358fe7e2bfb8fd450e7d59f170a
-  data.tar.gz: a58841046f2ef48192a3acc83d5aaadffb49f5e566e298593cf58efd6a3d03b8
+  metadata.gz: 8136cc320cf5d2ad265d4a3c966a7838f1c6a1d09d13a1e5aad7143fe510ea2a
+  data.tar.gz: b64999cabad79d62721921049e71f8b2e45b0b4c70256aeb8d496e157ae556bb
 SHA512:
-  metadata.gz: 36880ebea0df341c7ccfb68c969b0dea6c8b4afc8fa12476466d0e95e9087a3d6120a9d32704a9a247ec774fa27675344105d9015f863809d56c8bbd3d3e01ad
-  data.tar.gz: 3d676eed98ab33028a7d1ec7cc371474153c1dadeed6c406c5a8bd1a21816d7137f24cd097f1baa56e2d3ffa3f7b8ea7e695edec9896284a61bd5f052550485b
+  metadata.gz: c2fea0ace3f115d3c1168952f6664bf1f2618df80c4f7ce6b53dae51a41f81e91c3ae452f9d11afd050e9c5a219d3537d43cadb947155aa55e53a724e11fb5ad
+  data.tar.gz: 8e0c57d6e478f9db48e1e548a0ada0a9974e4cf68799d5adf55f00e8f4aeac9c47cd6c26252d030df9bba4c36895c188e098db5881282ae9311861ee1fad230a

data/lib/state/component/providers/influxdb.rb

@@ -6,9 +6,9 @@ module DTK::State
 
       attr_reader :client, :measurement
 
-      def initialize
+      def initialize(measurement_name)
         @client = Influxdb::Client.new('dtk')
-        @measurement = @client.measurement_helper(:attributes)
+        @measurement = @client.measurement_helper(measurement_name)
       end
 
       def get(namespace, crd_component_name, component_instance_name, attribute_name, opts = {})
@@ -25,6 +25,55 @@ module DTK::State
           puts error
         end
       end
+
+      def write_event(event_id, pod_name, pod_namespace, event_source, event_message, component_name, attribute_name, task_id, timestamp)
+        begin
+          fail "Bad timestamp input, write operation wont be completed" if timestamp > Time.new
+          value_to_write = { event_source: event_source, event_message: event_message }
+          required_tags = {
+            event_id: event_id,
+            pod_name: pod_name,
+            pod_namespace: pod_namespace,
+            component_name: component_name,
+            attribute_name: attribute_name,
+            task_id: task_id
+          }
+          measurement.write(value_to_write.to_s, required_tags, timestamp)
+        rescue => error
+          puts error
+        end
+      end
+
+      def get_event(event_id, pod_name, pod_namespace, component_name, attribute_name, task_id)
+        required_tags = {
+            event_id: event_id,
+            pod_name: pod_name,
+            pod_namespace: pod_namespace,
+            component_name: component_name,
+            attribute_name: attribute_name,
+            task_id: task_id
+          }
+        last_point = measurement.get_last_point(required_tags)
+      end
+
+      def write_state(type, name, namespace, object_state, spec, status, component_name, attribute_name, task_id, timestamp)
+        begin
+          fail "Bad timestamp input, write operation wont be completed" if timestamp > Time.new
+          value_to_write = { spec: spec, status: status }
+          required_tags = {
+            type: type,
+            name: name,
+            namespace: namespace,
+            object_state: object_state,
+            component_name: component_name,
+            attribute_name: attribute_name,
+            task_id: task_id
+          }
+          measurement.write(value_to_write.to_s, required_tags, timestamp)
+        rescue => error
+          puts error
+        end
+      end
       
       private 
 

data/lib/state/component/providers/influxdb/measurement.rb

@@ -2,6 +2,8 @@ module DTK::State
   class Component::Attribute::Influxdb
     class Measurement
       require_relative('measurement/attribute_measurement')
+      require_relative('measurement/events')
+      require_relative('measurement/states')
       
       def initialize(name, client)
         @name   = name

data/lib/state/component/providers/influxdb/measurement/attribute_measurement.rb

@@ -1,7 +1,6 @@
 module DTK::State
   class Component::Attribute::Influxdb
     class Measurement
-      # DTK Attributes measurement
       class Attributes < self
 
         def write(value, params_hash = {}, timestamp)

data/lib/state/component/providers/influxdb/measurement/events.rb

@@ -0,0 +1,20 @@
+module DTK::State
+  class Component::Attribute::Influxdb
+    class Measurement
+      class Events < self
+
+        def write(value, params_hash = {}, timestamp)
+          checked_params_hash = check_params_hash(params_hash)
+          write_point(value, checked_params_hash, timestamp)
+        end
+
+        protected
+
+        def required_params
+          [:event_id, :pod_name, :pod_namespace, :component_name, :attribute_name, :task_id]
+        end
+
+      end                       
+    end
+  end
+end

data/lib/state/component/providers/influxdb/measurement/states.rb

@@ -0,0 +1,20 @@
+module DTK::State
+  class Component::Attribute::Influxdb
+    class Measurement
+      class States < self
+
+        def write(value, params_hash = {}, timestamp)
+          checked_params_hash = check_params_hash(params_hash)
+          write_point(value, checked_params_hash, timestamp)
+        end
+
+        protected
+
+        def required_params
+          [:type, :name, :namespace, :object_state, :component_name, :attribute_name, :task_id]
+        end
+
+      end                       
+    end
+  end
+end

data/persistence-providers.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name        = 'persistence-providers'
-  spec.version     = '0.0.2.1'
+  spec.version     = '0.0.2.2'
   spec.author      = 'Reactor8'
   spec.email       = 'support@reactor8.com'
   spec.description = %q{Persistence providers plugin}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: persistence-providers
 version: !ruby/object:Gem::Version
-  version: 0.0.2.1
+  version: 0.0.2.2
 platform: ruby
 authors:
 - Reactor8
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-03-20 00:00:00.000000000 Z
+date: 2020-03-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kubeclient
@@ -67,6 +67,8 @@ files:
 - lib/state/component/providers/influxdb/client.rb
 - lib/state/component/providers/influxdb/measurement.rb
 - lib/state/component/providers/influxdb/measurement/attribute_measurement.rb
+- lib/state/component/providers/influxdb/measurement/events.rb
+- lib/state/component/providers/influxdb/measurement/states.rb
 - lib/state/component/providers/kube_crd.rb
 - lib/state/component_def.rb
 - lib/state/component_def/attribute_type_info.rb