实现ReleaseFeedWeightTask

internal/app/service/task/release_feed_weight_task.go

@@ -1,36 +1,149 @@
 package task
 
 import (
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/app/service/device"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport"
 )
 
+const (
+	// 这个参数是 TaskTaskReleaseFeedWeight 类型的 Task Parameters 中用于记录释放的重量字段的key
+	ParamsReleaseWeight = "release_weight"
+	// 这个参数是 TaskTaskReleaseFeedWeight 类型的 Task Parameters 中用于记录下料口ID字段的key
+	ParamsFeedPortDeviceID = "feed_port_device_id"
+	// 这个参数是 TaskTaskReleaseFeedWeight 类型的 Task Parameters 中用于记录称重传感器ID字段的key
+	ParamsMixingTankDeviceID = "mixing_tank_device_id"
+)
+
 // ReleaseFeedWeightTask 是一个控制下料口释放指定重量的任务
 type ReleaseFeedWeightTask struct {
 	deviceRepo     repository.DeviceRepository
 	sensorDataRepo repository.SensorDataRepository
+	claimedLog     *models.TaskExecutionLog
 
-	comm transport.Communicator
+	feedPortDevice     *models.Device // 下料口基本信息
+	releaseWeight      float64        // 需要释放的重量
+	mixingTankDeviceID uint           // 搅拌罐称重传感器ID
+
+	comm     transport.Communicator
+	feedPort *device.GeneralDeviceService // 下料口指令下发器
 
 	logger *logs.Logger
 }
 
-func (r *ReleaseFeedWeightTask) Execute() error {
-	//TODO implement me
-	panic("implement me")
-}
-
-func (r *ReleaseFeedWeightTask) OnFailure(executeErr error) {
-	//TODO implement me
-	panic("implement me")
-}
-
 // NewReleaseFeedWeightTask 创建一个新的 ReleaseFeedWeightTask 实例
