forked from JointCloud/pcm-coordinator
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package inference
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/updater"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/models"
|
|
"time"
|
|
)
|
|
|
|
type DeployInstanceListLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewDeployInstanceListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeployInstanceListLogic {
|
|
return &DeployInstanceListLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *DeployInstanceListLogic) DeployInstanceList(req *types.DeployInstanceListReq) (resp *types.DeployInstanceListResp, err error) {
|
|
limit := req.PageSize
|
|
offset := req.PageSize * (req.PageNum - 1)
|
|
resp = &types.DeployInstanceListResp{}
|
|
|
|
var list []*models.AiInferDeployInstance
|
|
|
|
tx := l.svcCtx.DbEngin.Raw("select * from ai_infer_deploy_instance").Scan(&list)
|
|
if tx.Error != nil {
|
|
logx.Errorf(tx.Error.Error())
|
|
return nil, tx.Error
|
|
}
|
|
|
|
if len(list) == 0 {
|
|
return
|
|
}
|
|
|
|
go updater.UpdateDeployInstanceStatusBatch(l.svcCtx, list)
|
|
|
|
ins := list[0]
|
|
for i := range list {
|
|
uTime, _ := time.Parse(time.RFC3339, ins.UpdateTime)
|
|
latest, _ := time.Parse(time.RFC3339, list[i].UpdateTime)
|
|
if latest.After(uTime) {
|
|
ins = list[i]
|
|
}
|
|
}
|
|
|
|
go updater.UpdateDeployInstanceStatus(l.svcCtx, ins, true)
|
|
go updater.UpdateDeployTaskStatus(l.svcCtx)
|
|
//count total
|
|
var total int64
|
|
err = tx.Count(&total).Error
|
|
tx.Limit(limit).Offset(offset)
|
|
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
|
|
err = tx.Order("create_time desc").Find(&list).Error
|
|
if err != nil {
|
|
return nil, errors.New(err.Error())
|
|
}
|
|
resp.List = &list
|
|
resp.PageSize = req.PageSize
|
|
resp.PageNum = req.PageNum
|
|
resp.Total = total
|
|
|
|
return
|
|
}
|