dimgigov
011bc5bef4
- config.nim: Add baradb.json parsing + full env var support (BARADB_*) - logging.nim: Wire structured JSON logger into server.nim and baradadb.nim - server.nim: Replace echo with info/warn/errorMsg logging calls - baradadb.nim: Init logger from config (logLevel, logFile) - PLAN.md: Update honest assessment — deadlock, compaction, JOIN, TLS, slow query all verified working
88 lines
2.8 KiB
Nim
88 lines
2.8 KiB
Nim
## BaraDB — Multimodal Database Engine
|
|
## Main entry point
|
|
import std/asyncdispatch
|
|
import std/threadpool
|
|
import std/locks
|
|
import std/os
|
|
import std/strutils
|
|
import barabadb/core/server
|
|
import barabadb/core/httpserver
|
|
import barabadb/core/config
|
|
import barabadb/core/logging
|
|
import barabadb/protocol/ssl
|
|
import barabadb/storage/lsm
|
|
import barabadb/storage/compaction
|
|
|
|
type
|
|
CompactionManager* = ref object
|
|
db*: LSMTree
|
|
strategy*: compaction.CompactionStrategy
|
|
|
|
proc newCompactionManager*(db: LSMTree): CompactionManager =
|
|
result = CompactionManager(db: db, strategy: compaction.newCompactionStrategy(db.dir))
|
|
for sst in db.sstables:
|
|
let meta = compaction.SSTableMeta(
|
|
path: sst.path,
|
|
level: sst.level,
|
|
minKey: sst.minKey,
|
|
maxKey: sst.maxKey,
|
|
entryCount: sst.entryCount,
|
|
sizeBytes: sst.entryCount * 64,
|
|
createdAt: 0,
|
|
)
|
|
result.strategy.addTable(meta)
|
|
|
|
proc compact*(cm: CompactionManager) =
|
|
acquire(cm.db.lock)
|
|
defer: release(cm.db.lock)
|
|
for level in 0 ..< compaction.MaxLevel:
|
|
if cm.strategy.needsCompaction(level):
|
|
discard cm.strategy.compact(level)
|
|
|
|
proc startCompactionLoop*(cm: CompactionManager, intervalMs: int = 60000) {.async.} =
|
|
while true:
|
|
await sleepAsync(intervalMs)
|
|
cm.compact()
|
|
|
|
proc runTcpServer(config: BaraConfig) {.async.} =
|
|
info("BaraDB TCP listening on " & config.address & ":" & $config.port)
|
|
var server = newServer(config)
|
|
await server.run()
|
|
|
|
proc main() =
|
|
var config = loadConfig()
|
|
# Init structured logger from config
|
|
let logLvl = parseEnum[LogLevel]("ll" & capitalizeAscii(config.logLevel))
|
|
defaultLogger = newLogger(logLvl, config.logFile)
|
|
info("BaraDB v0.1.0 — Multimodal Database Engine")
|
|
|
|
if config.tlsEnabled:
|
|
if config.certFile.len == 0 or config.keyFile.len == 0 or
|
|
not fileExists(config.certFile) or not fileExists(config.keyFile):
|
|
info("TLS enabled but no certificate found. Generating self-signed certificate...")
|
|
let (cert, key) = generateSelfSignedCert(config.dataDir / "certs")
|
|
if cert.len > 0 and key.len > 0:
|
|
config.certFile = cert
|
|
config.keyFile = key
|
|
info("Generated self-signed certificate: Cert=" & cert & " Key=" & key)
|
|
else:
|
|
warn("Failed to generate self-signed certificate. TLS disabled.")
|
|
config.tlsEnabled = false
|
|
|
|
# Start HTTP server (blocking, multi-threaded via hunos) in background thread
|
|
var httpServer = newHttpServer(config)
|
|
spawn httpServer.run(config.port + 440) # HTTP port = TCP port + 440
|
|
|
|
# Start background compaction loop
|
|
let cm = newCompactionManager(httpServer.db)
|
|
asyncCheck cm.startCompactionLoop()
|
|
|
|
# Start TCP wire protocol server on main thread with async event loop
|
|
waitFor runTcpServer(config)
|
|
|
|
# Shutdown
|
|
httpServer.stop()
|
|
|
|
when isMainModule:
|
|
main()
|