SunYF
ebf901ad16
1. 通用优化： - 统一系统标题为"库房客户端"/"MES产线控制"，移除北自所前缀 - 调整内置账号密码为统一123456，优化密码校验逻辑 - 新增数据库自动创建逻辑，简化部署流程 - 修复日志时间格式配置，优化日志输出 - 新增纯数字密码安全提示 2. MES系统优化： - 新增日排产管理功能，支持增删改查与自动算料生成备料单 - 重构BOM模块，改为按产品编码维护而非工单维度 - 新增工艺参数模板配置与手动报工页面 - 新增产品类型自动编码功能 - 优化菜单结构，调整工单管理、扫码报工等页面名称与路由 - 新增删除日排产接口与权限保护 - 修复物料清单查询逻辑，适配新BOM结构 - 新增refreshToken支持，优化登录会话管理 3. WMS系统优化： - 新增基础数据维护页面，支持区域与物料档案管理 - 新增工单列表下拉接口，对接MES获取未完成工单优先展示 - 优化入库管理提示文案，替换英文提示为中文 - 修复精密件入库校验逻辑，优化错误提示 - 优化Excel导入功能，新增备注字段支持 - 优化登录页面与菜单文案，统一备料台账名称 - 新增自动打开浏览器功能，优化客户端启动体验 - 修复备料出库页面查询提示文案 - 新增WMS与MES对接配置，完善内部API调用逻辑 4. 其他优化： - 删除冗余的旧版静态资源文件，更新资源引用路径 - 新增帮助文档，完善基础数据模块说明 - 修复多处文案不统一、英文残留问题
128 lines
3.3 KiB
Go
128 lines
3.3 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"bj_power_mes/ent"
|
|
|
|
"entgo.io/ent/dialect"
|
|
entsql "entgo.io/ent/dialect/sql"
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
|
)
|
|
|
|
// DatabaseConf 数据库配置，DSN 由调用方从 yaml 读取
|
|
type DatabaseConf struct {
|
|
Host string `json:",default=127.0.0.1"`
|
|
Port int `json:",default=5432"`
|
|
User string `json:",default=postgres"`
|
|
Password string `json:",default=postgres"`
|
|
Dbname string `json:",default=bj_power_mes"`
|
|
MaxIdle int `json:",default=10"`
|
|
MaxOpen int `json:",default=20"`
|
|
}
|
|
|
|
// DSN 从配置拼装连接串（不在代码中硬编码口令）
|
|
func (c DatabaseConf) DSN() string {
|
|
return fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable",
|
|
c.Host, c.Port, c.User, c.Password, c.Dbname)
|
|
}
|
|
|
|
// MustNewDB 打开 PostgreSQL 连接并包装为 ent.Client
|
|
func MustNewDB(c DatabaseConf) (*ent.Client, *sql.DB) {
|
|
sqlDB, err := sql.Open("pgx", c.DSN())
|
|
if err != nil {
|
|
panic(fmt.Sprintf("open db failed: %v", err))
|
|
}
|
|
sqlDB.SetMaxIdleConns(c.MaxIdle)
|
|
sqlDB.SetMaxOpenConns(c.MaxOpen)
|
|
sqlDB.SetConnMaxLifetime(time.Hour)
|
|
|
|
drv := entsql.OpenDB(dialect.Postgres, sqlDB)
|
|
client := ent.NewClient(ent.Driver(drv))
|
|
return client, sqlDB
|
|
}
|
|
|
|
// NewDB 打开连接，失败返回 error（用于 migrate 等工具）
|
|
func NewDB(c DatabaseConf) (*sql.DB, error) {
|
|
sqlDB, err := sql.Open("pgx", c.DSN())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := sqlDB.Ping(); err != nil {
|
|
return nil, err
|
|
}
|
|
return sqlDB, nil
|
|
}
|
|
|
|
// EnsureDB 若目标数据库不存在则自动创建。
|
|
// 先连接 PostgreSQL 内置维护库 maintenance db（默认 postgres），检查目标库，
|
|
// 不存在则执行 CREATE DATABASE，便于重置/建表命令在空环境一键运行。
|
|
func EnsureDB(c DatabaseConf) error {
|
|
maintenanceDSN := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=postgres sslmode=disable",
|
|
c.Host, c.Port, c.User, c.Password)
|
|
sqlDB, err := sql.Open("pgx", maintenanceDSN)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer sqlDB.Close()
|
|
|
|
var exists bool
|
|
if err := sqlDB.QueryRow(
|
|
`SELECT EXISTS(SELECT 1 FROM pg_database WHERE datname = $1)`, c.Dbname,
|
|
).Scan(&exists); err != nil {
|
|
return err
|
|
}
|
|
|
|
if exists {
|
|
return nil
|
|
}
|
|
|
|
// 库名需要用双引号包裹以支持大写/特殊字符
|
|
if _, err := sqlDB.Exec(`CREATE DATABASE "` + c.Dbname + `"`); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// AutoMigrate 执行 ent 自动建表/补齐（只增，不删数据）。供命令行 migrate 使用。
|
|
|
|
func AutoMigrate(c DatabaseConf) error {
|
|
client, sqlDB := MustNewDB(c)
|
|
defer sqlDB.Close()
|
|
return client.Schema.Create(context.Background())
|
|
}
|
|
|
|
// DropAllTables 清空 public schema 下所有业务表（破坏性，仅供命令行 reset-all 使用）。
|
|
func DropAllTables(c DatabaseConf) error {
|
|
sqlDB, err := NewDB(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer sqlDB.Close()
|
|
|
|
rows, err := sqlDB.Query(`SELECT tablename FROM pg_tables WHERE schemaname='public'`)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var names []string
|
|
for rows.Next() {
|
|
var t string
|
|
if err := rows.Scan(&t); err != nil {
|
|
rows.Close()
|
|
return err
|
|
}
|
|
names = append(names, t)
|
|
}
|
|
rows.Close()
|
|
|
|
for _, t := range names {
|
|
if _, err := sqlDB.Exec(`DROP TABLE IF EXISTS "` + t + `" CASCADE`); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|