twiglet 3.10.0 → 3.11.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6362109470f346c6703ed9e13e82904195f201cac31e1edf1dad8714c4312d79
-  data.tar.gz: a06b359e837cf995f78efdab94c53e5017ccc4ca6d06373a05148f0e4b93621c
+  metadata.gz: 99e8d740d7d9a800eef5d22fe68252f41bd6d75042d2a8f69596b9361a6daf4d
+  data.tar.gz: fe9ac8d41fed72423a4ee2e9b209f0c10f742546fbd36704a1c391dba6004e64
 SHA512:
-  metadata.gz: cd56a05ee8a42910901a85de11940ea16b1e6a4578fd2501aeb4e794674561dd9594bb10fedc0b07c207df5823c44361cf1a72d753b241fe4e564e786464f459
-  data.tar.gz: 7fa75694d0ff55caa28df71703fd4e8c14e00185149c456b26c8030315519e0caafcc77e0e10efe1d04aae1093dd53c9f1b4462b97f04d8352c996241a271662
+  metadata.gz: d14b3282ec480b9d628902261ac467f89bf5f635723d420a1b05607b7c6c4f549f6aaa70d0eba8513c23100e1f8142ed2099b07295c0ce288b85108478861daa
+  data.tar.gz: 9d8119f8ce88a5fed9a0cc1e159180d623a194ee2eacfa0ac898a3d289b6318343c3583764223d132bc0cc1550b7c37bac2bcd2f6b39e530747f9f1d1a201f6f

data/lib/twiglet/formatter.rb

@@ -11,12 +11,13 @@ module Twiglet
       validator:,
       default_properties: {},
       context_provider: nil,
+      context_providers: [],
       now: -> { Time.now.utc }
     )
       @service_name = service_name
       @now = now
       @default_properties = default_properties
-      @context_provider = context_provider
+      @context_providers = context_provider ? [context_provider] : context_providers
       @validator = validator
 
       super()
@@ -45,7 +46,9 @@ module Twiglet
         }
       }
 
-      context = @context_provider&.call || {}
+      context = @context_providers.reduce({}) do |c, context_provider|
+        c.deep_merge(context_provider.call)
+      end
 
       JSON.generate(
         base_message

data/lib/twiglet/logger.rb

@@ -33,7 +33,7 @@ module Twiglet
       formatter = Twiglet::Formatter.new(
         service_name,
         default_properties: args.fetch(:default_properties, {}),
-        context_provider: args[:context_provider],
+        context_providers: Array(args[:context_provider] || args[:context_providers]),
         now: now,
         validator: @validator
       )
@@ -82,15 +82,12 @@ module Twiglet
     end
 
     def context_provider(&blk)
-      new_context_provider = blk
-      if @args[:context_provider]
-        new_context_provider = lambda do
-          @args[:context_provider].call.merge(blk.call)
-        end
-      end
+      new_context_providers = Array(@args[:context_providers])
+      new_context_providers << blk
+
       self.class.new(
         @service_name,
-        **@args.merge(context_provider: new_context_provider)
+        **@args.merge(context_providers: new_context_providers)
       )
     end
 

data/lib/twiglet/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Twiglet
-  VERSION = '3.10.0'
+  VERSION = '3.11.0'
 end

data/test/formatter_test.rb

@@ -58,4 +58,32 @@ describe Twiglet::Formatter do
     }
     assert_equal JSON.parse(msg), expected_log
   end
+
+  it 'merges the outputs of all context providers into the messages log' do
+    provider_1 = -> { { 'request' => { 'id' => '1234567890' } } }
+    provider_2 = -> { { 'request' => { 'type' => 'test' } } }
+    formatter = Twiglet::Formatter.new(
+      'petshop', now: @now, validator: Twiglet::Validator.new({}.to_json),
+                 context_providers: [provider_1, provider_2]
+    )
+    msg = formatter.call('warn', nil, nil, 'shop is running low on dog food')
+    expected_log = {
+      "ecs" => {
+        "version" => '1.5.0'
+      },
+      "@timestamp" => '2020-05-11T15:01:01.000Z',
+      "service" => {
+        "name" => 'petshop'
+      },
+      "log" => {
+        "level" => 'warn'
+      },
+      "message" => 'shop is running low on dog food',
+      "request" => {
+        'id' => '1234567890',
+        'type' => 'test'
+      }
+    }
+    assert_equal expected_log, JSON.parse(msg)
+  end
 end

data/test/logger_test.rb

@@ -189,9 +189,37 @@ describe Twiglet::Logger do
     end
 
     it "should be able to add contextual information to events with the context_provider" do
-      purchase_logger = @logger.context_provider do
-        { 'context' => { 'id' => 'my-context-id' } }
-      end
+      provider = -> { { 'context' => { 'id' => 'my-context-id' } } }
+      purchase_logger = Twiglet::Logger.new(
+        'petshop',
+        now: @now,
+        output: @buffer,
+        context_provider: provider
+      )
+
+      # do stuff
+      purchase_logger.info(
+        {
+          message: 'customer bought a dog',
+          pet: { name: 'Barker', species: 'dog', breed: 'Bitsa' }
+        }
+      )
+
+      log = read_json @buffer
+
+      assert_equal 'customer bought a dog', log[:message]
+      assert_equal 'my-context-id', log[:context][:id]
+    end
+
+    it "should be able to add contextual information to events with multiple context providers" do
+      provider_1 = -> { { 'context' => { 'id' => 'my-context-id' } } }
+      provider_2 = -> { { 'context' => { 'type' => 'test' } } }
+      purchase_logger = Twiglet::Logger.new(
+        'petshop',
+        now: @now,
+        output: @buffer,
+        context_providers: [provider_1, provider_2]
+      )
 
       # do stuff
       purchase_logger.info(
@@ -205,6 +233,7 @@ describe Twiglet::Logger do
 
       assert_equal 'customer bought a dog', log[:message]
       assert_equal 'my-context-id', log[:context][:id]
+      assert_equal 'test', log[:context][:type]
     end
 
     it "chaining .with and .context_provider is possible" do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: twiglet
 version: !ruby/object:Gem::Version
-  version: 3.10.0
+  version: 3.11.0
 platform: ruby
 authors:
 - Simply Business
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-08-02 00:00:00.000000000 Z
+date: 2024-08-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json-schema