Commit a002bc52 authored by duanjinfei's avatar duanjinfei

handler task exec

parent 58123d00
...@@ -12,6 +12,9 @@ const ( ...@@ -12,6 +12,9 @@ const (
UseFileCache = "USE-FILE-CACHE" UseFileCache = "USE-FILE-CACHE"
Prefer = "Prefer" Prefer = "Prefer"
Async = "respond-async" Async = "respond-async"
HealthCheckAPI = "/health-check"
ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin"
READY = "READY"
ModelPublishStatusYes = 1 ModelPublishStatusYes = 1
ModelPublishStatusNo = 2 ModelPublishStatusNo = 2
) )
...@@ -77,6 +77,10 @@ type ModelInfo struct { ...@@ -77,6 +77,10 @@ type ModelInfo struct {
PublishStatus int `json:"publish_status"` PublishStatus int `json:"publish_status"`
} }
type HealthyCheck struct {
Status string `json:"status"`
}
type FileCacheResult struct { type FileCacheResult struct {
Code int `json:"code"` Code int `json:"code"`
Msg string `json:"msg"` Msg string `json:"msg"`
......
...@@ -303,12 +303,17 @@ func handlerMsg(nodeManager *models.NodeManagerClient, ...@@ -303,12 +303,17 @@ func handlerMsg(nodeManager *models.NodeManagerClient,
taskExecRes = taskExecResInterface.(*models.TaskResult) taskExecRes = taskExecResInterface.(*models.TaskResult)
} }
isSuccess := taskExecRes.TaskIsSuccess isSuccess := taskExecRes.TaskIsSuccess
containerSign := taskMsgWorker.DockerOp.GetContainerSign(taskMsg, taskExecRes.TaskRespBody) containerSign := make([]byte, 0)
if taskExecRes.TaskRespBody != nil {
containerSign = taskMsgWorker.DockerOp.GetContainerSign(taskMsg, taskExecRes.TaskRespBody)
if containerSign == nil || len(containerSign) == 0 { if containerSign == nil || len(containerSign) == 0 {
log.Error("Container signing failed................") log.Error("Container signing failed................")
isSuccess = false isSuccess = false
taskExecRes.TaskExecError = fmt.Sprintf("%s-%s", "Container sign failed", taskExecRes.TaskExecError) taskExecRes.TaskExecError = fmt.Sprintf("%s-%s", "Container sign failed", taskExecRes.TaskExecError)
} }
} else {
taskExecRes.TaskExecError = fmt.Sprintf("%s-%s", "Task exec error", taskExecRes.TaskExecError)
}
reqHash, respHash, minerSign := taskMsgWorker.GetMinerSign(taskMsg, taskExecRes.TaskRespBody) reqHash, respHash, minerSign := taskMsgWorker.GetMinerSign(taskMsg, taskExecRes.TaskRespBody)
params := buildParams(taskMsg.TaskId, containerSign, minerSign, taskExecRes, isSuccess) params := buildParams(taskMsg.TaskId, containerSign, minerSign, taskExecRes, isSuccess)
taskMsgWorker.Mutex.Lock() taskMsgWorker.Mutex.Lock()
......
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment