xlgo-core
wehub-resource-sync
117 行
4.7 KiB
Go
117 行
4.7 KiB
Go
package xlgo_test
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"database/sql/driver"
|
||
"errors"
|
||
"sync"
|
||
"testing"
|
||
"time"
|
||
|
||
xlgo "github.com/EthanCodeCraft/xlgo-core"
|
||
"github.com/EthanCodeCraft/xlgo-core/config"
|
||
"github.com/EthanCodeCraft/xlgo-core/database"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/clause"
|
||
"gorm.io/gorm/schema"
|
||
)
|
||
|
||
// appHungDriver 是测试用 database/sql 驱动，Conn.Ping 阻塞到 ctx 取消，模拟挂起 DB。
|
||
// 与 database 包内部的 hungDriver 同构（app_test 为外部包，需自带一份）。
|
||
type appHungDriver struct{}
|
||
|
||
func (appHungDriver) Open(string) (driver.Conn, error) { return appHungConn{}, nil }
|
||
|
||
type appHungConn struct{}
|
||
|
||
func (appHungConn) Prepare(string) (driver.Stmt, error) { return nil, errors.New("hung: not implemented") }
|
||
func (appHungConn) Close() error { return nil }
|
||
func (appHungConn) Begin() (driver.Tx, error) { return nil, errors.New("hung: not implemented") }
|
||
|
||
// Ping 阻塞到 ctx 取消（实现 driver.Pinger，方法名 Ping 而非 PingContext）。
|
||
func (appHungConn) Ping(ctx context.Context) error {
|
||
<-ctx.Done()
|
||
return ctx.Err()
|
||
}
|
||
|
||
var appHungRegisterOnce sync.Once
|
||
|
||
func registerAppHungDialect() {
|
||
appHungRegisterOnce.Do(func() {
|
||
sql.Register("xlgo_app_hung_hdb1", appHungDriver{})
|
||
database.RegisterDialect(database.DialectSpec{
|
||
Name: "app_hung_hdb1",
|
||
Dialector: func(string) gorm.Dialector {
|
||
db, _ := sql.Open("xlgo_app_hung_hdb1", "")
|
||
return appHungDialector{sqlDB: db}
|
||
},
|
||
DSN: func(*config.DatabaseConfig) string { return "app_hung://" },
|
||
})
|
||
})
|
||
}
|
||
|
||
// appHungDialector 让 gorm.Open 成功并注入挂起 *sql.DB，使 initDB 的 pingWithTimeout 触发。
|
||
type appHungDialector struct{ sqlDB *sql.DB }
|
||
|
||
func (d appHungDialector) Name() string { return "app_hung_hdb1" }
|
||
func (d appHungDialector) Initialize(db *gorm.DB) error { db.Config.ConnPool = d.sqlDB; return nil }
|
||
func (d appHungDialector) Migrator(*gorm.DB) gorm.Migrator { return nil }
|
||
func (d appHungDialector) DataTypeOf(*schema.Field) string { return "" }
|
||
func (d appHungDialector) DefaultValueOf(*schema.Field) clause.Expression { return nil }
|
||
func (d appHungDialector) BindVarTo(clause.Writer, *gorm.Statement, any) {}
|
||
func (d appHungDialector) QuoteTo(w clause.Writer, s string) { _, _ = w.WriteString(s) }
|
||
func (d appHungDialector) Explain(s string, _ ...any) string { return s }
|
||
|
||
// TestAppInitHungDBBounded_Hdb1 端到端回归 H-db-1（App 级闭环）：
|
||
// App.Init 启用 MySQL 且 DB 挂起时，initDB 的 pingWithTimeout 3s 失败 -> InitDB 返回错误
|
||
// -> App.Init 失败（failAfterInit 回滚）。全程有界，不因挂起 DB 无限卡死 App.Init。
|
||
// 修复前 gorm.Open 的 automatic Ping() 在 pingWithTimeout 之前就无限阻塞 -> App.Init 永久 hang。
|
||
// 同时验证 App.Init 失败后无 goroutine 泄漏（closeResources 正常）。
|
||
func TestAppInitHungDBBounded_Hdb1(t *testing.T) {
|
||
registerAppHungDialect()
|
||
cfg := &config.Config{
|
||
App: config.AppConfig{Env: "test"},
|
||
Server: config.ServerConfig{Port: 18091},
|
||
Database: config.DatabaseConfig{
|
||
Driver: "app_hung_hdb1",
|
||
Host: "localhost", Port: 3306, User: "u", Password: "p", Name: "n",
|
||
},
|
||
}
|
||
app := xlgo.New(xlgo.WithConfig(cfg), xlgo.WithMySQL())
|
||
|
||
done := make(chan error, 1)
|
||
start := time.Now()
|
||
go func() { done <- app.Init() }()
|
||
select {
|
||
case err := <-done:
|
||
elapsed := time.Since(start)
|
||
if err == nil {
|
||
t.Fatalf("挂起 DB 下 App.Init 应失败，got nil")
|
||
}
|
||
// initDB pingWithTimeout 3s 失败后重试 5 次（每次 ping 3s + retryDelay 1s 起步翻倍），
|
||
// 全跑约 5*4s=20s+。放宽到 35s 上限。关键是不触发下方 45s watchdog 的"无限阻塞"。
|
||
if elapsed > 35*time.Second {
|
||
t.Fatalf("App.Init 耗时 %v 超过 35s 上限（H-db-1：initDB ping 应被 3s 约束，重试 5 次应 ~20s）", elapsed)
|
||
}
|
||
case <-time.After(45 * time.Second):
|
||
t.Fatalf("App.Init 在挂起 DB 上无限阻塞（H-db-1：gorm.Open automatic ping 或 initDB ping 未被超时约束）")
|
||
}
|
||
|
||
// App.Init 失败后应能正常 Shutdown（failAfterInit 已回滚资源，Shutdown 幂等）。
|
||
shutdownDone := make(chan error, 1)
|
||
go func() { shutdownDone <- app.Shutdown() }()
|
||
select {
|
||
case err := <-shutdownDone:
|
||
if err != nil {
|
||
t.Logf("App.Shutdown after failed Init returned err: %v（可接受，资源已回滚）", err)
|
||
}
|
||
case <-time.After(10 * time.Second):
|
||
t.Fatalf("App.Shutdown 在 Init 失败后无限阻塞（closeResources 未正常收口）")
|
||
}
|
||
|
||
// rootCtx 已被 failAfterInit cancel，无残留 goroutine（StartProbing 未启动因 Init 失败）。
|
||
// 此处不直接断言 goroutine 数（受测试 runtime 噪声影响），靠 Shutdown 有界退出间接证明。
|
||
_ = context.Background()
|
||
}
|