pig/pig-farm-controller
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

定义ack表

Browse Source
This commit is contained in:
huang
2025-09-24 23:08:59 +08:00
parent cf53cdfe28
commit 6b931648dc
3 changed files with 71 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
internal/infra/database/postgres.go
View File

@@ -126,28 +126,47 @@ func (ps *PostgresStorage) Migrate(models ...interface{}) error {
// 如果是 TimescaleDB, 则将 sensor_data 转换为 hypertable
if ps.isTimescaleDB {
ps.logger.Info("检测到 TimescaleDB, 准备转换 sensor_data 为超表")
// 使用 if_not_exists => TRUE 保证幂等性
// 如果 sensor_data 已经是超表，此命令不会报错
// 'time' 是 SensorData 模型中定义的时间列
sql := "SELECT create_hypertable('sensor_data', 'time', if_not_exists => TRUE);"
if err := ps.db.Exec(sql).Error; err != nil {
ps.logger.Errorw("将 sensor_data 转换为超表失败", "error", err)
return fmt.Errorf("将 sensor_data 转换为超表失败: %w", err)
ps.logger.Info("检测到 TimescaleDB, 准备进行超表转换")
if err := ps.creatingHyperTable(); err != nil {
return err
}
ps.logger.Info("成功将 sensor_data 转换为超表 (或已转换)")
}
return nil
}
// creatingHyperTable 用于在数据库是 TimescaleDB 时创建超表
func (ps *PostgresStorage) creatingHyperTable() error {
// 将 sensor_data 转换为超表
// 使用 if_not_exists => TRUE 保证幂等性
// 'time' 是 SensorData 模型中定义的时间列
sqlSensorData := "SELECT create_hypertable('sensor_data', 'time', if_not_exists => TRUE);"
if err := ps.db.Exec(sqlSensorData).Error; err != nil {
ps.logger.Errorw("将 sensor_data 转换为超表失败", "error", err)
return fmt.Errorf("将 sensor_data 转换为超表失败: %w", err)
}
ps.logger.Info("成功将 sensor_data 转换为超表 (或已转换)")
// 将 downlink_task_records 转换为超表
// 'sent_at' 是 DownlinkTaskRecord 模型中定义的时间列
sqlDownlinkTaskRecords := "SELECT create_hypertable('downlink_task_records', 'sent_at', if_not_exists => TRUE);"
if err := ps.db.Exec(sqlDownlinkTaskRecords).Error; err != nil {
ps.logger.Errorw("将 downlink_task_records 转换为超表失败", "error", err)
return fmt.Errorf("将 downlink_task_records 转换为超表失败: %w", err)
}
ps.logger.Info("成功将 downlink_task_records 转换为超表 (或已转换)")
return nil
}
// creatingIndex 用于创建gorm无法处理的索引, 如gin索引
func (ps *PostgresStorage) creatingIndex() error {
// 创建GIN索引(用于优化JSONB查询)
ps.logger.Info("正在为 sensor_data 表的 data 字段创建 GIN 索引")
// 使用 IF NOT EXISTS 保证幂等性
// 如果索引已存在，此命令不会报错
ginIndexSQL := "CREATE INDEX IF NOT EXISTS idx_sensor_data_data_gin ON sensor_data USING GIN (data);"
if err := ps.db.Exec(ginIndexSQL).Error; err != nil {
// 为 sensor_data 表的 data 字段创建 GIN 索引
ps.logger.Info("正在为 sensor_data 表的 data 字段创建 GIN 索引")
ginSensorDataIndexSQL := "CREATE INDEX IF NOT EXISTS idx_sensor_data_data_gin ON sensor_data USING GIN (data);"
if err := ps.db.Exec(ginSensorDataIndexSQL).Error; err != nil {
ps.logger.Errorw("为 sensor_data 的 data 字段创建 GIN 索引失败", "error", err)
return fmt.Errorf("为 sensor_data 的 data 字段创建 GIN 索引失败: %w", err)
}
@@ -161,5 +180,15 @@ func (ps *PostgresStorage) creatingIndex() error {
return fmt.Errorf("为 tasks 的 parameters 字段创建 GIN 索引失败: %w", err)
}
ps.logger.Info("成功为 tasks 的 parameters 字段创建 GIN 索引 (或已存在)")
// 为 devices 表的 properties 字段创建 GIN 索引
ps.logger.Info("正在为 devices 表的 properties 字段创建 GIN 索引")
ginDevicePropertiesIndexSQL := "CREATE INDEX IF NOT EXISTS idx_devices_properties_gin ON devices USING GIN (properties);"
if err := ps.db.Exec(ginDevicePropertiesIndexSQL).Error; err != nil {
ps.logger.Errorw("为 devices 的 properties 字段创建 GIN 索引失败", "error", err)
return fmt.Errorf("为 devices 的 properties 字段创建 GIN 索引失败: %w", err)
}
ps.logger.Info("成功为 devices 的 properties 字段创建 GIN 索引 (或已存在)")
return nil
}