fix: Format gat commands

open-telemetry · Dec 19, 2024 · 97264dd · 97264dd
1 parent b1ea804
commit 97264dd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/instrumentation/dalli/lib/opentelemetry/instrumentation/dalli/utils.rb b/instrumentation/dalli/lib/opentelemetry/instrumentation/dalli/utils.rb
@@ -42,6 +42,7 @@ module Utils
         'multi_replace' => 'replaceq',
         'multi_delete' => 'deleteq',
         'touch' => 'touch'
+        'gat' => 'gat'
         # 'sasl_authentication' => 'auth_negotiation',
         # 'sasl_authentication' => 'auth_request',
       }.freeze

diff --git a/instrumentation/dalli/test/opentelemetry/instrumentation/dalli/instrumentation_test.rb b/instrumentation/dalli/test/opentelemetry/instrumentation/dalli/instrumentation_test.rb
@@ -127,6 +127,14 @@
       _(span.name).must_equal 'set'
       _(span.attributes['db.statement']).must_equal 'set ?'
     end
+
+    it 'supports gat' do
+      dalli.gat('foo')
+
+      _(exporter.finished_spans.size).must_equal 1
+      _(span.name).must_equal 'gat'
+      _(span.attributes['db.statement']).must_equal 'gat foo 0'
+    end
   end
 
   # Dalli 3.x has different behavior than 2.x versions and attempts to retry on network errors