优化索引

internal/infra/database/postgres.go

@@ -120,16 +120,9 @@ func (ps *PostgresStorage) Migrate(models ...interface{}) error {
 	ps.logger.Info("数据库表结构迁移完成")
 
 	// -- 处理gorm做不到的初始化逻辑 --
-	// 创建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 {
-		ps.logger.Errorw("为 sensor_data 的 data 字段创建 GIN 索引失败", "error", err)
-		return fmt.Errorf("为 sensor_data 的 data 字段创建 GIN 索引失败: %w", err)
+	if err := ps.creatingIndex(); err != nil {
+		return err
 	}
-	ps.logger.Info("成功为 sensor_data 的 data 字段创建 GIN 索引 (或已存在)")
 
 	// 如果是 TimescaleDB, 则将 sensor_data 转换为 hypertable
 	if ps.isTimescaleDB {
@@ -146,3 +139,27 @@ func (ps *PostgresStorage) Migrate(models ...interface{}) error {
 	}
 	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 {
+		ps.logger.Errorw("为 sensor_data 的 data 字段创建 GIN 索引失败", "error", err)
+		return fmt.Errorf("为 sensor_data 的 data 字段创建 GIN 索引失败: %w", err)
+	}
+	ps.logger.Info("成功为 sensor_data 的 data 字段创建 GIN 索引 (或已存在)")
+
+	// 为 tasks.parameters 创建 GIN 索引
+	ps.logger.Info("正在为 tasks 表的 parameters 字段创建 GIN 索引")
+	taskGinIndexSQL := "CREATE INDEX IF NOT EXISTS idx_tasks_parameters_gin ON tasks USING GIN (parameters);"
+	if err := ps.db.Exec(taskGinIndexSQL).Error; err != nil {
+		ps.logger.Errorw("为 tasks 的 parameters 字段创建 GIN 索引失败", "error", err)
+		return fmt.Errorf("为 tasks 的 parameters 字段创建 GIN 索引失败: %w", err)
+	}
+	ps.logger.Info("成功为 tasks 的 parameters 字段创建 GIN 索引 (或已存在)")
+	return nil
+}