Merge 8f8630540a into d0fb0f3ac1

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
@@ -50,6 +51,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
@@ -107,6 +110,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"
@@ -155,6 +160,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
@@ -180,6 +187,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
@@ -250,6 +259,8 @@ class Vmfloaty
         delete_all = options.all
         force = options.f
 
+        Otel.configure_tracing(service)
+
         failures = []
         successes = []
 
@@ -338,6 +349,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
@@ -366,6 +379,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
@@ -395,6 +410,9 @@ class Vmfloaty
           FloatyLogger.setlevel = options.loglevel
         end
         service = Service.new(options, config)
+
+        Otel.configure_tracing(service)
+
         if options.json
           pp service.status(verbose)
         else
@@ -415,6 +433,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
@@ -436,6 +456,8 @@ class Vmfloaty
         service = Service.new(options, config)
         action = args.first
 
+        Otel.configure_tracing(service)
+
         begin
           case action
           when 'get'
@@ -480,6 +502,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