lib/twiglet/logger.rb in twiglet-2.3.10 vs lib/twiglet/logger.rb in twiglet-2.3.11

@@ -3,10 +3,11 @@
 require 'logger'
 require 'time'
 require 'json'
 require_relative 'formatter'
 require_relative '../hash_extensions'
+require_relative 'message'
 
 module Twiglet
   class Logger < ::Logger
     Hash.include HashExtensions
 
@@ -35,10 +36,10 @@
           'error': {
             'message': error.message
           }
         }
         add_stack_trace(error_fields, error)
-        message = message.is_a?(Hash) ? message.merge(error_fields) : error_fields.merge(message: message)
+        message = Message.new(message).merge(error_fields)
       end
 
       super(message, &block)
     end