forked from JointCloud/pcm-coordinator
76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
/*
|
|
|
|
Copyright (c) [2023] [pcm]
|
|
[pcm-coordinator] is licensed under Mulan PSL v2.
|
|
You can use this software according to the terms and conditions of the Mulan PSL v2.
|
|
You may obtain a copy of Mulan PSL v2 at:
|
|
http://license.coscl.org.cn/MulanPSL2
|
|
THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
EITHER EXPaRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
See the Mulan PSL v2 for more details.
|
|
|
|
*/
|
|
|
|
package storelink
|
|
|
|
import (
|
|
"context"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/storeLink"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/models"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type GetLinkTaskLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGetLinkTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetLinkTaskLogic {
|
|
return &GetLinkTaskLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetLinkTaskLogic) GetLinkTask(req *types.GetLinkTaskReq) (resp *types.GetLinkTaskResp, err error) {
|
|
participant := storeLink.GetParticipantById(req.PartId, l.svcCtx.DbEngin)
|
|
if *participant == (models.StorelinkCenter{}) {
|
|
resp = &types.GetLinkTaskResp{}
|
|
resp.Success = false
|
|
resp.Task = nil
|
|
resp.ErrorMsg = "partId不存在"
|
|
return resp, nil
|
|
}
|
|
|
|
storelink := storeLink.NewStoreLink(l.svcCtx.OctopusRpc, l.svcCtx.ModelArtsRpc, l.svcCtx.ModelArtsImgRpc, l.svcCtx.ACRpc, participant)
|
|
if storelink == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
task, err := storelink.ILinkage.QueryTask(l.ctx, req.TaskId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if task == nil {
|
|
return nil, storeLink.ERROR_RESP_EMPTY
|
|
}
|
|
|
|
resp = &types.GetLinkTaskResp{}
|
|
//转换成统一返回类型
|
|
taskResp, err := storeLink.ConvertType(task, resp, participant)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if taskResp == nil {
|
|
return nil, storeLink.ERROR_CONVERT_EMPTY
|
|
}
|
|
|
|
return taskResp.(*types.GetLinkTaskResp), nil
|
|
}
|