-func NewReleaseFeedWeightTask(deviceRepo repository.DeviceRepository, sensorDataRepo repository.SensorDataRepository, comm transport.Communicator) Task {
+func NewReleaseFeedWeightTask(claimedLog *models.TaskExecutionLog, deviceRepo repository.DeviceRepository, sensorDataRepo repository.SensorDataRepository, comm transport.Communicator, logger *logs.Logger) Task {
 	return &ReleaseFeedWeightTask{
+		claimedLog:     claimedLog,
 		deviceRepo:     deviceRepo,
 		sensorDataRepo: sensorDataRepo,
 		comm:           comm,
+		logger:         logger,
 	}
 }
+
+func (r *ReleaseFeedWeightTask) Execute() error {
+	r.logger.Infof("任务 %v: 开始执行, 日志ID: %v", r.claimedLog.TaskID, r.claimedLog.ID)
+	if err := r.parseParameters(); err != nil {
+		return err
+	}
+
+	weight, err := r.getNowWeight()
+	if err != nil {
+		return err
+	}
+
+	if err = r.feedPort.Switch(r.feedPortDevice, device.DeviceActionStart); err != nil {
+		r.logger.Errorf("启动下料口(id=%v)失败: %v , 日志ID: %v", r.feedPortDevice.ID, err, r.claimedLog.ID)
+		return err
+	}
+
+	targetWeight := weight - r.releaseWeight
+	errCount := 1
+
+	// TODO 这个判断有延迟, 尤其是LoRa通信本身延迟较高, 可以考虑根据信号质量或其他指标提前发送停止命令
+	for targetWeight <= weight {
+		weight, err = r.getNowWeight()
+		if err != nil {
+			errCount++
+			if errCount > 3 { // 如果连续三次没成功采集到重量数据，则认为计划执行失败
+				r.logger.Errorf("获取当前计划执行日志(id=%v)的当前搅拌罐重量失败: %v, 任务结束", r.claimedLog.ID, err)
+				return err
+			}
+			r.logger.Warnf("第%v次尝试获取当前计划执行日志(id=%v)的当前搅拌罐重量失败: %v", errCount, r.claimedLog.ID, err)
+			continue
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+
+	if err = r.feedPort.Switch(r.feedPortDevice, device.DeviceActionStop); err != nil {
+		r.logger.Errorf("关闭下料口(id=%v)失败: %v , 日志ID: %v", r.feedPortDevice.ID, err, r.claimedLog.ID)
+		return err
+	}
+
+	r.logger.Infof("完成计划执行日志(id=%v)的当前计划, 完成下料 %vkg, 搅拌罐剩余重量 %vkg", r.claimedLog.ID, r.releaseWeight, weight)
+	return nil
+}
+
+// 获取当前搅拌罐重量
+func (r *ReleaseFeedWeightTask) getNowWeight() (float64, error) {
+	sensorData, err := r.sensorDataRepo.GetLatestSensorDataByDeviceIDAndSensorType(r.mixingTankDeviceID, models.SensorDataTypeWeight)
+	if err != nil {
+		r.logger.Errorf("获取设备 %v 最新传感器数据失败: %v , 日志ID: %v", r.mixingTankDeviceID, err, r.claimedLog.ID)
+		return 0, err
+	}
+
+	wg := &models.WeightData{}
+	err = json.Unmarshal(sensorData.Data, wg)
+	if err != nil {
+		r.logger.Errorf("反序列化设备 %v 最新传感器数据失败: %v , 日志ID: %v", r.mixingTankDeviceID, err, r.claimedLog.ID)
+		return 0, err
+	}
+
+	return wg.WeightKilograms, nil
+}
+
+func (r *ReleaseFeedWeightTask) parseParameters() error {
+	if r.claimedLog.Task.Parameters == nil {
+		r.logger.Errorf("任务 %v: 缺少参数", r.claimedLog.TaskID)
+		return fmt.Errorf("任务 %v: 参数不全", r.claimedLog.TaskID)
+	}
+
+	// TODO 定义成结构体放model包中
+	var params map[string]interface{}
+	err := json.Unmarshal(r.claimedLog.Task.Parameters, &params)
+	if err != nil {
+		r.logger.Errorf("任务 %v: 解析参数失败: %v", r.claimedLog.TaskID, err)
+		return fmt.Errorf("任务 %v: 解析参数失败: %v", r.claimedLog.TaskID, err)
+	}
+
+	r.releaseWeight = params[ParamsReleaseWeight].(float64)
+	r.mixingTankDeviceID = params[ParamsMixingTankDeviceID].(uint)
+	r.feedPortDevice, err = r.deviceRepo.FindByID(params[ParamsFeedPortDeviceID].(uint))
+	if err != nil {
+		r.logger.Errorf("任务 %v: 获取设备信息失败: %v", r.claimedLog.TaskID, err)
+		return fmt.Errorf("任务 %v: 获取设备信息失败: %v", r.claimedLog.TaskID, err)
+	}
+	r.feedPort = device.NewGeneralDeviceService(r.deviceRepo, r.logger, r.comm)
+
+	return nil
+}
+
+func (r *ReleaseFeedWeightTask) OnFailure(executeErr error) {
+	r.logger.Errorf("开始善后处理, 日志ID:%v", r.claimedLog.ID)
+	if r.feedPort != nil {
+		err := r.feedPort.Switch(r.feedPortDevice, device.DeviceActionStop)
+		if err != nil {
+			r.logger.Errorf("[严重] 下料口停止失败, 日志ID: %v, 错误: %v", r.claimedLog.ID, err)
+		}
+	} else {
+		r.logger.Warnf("[警告] 下料口通信器尚未初始化, 不进行任何操作, 日志ID: %v", r.claimedLog.ID)
+	}
+	r.logger.Errorf("善后处理完成, 日志ID:%v", r.claimedLog.ID)
+}