Add OpenTelemetry client instrumentation

lib/vmfloaty.rb

@@ -14,6 +14,7 @@ require 'vmfloaty/utils'
 require 'vmfloaty/service'
 require 'vmfloaty/ssh'
 require 'vmfloaty/logger'
+require 'vmfloaty/otel'
 
 class Vmfloaty
   include Commander::Methods
@@ -45,6 +46,8 @@ class Vmfloaty
         use_token = !options.notoken
         force = options.force
 
+        Otel.configure_tracing(service)
+
         if args.empty?
           FloatyLogger.error 'No operating systems provided to obtain. See `floaty get --help` for more information on how to get VMs.'
           exit 1
@@ -98,6 +101,8 @@ class Vmfloaty
         service = Service.new(options, config)
         filter = args[0]
 
+        Otel.configure_tracing(service)
+
         if options.active
           # list active vms
           if service.type == "ABS"
@@ -146,6 +151,8 @@ class Vmfloaty
         service = Service.new(options, config)
         hostname = args[0]
 
+        Otel.configure_tracing(service)
+
         query_req = service.query(verbose, hostname)
         pp query_req
       end
@@ -171,6 +178,8 @@ class Vmfloaty
         hostname = args[0]
         modify_all = options.all
 
+        Otel.configure_tracing(service)
+
         if hostname.nil? && !modify_all
           FloatyLogger.error 'ERROR: Provide a hostname or specify --all.'
           exit 1
@@ -236,6 +245,8 @@ class Vmfloaty
         delete_all = options.all
         force = options.f
 
+        Otel.configure_tracing(service)
+
         failures = []
         successes = []
 
@@ -324,6 +335,8 @@ class Vmfloaty
         service = Service.new(options, config)
         hostname = args[0]
 
+        Otel.configure_tracing(service)
+
         begin
           snapshot_req = service.snapshot(verbose, hostname)
         rescue TokenError, ModifyError => e
@@ -352,6 +365,8 @@ class Vmfloaty
         hostname = args[0]
         snapshot_sha = args[1] || options.snapshot
 
+        Otel.configure_tracing(service)
+
         FloatyLogger.info "Two snapshot arguments were given....using snapshot #{snapshot_sha}" if args[1] && options.snapshot
 
         begin
@@ -377,6 +392,9 @@ class Vmfloaty
       c.action do |_, options|
         verbose = options.verbose || config['verbose']
         service = Service.new(options, config)
+
+        Otel.configure_tracing(service)
+
         if options.json
           pp service.status(verbose)
         else
@@ -397,6 +415,8 @@ class Vmfloaty
         verbose = options.verbose || config['verbose']
         service = Service.new(options, config)
 
+        Otel.configure_tracing(service)
+
         summary = service.summary(verbose)
         pp summary
         exit 0
@@ -418,6 +438,8 @@ class Vmfloaty
         service = Service.new(options, config)
         action = args.first
 
+        Otel.configure_tracing(service)
+
         begin
           case action
           when 'get'
@@ -462,6 +484,8 @@ class Vmfloaty
         service = Service.new(options, config)
         use_token = !options.notoken
 
+        Otel.configure_tracing(service)
+
         if args.empty?
           FloatyLogger.error 'No operating systems provided to obtain. See `floaty ssh --help` for more information on how to get VMs.'
           exit 1

lib/vmfloaty/otel.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require 'opentelemetry-api'
+require 'opentelemetry-instrumentation-faraday'
+require 'opentelemetry-sdk'
+require 'opentelemetry/exporter/jaeger'
+require 'socket'
+require 'vmfloaty/version'
+
+class Otel
+  def configure_tracing(service)
+    span_processor = if service.config['jaeger_endpoint']
+                       OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor.new(
+                         exporter: OpenTelemetry::Exporter::Jaeger::CollectorExporter.new(endpoint: tracing_jaeger_host)
+                       )
+                     else
+                       OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor.new(
+                         exporter: OpenTelemetry::SDK::Trace::Export::NoopSpanExporter.new
+                       )
+                     end
+
+    OpenTelemetry::SDK.configure do |c|
+      c.add_span_processor(span_processor)
+      c.resource = OpenTelemetry::SDK::Resources::Resource.create(
+        {
+          'net.host.name' => Socket.gethostname,
+          'net.peer.name' => URI.parse(service.config['url']).host
+        }
+      )
+      c.service_name = 'vmfloaty'
+      c.service_version = Vmfloaty::VERSION
+      c.use 'OpenTelemetry::Instrumentation::Faraday'
+    end
+  end
+end

vmfloaty.gemspec

@@ -27,4 +27,8 @@ Gem::Specification.new do |s|
   s.add_dependency 'colorize', '~> 0.8.1'
   s.add_dependency 'commander', '>= 4.4.3', '< 4.6.0'
   s.add_dependency 'faraday', '~> 0.17.0'
+  s.add_dependency 'opentelemetry-api', '0.7.0'
+  s.add_dependency 'opentelemetry-exporter-jaeger', '~> 0.7.0'
+  s.add_dependency 'opentelemetry-instrumentation-faraday', '~> 0.7.0'
+  s.add_dependency 'opentelemetry-sdk', '0.7.0'
 end