freefire
429956c594
请求日志按日 JSONL；page_views 不进主库，避免统计数据撑大 jiang13.db。 Co-authored-by: Cursor <cursoragent@cursor.com>
55 lines
1013 B
Go
55 lines
1013 B
Go
package model
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestInitMonitorDBUsesSeparateFile(t *testing.T) {
|
|
dir := t.TempDir()
|
|
mainPath := filepath.Join(dir, "jiang13.db")
|
|
monPath := filepath.Join(dir, "monitor.db")
|
|
|
|
prevDB, prevMon := DB, MonitorDB
|
|
t.Cleanup(func() {
|
|
closeGorm(MonitorDB)
|
|
closeGorm(DB)
|
|
DB, MonitorDB = prevDB, prevMon
|
|
})
|
|
|
|
if err := InitDB(mainPath); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := InitMonitorDB(monPath); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if tableExists(DB, "page_views") {
|
|
t.Fatal("主库不应创建 page_views")
|
|
}
|
|
if !tableExists(MonitorDB, "page_views") {
|
|
t.Fatal("监控库缺少 page_views")
|
|
}
|
|
}
|
|
|
|
func tableExists(db *gorm.DB, name string) bool {
|
|
if db == nil {
|
|
return false
|
|
}
|
|
var n int
|
|
_ = db.Raw("SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name=?", name).Scan(&n).Error
|
|
return n > 0
|
|
}
|
|
|
|
func closeGorm(db *gorm.DB) {
|
|
if db == nil {
|
|
return
|
|
}
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return
|
|
}
|
|
_ = sqlDB.Close()
|
|
}
|