mirror of https://github.com/xSmurf/oz.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.2 KiB
93 lines
2.2 KiB
package daemon
|
|
|
|
import (
|
|
"github.com/op/go-logging"
|
|
"github.com/subgraph/oz/ipc"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
func (d *daemonState) Debug(format string, args ...interface{}) {
|
|
d.log.Debug(format, args...)
|
|
}
|
|
func (d *daemonState) Info(format string, args ...interface{}) {
|
|
d.log.Info(format, args...)
|
|
}
|
|
func (d *daemonState) Notice(format string, args ...interface{}) {
|
|
d.log.Notice(format, args...)
|
|
}
|
|
func (d *daemonState) Warning(format string, args ...interface{}) {
|
|
d.log.Warning(format, args...)
|
|
}
|
|
func (d *daemonState) Error(format string, args ...interface{}) {
|
|
d.log.Error(format, args...)
|
|
}
|
|
func (d *daemonState) Critical(format string, args ...interface{}) {
|
|
d.log.Critical(format, args...)
|
|
}
|
|
|
|
func (d *daemonState) initializeLogging() {
|
|
d.log = logging.MustGetLogger("oz")
|
|
be := logging.NewChannelMemoryBackend(100)
|
|
fbe := logging.NewBackendFormatter(be, format)
|
|
d.memBackend = be
|
|
stderr := logging.NewLogBackend(os.Stderr, "", log.LstdFlags)
|
|
d.backends = []logging.Backend{
|
|
stderr,
|
|
fbe,
|
|
}
|
|
d.installBackends()
|
|
}
|
|
|
|
var format = logging.MustStringFormatter(
|
|
"%{color}%{time:15:04:05} ▶ %{level:.4s} %{id:03x}%{color:reset} %{message}",
|
|
)
|
|
|
|
func (d *daemonState) addBackend(be logging.Backend) {
|
|
d.backends = append(d.backends, be)
|
|
d.installBackends()
|
|
}
|
|
|
|
func (d *daemonState) removeBackend(be logging.Backend) {
|
|
newBackends := []logging.Backend{}
|
|
for _, b := range d.backends {
|
|
if b != be {
|
|
newBackends = append(newBackends, b)
|
|
}
|
|
}
|
|
d.backends = newBackends
|
|
d.installBackends()
|
|
}
|
|
|
|
func (d *daemonState) installBackends() {
|
|
if len(d.backends) == 1 {
|
|
d.log.SetBackend(logging.AddModuleLevel(d.backends[0]))
|
|
return
|
|
}
|
|
d.log.SetBackend(logging.MultiLogger(d.backends...))
|
|
}
|
|
|
|
type logFollower struct {
|
|
daemon *daemonState
|
|
wrapper logging.Backend
|
|
m *ipc.Message
|
|
}
|
|
|
|
func (lf *logFollower) Log(level logging.Level, calldepth int, rec *logging.Record) error {
|
|
s := rec.Formatted(calldepth)
|
|
if err := lf.m.Respond(&LogData{[]string{s}}); err != nil {
|
|
lf.remove()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (lf *logFollower) remove() {
|
|
lf.daemon.removeBackend(lf.wrapper)
|
|
}
|
|
|
|
func (d *daemonState) followLogs(m *ipc.Message) {
|
|
be := &logFollower{m: m, daemon: d}
|
|
be.wrapper = logging.NewBackendFormatter(be, format)
|
|
d.addBackend(be.wrapper)
|
|
}
|