Commit a002bc52 authored by duanjinfei's avatar duanjinfei

handler task exec

parent 58123d00
package models package models
const ( const (
TaskType = "taskType" TaskType = "taskType"
ContainerSign = "container" ContainerSign = "container"
MinerSign = "miner" MinerSign = "miner"
ReqHash = "reqHash" ReqHash = "reqHash"
RespHash = "respHash" RespHash = "respHash"
ResultFileExpiresDB = "expires" ResultFileExpiresDB = "expires"
ContentType = "type" ContentType = "type"
RedirectCode = 303 RedirectCode = 303
UseFileCache = "USE-FILE-CACHE" UseFileCache = "USE-FILE-CACHE"
Prefer = "Prefer" Prefer = "Prefer"
Async = "respond-async" Async = "respond-async"
ModelPublishStatusYes = 1 HealthCheckAPI = "/health-check"
ModelPublishStatusNo = 2 ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin"
READY = "READY"
ModelPublishStatusYes = 1
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,11 +303,16 @@ func handlerMsg(nodeManager *models.NodeManagerClient, ...@@ -303,11 +303,16 @@ 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 containerSign == nil || len(containerSign) == 0 { if taskExecRes.TaskRespBody != nil {
log.Error("Container signing failed................") containerSign = taskMsgWorker.DockerOp.GetContainerSign(taskMsg, taskExecRes.TaskRespBody)
isSuccess = false if containerSign == nil || len(containerSign) == 0 {
taskExecRes.TaskExecError = fmt.Sprintf("%s-%s", "Container sign failed", taskExecRes.TaskExecError) log.Error("Container signing failed................")
isSuccess = false
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)
......
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