package service
|
|
import (
|
"apsClient/conf"
|
"apsClient/constvar"
|
"apsClient/model"
|
"apsClient/model/common"
|
"apsClient/model/response"
|
"apsClient/nsq"
|
"apsClient/pkg/ecode"
|
"apsClient/pkg/logx"
|
"apsClient/pkg/structx"
|
"encoding/json"
|
"fmt"
|
"gorm.io/gorm"
|
"time"
|
)
|
|
type TaskService struct {
|
}
|
|
func NewTaskService() *TaskService {
|
return &TaskService{}
|
}
|
|
// GetTask 获取任务，未完成的开始时间最早的任务
|
func (slf TaskService) GetTask(page, pageSize int) (taskDataList []*response.TaskData, count int64, code int) {
|
nowTs := time.Now().Unix()
|
var (
|
err error
|
orders []*model.Order
|
procedures []*model.Procedures
|
workOrderIds []string
|
)
|
procedures, err = model.NewProceduresSearch(nil).SetOrder("start_time asc").
|
SetStartTimeMax(nowTs).
|
SetDeviceId(conf.Conf.System.DeviceId).
|
SetStatusNot(model.ProcedureStatusFinished).
|
SetPage(page, pageSize).
|
FindNotTotal()
|
if err != nil {
|
return nil, 0, ecode.DBErr
|
}
|
count, err = model.NewProceduresSearch(nil).
|
SetDeviceId(conf.Conf.System.DeviceId).
|
SetStatusNot(model.ProcedureStatusFinished).
|
Count()
|
if err != nil {
|
return nil, 0, ecode.DBErr
|
}
|
if len(procedures) == 0 {
|
return nil, 0, ecode.OK
|
}
|
|
for _, procedure := range procedures {
|
workOrderIds = append(workOrderIds, procedure.WorkOrderID)
|
}
|
orders, err = model.NewOrderSearch(nil).SetWorkOrderIds(workOrderIds).FindNotTotal() //由工序反推工单
|
if err != nil {
|
return nil, 0, ecode.DBErr
|
}
|
|
orderMap := make(map[string]*model.Order, len(workOrderIds))
|
for _, order := range orders {
|
orderMap[order.WorkOrderID] = order
|
}
|
for _, procedure := range procedures {
|
taskData := response.TaskData{
|
Order: orderMap[procedure.WorkOrderID],
|
Procedure: procedure,
|
AllProcedures: nil,
|
CurrentProcedureIndex: 0,
|
Position: procedure.Position,
|
}
|
taskDataList = append(taskDataList, &taskData)
|
}
|
return taskDataList, count, ecode.OK
|
}
|
|
func (slf TaskService) GetProcedureById(id int) (*model.Procedures, int) {
|
procedure, err := model.NewProceduresSearch(nil).SetId(id).First()
|
if err != nil {
|
return nil, ecode.DBErr
|
}
|
return procedure, ecode.OK
|
}
|
|
func (slf TaskService) UpdateProcedureStatusAndPosition(db *gorm.DB, id int, status model.ProcedureStatus, position int) error {
|
if status == model.ProcedureStatusFinished {
|
ProgressCacheUnset()
|
}
|
return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
|
"status": status,
|
"position": position,
|
})
|
}
|
|
func (slf TaskService) UpdateProcedureStatus(db *gorm.DB, id int, status model.ProcedureStatus) error {
|
if status == model.ProcedureStatusFinished {
|
ProgressCacheUnset()
|
}
|
return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
|
"status": status,
|
})
|
}
|
|
func (slf TaskService) UpdateOrderStatus(db *gorm.DB, id uint, status model.OrderStatus) error {
|
return model.NewOrderSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
|
"status": status,
|
})
|
}
|
|
func (slf TaskService) GetOrderByWorkOrderId(workOrderId string) (order *model.Order, err error) {
|
return model.NewOrderSearch(nil).SetWorkOrderId(workOrderId).First()
|
}
|
|
func (slf TaskService) GetProcessParams(procedure *model.Procedures, order *model.Order) (processModel *model.ProcessModel, err error) {
|
data, err := model.NewProcessModelSearch().
|
SetWorkOrder(procedure.WorkOrderID).
|
SetDevice(procedure.ProceduresInfo.DeviceName).
|
SetOrderId(procedure.OrderID).
|
SetProcedure(procedure.ProceduresInfo.ProcedureName).
|
SetProduct(order.ProductName).
|
SetOrder("id desc").First()
|
if err != nil && err != gorm.ErrRecordNotFound {
|
logx.Errorf("TaskStart GetProcessModel err:%v", err)
|
return nil, err
|
}
|
if err == nil {
|
err = json.Unmarshal([]byte(data.Params), &data.ParamsMap)
|
if err != nil {
|
logx.Errorf("process model json.Unmarshal:%v", err)
|
return
|
}
|
return data, nil
|
}
|
|
if err == gorm.ErrRecordNotFound { //如果数据库没有从云端获取
|
caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicProcessParamsRequest, conf.Conf.NsqConf.NodeId), fmt.Sprintf(constvar.NsqTopicProcessParamsResponse, conf.Conf.NsqConf.NodeId))
|
var result common.ResponseProcessParams
|
err = caller.Call(common.RequestProcessParams{
|
WorkOrder: procedure.WorkOrderID,
|
OrderId: procedure.OrderID,
|
Product: order.ProductName,
|
Procedure: procedure.ProceduresInfo.ProcedureName,
|
Device: procedure.ProceduresInfo.DeviceName,
|
DeviceId: conf.Conf.System.DeviceId,
|
}, &result, time.Second*3)
|
if err != nil {
|
logx.Errorf("TaskStart GetProcessModel error:%v", err.Error())
|
return
|
}
|
if result.ParamsMap == nil {
|
logx.Errorf("TaskStart GetProcessModel response miss process params:%v", result)
|
return
|
}
|
processModel = new(model.ProcessModel)
|
err = structx.AssignTo(result, &processModel)
|
return processModel, err
|
}
|
return
|
}
|
|
// GetNextTask 获取未开始的任务
|
func (slf TaskService) GetNextTask() (workOrder *model.Order, err error) {
|
nowTs := time.Now().Unix()
|
orderSearch := model.NewOrderSearch(nil)
|
return orderSearch.SetOrder("start_time asc").
|
SetStartTimeMin(nowTs).
|
SetStatus(model.OrderStatusWaitProcess).First()
|
}
|
