package service
|
|
import (
|
"apsClient/conf"
|
"apsClient/constvar"
|
"apsClient/model"
|
"apsClient/model/common"
|
"apsClient/model/request"
|
"apsClient/model/response"
|
"apsClient/pkg/ecode"
|
"apsClient/pkg/logx"
|
"apsClient/pkg/timex"
|
"fmt"
|
"github.com/jinzhu/gorm"
|
"time"
|
)
|
|
type TaskService struct {
|
}
|
|
func NewTaskService() *TaskService {
|
return &TaskService{}
|
}
|
|
// GetTask 获取任务，未完成的开始时间小于等于当前时间，结束时间大于当前时间的任务
|
func (slf TaskService) GetTask(deviceID string, page, pageSize int, mode constvar.TaskMode, channels []int32) (taskResp *response.TaskResponse, code int) {
|
var taskList []*response.TaskData
|
var count int64
|
var workers []*common.ProcedureWorker
|
|
taskResp = &response.TaskResponse{
|
Tasks: taskList,
|
TaskCount: count,
|
Workers: workers,
|
}
|
|
var (
|
err error
|
orders []*model.Order
|
procedures []*model.Procedures
|
workOrderIds []string
|
)
|
search := model.NewProceduresSearch(nil).
|
SetDeviceId(deviceID).
|
SetPage(page, pageSize)
|
|
if mode == constvar.TaskModeUnStarted {
|
search.SetStatus(model.ProcedureStatusWaitProcess).
|
SetOrder("start_time asc")
|
} else if mode == constvar.TaskModeCurrent {
|
search.SetStatus(model.ProcedureStatusProcessing).
|
SetOrder("start_time asc")
|
} else if mode == constvar.TaskModeLastFinished {
|
search.SetStatus(model.ProcedureStatusFinished).SetOrder("updated_at desc")
|
if len(channels) > 0 {
|
search.SetChannels(channels)
|
}
|
}
|
|
procedures, err = search.FindNotTotal()
|
if err != nil {
|
return taskResp, ecode.DBErr
|
}
|
|
if err != nil {
|
return taskResp, ecode.DBErr
|
}
|
if len(procedures) == 0 {
|
return taskResp, ecode.OK
|
}
|
|
for _, procedure := range procedures {
|
workOrderIds = append(workOrderIds, procedure.WorkOrderID)
|
}
|
orders, err = model.NewOrderSearch(nil).SetWorkOrderIds(workOrderIds).FindNotTotal() //由工序反推工单
|
if err != nil {
|
return taskResp, 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: procedure.ProceduresInfo.AllProcedureNames,
|
CurrentProcedureIndex: 0,
|
Channel: procedure.Channel,
|
}
|
index := 0
|
for _, name := range procedure.ProceduresInfo.AllProcedureNames {
|
if name == procedure.ProceduresInfo.ProcedureName {
|
break
|
}
|
index++
|
}
|
taskData.CurrentProcedureIndex = index
|
taskList = append(taskList, &taskData)
|
workers = append(workers, procedure.ProceduresInfo.Workers...)
|
}
|
taskResp = &response.TaskResponse{
|
Tasks: taskList,
|
TaskCount: count,
|
Workers: slf.WorkerDistinct(workers),
|
}
|
return taskResp, ecode.OK
|
}
|
func (slf TaskService) NewTaskCount(deviceId string) (count int64) {
|
count, _ = model.NewProceduresSearch(nil).SetDeviceId(deviceId).SetStatus(model.ProcedureStatusWaitProcess).Count()
|
return count
|
}
|
|
func (slf TaskService) WorkerDistinct(workers []*common.ProcedureWorker) (NoRepeatedWorkers []*common.ProcedureWorker) {
|
exists := make(map[string]bool, 0)
|
for _, worker := range workers {
|
key := fmt.Sprintf("%v%v", worker.WorkerID, worker.StartTime)
|
if !exists[key] {
|
NoRepeatedWorkers = append(NoRepeatedWorkers, worker)
|
exists[key] = true
|
}
|
}
|
return NoRepeatedWorkers
|
}
|
|
func (slf TaskService) GetProcedureById(id uint) (*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) GetProcedureAndWorkOrder(procedureId uint) (procedure *model.Procedures, workOrder *model.Order, err error) {
|
procedure, err = model.NewProceduresSearch(nil).SetId(procedureId).First()
|
if err != nil {
|
return nil, nil, err
|
}
|
workOrder, err = model.NewOrderSearch(nil).SetWorkOrderId(procedure.WorkOrderID).First()
|
if err != nil {
|
return nil, nil, err
|
}
|
return procedure, workOrder, nil
|
}
|
|
func (slf TaskService) UpdateProcedureStatusAndChannel(db *gorm.DB, id uint, status model.ProcedureStatus, channel int32, processModelNumber string) error {
|
if status == model.ProcedureStatusFinished || status == model.ProcedureStatusWaitProcess {
|
ProgressCacheUnset(conf.Conf.CurrentDeviceID, channel)
|
}
|
|
upMap := map[string]interface{}{"status": status}
|
if status == model.ProcedureStatusProcessing {
|
upMap["process_model_number"] = processModelNumber
|
upMap["real_start_time"] = time.Now().Unix()
|
} else if status == model.ProcedureStatusWaitProcess {
|
upMap["process_model_number"] = ""
|
} else if status == model.ProcedureStatusFinished {
|
upMap["real_end_time"] = time.Now().Unix()
|
}
|
|
return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
|
}
|
|
func (slf TaskService) UpdateProcessModelNumber(db *gorm.DB, id uint, processModelNumber string) error {
|
upMap := map[string]interface{}{"process_model_number": processModelNumber}
|
return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
|
}
|
|
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().
|
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 == gorm.ErrRecordNotFound {
|
return nil, err
|
}
|
if err == nil {
|
return data, nil
|
}
|
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()
|
}
|
|
// SaveTaskStatusSync 保存任务同步信息
|
func (slf TaskService) SaveTaskStatusSync(db *gorm.DB, record *model.TaskStatusSync) (err error) {
|
if record.IsProcessing {
|
record.StartTs = time.Now().Unix()
|
}
|
if record.IsFinish {
|
record.FinishTs = time.Now().Unix()
|
startedRecord, _ := model.NewTaskStatusSyncSearch(db).SetProductProcedureId(record.ProductProcedureID).SetWorkOrderId(record.WorkOrderId).First()
|
record.StartTs = startedRecord.StartTs
|
}
|
return model.NewTaskStatusSyncSearch(db).Create(record)
|
}
|
|
// GetTaskStatusSync 获取待同步任务信息
|
func (slf TaskService) GetTaskStatusSync(limit int) ([]*model.TaskStatusSync, error) {
|
return model.NewTaskStatusSyncSearch(nil).SetOrder("id asc").SetSyncWait().SetPage(1, limit).FindNotTotal()
|
}
|
|
// RemoveTaskStatusSync 删除同步完的记录
|
func (slf TaskService) RemoveTaskStatusSync(ids []uint) (err error) {
|
return model.NewTaskStatusSyncSearch(nil).SetIDs(ids).Delete()
|
}
|
|
// FinishTaskStatusSync 设置已同步完成
|
func (slf TaskService) FinishTaskStatusSync(ids []uint) (err error) {
|
return model.NewTaskStatusSyncSearch(nil).SetIDs(ids).UpdateByMap(map[string]interface{}{"sync_status": constvar.SyncStatusOk})
|
}
|
|
// GetTask2 获取任务列表2
|
func (slf TaskService) GetTask2(deviceID string, offset, limit int, channels []int32, queryType request.QueryType) (taskResp *response.TaskResponse, err error) {
|
var taskList []*response.TaskData
|
var count int64
|
var workers []*common.ProcedureWorker
|
|
taskResp = &response.TaskResponse{
|
Tasks: taskList,
|
TaskCount: count,
|
Workers: workers,
|
}
|
|
var (
|
orders []*model.Order
|
procedures []*model.Procedures
|
workOrderIds []string
|
)
|
search := model.NewProceduresSearch(nil).
|
SetDeviceId(deviceID).
|
SetOffset(offset, limit).SetChannels(channels)
|
todayStartTs := timex.TodayStartTs()
|
switch queryType {
|
case request.QueryTypeUnFinish:
|
search.SetStatusNot(model.ProcedureStatusFinished).SetOrder("status desc, start_time asc")
|
case request.QueryTypeToday:
|
search.SetStatusNot(model.ProcedureStatusFinished).SetStartTimeMin(todayStartTs).SetEndTimeMax(todayStartTs + 86400).SetOrder("status desc, start_time asc")
|
case request.QueryTypeFinished:
|
search.SetStatus(model.ProcedureStatusFinished).SetOrder("start_time desc")
|
}
|
|
procedures, total, err := search.Find()
|
if err != nil {
|
return taskResp, err
|
}
|
|
if err != nil {
|
return taskResp, err
|
}
|
if len(procedures) == 0 {
|
return taskResp, nil
|
}
|
|
for _, procedure := range procedures {
|
workOrderIds = append(workOrderIds, procedure.WorkOrderID)
|
}
|
orders, err = model.NewOrderSearch(nil).SetWorkOrderIds(workOrderIds).FindNotTotal() //由工序反推工单
|
if err != nil {
|
return taskResp, err
|
}
|
|
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: procedure.ProceduresInfo.AllProcedureNames,
|
CurrentProcedureIndex: 0,
|
Channel: procedure.Channel,
|
}
|
index := 0
|
for _, name := range procedure.ProceduresInfo.AllProcedureNames {
|
if name == procedure.ProceduresInfo.ProcedureName {
|
break
|
}
|
index++
|
}
|
taskData.CurrentProcedureIndex = index
|
taskList = append(taskList, &taskData)
|
workers = append(workers, procedure.ProceduresInfo.Workers...)
|
}
|
taskResp = &response.TaskResponse{
|
Tasks: taskList,
|
TaskCount: total,
|
Workers: slf.WorkerDistinct(workers),
|
}
|
return taskResp, err
|
}
|
