output.go

package utils

import (
	"io"
	"log/slog"
	"os"
	"sync"
)

func InitLogger(level slog.Level) {
	slog.SetDefault(slog.New(slog.NewTextHandler(Stdout, &slog.HandlerOptions{
		ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
			// remove time
			if a.Key == "time" {
				return slog.Attr{}
			}
			return a
		},
		Level: level,
	})))
}

type SyncOutput struct {
	sync.Mutex
	output io.Writer
}

func (s *SyncOutput) Write(p []byte) (n int, err error) {
	s.Lock()
	defer s.Unlock()
	return s.output.Write(p)
}

func (s *SyncOutput) SetOutput(output io.Writer) {
	s.Lock()
	defer s.Unlock()
	s.output = output
}

var Stdout = &SyncOutput{
	Mutex:  sync.Mutex{},
	output: os.Stderr,
}

var _ io.Writer = Stdout