54b5cb56e7
This cleans up btcd.go a bit and consolidates the logging related functions. It also paves the way for upcoming message summaries.
100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
// Copyright (c) 2013 Conformal Systems LLC.
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/conformal/btcchain"
|
|
"github.com/conformal/btcdb"
|
|
"github.com/conformal/btcscript"
|
|
"github.com/conformal/seelog"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
log = seelog.Disabled
|
|
)
|
|
|
|
// logClosure is used to provide a closure over expensive logging operations
|
|
// so don't have to be performed when the logging level doesn't warrant it.
|
|
type logClosure func() string
|
|
|
|
// String invokes the underlying function and returns the result.
|
|
func (c logClosure) String() string {
|
|
return c()
|
|
}
|
|
|
|
// newLogClosure returns a new closure over a function that returns a string
|
|
// which itself provides a Stringer interface so that it can be used with the
|
|
// logging system.
|
|
func newLogClosure(c func() string) logClosure {
|
|
return logClosure(c)
|
|
}
|
|
|
|
// newLogger creates a new seelog logger using the provided logging level and
|
|
// log message prefix.
|
|
func newLogger(level string, prefix string) seelog.LoggerInterface {
|
|
fmtstring := `
|
|
<seelog type="adaptive" mininterval="2000000" maxinterval="100000000"
|
|
critmsgcount="500" minlevel="%s">
|
|
<outputs formatid="all">
|
|
<console/>
|
|
</outputs>
|
|
<formats>
|
|
<format id="all" format="[%%Time %%Date] [%%LEV] [%s] %%Msg%%n" />
|
|
</formats>
|
|
</seelog>`
|
|
config := fmt.Sprintf(fmtstring, level, prefix)
|
|
|
|
logger, err := seelog.LoggerFromConfigAsString(config)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to create logger: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
return logger
|
|
}
|
|
|
|
// useLogger sets the btcd logger to the passed logger.
|
|
func useLogger(logger seelog.LoggerInterface) {
|
|
log = logger
|
|
}
|
|
|
|
// setLogLevel sets the log level for the logging system. It initializes a
|
|
// logger for each subsystem at the provided level.
|
|
func setLogLevel(logLevel string) []seelog.LoggerInterface {
|
|
var loggers []seelog.LoggerInterface
|
|
|
|
// Define sub-systems.
|
|
subSystems := []struct {
|
|
level string
|
|
prefix string
|
|
useLogger func(seelog.LoggerInterface)
|
|
}{
|
|
{logLevel, "BTCD", useLogger},
|
|
{logLevel, "BCDB", btcdb.UseLogger},
|
|
{logLevel, "CHAN", btcchain.UseLogger},
|
|
{logLevel, "SCRP", btcscript.UseLogger},
|
|
}
|
|
|
|
// Configure all sub-systems with new loggers while keeping track of
|
|
// the created loggers to return so they can be flushed.
|
|
for _, s := range subSystems {
|
|
newLog := newLogger(s.level, s.prefix)
|
|
loggers = append(loggers, newLog)
|
|
s.useLogger(newLog)
|
|
}
|
|
|
|
return loggers
|
|
}
|
|
|
|
// directionString is a helper function that returns a string that represents
|
|
// the direction of a connection (inbound or outbound).
|
|
func directionString(inbound bool) string {
|
|
if inbound {
|
|
return "inbound"
|
|
}
|
|
return "outbound"
|
|
}
|