1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-10 01:40:44 +00:00
v2fly/app/log/log.go

123 lines
2.6 KiB
Go
Raw Normal View History

package log
2017-12-03 00:04:57 +00:00
//go:generate go run $GOPATH/src/v2ray.com/core/common/errors/errorgen/main.go -pkg log -path App,Log
2017-04-08 23:43:25 +00:00
import (
2017-02-01 20:35:40 +00:00
"context"
2017-12-19 20:28:12 +00:00
"sync"
2017-02-01 20:35:40 +00:00
"v2ray.com/core/common"
2017-12-10 22:33:23 +00:00
"v2ray.com/core/common/log"
)
2017-12-19 23:09:52 +00:00
// Instance is an app.Application that handles logs.
2017-12-19 20:28:12 +00:00
type Instance struct {
sync.RWMutex
config *Config
2017-12-21 23:41:40 +00:00
accessLogger log.Handler
errorLogger log.Handler
active bool
2017-12-19 20:28:12 +00:00
}
2017-12-19 23:09:52 +00:00
// New creates a new log.Instance based on the given config.
2017-12-19 20:28:12 +00:00
func New(ctx context.Context, config *Config) (*Instance, error) {
2017-12-21 23:41:40 +00:00
g := &Instance{
2017-12-19 20:28:12 +00:00
config: config,
2017-12-21 23:41:40 +00:00
active: true,
}
if err := g.initAccessLogger(); err != nil {
return nil, newError("failed to initialize access logger").Base(err).AtWarning()
}
if err := g.initErrorLogger(); err != nil {
return nil, newError("failed to initialize error logger").Base(err).AtWarning()
}
log.RegisterHandler(g)
return g, nil
2017-12-19 20:28:12 +00:00
}
2017-12-19 23:09:52 +00:00
// Interface implements app.Application.Interface().
2017-12-19 20:28:12 +00:00
func (*Instance) Interface() interface{} {
return (*Instance)(nil)
}
2017-12-19 20:28:12 +00:00
func (g *Instance) initAccessLogger() error {
switch g.config.AccessLogType {
case LogType_File:
2017-12-21 23:41:40 +00:00
creator, err := log.CreateFileLogWriter(g.config.AccessLogPath)
2017-12-19 20:28:12 +00:00
if err != nil {
return err
}
2017-12-21 23:41:40 +00:00
g.accessLogger = log.NewLogger(creator)
2017-12-19 20:28:12 +00:00
case LogType_Console:
2017-12-21 23:41:40 +00:00
g.accessLogger = log.NewLogger(log.CreateStdoutLogWriter())
2017-12-19 20:28:12 +00:00
default:
2015-12-05 20:10:14 +00:00
}
return nil
}
2017-12-19 20:28:12 +00:00
func (g *Instance) initErrorLogger() error {
switch g.config.ErrorLogType {
case LogType_File:
2017-12-21 23:41:40 +00:00
creator, err := log.CreateFileLogWriter(g.config.ErrorLogPath)
2017-12-19 20:28:12 +00:00
if err != nil {
return err
}
2017-12-21 23:41:40 +00:00
g.errorLogger = log.NewLogger(creator)
2017-12-19 20:28:12 +00:00
case LogType_Console:
2017-12-21 23:41:40 +00:00
g.errorLogger = log.NewLogger(log.CreateStdoutLogWriter())
2017-04-06 13:13:09 +00:00
default:
}
2017-12-19 20:28:12 +00:00
return nil
2017-04-06 13:13:09 +00:00
}
2017-12-19 23:09:52 +00:00
// Start implements app.Application.Start().
2017-12-19 20:28:12 +00:00
func (g *Instance) Start() error {
2017-12-21 23:41:40 +00:00
g.Lock()
defer g.Unlock()
g.active = true
2017-12-19 20:28:12 +00:00
return nil
2017-02-01 20:35:40 +00:00
}
2017-12-21 23:41:40 +00:00
func (g *Instance) isActive() bool {
g.RLock()
defer g.RUnlock()
return g.active
}
2017-12-19 23:09:52 +00:00
// Handle implements log.Handler.
2017-12-19 20:28:12 +00:00
func (g *Instance) Handle(msg log.Message) {
2017-12-21 23:41:40 +00:00
if !g.isActive() {
return
}
2017-12-19 20:28:12 +00:00
switch msg := msg.(type) {
case *log.AccessMessage:
if g.accessLogger != nil {
2017-12-21 23:41:40 +00:00
g.accessLogger.Handle(msg)
2017-02-01 20:35:40 +00:00
}
2017-12-19 20:28:12 +00:00
case *log.GeneralMessage:
2017-12-21 23:41:40 +00:00
if g.errorLogger != nil && msg.Severity <= g.config.ErrorLogLevel {
g.errorLogger.Handle(msg)
2017-02-01 20:35:40 +00:00
}
2017-12-19 20:28:12 +00:00
default:
// Swallow
2017-02-01 20:35:40 +00:00
}
}
2017-12-19 23:09:52 +00:00
// Close implement app.Application.Close().
2017-12-19 20:28:12 +00:00
func (g *Instance) Close() {
g.Lock()
defer g.Unlock()
2017-12-21 23:41:40 +00:00
g.active = true
2016-07-10 21:11:42 +00:00
}
2017-02-01 20:35:40 +00:00
func init() {
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
return New(ctx, config.(*Config))
}))
}