m7s/m7s.live/engine/v4@v4.15.2-0.20240315012939-f677b9885c3c/log/writer.go

package log

import (
	"io"
	"os"
	"sync"
)

type MultipleWriter struct {
	io.Writer // 默认输出到标准输出
	sync.Map  // 用于存储多个输出
}

func (m *MultipleWriter) Write(p []byte) (n int, err error) {
	n, err = m.Writer.Write(p)
	m.Range(func(key, value any) bool {
		if _, err := key.(io.Writer).Write(p); err != nil {
			m.Delete(key)
		}
		return true
	})
	return
}

func (m *MultipleWriter) Add(writer io.Writer) {
	m.Map.Store(writer, struct{}{})
}

var multipleWriter = &MultipleWriter{Writer: os.Stdout}

func AddWriter(writer io.Writer) {
	multipleWriter.Add(writer)
}
func DeleteWriter(writer io.Writer) {
	multipleWriter.Delete(writer)
}