Browse Source

updated statussync

pull/483/head
tzwang 5 months ago
parent
commit
d637d3e75b
4 changed files with 38 additions and 22 deletions
  1. +1
    -1
      internal/logic/inference/createinferencetasklogic.go
  2. +10
    -7
      internal/scheduler/service/utils/jcs/middleware.go
  3. +6
    -6
      internal/scheduler/service/utils/status/deployInstance.go
  4. +21
    -8
      internal/scheduler/service/utils/status/statusSync.go

+ 1
- 1
internal/logic/inference/createinferencetasklogic.go View File

@@ -80,7 +80,7 @@ func (l *CreateInferenceTaskLogic) CreateInferenceTask(req *types.CreateInferenc
err = l.createInferenceTask(taskId, adapterClusterMap, opt)
if err != nil {
if len(assignedClusters) != 0 {
_ = status.ReportStatus(l.svcCtx, taskName, strconv.FormatInt(taskId, 10), assignedClusters[0].ClusterId, "", false, "")
_ = status.ReportInferenceStatusMessages(l.svcCtx, nil, taskName, strconv.FormatInt(taskId, 10), assignedClusters[0].ClusterId, "", false, "")
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}
return nil, err


+ 10
- 7
internal/scheduler/service/utils/jcs/middleware.go View File

@@ -24,13 +24,16 @@ type TrainReportMessage struct {
Output string `json:"output"`
}
type InferReportMessage struct {
Type string `json:"type"`
TaskName string `json:"taskName"`
TaskID string `json:"taskID"`
Status bool `json:"status"`
Message string `json:"message"`
ClusterID string `json:"clusterID"`
Url string `json:"url"`
Type string `json:"type"`
TaskName string `json:"taskName"`
TaskID string `json:"taskID"`
Status bool `json:"status"`
Message string `json:"message"`
Url string `json:"url"`
ID string `json:"id"`
AdapterID string `json:"adapterId"`
ClusterID string `json:"clusterId"`
InstanceID string `json:"instanceId"`
}

func StatusReport(url string, report interface{}) error {


+ 6
- 6
internal/scheduler/service/utils/status/deployInstance.go View File

@@ -99,7 +99,7 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
return
}
url := ins.InferUrl
err := ReportStatus(svc, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), url, true, "")
err := ReportInferenceStatusMessages(svc, instance, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), url, true, "")
if err != nil {
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}
@@ -127,7 +127,7 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
return
}
url := ins.InferUrl
err := ReportStatus(svc, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), url, true, "")
err := ReportInferenceStatusMessages(svc, instance, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), url, true, "")
if err != nil {
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}
@@ -149,7 +149,7 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
}
return
}
err := ReportStatus(svc, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), "", false, ins.Status)
err := ReportInferenceStatusMessages(svc, instance, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), "", false, ins.Status)
if err != nil {
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}
@@ -191,7 +191,7 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
return
}
url := ins.InferUrl
err := ReportStatus(svc, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), url, true, "")
err := ReportInferenceStatusMessages(svc, instance, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), url, true, "")
if err != nil {
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}
@@ -213,7 +213,7 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
}
return
}
err := ReportStatus(svc, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), "", false, ins.Status)
err := ReportInferenceStatusMessages(svc, instance, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), "", false, ins.Status)
if err != nil {
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}
@@ -226,7 +226,7 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
}
return
}
err := ReportStatus(svc, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), "", false, ins.Status)
err := ReportInferenceStatusMessages(svc, instance, instance.InstanceName, strconv.FormatInt(instance.DeployInstanceTaskId, 10), strconv.FormatInt(instance.ClusterId, 10), "", false, ins.Status)
if err != nil {
logx.Errorf("############ Report Infer Task Status Message Error %s", err.Error())
}


+ 21
- 8
internal/scheduler/service/utils/status/statusSync.go View File

@@ -226,16 +226,29 @@ func (s *TaskStatus) reportStatusMessages(task *types.TaskModel, aiTask *models.
return nil
}

func ReportStatus(svc *svc.ServiceContext, taskName string, taskId string, clusterId string, url string, status bool, msg string) error {
func ReportInferenceStatusMessages(svc *svc.ServiceContext, ins *models.AiInferDeployInstance, taskName string, taskId string, clusterId string, url string, status bool, msg string) error {
var id string
var adapterID string
var clusterID string
var instanceID string
if ins != nil {
id = strconv.FormatInt(ins.Id, 10)
adapterID = strconv.FormatInt(ins.AdapterId, 10)
clusterID = strconv.FormatInt(ins.ClusterId, 10)
instanceID = ins.InstanceId
}
report := &jcs.JobStatusReportReq{}
reportMsg := &jcs.InferReportMessage{
Type: "Inference",
TaskName: taskName,
TaskID: taskId,
Status: status,
Message: msg,
ClusterID: clusterId,
Url: url,
Type: "Inference",
TaskName: taskName,
TaskID: taskId,
Status: status,
Message: msg,
Url: url,
ID: id,
AdapterID: adapterID,
ClusterID: clusterID,
InstanceID: instanceID,
}
report.Report = reportMsg



Loading…
Cancel
Save