fix: 修复应用日志异常膨胀

This commit is contained in:
ryan
2026-06-21 10:21:18 +08:00
parent a343c7a605
commit b12a9b0185
12 changed files with 358 additions and 53 deletions
+34
View File
@@ -6,6 +6,7 @@ package model
import (
"context"
"errors"
"strings"
"time"
"github.com/Rain-kl/Wavelet/internal/db"
@@ -83,6 +84,39 @@ func CountOpenFlareApplyLogs(ctx context.Context, nodeID string) (int64, error)
return total, nil
}
// GetLatestOpenFlareApplyLogByNodeID returns the most recent apply log for a node.
func GetLatestOpenFlareApplyLogByNodeID(ctx context.Context, nodeID string) (*OpenFlareApplyLog, error) {
nodeID = strings.TrimSpace(nodeID)
if nodeID == "" {
return nil, errors.New("node_id is required")
}
conn := db.DB(ctx)
if conn == nil {
return nil, errors.New(errDatabaseNotInitialized)
}
var log OpenFlareApplyLog
err := conn.Where("node_id = ?", nodeID).Order("id desc").First(&log).Error
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, nil
}
if err != nil {
return nil, err
}
return &log, nil
}
// IsRepeatSuccessApplyLog reports whether the payload repeats an already-recorded success entry.
func IsRepeatSuccessApplyLog(latest *OpenFlareApplyLog, version, checksum, result string) bool {
if latest == nil || result != "success" {
return false
}
return latest.Result == "success" &&
strings.TrimSpace(latest.Version) == strings.TrimSpace(version) &&
strings.TrimSpace(latest.Checksum) == strings.TrimSpace(checksum)
}
// GetLatestOpenFlareApplyLogsByNodeIDs returns the latest apply log per node id.
func GetLatestOpenFlareApplyLogsByNodeIDs(ctx context.Context, nodeIDs []string) (map[string]*OpenFlareApplyLog, error) {
result := make(map[string]*OpenFlareApplyLog)
@@ -0,0 +1,73 @@
package model
import (
"context"
"testing"
"time"
"github.com/Rain-kl/Wavelet/internal/db"
"github.com/glebarez/sqlite"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gorm.io/gorm"
)
func setupApplyLogModelTestDB(t *testing.T) func() {
t.Helper()
sqliteDB, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{
DisableForeignKeyConstraintWhenMigrating: true,
})
require.NoError(t, err)
require.NoError(t, sqliteDB.AutoMigrate(&OpenFlareApplyLog{}))
db.SetDB(sqliteDB)
return func() {
db.SetDB(nil)
}
}
func TestIsRepeatSuccessApplyLog(t *testing.T) {
latest := &OpenFlareApplyLog{
Version: "20260615-001",
Checksum: "checksum-a",
Result: "success",
}
assert.True(t, IsRepeatSuccessApplyLog(latest, "20260615-001", "checksum-a", "success"))
assert.False(t, IsRepeatSuccessApplyLog(latest, "20260615-002", "checksum-a", "success"))
assert.False(t, IsRepeatSuccessApplyLog(latest, "20260615-001", "checksum-b", "success"))
assert.False(t, IsRepeatSuccessApplyLog(latest, "20260615-001", "checksum-a", "failed"))
assert.False(t, IsRepeatSuccessApplyLog(nil, "20260615-001", "checksum-a", "success"))
}
func TestGetLatestOpenFlareApplyLogByNodeID(t *testing.T) {
cleanup := setupApplyLogModelTestDB(t)
defer cleanup()
ctx := context.Background()
now := time.Now().UTC()
require.NoError(t, db.DB(ctx).Create(&OpenFlareApplyLog{
NodeID: "node-1",
Version: "v1",
Result: "success",
Checksum: "checksum-1",
CreatedAt: now.Add(-time.Hour),
}).Error)
require.NoError(t, db.DB(ctx).Create(&OpenFlareApplyLog{
NodeID: "node-1",
Version: "v2",
Result: "success",
Checksum: "checksum-2",
CreatedAt: now,
}).Error)
latest, err := GetLatestOpenFlareApplyLogByNodeID(ctx, "node-1")
require.NoError(t, err)
require.NotNil(t, latest)
assert.Equal(t, "v2", latest.Version)
missing, err := GetLatestOpenFlareApplyLogByNodeID(ctx, "node-missing")
require.NoError(t, err)
assert.Nil(t, missing)
}