nim-chronos/chronos/profiler.nim

import ./config

when chronosProfiling:
  import futures
  import ./profiler/[events, metrics]

  export futures, events, metrics

  when not chronosFutureId:
    {.error: "chronosProfiling requires chronosFutureId to be enabled".}

  var futureMetrics {.threadvar.}: ProfilerMetrics 

  proc getMetrics*(): ProfilerMetrics = 
    ## Returns metrics for the current event loop.
    result = futureMetrics

  proc enableEventCallbacks*(): void =
    onBaseFutureEvent = handleBaseFutureEvent
    onAsyncFutureEvent = handleAsyncFutureEvent
    
  proc enableProfiling*() =
    ## Enables profiling on the current event loop.
    if not isNil(handleFutureEvent): return

    enableEventCallbacks()
    handleFutureEvent = proc (e: Event) {.nimcall.} = 
      futureMetrics.processEvent(e)