fix: Format gat commands

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

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/instrumentation/dalli/lib/opentelemetry/instrumentation/dalli/utils.rb b/instrumentation/dalli/lib/opentelemetry/instrumentation/dalli/utils.rb
@@ -41,7 +41,8 @@ module Utils
         'multi_add' => 'addq',
         'multi_replace' => 'replaceq',
         'multi_delete' => 'deleteq',
-        'touch' => 'touch'
+        '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