-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
138 additions
and
130 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,138 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "io/fs" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
| "sync" | ||
|
|
||
| "github.com/gin-gonic/gin" | ||
| cfg "github.com/golobby/config/v3" | ||
| "github.com/golobby/config/v3/pkg/feeder" | ||
| log "github.com/sirupsen/logrus" | ||
| "gorm.io/gorm/logger" | ||
| ) | ||
|
|
||
| // InitLogging sets up logging | ||
| func (w *Watcher) InitLogging() { | ||
| // Read from .env and override from the local environment | ||
| dotEnvFeeder := feeder.DotEnv{Path: ".env"} | ||
| envFeeder := feeder.Env{} | ||
|
|
||
| // We need to use w.Config because we're using pointers | ||
| _ = cfg.New().AddFeeder(dotEnvFeeder).AddStruct(&w.Config).Feed() | ||
| _ = cfg.New().AddFeeder(envFeeder).AddStruct(&w.Config).Feed() | ||
|
|
||
| // Info level by default | ||
| LogLevelSelection := log.InfoLevel | ||
| switch { | ||
| case strings.EqualFold(w.Config.LogLevel, "trace"): | ||
| LogLevelSelection = log.TraceLevel | ||
| log.SetReportCaller(true) | ||
| case strings.EqualFold(w.Config.LogLevel, "debug"): | ||
| LogLevelSelection = log.DebugLevel | ||
| log.SetReportCaller(true) | ||
| case strings.EqualFold(w.Config.LogLevel, "info"): | ||
| LogLevelSelection = log.InfoLevel | ||
| case strings.EqualFold(w.Config.LogLevel, "warn"): | ||
| LogLevelSelection = log.WarnLevel | ||
| case strings.EqualFold(w.Config.LogLevel, "error"): | ||
| LogLevelSelection = log.ErrorLevel | ||
| } | ||
| log.SetLevel(LogLevelSelection) | ||
| log.SetFormatter(&log.JSONFormatter{}) | ||
|
|
||
| if log.GetLevel() > log.DebugLevel { | ||
| w.LogConfig = logger.Default.LogMode(logger.Info) | ||
| } | ||
| } | ||
|
|
||
| func GinJSONFormatter(param gin.LogFormatterParams) string { | ||
| jsonFormat := `{"time":"%s","clientip":"%s","method":"%s","uri":"%s","status":%3d,"latency":"%v","message":"%s","host":"%s","useragent":"%s","proto","%s","error_msg":"%s","size":%d}` + "\n" | ||
| return fmt.Sprintf(jsonFormat, | ||
| param.TimeStamp.Format(TimeFormatter), | ||
| param.ClientIP, | ||
| param.Method, | ||
| param.Request.RequestURI, | ||
| param.StatusCode, | ||
| param.Latency, | ||
| fmt.Sprintf("%s %s %s", param.Method, param.Request.URL, param.Request.Proto), | ||
| param.Request.Host, | ||
| param.Request.UserAgent(), | ||
| param.Request.Proto, | ||
| param.ErrorMessage, | ||
| param.BodySize, | ||
| ) | ||
| } | ||
|
|
||
| // FillDefaults sets a Watcher to default | ||
| // values defined in the constants | ||
| func (w *Watcher) FillDefaults() { | ||
| // Set unitialized values to preset defaults | ||
| if w.Config.BTCAPIEndpoint == "" { | ||
| w.Config.BTCAPIEndpoint = DefaultApi | ||
| } | ||
| if w.Config.LogLevel == "" { | ||
| w.Config.LogLevel = "info" | ||
| } | ||
| if w.Config.SleepInterval == 0 { | ||
| w.Config.SleepInterval = DefaultSleepInterval | ||
| } | ||
| if w.Config.Lookahead == 0 { | ||
| w.Config.Lookahead = DefaultLookahead | ||
| } | ||
| if w.Config.PageSize == 0 { | ||
| w.Config.PageSize = DefaultPageSize | ||
| } | ||
| if w.Config.DBPath == "" { | ||
| w.Config.DBPath = DefaultDBPath | ||
| } | ||
| if w.Config.Port == "" { | ||
| w.Config.Port = "80" | ||
| } | ||
| if w.Config.Currency == "" { | ||
| w.Config.Currency = CurrencyUSD | ||
| } | ||
|
|
||
| // Set up DB path | ||
| // Create the folder path if it doesn't exist | ||
| if _, err := os.Stat(w.Config.DBPath); errors.Is(err, fs.ErrNotExist) { | ||
| dirPath := filepath.Dir(w.Config.DBPath) | ||
| if err := os.MkdirAll(dirPath, 0660); err != nil { | ||
| log.Warn("unable to make directory path ", dirPath, " err: ", err) | ||
| w.Config.DBPath = "./local.db" | ||
| } | ||
| } | ||
| } | ||
|
|
||
| // StartWatches starts goroutines for watching all of the known | ||
| // addresses in the database. | ||
| func (w *Watcher) StartWatches() { | ||
| w.CancelWaitGroup = &sync.WaitGroup{} | ||
| // Check balance of each address | ||
| addresses := []AddressInfo{} | ||
| w.DB.Model(&AddressInfo{}).Scan(&addresses) | ||
| w.CancelSignals = map[string]chan bool{} | ||
| for _, address := range addresses { | ||
| // This channel is used to send a signal to stop watching the address | ||
| cancel := make(chan bool, 1) | ||
| w.CancelWaitGroup.Add(1) | ||
| w.AddCancelSignal(address.Address, cancel) | ||
| go w.WatchAddress(cancel, address.Address) | ||
| } | ||
|
|
||
| // Check balance of each key of each pubkey | ||
| pubkeys := []PubkeyInfo{} | ||
| w.DB.Model(&PubkeyInfo{}).Scan(&pubkeys) | ||
| for _, pubkey := range pubkeys { | ||
| // This channel is used to send a signal to stop watching the pubkey | ||
| cancel := make(chan bool, 1) | ||
| w.CancelWaitGroup.Add(1) | ||
| w.AddCancelSignal(pubkey.Pubkey, cancel) | ||
| go w.WatchPubkey(cancel, pubkey.Pubkey) | ||
| } | ||
| log.Infof("watching %d addresses and %d pubkeys", len(addresses), len(pubkeys)) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters