buyfakett/server_torii
1
0
Fork 0
mirror of https://github.com/Rayzggz/server_torii.git synced 2025-06-16 20:21:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: global logger

Browse Source
This commit is contained in:
Roi Feng
2025-04-28 22:08:28 -04:00
parent af4af7346a
commit 705230dd0d
2 changed files with 27 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
internal/utils/logx.go
View File

@ -19,7 +19,13 @@ type LogxManager struct {
mu sync.RWMutex mu sync.RWMutex
} }
func NewManager(base string) *LogxManager { var logger *LogxManager
func InitLogx(base string) {
logger = newManager(base)
}
func newManager(base string) *LogxManager {
m := &LogxManager{basePath: base, loggers: make(map[string]*zap.Logger)} m := &LogxManager{basePath: base, loggers: make(map[string]*zap.Logger)}
if err := os.MkdirAll(m.basePath, 0744); err != nil { if err := os.MkdirAll(m.basePath, 0744); err != nil {
@ -72,7 +78,12 @@ func (m *LogxManager) openLogFile(path string) *os.File {
return f return f
} }
func (m *LogxManager) LogInfo(reqData dataType.UserRequest, msg, msg2 string) { func LogInfo(reqData dataType.UserRequest, msg, msg2 string) {
m := logger
if m == nil {
log.Printf("LogxManager is not initialized")
return
}
lg := m.getLogger(reqData.Host) lg := m.getLogger(reqData.Host)
line := fmt.Sprintf("%s - - [%s] %s %s %s %s %s", line := fmt.Sprintf("%s - - [%s] %s %s %s %s %s",
reqData.RemoteIP, reqData.RemoteIP,
@ -86,7 +97,12 @@ func (m *LogxManager) LogInfo(reqData dataType.UserRequest, msg, msg2 string) {
lg.Info(line) lg.Info(line)
} }
func (m *LogxManager) LogError(reqData dataType.UserRequest, msg, msg2 string) { func LogError(reqData dataType.UserRequest, msg, msg2 string) {
m := logger
if m == nil {
log.Printf("LogxManager is not initialized")
return
}
lg := m.getLogger(reqData.Host) lg := m.getLogger(reqData.Host)
line := fmt.Sprintf("%s - - [%s] %s %s %s %s %s", line := fmt.Sprintf("%s - - [%s] %s %s %s %s %s",
reqData.RemoteIP, reqData.RemoteIP,
@ -100,7 +116,12 @@ func (m *LogxManager) LogError(reqData dataType.UserRequest, msg, msg2 string) {
lg.Error(line) lg.Error(line)
} }
func (m *LogxManager) LogDebug(reqData dataType.UserRequest, msg, msg2 string) { func LogDebug(reqData dataType.UserRequest, msg, msg2 string) {
m := logger
if m == nil {
log.Printf("LogxManager is not initialized")
return
}
lg := m.getLogger(reqData.Host) lg := m.getLogger(reqData.Host)
line := fmt.Sprintf("%s - - [%s] %s %s %s %s %s", line := fmt.Sprintf("%s - - [%s] %s %s %s %s %s",
reqData.RemoteIP, reqData.RemoteIP,

2
main.go
View File

@ -52,6 +52,8 @@ func main() {
}(logFile) }(logFile)
log.SetOutput(logFile) log.SetOutput(logFile)
utils.InitLogx(cfg.LogPath)
//allocate shared memory //allocate shared memory
sharedMem := &dataType.SharedMemory{ sharedMem := &dataType.SharedMemory{
HTTPFloodSpeedLimitCounter: dataType.NewCounter(max(runtime.NumCPU()*8, 16), utils.FindMaxRateTime(ruleSet.HTTPFloodRule.HTTPFloodSpeedLimit)), HTTPFloodSpeedLimitCounter: dataType.NewCounter(max(runtime.NumCPU()*8, 16), utils.FindMaxRateTime(ruleSet.HTTPFloodRule.HTTPFloodSpeedLimit)),