mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-08 00:26:37 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package logstore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPartitionNameMonth(t *testing.T) {
|
|
cases := []struct {
|
|
table string
|
|
name string
|
|
want string
|
|
}{
|
|
{"w_user_access_logs", "w_user_access_logs_202612", "2026-12"},
|
|
{"w_user_access_logs", "w_user_access_logs_202608", "2026-08"},
|
|
{"w_user_access_logs", "of_node_access_logs_202608", ""},
|
|
{"w_user_access_logs", "w_user_access_logs_20268", ""},
|
|
{"w_user_access_logs", "w_user_access_logs_202613", ""},
|
|
{"w_user_access_logs", "w_user_access_logs_default", ""},
|
|
}
|
|
for _, c := range cases {
|
|
got, ok := partitionNameMonth(c.table, c.name)
|
|
if c.want == "" {
|
|
if ok {
|
|
t.Fatalf("partitionNameMonth(%q, %q) ok = true, want false", c.table, c.name)
|
|
}
|
|
continue
|
|
}
|
|
if !ok || got.Format("2006-01") != c.want {
|
|
t.Fatalf("partitionNameMonth(%q, %q) = %v, want %s", c.table, c.name, got, c.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDropEligiblePartitionNames(t *testing.T) {
|
|
before := time.Date(2026, 10, 15, 0, 0, 0, 0, time.UTC)
|
|
names := []string{
|
|
"w_user_access_logs_202608",
|
|
"w_user_access_logs_202609",
|
|
"w_user_access_logs_202610",
|
|
"w_user_access_logs_202611",
|
|
"w_user_access_logs_default",
|
|
}
|
|
got := dropEligiblePartitionNames(userAccessLogTable, names, before)
|
|
want := []string{"w_user_access_logs_202608", "w_user_access_logs_202609"}
|
|
require.Equal(t, want, got)
|
|
|
|
first := time.Date(2026, 10, 1, 0, 0, 0, 0, time.UTC)
|
|
require.Empty(t, dropEligiblePartitionNames(userAccessLogTable, []string{"w_user_access_logs_202610"}, first))
|
|
}
|
|
|
|
func TestDropPartitionHelpersSQLiteNoop(t *testing.T) {
|
|
ua := newTestUserAccessStore(t)
|
|
ctx := context.Background()
|
|
require.NoError(t, ua.BatchInsert(ctx, []UserAccessLog{
|
|
{UserID: 1, Path: "/x", CreatedAt: time.Now().UTC()},
|
|
}))
|
|
require.NoError(t, ua.DropExpiredPartitions(ctx, time.Now().AddDate(0, 0, -90)))
|
|
require.NoError(t, ua.DropEmptyPartitions(ctx, time.Now()))
|
|
count, err := ua.Count(ctx, AccessLogFilter{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(1), count)
|
|
}
|