diff --git a/cmd/cmd.go b/cmd/cmd.go index 3753551..4ee9dc1 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -80,6 +80,8 @@ func init() { flagPkg.Token = os.Getenv("GITEA_ACCESS_TOKEN") } + flagPkg.Mode = transport + if debug { flagPkg.Debug = debug } diff --git a/pkg/flag/flag.go b/pkg/flag/flag.go index bfe27c5..20b032a 100644 --- a/pkg/flag/flag.go +++ b/pkg/flag/flag.go @@ -5,6 +5,7 @@ var ( Port int Token string Version string + Mode string Debug bool ) diff --git a/pkg/log/log.go b/pkg/log/log.go index 7ee3021..a7c4793 100644 --- a/pkg/log/log.go +++ b/pkg/log/log.go @@ -39,6 +39,10 @@ func Default() *zap.Logger { MaxAge: 30, })) + if flag.Mode == "sse" { + wss = append(wss, zapcore.AddSync(os.Stdout)) + } + ws = zapcore.NewMultiWriteSyncer(wss...) enc := zapcore.NewConsoleEncoder(ec)