175 lines
4.5 KiB
Go
175 lines
4.5 KiB
Go
package service // import "go.unistack.org/micro-logger-service/v4"
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/google/uuid"
|
|
pbmicro "go.unistack.org/micro-logger-service/v4/micro"
|
|
pb "go.unistack.org/micro-logger-service/v4/proto"
|
|
"go.unistack.org/micro/v4/client"
|
|
"go.unistack.org/micro/v4/logger"
|
|
"go.unistack.org/micro/v4/store"
|
|
)
|
|
|
|
var _ logger.Logger = &serviceLogger{}
|
|
|
|
type serviceLogger struct {
|
|
opts logger.Options
|
|
service string
|
|
client pbmicro.LoggerServiceClient
|
|
store store.Store
|
|
}
|
|
|
|
func (l *serviceLogger) Clone(opts ...logger.Option) logger.Logger {
|
|
nl := &serviceLogger{service: l.service, store: l.store, client: l.client, opts: l.opts}
|
|
for _, o := range opts {
|
|
o(&nl.opts)
|
|
}
|
|
return nl
|
|
}
|
|
|
|
func (l *serviceLogger) Level(lvl logger.Level) {
|
|
l.opts.Level = lvl
|
|
}
|
|
|
|
func (l *serviceLogger) Init(opts ...logger.Option) error {
|
|
// TODO: optimize to store only []string as fields
|
|
for _, o := range opts {
|
|
o(&l.opts)
|
|
}
|
|
|
|
var cli client.Client
|
|
if l.opts.Context != nil {
|
|
if v, ok := l.opts.Context.Value(clientKey{}).(client.Client); ok && v != nil {
|
|
cli = v
|
|
}
|
|
if v, ok := l.opts.Context.Value(serviceKey{}).(string); ok && v != "" {
|
|
l.service = v
|
|
}
|
|
}
|
|
|
|
if l.service == "" {
|
|
return fmt.Errorf("missing Service option")
|
|
}
|
|
|
|
if cli == nil {
|
|
return fmt.Errorf("missing Client option")
|
|
}
|
|
|
|
l.client = pbmicro.NewLoggerServiceClient(l.service, cli)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *serviceLogger) Fields(fields ...interface{}) logger.Logger {
|
|
// TODO: optimize to store only []string as fields
|
|
return l.Clone(logger.WithFields(fields...))
|
|
}
|
|
|
|
func (l *serviceLogger) V(level logger.Level) bool {
|
|
return l.opts.Level >= level
|
|
}
|
|
|
|
func (l *serviceLogger) Info(ctx context.Context, args ...interface{}) {
|
|
l.Log(ctx, logger.InfoLevel, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Error(ctx context.Context, args ...interface{}) {
|
|
l.Log(ctx, logger.ErrorLevel, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Warn(ctx context.Context, args ...interface{}) {
|
|
l.Log(ctx, logger.WarnLevel, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Debug(ctx context.Context, args ...interface{}) {
|
|
l.Log(ctx, logger.DebugLevel, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Trace(ctx context.Context, args ...interface{}) {
|
|
l.Log(ctx, logger.TraceLevel, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Fatal(ctx context.Context, args ...interface{}) {
|
|
l.Log(ctx, logger.FatalLevel, args...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func (l *serviceLogger) Infof(ctx context.Context, msg string, args ...interface{}) {
|
|
l.Logf(ctx, logger.InfoLevel, msg, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Errorf(ctx context.Context, msg string, args ...interface{}) {
|
|
l.Logf(ctx, logger.ErrorLevel, msg, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Warnf(ctx context.Context, msg string, args ...interface{}) {
|
|
l.Logf(ctx, logger.WarnLevel, msg, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Debugf(ctx context.Context, msg string, args ...interface{}) {
|
|
l.Logf(ctx, logger.DebugLevel, msg, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Tracef(ctx context.Context, msg string, args ...interface{}) {
|
|
l.Logf(ctx, logger.TraceLevel, msg, args...)
|
|
}
|
|
|
|
func (l *serviceLogger) Fatalf(ctx context.Context, msg string, args ...interface{}) {
|
|
l.Logf(ctx, logger.FatalLevel, msg, args...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func (l *serviceLogger) Log(ctx context.Context, level logger.Level, args ...interface{}) {
|
|
msg := l.newMessage(level, "", args...)
|
|
if _, err := l.client.Log(ctx, msg); err != nil {
|
|
_ = l.storeMessage(ctx, msg)
|
|
}
|
|
}
|
|
|
|
func (l *serviceLogger) Logf(ctx context.Context, level logger.Level, format string, args ...interface{}) {
|
|
msg := l.newMessage(level, "", args...)
|
|
if _, err := l.client.Log(ctx, msg); err != nil {
|
|
_ = l.storeMessage(ctx, msg)
|
|
}
|
|
}
|
|
|
|
func (l *serviceLogger) String() string {
|
|
return "service"
|
|
}
|
|
|
|
func (l *serviceLogger) Options() logger.Options {
|
|
return l.opts
|
|
}
|
|
|
|
// NewLogger builds a new logger based on options
|
|
func NewLogger(opts ...logger.Option) logger.Logger {
|
|
options := logger.NewOptions(opts...)
|
|
l := &serviceLogger{opts: options}
|
|
return l
|
|
}
|
|
|
|
func (l *serviceLogger) newMessage(level logger.Level, format string, args ...interface{}) *pb.LogReq {
|
|
msg := &pb.LogReq{Level: int32(level), Format: format, Fields: make([]*pb.Field, 0, len(l.opts.Fields)/2)}
|
|
for idx := 0; idx < len(l.opts.Fields); idx += 2 {
|
|
msg.Fields = append(msg.Fields, &pb.Field{
|
|
Key: fmt.Sprintf("%v", l.opts.Fields[idx]),
|
|
Val: fmt.Sprintf("%v", l.opts.Fields[idx+1]),
|
|
})
|
|
}
|
|
return msg
|
|
}
|
|
|
|
func (l *serviceLogger) storeMessage(ctx context.Context, msg *pb.LogReq) error {
|
|
if l.store == nil {
|
|
return nil
|
|
}
|
|
uid, err := uuid.NewRandom()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return l.store.Write(ctx, uid.String(), msg)
|
|
}
|