mirror of
https://github.com/logos-storage/nim-chronos.git
synced 2026-01-29 10:43:07 +00:00
65 lines
1.8 KiB
Nim
65 lines
1.8 KiB
Nim
import std/with
|
|
|
|
import ".."/".."/chronos
|
|
import ".."/".."/chronos/profiler/events
|
|
import ".."/".."/chronos/profiler/metrics
|
|
|
|
type
|
|
SimpleEvent* = object
|
|
procedure*: string
|
|
state*: ExtendedFutureState
|
|
|
|
# XXX this is sort of bad cause we get global state all over, but the fact we
|
|
# can't use closures on callbacks and that callbacks themselves are just
|
|
# global vars means we can't really do much better for now.
|
|
var recording*: seq[SimpleEvent]
|
|
var rawRecording*: seq[Event]
|
|
var fakeTime*: Moment = Moment.now()
|
|
|
|
# FIXME bad, this needs to be refactored into a callback interface for the profiler.
|
|
var oldHandleFutureEvent: proc(event: Event) {.nimcall, gcsafe, raises: [].} = nil
|
|
var installed: bool = false
|
|
|
|
proc recordEvent(event: Event) {.nimcall, gcsafe, raises: [].} =
|
|
{.cast(gcsafe).}:
|
|
recording.add(
|
|
SimpleEvent(procedure: $event.location.procedure, state: event.newState))
|
|
|
|
var timeShifted = event
|
|
timeshifted.timestamp = fakeTime
|
|
|
|
rawRecording.add(timeShifted)
|
|
|
|
proc recordSegment*(segment: string) =
|
|
{.cast(gcsafe).}:
|
|
recording.add(SimpleEvent(
|
|
procedure: segment,
|
|
state: ExtendedFutureState.Running
|
|
))
|
|
|
|
proc clearRecording*(): void =
|
|
recording = @[]
|
|
|
|
proc installCallbacks*() =
|
|
assert not installed, "Callbacks already installed"
|
|
oldHandleFutureEvent = handleFutureEvent
|
|
handleFutureEvent = recordEvent
|
|
|
|
installed = true
|
|
|
|
proc revertCallbacks*() =
|
|
assert installed, "Callbacks already uninstalled"
|
|
|
|
handleFutureEvent = oldHandleFutureEvent
|
|
installed = false
|
|
|
|
proc forProc*(self: var ProfilerMetrics, procedure: string): AggregateFutureMetrics =
|
|
for (key, aggMetrics) in self.totals.mpairs:
|
|
if key.procedure == procedure:
|
|
return aggMetrics
|
|
|
|
proc resetTime*() =
|
|
fakeTime = Moment.now()
|
|
|
|
proc advanceTime*(duration: Duration) =
|
|
fakeTime += duration |