48 lines
952 B
Go
48 lines
952 B
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/company/ai-ops/internal/config"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
// Pool 是全局数据库连接池
|
|
var Pool *pgxpool.Pool
|
|
|
|
// Init 初始化数据库连接
|
|
func Init(cfg config.DatabaseConfig) error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
poolConfig, err := pgxpool.ParseConfig(cfg.DSN())
|
|
if err != nil {
|
|
return fmt.Errorf("parse db config: %w", err)
|
|
}
|
|
|
|
poolConfig.MaxConns = int32(cfg.PoolSize)
|
|
poolConfig.MinConns = 2
|
|
poolConfig.MaxConnLifetime = 30 * time.Minute
|
|
poolConfig.MaxConnIdleTime = 10 * time.Minute
|
|
|
|
Pool, err = pgxpool.NewWithConfig(ctx, poolConfig)
|
|
if err != nil {
|
|
return fmt.Errorf("create db pool: %w", err)
|
|
}
|
|
|
|
if err := Pool.Ping(ctx); err != nil {
|
|
return fmt.Errorf("ping db: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Close 关闭数据库连接
|
|
func Close() {
|
|
if Pool != nil {
|
|
Pool.Close()
|
|
}
|
|
}
|