Be able to set a logger per context

ManageIQ · Oct 16, 2024 · af60de2 · af60de2
1 parent 4a61d68
commit af60de2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lib/floe/logging.rb b/lib/floe/logging.rb
@@ -7,7 +7,11 @@ def self.included(base)
     end
 
     def logger
-      Floe.logger
+      @logger || Floe.logger
+    end
+
+    def logger=(logger)
+      @logger = logger
     end
   end
 end
diff --git a/lib/floe/workflow/context.rb b/lib/floe/workflow/context.rb
@@ -9,7 +9,7 @@ class Context
 
       # @param context [Json|Hash] (default, create another with input and execution params)
       # @param input [Hash] (default: {})
-      def initialize(context = nil, input: nil, credentials: {})
+      def initialize(context = nil, input: nil, credentials: {}, logger: nil)
         context = JSON.parse(context) if context.kind_of?(String)
         input   = JSON.parse(input || "{}")
 
@@ -22,6 +22,8 @@ def initialize(context = nil, input: nil, credentials: {})
         self["Task"]               ||= {}
 
         @credentials = credentials || {}
+
+        self.logger = logger if logger
       rescue JSON::ParserError => err
         raise Floe::InvalidExecutionInput, "Invalid State Machine Execution Input: #{err}: was expecting (JSON String, Number, Array, Object or token 'null', 'true' or 'false')"
       end