mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-07-04 03:42:30 +00:00
b34894e4da
* [NOD-140] Converted DNSSeeder to use btclog. * [NOD-140] Converted MiningSimulator to use btclog. * [NOD-140] Converted TxGen to use btclog. * [NOD-140] Fixed log level in handlePanic in txgen. * [NOD-140] Renamed logger to log everywhere. Removed superfluous flag-setting to go-log.
51 lines
1002 B
Go
51 lines
1002 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/btcsuite/btclog"
|
|
"github.com/daglabs/btcd/rpcclient"
|
|
"github.com/jrick/logrotate/rotator"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
type logWriter struct{}
|
|
|
|
func (logWriter) Write(p []byte) (n int, err error) {
|
|
if initiated {
|
|
os.Stdout.Write(p)
|
|
LogRotator.Write(p)
|
|
}
|
|
return len(p), nil
|
|
}
|
|
|
|
var (
|
|
backendLog = btclog.NewBackend(logWriter{})
|
|
LogRotator *rotator.Rotator
|
|
log = backendLog.Logger("MNSM")
|
|
initiated = false
|
|
)
|
|
|
|
func initLogRotator(logFile string) {
|
|
initiated = true
|
|
logDir, _ := filepath.Split(logFile)
|
|
err := os.MkdirAll(logDir, 0700)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to create log directory: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
r, err := rotator.New(logFile, 10*1024, false, 3)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to create file rotator: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
LogRotator = r
|
|
}
|
|
|
|
func enableRPCLogging() {
|
|
rpclog := backendLog.Logger("RPCC")
|
|
rpclog.SetLevel(btclog.LevelTrace)
|
|
rpcclient.UseLogger(rpclog)
|
|
}
|