mirror of
https://gitlab.com/pulsechaincom/go-pulse.git
synced 2024-12-22 03:30:35 +00:00
log, internal/debug: delete RotatingFileHandler (#20586)
* log: delete RotatingFileHandler We added this for the dashboard, which is gone now. The handler never really worked well and had data race and file handling issues. * internal/debug: remove unused RotatingFileHandler setup code
This commit is contained in:
parent
ad2fc7c6a6
commit
31baf3a9af
@ -45,7 +45,7 @@ func init() {
|
||||
// Set up the CLI app.
|
||||
app.Flags = append(app.Flags, debug.Flags...)
|
||||
app.Before = func(ctx *cli.Context) error {
|
||||
return debug.Setup(ctx, "")
|
||||
return debug.Setup(ctx)
|
||||
}
|
||||
app.After = func(ctx *cli.Context) error {
|
||||
debug.Exit()
|
||||
|
@ -233,7 +233,7 @@ func init() {
|
||||
app.Flags = append(app.Flags, metricsFlags...)
|
||||
|
||||
app.Before = func(ctx *cli.Context) error {
|
||||
return debug.Setup(ctx, "")
|
||||
return debug.Setup(ctx)
|
||||
}
|
||||
app.After = func(ctx *cli.Context) error {
|
||||
debug.Exit()
|
||||
|
@ -112,20 +112,9 @@ func init() {
|
||||
|
||||
// Setup initializes profiling and logging based on the CLI flags.
|
||||
// It should be called as early as possible in the program.
|
||||
func Setup(ctx *cli.Context, logdir string) error {
|
||||
func Setup(ctx *cli.Context) error {
|
||||
// logging
|
||||
log.PrintOrigins(ctx.GlobalBool(debugFlag.Name))
|
||||
if logdir != "" {
|
||||
rfh, err := log.RotatingFileHandler(
|
||||
logdir,
|
||||
262144,
|
||||
log.JSONFormatOrderedEx(false, true),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
glogger.SetHandler(log.MultiHandler(ostream, rfh))
|
||||
}
|
||||
glogger.Verbosity(log.Lvl(ctx.GlobalInt(verbosityFlag.Name)))
|
||||
glogger.Vmodule(ctx.GlobalString(vmoduleFlag.Name))
|
||||
glogger.BacktraceAt(ctx.GlobalString(backtraceAtFlag.Name))
|
||||
|
110
log/handler.go
110
log/handler.go
@ -8,11 +8,6 @@ import (
|
||||
"reflect"
|
||||
"sync"
|
||||
|
||||
"io/ioutil"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/go-stack/stack"
|
||||
)
|
||||
|
||||
@ -75,111 +70,6 @@ func FileHandler(path string, fmtr Format) (Handler, error) {
|
||||
return closingHandler{f, StreamHandler(f, fmtr)}, nil
|
||||
}
|
||||
|
||||
// countingWriter wraps a WriteCloser object in order to count the written bytes.
|
||||
type countingWriter struct {
|
||||
w io.WriteCloser // the wrapped object
|
||||
count uint // number of bytes written
|
||||
}
|
||||
|
||||
// Write increments the byte counter by the number of bytes written.
|
||||
// Implements the WriteCloser interface.
|
||||
func (w *countingWriter) Write(p []byte) (n int, err error) {
|
||||
n, err = w.w.Write(p)
|
||||
w.count += uint(n)
|
||||
return n, err
|
||||
}
|
||||
|
||||
// Close implements the WriteCloser interface.
|
||||
func (w *countingWriter) Close() error {
|
||||
return w.w.Close()
|
||||
}
|
||||
|
||||
// prepFile opens the log file at the given path, and cuts off the invalid part
|
||||
// from the end, because the previous execution could have been finished by interruption.
|
||||
// Assumes that every line ended by '\n' contains a valid log record.
|
||||
func prepFile(path string) (*countingWriter, error) {
|
||||
f, err := os.OpenFile(path, os.O_RDWR|os.O_APPEND, 0600)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
_, err = f.Seek(-1, io.SeekEnd)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
buf := make([]byte, 1)
|
||||
var cut int64
|
||||
for {
|
||||
if _, err := f.Read(buf); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if buf[0] == '\n' {
|
||||
break
|
||||
}
|
||||
if _, err = f.Seek(-2, io.SeekCurrent); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
cut++
|
||||
}
|
||||
fi, err := f.Stat()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
ns := fi.Size() - cut
|
||||
if err = f.Truncate(ns); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &countingWriter{w: f, count: uint(ns)}, nil
|
||||
}
|
||||
|
||||
// RotatingFileHandler returns a handler which writes log records to file chunks
|
||||
// at the given path. When a file's size reaches the limit, the handler creates
|
||||
// a new file named after the timestamp of the first log record it will contain.
|
||||
func RotatingFileHandler(path string, limit uint, formatter Format) (Handler, error) {
|
||||
if err := os.MkdirAll(path, 0700); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
files, err := ioutil.ReadDir(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
re := regexp.MustCompile(`\.log$`)
|
||||
last := len(files) - 1
|
||||
for last >= 0 && (!files[last].Mode().IsRegular() || !re.MatchString(files[last].Name())) {
|
||||
last--
|
||||
}
|
||||
var counter *countingWriter
|
||||
if last >= 0 && files[last].Size() < int64(limit) {
|
||||
// Open the last file, and continue to write into it until it's size reaches the limit.
|
||||
if counter, err = prepFile(filepath.Join(path, files[last].Name())); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
if counter == nil {
|
||||
counter = new(countingWriter)
|
||||
}
|
||||
h := StreamHandler(counter, formatter)
|
||||
|
||||
return FuncHandler(func(r *Record) error {
|
||||
if counter.count > limit {
|
||||
counter.Close()
|
||||
counter.w = nil
|
||||
}
|
||||
if counter.w == nil {
|
||||
f, err := os.OpenFile(
|
||||
filepath.Join(path, fmt.Sprintf("%s.log", strings.Replace(r.Time.Format("060102150405.00"), ".", "", 1))),
|
||||
os.O_CREATE|os.O_APPEND|os.O_WRONLY,
|
||||
0600,
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
counter.w = f
|
||||
counter.count = 0
|
||||
}
|
||||
return h.Log(r)
|
||||
}), nil
|
||||
}
|
||||
|
||||
// NetHandler opens a socket to the given address and writes records
|
||||
// over the connection.
|
||||
func NetHandler(network, addr string, fmtr Format) (Handler, error) {
|
||||
|
Loading…
Reference in New Issue
Block a user