-
Notifications
You must be signed in to change notification settings - Fork 278
/
logcore.go
98 lines (84 loc) · 2.56 KB
/
logcore.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
/*
* Copyright 2020 The Dragonfly Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package logcore
import (
"os"
"path/filepath"
"strings"
"d7y.io/dragonfly/v2/pkg/unit"
"d7y.io/dragonfly/v2/pkg/util/fileutils"
"d7y.io/dragonfly/v2/pkg/util/fileutils/filerw"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"gopkg.in/natefinch/lumberjack.v2"
)
const (
CoreLogFileName = "core.log"
GrpcLogFileName = "grpc.log"
GCLogFileName = "gc.log"
)
var coreLevel = zap.NewAtomicLevelAt(zapcore.InfoLevel)
var grpcLevel = zap.NewAtomicLevelAt(zapcore.WarnLevel)
func CreateLogger(filePath string, maxSize int, maxAge int, maxBackups int, compress bool, stats bool) (*zap.Logger, error) {
var syncer zapcore.WriteSyncer
if maxAge < 0 || maxBackups < 0 {
if err := fileutils.MkdirAll(filepath.Dir(filePath)); err != nil {
return nil, err
}
fileInfo, err := os.Stat(filePath)
if err == nil && fileInfo.Size() >= int64(maxSize)*unit.MB.ToNumber() {
_, _ = filerw.CopyFile(filePath, filePath+".old")
_ = filerw.CleanFile(filePath)
}
if syncer, _, err = zap.Open(filePath); err != nil {
return nil, err
}
} else {
rotateConfig := &lumberjack.Logger{
Filename: filePath,
MaxSize: maxSize,
MaxAge: maxAge,
MaxBackups: maxBackups,
LocalTime: true,
Compress: compress,
}
syncer = zapcore.AddSync(rotateConfig)
}
encoderConfig := zap.NewProductionEncoderConfig()
encoderConfig.EncodeTime = zapcore.TimeEncoderOfLayout("2006-01-02 15:04:05.000")
level := zap.NewAtomicLevel()
if strings.HasSuffix(filePath, GrpcLogFileName) {
level = grpcLevel
} else if strings.HasSuffix(filePath, CoreLogFileName) {
level = coreLevel
}
core := zapcore.NewCore(
zapcore.NewJSONEncoder(encoderConfig),
syncer,
level,
)
var opts []zap.Option
if !stats {
opts = append(opts, zap.AddCaller(), zap.AddStacktrace(zap.WarnLevel), zap.AddCallerSkip(1))
}
return zap.New(core, opts...), nil
}
func SetCoreLevel(level zapcore.Level) {
coreLevel.SetLevel(level)
}
func SetGrpcLevel(level zapcore.Level) {
grpcLevel.SetLevel(level)
}