主要更新: 1. 数据库持久化功能 - 支持三种策略:仅落库、既落库又存证、仅存证 - 实现 Cursor Worker 异步扫描和存证机制 - 实现 Retry Worker 失败重试机制 - 支持 PostgreSQL、MySQL、SQLite 等多种数据库 - 添加 ClientIP 和 ServerIP 字段(可空,仅落库) 2. 集群并发安全 - 使用 SELECT FOR UPDATE SKIP LOCKED 防止重复处理 - 实现 CAS (Compare-And-Set) 原子状态更新 - 添加 updated_at 字段支持并发控制 3. Cursor 初始化优化 - 自动基于历史数据初始化 cursor - 确保不遗漏任何历史记录 - 修复 UPSERT 逻辑 4. 测试完善 - 添加 E2E 集成测试(含 Pulsar 消费者验证) - 添加 PostgreSQL 集成测试 - 添加 Pulsar 集成测试 - 添加集群并发安全测试 - 添加 Cursor 初始化验证测试 - 补充大量单元测试,提升覆盖率 5. 工具脚本 - 添加数据库迁移脚本 - 添加 Cursor 状态检查工具 - 添加 Cursor 初始化工具 - 添加 Pulsar 消息验证工具 6. 文档清理 - 删除冗余文档,只保留根目录 README 测试结果: - 所有 E2E 测试通过(100%) - 数据库持久化与异步存证流程验证通过 - 集群环境下的并发安全性验证通过 - Cursor 自动初始化和历史数据处理验证通过
256 lines
4.6 KiB
Go
256 lines
4.6 KiB
Go
package logger_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/go-logr/logr"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.yandata.net/iod/iod/go-trustlog/api/logger"
|
|
)
|
|
|
|
func TestNewLogger(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
result := logger.NewLogger(l)
|
|
assert.NotNil(t, result)
|
|
}
|
|
|
|
func TestNewDefaultLogger(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
result := logger.NewDefaultLogger()
|
|
assert.NotNil(t, result)
|
|
}
|
|
|
|
func TestNewNopLogger(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
result := logger.NewNopLogger()
|
|
assert.NotNil(t, result)
|
|
}
|
|
|
|
func TestNopLogger_Methods(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
nop := logger.NewNopLogger()
|
|
ctx := context.Background()
|
|
|
|
// All methods should not panic
|
|
assert.NotPanics(t, func() {
|
|
nop.DebugContext(ctx, "test")
|
|
nop.InfoContext(ctx, "test")
|
|
nop.WarnContext(ctx, "test")
|
|
nop.ErrorContext(ctx, "test")
|
|
nop.Debug("test")
|
|
nop.Info("test")
|
|
nop.Warn("test")
|
|
nop.Error("test")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_DebugContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
ctx := context.Background()
|
|
|
|
assert.NotPanics(t, func() {
|
|
adapter.DebugContext(ctx, "debug message", "key", "value")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_InfoContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
ctx := context.Background()
|
|
|
|
assert.NotPanics(t, func() {
|
|
adapter.InfoContext(ctx, "info message", "key", "value")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_WarnContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
ctx := context.Background()
|
|
|
|
assert.NotPanics(t, func() {
|
|
adapter.WarnContext(ctx, "warn message", "key", "value")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_ErrorContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
ctx := context.Background()
|
|
|
|
tests := []struct {
|
|
name string
|
|
args []any
|
|
}{
|
|
{
|
|
name: "with error",
|
|
args: []any{"error", errors.New("test error"), "key", "value"},
|
|
},
|
|
{
|
|
name: "without error",
|
|
args: []any{"key", "value"},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
assert.NotPanics(t, func() {
|
|
adapter.ErrorContext(ctx, "error message", tt.args...)
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestLogrAdapter_Debug(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
|
|
assert.NotPanics(t, func() {
|
|
adapter.Debug("debug message", "key", "value")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_Info(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
|
|
assert.NotPanics(t, func() {
|
|
adapter.Info("info message", "key", "value")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_Warn(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
|
|
assert.NotPanics(t, func() {
|
|
adapter.Warn("warn message", "key", "value")
|
|
})
|
|
}
|
|
|
|
func TestLogrAdapter_Error(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
l := logr.Discard()
|
|
adapter := logger.NewLogger(l)
|
|
|
|
tests := []struct {
|
|
name string
|
|
args []any
|
|
}{
|
|
{
|
|
name: "with error",
|
|
args: []any{"error", errors.New("test error"), "key", "value"},
|
|
},
|
|
{
|
|
name: "without error",
|
|
args: []any{"key", "value"},
|
|
},
|
|
{
|
|
name: "odd number of args",
|
|
args: []any{"key", "value", "extra"},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
assert.NotPanics(t, func() {
|
|
adapter.Error("error message", tt.args...)
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSetGlobalLogger(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
original := logger.GetGlobalLogger()
|
|
defer logger.SetGlobalLogger(original)
|
|
|
|
newLogger := logger.NewNopLogger()
|
|
logger.SetGlobalLogger(newLogger)
|
|
|
|
result := logger.GetGlobalLogger()
|
|
assert.Equal(t, newLogger, result)
|
|
}
|
|
|
|
func TestSetGlobalLogger_Nil(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
original := logger.GetGlobalLogger()
|
|
defer logger.SetGlobalLogger(original)
|
|
|
|
logger.SetGlobalLogger(nil)
|
|
result := logger.GetGlobalLogger()
|
|
assert.NotNil(t, result) // Should be NopLogger
|
|
}
|
|
|
|
func TestGetGlobalLogger(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
result := logger.GetGlobalLogger()
|
|
assert.NotNil(t, result)
|
|
}
|
|
|
|
func TestGlobalLogger_ConcurrentAccess(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
original := logger.GetGlobalLogger()
|
|
defer logger.SetGlobalLogger(original)
|
|
|
|
// Test concurrent reads
|
|
done := make(chan bool, 10)
|
|
for range 10 {
|
|
go func() {
|
|
_ = logger.GetGlobalLogger()
|
|
done <- true
|
|
}()
|
|
}
|
|
|
|
for range 10 {
|
|
<-done
|
|
}
|
|
|
|
// Test concurrent writes
|
|
newLogger := logger.NewNopLogger()
|
|
for range 5 {
|
|
go func() {
|
|
logger.SetGlobalLogger(newLogger)
|
|
done <- true
|
|
}()
|
|
}
|
|
|
|
for range 5 {
|
|
<-done
|
|
}
|
|
|
|
result := logger.GetGlobalLogger()
|
|
require.NotNil(t, result)
|
|
}
|