wrap trace in with_logger()

Co-Authored-by: Chris Foster <[email protected]>
JunoLab · Mar 31, 2021 · 960513f · 960513f
1 parent 6d118cf
commit 960513f
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/util.jl b/src/util.jl
@@ -9,12 +9,13 @@ for level in [:trace, :debug, :info, :warn, :error, :fatal]
             # NOTE: `@placeholder` in order to avoid hard-coding @__LINE__ etc
             macrocall.args[1] = Symbol($"@$level")
             quote
-                old_logger = global_logger()
-                global_logger(Logging.ConsoleLogger(Core.stderr, Logging.min_enabled_level(old_logger)))
-                ret = $(esc(macrocall))
-                global_logger(old_logger)
+                global_min_level = Logging.min_enabled_level(global_logger())
+                raw_logger = Logging.ConsoleLogger(Core.stderr, global_min_level)
+                Logging.with_logger(raw_logger) do
+	                global ret = $(esc(macrocall))
+                end
                 ret
             end
         end
     end
-end
+end