93 lines
2.7 KiB
Go
93 lines
2.7 KiB
Go
package gormx
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"strings"
|
||
"sync"
|
||
"testing"
|
||
"time"
|
||
|
||
"git.yuntuo.hk/k2safe/go-omni/logx"
|
||
"gorm.io/gorm"
|
||
gormlog "gorm.io/gorm/logger"
|
||
)
|
||
|
||
func TestSQLLoggerLevelsPrivacyAndCorrelation(t *testing.T) {
|
||
var out bytes.Buffer
|
||
base, _ := logx.New(logx.Config{Writer: &out})
|
||
l, _ := NewLogger(base, DefaultLogConfig())
|
||
ctx := logx.WithRequestID(logx.WithTraceID(context.Background(), "trace-1"), "request-1")
|
||
count := 0
|
||
fc := func() (string, int64) { count++; return "SELECT * FROM records WHERE token = ?", 1 }
|
||
l.Trace(ctx, time.Now(), fc, nil)
|
||
l.Trace(ctx, time.Now(), fc, fmt.Errorf("wrapped: %w", gorm.ErrRecordNotFound))
|
||
l.LogMode(gormlog.Silent).Trace(ctx, time.Now(), fc, errors.New("password-value"))
|
||
if count != 0 {
|
||
t.Fatal("ignored event evaluated SQL")
|
||
}
|
||
l.Trace(ctx, time.Now().Add(-time.Second), fc, nil)
|
||
l.Trace(ctx, time.Now(), fc, errors.New("password-value"))
|
||
l.Error(ctx, "failed: %v", errors.New("password-value"))
|
||
l.LogMode(gormlog.Info).Trace(ctx, time.Now(), fc, nil)
|
||
if count != 3 {
|
||
t.Fatal(count)
|
||
}
|
||
if strings.Contains(out.String(), "password-value") {
|
||
t.Fatal("driver detail leaked", out.String())
|
||
}
|
||
for _, text := range []string{"gorm.slowQuery", "gorm.queryFailed", "gorm.query", `"requestId":"request-1"`, `"traceId":"trace-1"`} {
|
||
if !strings.Contains(out.String(), text) {
|
||
t.Fatal(text, out.String())
|
||
}
|
||
}
|
||
q, args := l.(gorm.ParamsFilter).ParamsFilter(ctx, "SELECT ?", "private")
|
||
if q != "SELECT ?" || len(args) != 0 {
|
||
t.Fatal("default SQL arguments exposed")
|
||
}
|
||
// LogMode clones rather than mutating the shared instance.
|
||
l.Trace(ctx, time.Now(), fc, nil)
|
||
if count != 3 {
|
||
t.Fatal("session mode changed root logger")
|
||
}
|
||
}
|
||
|
||
func TestSQLLoggerExplicitDetailsAndConcurrentModes(t *testing.T) {
|
||
var out bytes.Buffer
|
||
base, _ := logx.New(logx.Config{Writer: &out})
|
||
cfg := DefaultLogConfig()
|
||
cfg.IncludeParameters = true
|
||
cfg.IncludeErrorDetails = true
|
||
cfg.Level = "info"
|
||
cfg.MaxSQLBytes = 8
|
||
l, err := NewLogger(base, cfg)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
_, args := l.(gorm.ParamsFilter).ParamsFilter(context.Background(), "SELECT ?", "value")
|
||
if len(args) != 1 {
|
||
t.Fatal("explicit arguments dropped")
|
||
}
|
||
var wg sync.WaitGroup
|
||
for i := 0; i < 20; i++ {
|
||
wg.Add(1)
|
||
go func() {
|
||
defer wg.Done()
|
||
l.LogMode(gormlog.Error).Trace(context.Background(), time.Now(), func() (string, int64) { return "SELECT 中文标识符", -1 }, errors.New("failure"))
|
||
}()
|
||
}
|
||
wg.Wait()
|
||
for _, line := range strings.Split(strings.TrimSpace(out.String()), "\n") {
|
||
var event map[string]any
|
||
if err = json.Unmarshal([]byte(line), &event); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if event["error"] != "failure" || strings.Contains(event["sql"].(string), "<22>") {
|
||
t.Fatal(event)
|
||
}
|
||
}
|
||
}
|