81 lines
5.2 KiB
Go
81 lines
5.2 KiB
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
||
|
|
)
|
||
|
|
|
||
|
|
// MigrateAndSeed 执行幂等结构迁移，并只在空库时创建演示数据。
|
||
|
|
func MigrateAndSeed(ctx context.Context, pool *pgxpool.Pool) error {
|
||
|
|
statements := []string{
|
||
|
|
`CREATE TABLE IF NOT EXISTS idn_account (
|
||
|
|
identity uuid PRIMARY KEY, phone varchar(32) NOT NULL UNIQUE, account_type varchar(32) NOT NULL,
|
||
|
|
status varchar(32) NOT NULL, service_area varchar(128) NOT NULL, created_at timestamptz NOT NULL,
|
||
|
|
updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS org_gas_station (
|
||
|
|
identity uuid PRIMARY KEY, station_code varchar(32) NOT NULL UNIQUE, name varchar(128) NOT NULL,
|
||
|
|
principal varchar(64) NOT NULL, service_area varchar(128) NOT NULL, status varchar(32) NOT NULL,
|
||
|
|
created_at timestamptz NOT NULL, updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS org_delivery_point (
|
||
|
|
identity uuid PRIMARY KEY, delivery_code varchar(32) NOT NULL UNIQUE, gas_station_identity uuid REFERENCES org_gas_station(identity),
|
||
|
|
name varchar(128) NOT NULL, service_area varchar(128) NOT NULL, status varchar(32) NOT NULL,
|
||
|
|
created_at timestamptz NOT NULL, updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS org_service_person (
|
||
|
|
identity uuid PRIMARY KEY, account_identity uuid UNIQUE REFERENCES idn_account(identity), gas_station_identity uuid REFERENCES org_gas_station(identity),
|
||
|
|
delivery_point_identity uuid REFERENCES org_delivery_point(identity), name varchar(64) NOT NULL, roles varchar(128) NOT NULL,
|
||
|
|
work_status varchar(32) NOT NULL, credential_status varchar(32) NOT NULL, created_at timestamptz NOT NULL,
|
||
|
|
updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS org_user_service_relation (
|
||
|
|
identity uuid PRIMARY KEY, account_identity uuid NOT NULL REFERENCES idn_account(identity), gas_station_identity uuid REFERENCES org_gas_station(identity),
|
||
|
|
delivery_point_identity uuid REFERENCES org_delivery_point(identity), source varchar(32) NOT NULL, status varchar(32) NOT NULL,
|
||
|
|
created_at timestamptz NOT NULL, updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS dev_device (
|
||
|
|
identity uuid PRIMARY KEY, device_code varchar(64) NOT NULL UNIQUE, account_identity uuid REFERENCES idn_account(identity),
|
||
|
|
online_status varchar(32) NOT NULL, valve_status varchar(32) NOT NULL, created_at timestamptz NOT NULL,
|
||
|
|
updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS saf_event (
|
||
|
|
identity uuid PRIMARY KEY, event_code varchar(32) NOT NULL UNIQUE, device_identity uuid REFERENCES dev_device(identity),
|
||
|
|
level integer NOT NULL CHECK (level BETWEEN 1 AND 3), title varchar(256) NOT NULL, status varchar(32) NOT NULL,
|
||
|
|
created_at timestamptz NOT NULL, updated_at timestamptz NOT NULL, version integer NOT NULL DEFAULT 1
|
||
|
|
)`,
|
||
|
|
`CREATE TABLE IF NOT EXISTS aud_operation_log (
|
||
|
|
identity uuid PRIMARY KEY, operator_identity uuid, action varchar(64) NOT NULL, object_type varchar(64) NOT NULL,
|
||
|
|
object_identity uuid, detail jsonb NOT NULL DEFAULT '{}'::jsonb, created_at timestamptz NOT NULL
|
||
|
|
)`,
|
||
|
|
`CREATE INDEX IF NOT EXISTS idx_org_delivery_point_gas_station_identity ON org_delivery_point(gas_station_identity)`,
|
||
|
|
`CREATE INDEX IF NOT EXISTS idx_org_service_person_gas_station_identity ON org_service_person(gas_station_identity)`,
|
||
|
|
`CREATE INDEX IF NOT EXISTS idx_saf_event_status_level ON saf_event(status, level)`,
|
||
|
|
`COMMENT ON TABLE idn_account IS '身份账户主表'`,
|
||
|
|
`COMMENT ON COLUMN idn_account.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE org_gas_station IS '可燃气体站主表'`,
|
||
|
|
`COMMENT ON COLUMN org_gas_station.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE org_delivery_point IS '配送点主表'`,
|
||
|
|
`COMMENT ON COLUMN org_delivery_point.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE org_service_person IS '服务人员主表'`,
|
||
|
|
`COMMENT ON COLUMN org_service_person.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE org_user_service_relation IS '用户服务关系表'`,
|
||
|
|
`COMMENT ON COLUMN org_user_service_relation.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE dev_device IS '智能瓶阀设备主表'`,
|
||
|
|
`COMMENT ON COLUMN dev_device.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE saf_event IS '安全事件主表'`,
|
||
|
|
`COMMENT ON COLUMN saf_event.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
`COMMENT ON TABLE aud_operation_log IS '不可变操作审计日志表'`,
|
||
|
|
`COMMENT ON COLUMN aud_operation_log.identity IS '主键，应用生成的 UUID V7'`,
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, statement := range statements {
|
||
|
|
if _, err := pool.Exec(ctx, statement); err != nil {
|
||
|
|
return fmt.Errorf("执行数据库迁移失败: %w", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return seed(ctx, pool)
|
||
|
|
}
|