Add 'ocis-pkg/' from commit '72d605ba3857d0b972ddd72e226d8a5360fb480d'

git-subtree-dir: ocis-pkg
git-subtree-mainline: 4c12bed11b
git-subtree-split: 72d605ba38

ocis-pkg/log/log.go

@@ -0,0 +1,172 @@
+package log
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	mdlog "github.com/micro/go-micro/v2/debug/log"
+	mlog "github.com/micro/go-micro/v2/util/log"
+	"github.com/micro/go-micro/v2/util/ring"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+)
+
+// Logger simply wraps the zerolog logger.
+type Logger struct {
+	zerolog.Logger
+}
+
+// NewLogger initializes a new logger instance.
+func NewLogger(opts ...Option) Logger {
+	options := newOptions(opts...)
+
+	switch strings.ToLower(options.Level) {
+	case "panic":
+		zerolog.SetGlobalLevel(zerolog.PanicLevel)
+		mlog.SetLevel(mlog.LevelFatal)
+	case "fatal":
+		zerolog.SetGlobalLevel(zerolog.FatalLevel)
+		mlog.SetLevel(mlog.LevelFatal)
+	case "error":
+		zerolog.SetGlobalLevel(zerolog.ErrorLevel)
+		mlog.SetLevel(mlog.LevelError)
+	case "warn":
+		zerolog.SetGlobalLevel(zerolog.WarnLevel)
+		mlog.SetLevel(mlog.LevelWarn)
+	case "info":
+		zerolog.SetGlobalLevel(zerolog.InfoLevel)
+		mlog.SetLevel(mlog.LevelInfo)
+	case "debug":
+		zerolog.SetGlobalLevel(zerolog.DebugLevel)
+		mlog.SetLevel(mlog.LevelDebug)
+	case "trace":
+		zerolog.SetGlobalLevel(zerolog.DebugLevel)
+		mlog.SetLevel(mlog.LevelTrace)
+	default:
+		zerolog.SetGlobalLevel(zerolog.InfoLevel)
+		mlog.SetLevel(mlog.LevelInfo)
+	}
+
+	var logger zerolog.Logger
+
+	if options.Pretty {
+		logger = log.Output(
+			zerolog.NewConsoleWriter(
+				func(w *zerolog.ConsoleWriter) {
+					w.TimeFormat = time.RFC3339
+					w.Out = os.Stderr
+					w.NoColor = !options.Color
+				},
+			),
+		)
+	} else {
+		logger = zerolog.New(os.Stderr)
+	}
+
+	logger = logger.With().
+		Str("service", options.Name).
+		Timestamp().
+		Logger()
+
+	mlog.SetLogger(
+		microZerolog{
+			logger: logger,
+			buffer: ring.New(mdlog.DefaultSize),
+		},
+	)
+
+	return Logger{
+		logger,
+	}
+}
+
+// microZerolog implements the required interface for the go-micro logger.
+type microZerolog struct {
+	logger zerolog.Logger
+	buffer *ring.Buffer
+}
+
+func (mz microZerolog) Read(opts ...mdlog.ReadOption) ([]mdlog.Record, error) {
+	options := mdlog.ReadOptions{}
+	for _, o := range opts {
+		o(&options)
+	}
+
+	var entries []*ring.Entry
+
+	if !options.Since.IsZero() {
+		entries = mz.buffer.Since(options.Since)
+	}
+
+	if options.Count > 0 {
+		switch len(entries) > 0 {
+		case true:
+			if options.Count > len(entries) {
+				entries = entries[0:options.Count]
+			}
+		default:
+			entries = mz.buffer.Get(options.Count)
+		}
+	}
+
+	records := make([]mdlog.Record, 0, len(entries))
+	for _, entry := range entries {
+		record := mdlog.Record{
+			Timestamp: entry.Timestamp,
+			Message:   entry.Value,
+		}
+		records = append(records, record)
+	}
+
+	return records, nil
+}
+
+func (mz microZerolog) Write(record mdlog.Record) error {
+	level := record.Metadata["level"]
+	mz.log(level, fmt.Sprint(record.Message))
+	mz.buffer.Put(record.Message)
+	return nil
+}
+
+func (mz microZerolog) Stream() (mdlog.Stream, error) {
+	stream, stop := mz.buffer.Stream()
+	records := make(chan mdlog.Record, 128)
+	last10 := mz.buffer.Get(10)
+
+	go func() {
+		for _, entry := range last10 {
+			records <- mdlog.Record{
+				Timestamp: entry.Timestamp,
+				Message:   entry.Value,
+				Metadata:  make(map[string]string),
+			}
+		}
+		for entry := range stream {
+			records <- mdlog.Record{
+				Timestamp: entry.Timestamp,
+				Message:   entry.Value,
+				Metadata:  make(map[string]string),
+			}
+		}
+	}()
+	return &logStream{
+		stream: records,
+		stop:   stop,
+	}, nil
+}
+
+func (mz microZerolog) log(level string, msg string) {
+	l, err := zerolog.ParseLevel(level)
+	if err != nil {
+		l = zerolog.InfoLevel
+	}
+
+	mz.logger.WithLevel(l).Msg(msg)
+
+	// Invoke os.Exit because unlike zerolog.Logger.Fatal zerolog.Logger.WithLevel won't stop the execution.
+	if l == zerolog.FatalLevel {
+		os.Exit(1)
+	}
+}

ocis-pkg/log/option.go

@@ -0,0 +1,56 @@
+package log
+
+// Option defines a single option function.
+type Option func(o *Options)
+
+// Options defines the available options for this package.
+type Options struct {
+	Name   string
+	Level  string
+	Pretty bool
+	Color  bool
+}
+
+// newOptions initializes the available default options.
+func newOptions(opts ...Option) Options {
+	opt := Options{
+		Name:   "ocis",
+		Level:  "info",
+		Pretty: true,
+		Color:  true,
+	}
+
+	for _, o := range opts {
+		o(&opt)
+	}
+
+	return opt
+}
+
+// Name provides a function to set the name option.
+func Name(val string) Option {
+	return func(o *Options) {
+		o.Name = val
+	}
+}
+
+// Level provides a function to set the level option.
+func Level(val string) Option {
+	return func(o *Options) {
+		o.Level = val
+	}
+}
+
+// Pretty provides a function to set the pretty option.
+func Pretty(val bool) Option {
+	return func(o *Options) {
+		o.Pretty = val
+	}
+}
+
+// Color provides a function to set the color option.
+func Color(val bool) Option {
+	return func(o *Options) {
+		o.Color = val
+	}
+}

ocis-pkg/log/stream.go

@@ -0,0 +1,24 @@
+package log
+
+import (
+	mdlog "github.com/micro/go-micro/v2/debug/log"
+)
+
+type logStream struct {
+	stream <-chan mdlog.Record
+	stop   chan bool
+}
+
+func (l *logStream) Chan() <-chan mdlog.Record {
+	return l.stream
+}
+
+func (l *logStream) Stop() error {
+	select {
+	case <-l.stop:
+		return nil
+	default:
+		close(l.stop)
+	}
+	return nil
+}