mirror of
https://github.com/logos-storage/logos-storage-network-crawler.git
synced 2026-01-02 13:33:08 +00:00
84 lines
2.1 KiB
Nim
84 lines
2.1 KiB
Nim
import std/os
|
|
import pkg/chronicles
|
|
import pkg/chronos
|
|
import pkg/questionable
|
|
import pkg/questionable/results
|
|
|
|
import ./codexcrawler/main
|
|
import ./codexcrawler/config
|
|
import ./codexcrawler/metrics
|
|
import ./codexcrawler/logging
|
|
|
|
when defined(posix):
|
|
import system/ansi_c
|
|
|
|
type
|
|
ApplicationStatus {.pure.} = enum
|
|
Stopped
|
|
Stopping
|
|
Running
|
|
|
|
Application = ref object
|
|
status: ApplicationStatus
|
|
|
|
proc run(app: Application) =
|
|
let config = parseConfig()
|
|
info "Loaded configuration", config
|
|
|
|
# Configure loglevel
|
|
updateLogLevel(config.logLevel)
|
|
|
|
# Ensure datadir path exists:
|
|
if not existsDir(config.dataDir):
|
|
createDir(config.dataDir)
|
|
|
|
setupMetrics(config.metricsAddress, config.metricsPort)
|
|
info "Metrics endpoint initialized"
|
|
|
|
info "Starting application"
|
|
app.status = ApplicationStatus.Running
|
|
if err =? (waitFor startApplication(config)).errorOption:
|
|
app.status = ApplicationStatus.Stopping
|
|
error "Failed to start application", err = err.msg
|
|
|
|
while app.status == ApplicationStatus.Running:
|
|
try:
|
|
chronos.poll()
|
|
except Exception as exc:
|
|
error "Unhandled exception", msg = exc.msg
|
|
quit QuitFailure
|
|
notice "Application closed"
|
|
|
|
when isMainModule:
|
|
let app = Application()
|
|
|
|
# Stopping code must be in scope of app declaration.
|
|
# Else capture of the instance is not allowed due to {.noconv.}.
|
|
proc onStopSignal() =
|
|
app.status = ApplicationStatus.Stopping
|
|
notice "Stopping Crawler..."
|
|
|
|
proc controlCHandler() {.noconv.} =
|
|
when defined(windows):
|
|
# workaround for https://github.com/nim-lang/Nim/issues/4057
|
|
try:
|
|
setupForeignThreadGc()
|
|
except Exception as exc:
|
|
raiseAssert exc.msg
|
|
notice "Shutting down after having received SIGINT"
|
|
onStopSignal()
|
|
|
|
try:
|
|
setControlCHook(controlCHandler)
|
|
except Exception as exc:
|
|
warn "Cannot set ctrl-c handler", msg = exc.msg
|
|
|
|
when defined(posix):
|
|
proc SIGTERMHandler(signal: cint) {.noconv.} =
|
|
notice "Shutting down after having received SIGTERM"
|
|
onStopSignal()
|
|
|
|
c_signal(ansi_c.SIGTERM, SIGTERMHandler)
|
|
|
|
app.run()
|