feat: 五项目业务实现并对接完成
- MES(B): 工单/BOM/备料/工序字典/PLC下发/拧紧/扫码报工/半成品/AGV/追溯逻辑,WMS与海康RCS客户端,看板Redis缓存API(概览/设备/进度/报警/趋势)+SSE - WMS(C): JWT滑动续签、Excel导入、盘点、内部API、种子数据、独立Postgres配置 - WMS客户端(E): Go网关8891反向代理+内嵌Vue3十页 - 工位终端(D): SQLite本地缓存+模拟拧紧源+内嵌Vue3页面 - Dashboard(A): 看板数据改接MES内部缓存API,SSE实时刷新+vite代理 - 清理各项目球形磨遗留代码,新增部署手册.md
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
|
||||
"bj_power_wms/ent"
|
||||
|
||||
"entgo.io/ent/dialect"
|
||||
entsql "entgo.io/ent/dialect/sql"
|
||||
_ "github.com/jackc/pgx/v5/stdlib"
|
||||
)
|
||||
|
||||
// MustNewDB 创建 Ent 客户端并验证连接
|
||||
func MustNewDB(c DatabaseConf) *ent.Client {
|
||||
ec, err := NewDB(c)
|
||||
if err != nil {
|
||||
panic(fmt.Sprintf("数据库连接失败: %v", err))
|
||||
}
|
||||
return ec
|
||||
}
|
||||
|
||||
func NewDB(c DatabaseConf) (*ent.Client, error) {
|
||||
dsn := fmt.Sprintf("postgresql://%s:%s@%s:%d/%s",
|
||||
c.User, c.Password, c.Host, c.Port, c.Dbname,
|
||||
)
|
||||
|
||||
db, err := sql.Open("pgx", dsn)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err := db.Ping(); err != nil {
|
||||
return nil, fmt.Errorf("ping db: %w", err)
|
||||
}
|
||||
|
||||
drv := entsql.OpenDB(dialect.Postgres, db)
|
||||
return ent.NewClient(ent.Driver(drv)), nil
|
||||
}
|
||||
|
||||
// AutoMigrate 启动时自动建表/迁移
|
||||
func AutoMigrate(client *ent.Client) error {
|
||||
ctx := contextBackdrop()
|
||||
if err := client.Schema.Create(ctx); err != nil {
|
||||
return fmt.Errorf("自动迁移失败: %w", err)
|
||||
}
|
||||
slog.Info("数据库迁移完成")
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user