TimespanLogging: Remove worker Id Nesting

lib/TimespanLogging/src/core.jl

@@ -233,28 +233,35 @@ function write_event(ml::MultiEventLog, event::Event)
     end
 end
 
-function get_logs!(ml::MultiEventLog; only_local=false)
-    logs = Dict{Int,Dict{Symbol,Vector}}()
-    wkrs = only_local ? myid() : procs()
-    # FIXME: Log this logic
-    @sync for p in wkrs
-        @async begin
-            logs[p] = remotecall_fetch(p, ml) do ml
-                mls = get_state(ml)
-                lock(event_log_lock) do
-                    sublogs = Dict{Symbol,Vector}()
-                    for name in keys(mls.consumers)
-                        sublogs[name] = mls.consumer_logs[name]
-                        mls.consumer_logs[name] = []
+function get_logs!(ml::MultiEventLog; only_local=true)
+    if only_local
+        # Only return logs from current process
+        mls = get_state(ml)
+        return mls.consumer_logs
+    else
+        # Return logs from all processes
+        logs = Dict{Int,Dict{Symbol,Vector}}()
+        wkrs = procs()
+        @sync for p in wkrs 
+            @async begin
+                logs[p] = remotecall_fetch(p, ml) do ml
+                    mls = get_state(ml)
+                    lock(event_log_lock) do
+                        sublogs = Dict{Symbol,Vector}()
+                        for name in keys(mls.consumers)
+                            sublogs[name] = mls.consumer_logs[name]
+                            mls.consumer_logs[name] = []
+                        end
+                        sublogs
                     end
-                    sublogs
                 end
             end
         end
-    end
-    return logs
+        return logs
+    end       
 end
 
+
 # Core logging operations
 
 empty_prof() = ProfilerResult(UInt[], Dict{UInt64, Vector{Base.StackTraces.StackFrame}}(), UInt[])
@@ -471,3 +478,4 @@ function summarize_events(time_spent, gc_diff, profiler_samples)
 end
 
 summarize_events(xs) = summarize_events(aggregate_events(xs)...)
+