prune util/log and user logger (#1237)
* prune util/log and user logger Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org> * plaintext logger Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org> * add newline Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
123
debug/log/os.go
123
debug/log/os.go
@@ -1,13 +1,7 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/micro/go-micro/v2/util/ring"
|
||||
@@ -25,106 +19,10 @@ type osLog struct {
|
||||
|
||||
type osStream struct {
|
||||
stream chan Record
|
||||
stop chan bool
|
||||
}
|
||||
|
||||
// watch io stream
|
||||
func (o *osLog) run() {
|
||||
// save outputs
|
||||
stdout := *os.Stdout
|
||||
stderr := *os.Stderr
|
||||
|
||||
// new os pipe
|
||||
r, w := io.Pipe()
|
||||
|
||||
// create new iopipes
|
||||
r1, w1, _ := os.Pipe()
|
||||
r2, w2, _ := os.Pipe()
|
||||
|
||||
// create tea readers
|
||||
tee1 := io.TeeReader(r1, &stdout)
|
||||
tee2 := io.TeeReader(r2, &stderr)
|
||||
|
||||
// start copying
|
||||
go io.Copy(w, tee1)
|
||||
go io.Copy(w, tee2)
|
||||
|
||||
// set default go log output
|
||||
//log.SetOutput(w2)
|
||||
|
||||
// replace os stdout and os stderr
|
||||
*os.Stdout = *w1
|
||||
*os.Stderr = *w2
|
||||
|
||||
// this should short circuit everything
|
||||
defer func() {
|
||||
// reset stdout and stderr
|
||||
*os.Stdout = stdout
|
||||
*os.Stderr = stderr
|
||||
//log.SetOutput(stderr)
|
||||
|
||||
// close all the outputs
|
||||
r.Close()
|
||||
r1.Close()
|
||||
r2.Close()
|
||||
w.Close()
|
||||
w1.Close()
|
||||
w2.Close()
|
||||
}()
|
||||
|
||||
// read from standard error
|
||||
scanner := bufio.NewReader(r)
|
||||
|
||||
for {
|
||||
// read the line
|
||||
line, err := scanner.ReadString('\n')
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
// check if the line exists
|
||||
if len(line) == 0 {
|
||||
continue
|
||||
}
|
||||
// parse the record
|
||||
var r Record
|
||||
if line[0] == '{' {
|
||||
json.Unmarshal([]byte(line), &r)
|
||||
} else {
|
||||
r = Record{
|
||||
Timestamp: time.Now(),
|
||||
Message: strings.TrimSuffix(line, "\n"),
|
||||
Metadata: make(map[string]string),
|
||||
}
|
||||
}
|
||||
|
||||
o.Lock()
|
||||
|
||||
// write to the buffer
|
||||
o.buffer.Put(r)
|
||||
|
||||
// check subs and send to stream
|
||||
for id, sub := range o.subs {
|
||||
// send to stream
|
||||
select {
|
||||
case <-sub.stop:
|
||||
delete(o.subs, id)
|
||||
case sub.stream <- r:
|
||||
// send to stream
|
||||
default:
|
||||
// do not block
|
||||
}
|
||||
}
|
||||
|
||||
o.Unlock()
|
||||
}
|
||||
}
|
||||
|
||||
// Read reads log entries from the logger
|
||||
func (o *osLog) Read(...ReadOption) ([]Record, error) {
|
||||
o.once.Do(func() {
|
||||
go o.run()
|
||||
})
|
||||
|
||||
var records []Record
|
||||
|
||||
// read the last 100 records
|
||||
@@ -137,29 +35,18 @@ func (o *osLog) Read(...ReadOption) ([]Record, error) {
|
||||
|
||||
// Write writes records to log
|
||||
func (o *osLog) Write(r Record) error {
|
||||
o.once.Do(func() {
|
||||
go o.run()
|
||||
})
|
||||
|
||||
// generate output
|
||||
out := o.format(r) + "\n"
|
||||
_, err := os.Stderr.Write([]byte(out))
|
||||
return err
|
||||
o.buffer.Put(r)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Stream log records
|
||||
func (o *osLog) Stream() (Stream, error) {
|
||||
o.once.Do(func() {
|
||||
go o.run()
|
||||
})
|
||||
|
||||
o.Lock()
|
||||
defer o.Unlock()
|
||||
|
||||
// create stream
|
||||
st := &osStream{
|
||||
stream: make(chan Record, 128),
|
||||
stop: make(chan bool),
|
||||
}
|
||||
|
||||
// save stream
|
||||
@@ -173,12 +60,6 @@ func (o *osStream) Chan() <-chan Record {
|
||||
}
|
||||
|
||||
func (o *osStream) Stop() error {
|
||||
select {
|
||||
case <-o.stop:
|
||||
return nil
|
||||
default:
|
||||
close(o.stop)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user