294 lines
11 KiB
Nim
294 lines
11 KiB
Nim
import
|
|
macros, tables, strutils, strformat,
|
|
chronicles/[scope_helpers, dynamic_scope, log_output, options]
|
|
|
|
export
|
|
dynamic_scope, log_output, options
|
|
|
|
# So, how does Chronicles work?
|
|
#
|
|
# The tricky part is understanding how the lexical scopes are implemened.
|
|
# For them to work, we need to be able to associate a mutable compile-time
|
|
# data with a lexical scope (with a different value for each scope).
|
|
# The regular compile-time variable are not suited for this, because they
|
|
# offer us only a single global value that can be mutated.
|
|
#
|
|
# Luckily, we can use the body of a template as the storage mechanism for
|
|
# our data. This works, because template names bound to particular scopes
|
|
# and templates can be freely redefined as many times as necessary.
|
|
#
|
|
# `activeChroniclesScope` stores the current lexical scope.
|
|
#
|
|
# `logScopeIMPL` is used to merge a previously defined scope with some
|
|
# new definition in order to produce a new scope:
|
|
#
|
|
|
|
template activeChroniclesScope* =
|
|
0 # track the scope revision
|
|
|
|
macro logScopeIMPL(prevScopes: typed,
|
|
newBindings: untyped,
|
|
isPublic: static[bool]): untyped =
|
|
result = newStmtList()
|
|
var
|
|
bestScope = prevScopes.lastScopeBody
|
|
bestScopeRev = bestScope.scopeRevision
|
|
newRevision = newLit(bestScopeRev + 1)
|
|
finalBindings = initTable[string, NimNode]()
|
|
newAssingments = newStmtList()
|
|
chroniclesExportNode: NimNode = if not isPublic: nil
|
|
else: newTree(nnkExportExceptStmt,
|
|
id"chronicles",
|
|
id"activeChroniclesScope")
|
|
|
|
for k, v in assignments(bestScope.scopeAssignments):
|
|
finalBindings[k] = v
|
|
|
|
for k, v in assignments(newBindings, false):
|
|
finalBindings[k] = v
|
|
|
|
for k, v in finalBindings:
|
|
if k == "stream":
|
|
let streamId = id($v)
|
|
let errorMsg = &"{v.lineInfo}: {$streamId} is not a recognized stream name"
|
|
let templateName = id("activeChroniclesStream", isPublic)
|
|
|
|
result.add quote do:
|
|
when not declared(`streamId`):
|
|
# XXX: how to report the proper line info here?
|
|
{.error: `errorMsg`.}
|
|
#elif not isStreamSymbolIMPL(`streamId`):
|
|
# {.error: `errorMsg`.}
|
|
template `templateName`: typedesc = `streamId`
|
|
|
|
if isPublic:
|
|
chroniclesExportNode.add id"activeChroniclesStream"
|
|
|
|
else:
|
|
newAssingments.add newAssignment(id(k), v)
|
|
|
|
if isPublic:
|
|
result.add chroniclesExportNode
|
|
|
|
let activeScope = id("activeChroniclesScope", isPublic)
|
|
result.add quote do:
|
|
template `activeScope` =
|
|
`newRevision`
|
|
`newAssingments`
|
|
|
|
template logScope*(newBindings: untyped) {.dirty.} =
|
|
bind bindSym, logScopeIMPL, brForceOpen
|
|
logScopeIMPL(bindSym("activeChroniclesScope", brForceOpen),
|
|
newBindings, false)
|
|
|
|
template publicLogScope*(newBindings: untyped) {.dirty.} =
|
|
bind bindSym, logScopeIMPL, brForceOpen
|
|
logScopeIMPL(bindSym("activeChroniclesScope", brForceOpen),
|
|
newBindings, true)
|
|
|
|
template dynamicLogScope*(recordType: typedesc,
|
|
bindings: varargs[untyped]) {.dirty.} =
|
|
bind bindSym, brForceOpen
|
|
dynamicLogScopeIMPL(recordType,
|
|
bindSym("activeChroniclesScope", brForceOpen),
|
|
bindings)
|
|
|
|
template dynamicLogScope*(bindings: varargs[untyped]) {.dirty.} =
|
|
bind bindSym, brForceOpen
|
|
dynamicLogScopeIMPL(activeChroniclesStream(),
|
|
bindSym("activeChroniclesScope", brForceOpen),
|
|
bindings)
|
|
|
|
when runtimeFilteringEnabled:
|
|
import chronicles/topics_registry
|
|
export setTopicState, TopicState
|
|
|
|
var gActiveLogLevel: LogLevel
|
|
|
|
proc setLogLevel*(lvl: LogLevel) =
|
|
gActiveLogLevel = lvl
|
|
|
|
proc topicStateIMPL(topicName: static[string]): ptr TopicState =
|
|
var state {.global.}: TopicState
|
|
var dummy {.global.} = registerTopic(topicName, addr(state))
|
|
return addr(state)
|
|
|
|
proc runtimeTopicFilteringCode*(logLevel: LogLevel, topics: seq[string]): NimNode =
|
|
# This proc generates the run-time code used for topic filtering.
|
|
# Each logging statement has a statically known list of associated topics.
|
|
# For each of the topics in the list, we consult a TLS TopicState value
|
|
# created in topicStateIMPL. `break chroniclesLogStmt` exits a named
|
|
# block surrounding the entire log statement.
|
|
result = newStmtList()
|
|
var
|
|
matchEnabledTopics = genSym(nskVar, "matchEnabledTopics")
|
|
requiredTopicsCount = genSym(nskVar, "requiredTopicsCount")
|
|
topicChecks = newStmtList()
|
|
|
|
result.add quote do:
|
|
if LogLevel(`logLevel`) < gActiveLogLevel:
|
|
break chroniclesLogStmt
|
|
|
|
var `matchEnabledTopics` = registry.totalEnabledTopics == 0
|
|
var `requiredTopicsCount` = registry.totalRequiredTopics
|
|
|
|
for topic in topics:
|
|
result.add quote do:
|
|
let s = topicStateIMPL(`topic`)
|
|
case s[]
|
|
of Normal: discard
|
|
of Enabled: `matchEnabledTopics` = true
|
|
of Disabled: break chroniclesLogStmt
|
|
of Required: dec `requiredTopicsCount`
|
|
|
|
result.add quote do:
|
|
if not `matchEnabledTopics` or `requiredTopicsCount` > 0:
|
|
break chroniclesLogStmt
|
|
else:
|
|
template runtimeFilteringDisabledError =
|
|
{.error: "Run-time topic filtering is currently disabled. " &
|
|
"You can enable it by specifying '-d:chronicles_runtime_filtering:on'".}
|
|
|
|
template setTopicState*(name, state) = runtimeFilteringDisabledError
|
|
template setLogLevel*(name, state) = runtimeFilteringDisabledError
|
|
|
|
macro logIMPL(recordType: typedesc,
|
|
eventName: static[string],
|
|
severity: static[LogLevel],
|
|
scopes: typed,
|
|
logStmtBindings: varargs[untyped]): untyped =
|
|
if not loggingEnabled or severity < enabledLogLevel: return
|
|
clearEmptyVarargs logStmtBindings
|
|
|
|
# First, we merge the lexical bindings with the additional
|
|
# bindings passed to the logging statement itself:
|
|
let lexicalBindings = scopes.finalLexicalBindings
|
|
var finalBindings = initOrderedTable[string, NimNode]()
|
|
|
|
for k, v in assignments(lexicalBindings):
|
|
finalBindings[k] = v
|
|
|
|
for k, v in assignments(logStmtBindings):
|
|
finalBindings[k] = v
|
|
|
|
finalBindings.sort do (lhs, rhs: auto) -> int: cmp(lhs[0], rhs[0])
|
|
|
|
# This is the compile-time topic filtering code, which has a similar
|
|
# logic to the generated run-time filtering code:
|
|
var enabledTopicsMatch = enabledTopics.len == 0
|
|
var requiredTopicsCount = requiredTopics.len
|
|
var currentTopics: seq[string] = @[]
|
|
|
|
if finalBindings.hasKey("topics"):
|
|
let topicsNode = finalBindings["topics"]
|
|
if topicsNode.kind notin {nnkStrLit, nnkTripleStrLit}:
|
|
error "Please specify the 'topics' list as a space separated string literal", topicsNode
|
|
|
|
currentTopics = topicsNode.strVal.split(Whitespace)
|
|
|
|
for t in currentTopics:
|
|
if t in disabledTopics:
|
|
return
|
|
elif t in enabledTopics:
|
|
enabledTopicsMatch = true
|
|
elif t in requiredTopics:
|
|
dec requiredTopicsCount
|
|
|
|
if not enabledTopicsMatch or requiredTopicsCount > 0:
|
|
return
|
|
|
|
var code = newStmtList()
|
|
when runtimeFilteringEnabled:
|
|
code.add runtimeTopicFilteringCode(severity, currentTopics)
|
|
|
|
# The rest of the code selects the active LogRecord type (which can
|
|
# be a tuple when the sink has multiple destinations) and then
|
|
# translates the log statement to a set of calls to `initLogRecord`,
|
|
# `setProperty` and `flushRecord`.
|
|
let
|
|
recordTypeSym = skipTypedesc(recordType.getTypeImpl())
|
|
recordTypeNodes = recordTypeSym.getTypeImpl()
|
|
recordArity = if recordTypeNodes.kind != nnkTupleConstr: 1
|
|
else: recordTypeNodes.len
|
|
record = genSym(nskVar, "record")
|
|
threadId = when compileOption("threads"): newCall("getThreadId")
|
|
else: newLit(0)
|
|
|
|
code.add quote do:
|
|
var `record`: `recordType`
|
|
|
|
for i in 0 ..< recordArity:
|
|
# We do something complicated here on purpose.
|
|
# We want to produce the setProperty calls for each record in turn
|
|
# because this would allow for the write optimization rules defined
|
|
# in `log_output` to kick in.
|
|
let recordRef = if recordArity == 1: record
|
|
else: newTree(nnkBracketExpr, record, newLit(i))
|
|
code.add quote do:
|
|
initLogRecord(`recordRef`, LogLevel(`severity`), `eventName`)
|
|
setFirstProperty(`recordRef`, "thread", `threadId`)
|
|
|
|
for k, v in finalBindings:
|
|
code.add newCall("setProperty", recordRef, newLit(k), v)
|
|
|
|
code.add newCall("logAllDynamicProperties", record)
|
|
code.add newCall("flushRecord", record)
|
|
|
|
result = newBlockStmt(id"chroniclesLogStmt", code)
|
|
|
|
# Translate all the possible overloads to `logIMPL`:
|
|
template log*(severity: LogLevel,
|
|
eventName: static[string],
|
|
props: varargs[untyped]) {.dirty.} =
|
|
|
|
bind logIMPL, bindSym, brForceOpen
|
|
logIMPL(activeChroniclesStream(), eventName, severity,
|
|
bindSym("activeChroniclesScope", brForceOpen), props)
|
|
|
|
template log*(recordType: typedesc,
|
|
severity: static[LogLevel],
|
|
eventName: static[string],
|
|
props: varargs[untyped]) {.dirty.} =
|
|
|
|
bind logIMPL, bindSym, brForceOpen
|
|
logIMPL(recordType, eventName, severity,
|
|
bindSym("activeChroniclesScope", brForceOpen), props)
|
|
|
|
template logFn(name, severity) {.dirty.} =
|
|
template `name`*(eventName: static[string],
|
|
props: varargs[untyped]) {.dirty.} =
|
|
|
|
bind logIMPL, bindSym, brForceOpen
|
|
logIMPL(activeChroniclesStream(), eventName, severity,
|
|
bindSym("activeChroniclesScope", brForceOpen), props)
|
|
|
|
template `name`*(recordType: typedesc,
|
|
eventName: static[string],
|
|
props: varargs[untyped]) {.dirty.} =
|
|
|
|
bind logIMPL, bindSym, brForceOpen
|
|
logIMPL(recordType, eventName, severity,
|
|
bindSym("activeChroniclesScope", brForceOpen), props)
|
|
|
|
logFn debug , LogLevel.DEBUG
|
|
logFn info , LogLevel.INFO
|
|
logFn notice, LogLevel.NOTICE
|
|
logFn warn , LogLevel.WARN
|
|
logFn error , LogLevel.ERROR
|
|
logFn fatal , LogLevel.FATAL
|
|
|
|
# TODO:
|
|
#
|
|
# * dynamic scope overrides (plus maybe an option to control the priority
|
|
# between dynamic and lexical bindings)
|
|
# * evaluate the lexical expressions only once in the presence of multiple sinks
|
|
# * syslog logging, Android and iOS logging, mixed std streams (logging both to stdout and stderr?)
|
|
# * resource management scheme for custom streams
|
|
# * define a bounty for creating a better test suite
|
|
# * define a bounty for implementing chronicles-tail
|
|
# - cross platform
|
|
# - interactive (on-the-fly commands can be entered)
|
|
# - allow filtering with custom (and/or expressions)
|
|
# - on-the-fly transforms, perhaps using the Nim VM?
|
|
#
|