black-internet-days-ahead/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2024-11-24 09:31:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Append time to session log

Browse source
This commit is contained in:
世界 2023-03-29 13:07:08 +08:00
parent b3fb86d415
commit 187421c754
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
2 changed files with 34 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
log/format.go
View file

@ -36,15 +36,16 @@ func (f Formatter) Format(ctx context.Context, level Level, tag string, message
if tag != "" { if tag != "" {
message = tag + ": " + message message = tag + ": " + message
} }
var id uint32 var id ID
var hasId bool var hasId bool
if ctx != nil { if ctx != nil {
id, hasId = IDFromContext(ctx) id, hasId = IDFromContext(ctx)
} }
if hasId { if hasId {
activeDuration := formatDuration(time.Since(id.CreatedAt))
if !f.DisableColors { if !f.DisableColors {
var color aurora.Color var color aurora.Color
color = aurora.Color(uint8(id)) color = aurora.Color(uint8(id.ID))
color %= 215 color %= 215
row := uint(color / 36) row := uint(color / 36)
column := uint(color % 36) column := uint(color % 36)
@ -62,9 +63,9 @@ func (f Formatter) Format(ctx context.Context, level Level, tag string, message
color += 16 color += 16
color = color << 16 color = color << 16
color |= 1 << 14 color |= 1 << 14
message = F.ToString("[", aurora.Colorize(id, color).String(), "] ", message) message = F.ToString("[", aurora.Colorize(id.ID, color).String(), " ", activeDuration, "] ", message)
} else { } else {
message = F.ToString("[", id, "] ", message) message = F.ToString("[", id.ID, " ", activeDuration, "] ", message)
} }
} }
switch { switch {
@ -99,15 +100,16 @@ func (f Formatter) FormatWithSimple(ctx context.Context, level Level, tag string
message = tag + ": " + message message = tag + ": " + message
} }
messageSimple := message messageSimple := message
var id uint32 var id ID
var hasId bool var hasId bool
if ctx != nil { if ctx != nil {
id, hasId = IDFromContext(ctx) id, hasId = IDFromContext(ctx)
} }
if hasId { if hasId {
activeDuration := formatDuration(time.Since(id.CreatedAt))
if !f.DisableColors { if !f.DisableColors {
var color aurora.Color var color aurora.Color
color = aurora.Color(uint8(id)) color = aurora.Color(uint8(id.ID))
color %= 215 color %= 215
row := uint(color / 36) row := uint(color / 36)
column := uint(color % 36) column := uint(color % 36)
@ -125,11 +127,11 @@ func (f Formatter) FormatWithSimple(ctx context.Context, level Level, tag string
color += 16 color += 16
color = color << 16 color = color << 16
color |= 1 << 14 color |= 1 << 14
message = F.ToString("[", aurora.Colorize(id, color).String(), "] ", message) message = F.ToString("[", aurora.Colorize(id.ID, color).String(), " ", activeDuration, "] ", message)
} else { } else {
message = F.ToString("[", id, "] ", message) message = F.ToString("[", id.ID, " ", activeDuration, "] ", message)
} }
messageSimple = F.ToString("[", id, "] ", messageSimple) messageSimple = F.ToString("[", id.ID, " ", activeDuration, "] ", messageSimple)
} }
switch { switch {
@ -153,3 +155,13 @@ func xd(value int, x int) string {
} }
return message return message
} }
func formatDuration(duration time.Duration) string {
if duration < time.Second {
return F.ToString(duration.Milliseconds(), "ms")
} else if duration < time.Minute {
return F.ToString(int64(duration.Seconds()), ".", int64(duration.Seconds()*100)%100, "s")
} else {
return F.ToString(int64(duration.Minutes()), "m", int64(duration.Seconds())%60, "s")
}
}

17
log/id.go
View file

@ -3,6 +3,7 @@ package log
import ( import (
"context" "context"
"math/rand" "math/rand"
"time"
"github.com/sagernet/sing/common/random" "github.com/sagernet/sing/common/random"
) )
@ -13,11 +14,19 @@ func init() {
type idKey struct{} type idKey struct{}
func ContextWithNewID(ctx context.Context) context.Context { type ID struct {
return context.WithValue(ctx, (*idKey)(nil), rand.Uint32()) ID uint32
CreatedAt time.Time
} }
func IDFromContext(ctx context.Context) (uint32, bool) { func ContextWithNewID(ctx context.Context) context.Context {
id, loaded := ctx.Value((*idKey)(nil)).(uint32) return context.WithValue(ctx, (*idKey)(nil), ID{
ID: rand.Uint32(),
CreatedAt: time.Now(),
})
}
func IDFromContext(ctx context.Context) (ID, bool) {
id, loaded := ctx.Value((*idKey)(nil)).(ID)
return id, loaded return id, loaded
} }