expanded logging in router.go

This commit is contained in:
Haris Khan
2024-10-19 09:46:23 -04:00
parent c31e990995
commit aa98589f1c

View File

@@ -1,13 +1,16 @@
package router package router
import ( import (
"github.com/go-i2p/go-i2p/lib/util/logger"
"github.com/sirupsen/logrus"
"time" "time"
"github.com/go-i2p/go-i2p/lib/config" "github.com/go-i2p/go-i2p/lib/config"
"github.com/go-i2p/go-i2p/lib/netdb" "github.com/go-i2p/go-i2p/lib/netdb"
log "github.com/sirupsen/logrus"
) )
var log = logger.GetLogger()
// i2p router type // i2p router type
type Router struct { type Router struct {
cfg *config.RouterConfig cfg *config.RouterConfig
@@ -18,73 +21,93 @@ type Router struct {
// create router with default configuration // create router with default configuration
func CreateRouter() (r *Router, err error) { func CreateRouter() (r *Router, err error) {
log.Debug("Creating router with default configuration")
cfg := config.RouterConfigProperties cfg := config.RouterConfigProperties
r, err = FromConfig(cfg) r, err = FromConfig(cfg)
if err != nil {
log.WithError(err).Error("Failed to create router from default configuration")
} else {
log.Debug("Router created successfully with default configuration")
}
return return
} }
// create router from configuration // create router from configuration
func FromConfig(c *config.RouterConfig) (r *Router, err error) { func FromConfig(c *config.RouterConfig) (r *Router, err error) {
log.WithField("config", c).Debug("Creating router from configuration")
r = new(Router) r = new(Router)
r.cfg = c r.cfg = c
r.closeChnl = make(chan bool) r.closeChnl = make(chan bool)
log.Debug("Router created successfully from configuration")
return return
} }
// Wait blocks until router is fully stopped // Wait blocks until router is fully stopped
func (r *Router) Wait() { func (r *Router) Wait() {
log.Debug("Waiting for router to stop")
<-r.closeChnl <-r.closeChnl
log.Debug("Router has stopped")
} }
// Stop starts stopping internal state of router // Stop starts stopping internal state of router
func (r *Router) Stop() { func (r *Router) Stop() {
log.Debug("Stopping router")
r.closeChnl <- true r.closeChnl <- true
r.running = false r.running = false
log.Debug("Router stop signal sent")
} }
// Close closes any internal state and finallizes router resources so that nothing can start up again // Close closes any internal state and finallizes router resources so that nothing can start up again
func (r *Router) Close() error { func (r *Router) Close() error {
log.Warn("Closing router not implemented(?)")
return nil return nil
} }
// Start starts router mainloop // Start starts router mainloop
func (r *Router) Start() { func (r *Router) Start() {
if r.running { if r.running {
log.WithFields(log.Fields{ log.WithFields(logrus.Fields{
"at": "(Router) Start", "at": "(Router) Start",
"reason": "router is already running", "reason": "router is already running",
}).Error("Error Starting router") }).Error("Error Starting router")
return return
} }
log.Debug("Starting router")
r.running = true r.running = true
go r.mainloop() go r.mainloop()
} }
// run i2p router mainloop // run i2p router mainloop
func (r *Router) mainloop() { func (r *Router) mainloop() {
log.Debug("Entering router mainloop")
r.ndb = netdb.NewStdNetDB(r.cfg.NetDb.Path) r.ndb = netdb.NewStdNetDB(r.cfg.NetDb.Path)
log.WithField("netdb_path", r.cfg.NetDb.Path).Debug("Created StdNetDB")
// make sure the netdb is ready // make sure the netdb is ready
var e error var e error
if err := r.ndb.Ensure(); err != nil { if err := r.ndb.Ensure(); err != nil {
e = err e = err
log.WithError(err).Error("Failed to ensure NetDB")
} }
if sz := r.ndb.Size(); sz >= 0 { if sz := r.ndb.Size(); sz >= 0 {
log.Info("NetDB Size:", sz) log.WithField("size", sz).Debug("NetDB Size")
} else {
log.Warn("Unable to determine NetDB size")
} }
if e == nil { if e == nil {
// netdb ready // netdb ready
log.WithFields(log.Fields{ log.WithFields(logrus.Fields{
"at": "(Router) mainloop", "at": "(Router) mainloop",
}).Info("Router ready") }).Debug("Router ready")
for e == nil { for e == nil {
time.Sleep(time.Second) time.Sleep(time.Second)
} }
} else { } else {
// netdb failed // netdb failed
log.WithFields(log.Fields{ log.WithFields(logrus.Fields{
"at": "(Router) mainloop", "at": "(Router) mainloop",
"reason": e.Error(), "reason": e.Error(),
}).Error("Netdb Startup failed") }).Error("Netdb Startup failed")
r.Stop() r.Stop()
} }
log.Debug("Exiting router mainloop")
} }