mirror of
https://github.com/livekit/livekit.git
synced 2026-04-01 10:55:45 +00:00
faa870de3d
* move callbacks out of messageRouter * OCD * more OCD * fix forwarder test * even more OCD * maximum OCD * package name collision, copy lock by value
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package serverlogger
|
|
|
|
import (
|
|
"github.com/go-logr/logr"
|
|
"github.com/go-logr/zapr"
|
|
"github.com/livekit/protocol/logger"
|
|
"github.com/pion/logging"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
|
|
"github.com/livekit/livekit-server/pkg/sfu"
|
|
"github.com/livekit/livekit-server/pkg/sfu/buffer"
|
|
)
|
|
|
|
var (
|
|
// pion/webrtc, pion/turn
|
|
defaultFactory logging.LoggerFactory
|
|
)
|
|
|
|
func LoggerFactory() logging.LoggerFactory {
|
|
if defaultFactory == nil {
|
|
defaultFactory = logging.NewDefaultLoggerFactory()
|
|
}
|
|
return defaultFactory
|
|
}
|
|
|
|
// Note: only pass in logr.Logger with default depth
|
|
func SetLogger(l logr.Logger) {
|
|
logger.SetLogger(l, "livekit")
|
|
sfu.Logger = l.WithName("sfu")
|
|
buffer.Logger = sfu.Logger
|
|
}
|
|
|
|
func InitProduction(logLevel string) {
|
|
initLogger(zap.NewProductionConfig(), logLevel)
|
|
}
|
|
|
|
func InitDevelopment(logLevel string) {
|
|
initLogger(zap.NewDevelopmentConfig(), logLevel)
|
|
}
|
|
|
|
// valid levels: debug, info, warn, error, fatal, panic
|
|
func initLogger(config zap.Config, level string) {
|
|
if level != "" {
|
|
lvl := zapcore.Level(0)
|
|
if err := lvl.UnmarshalText([]byte(level)); err == nil {
|
|
config.Level = zap.NewAtomicLevelAt(lvl)
|
|
}
|
|
}
|
|
|
|
l, _ := config.Build()
|
|
zapLogger := zapr.NewLogger(l)
|
|
SetLogger(zapLogger)
|
|
}
|