Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
P
power-node
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Odysseus
power-node
Commits
a002bc52
Commit
a002bc52
authored
Mar 21, 2024
by
duanjinfei
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
handler task exec
parent
58123d00
Changes
4
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
175 additions
and
109 deletions
+175
-109
const.go
models/const.go
+16
-13
node_manager.go
models/node_manager.go
+4
-0
start.go
nm/start.go
+10
-5
task_msg.go
nm/task_msg.go
+145
-91
No files found.
models/const.go
View file @
a002bc52
package
models
const
(
TaskType
=
"taskType"
ContainerSign
=
"container"
MinerSign
=
"miner"
ReqHash
=
"reqHash"
RespHash
=
"respHash"
ResultFileExpiresDB
=
"expires"
ContentType
=
"type"
RedirectCode
=
303
UseFileCache
=
"USE-FILE-CACHE"
Prefer
=
"Prefer"
Async
=
"respond-async"
ModelPublishStatusYes
=
1
ModelPublishStatusNo
=
2
TaskType
=
"taskType"
ContainerSign
=
"container"
MinerSign
=
"miner"
ReqHash
=
"reqHash"
RespHash
=
"respHash"
ResultFileExpiresDB
=
"expires"
ContentType
=
"type"
RedirectCode
=
303
UseFileCache
=
"USE-FILE-CACHE"
Prefer
=
"Prefer"
Async
=
"respond-async"
HealthCheckAPI
=
"/health-check"
ReplicateImageNameSuffix
=
"docker.agicoin.ai/agicoin"
READY
=
"READY"
ModelPublishStatusYes
=
1
ModelPublishStatusNo
=
2
)
models/node_manager.go
View file @
a002bc52
...
...
@@ -77,6 +77,10 @@ type ModelInfo struct {
PublishStatus
int
`json:"publish_status"`
}
type
HealthyCheck
struct
{
Status
string
`json:"status"`
}
type
FileCacheResult
struct
{
Code
int
`json:"code"`
Msg
string
`json:"msg"`
...
...
nm/start.go
View file @
a002bc52
...
...
@@ -303,11 +303,16 @@ func handlerMsg(nodeManager *models.NodeManagerClient,
taskExecRes
=
taskExecResInterface
.
(
*
models
.
TaskResult
)
}
isSuccess
:=
taskExecRes
.
TaskIsSuccess
containerSign
:=
taskMsgWorker
.
DockerOp
.
GetContainerSign
(
taskMsg
,
taskExecRes
.
TaskRespBody
)
if
containerSign
==
nil
||
len
(
containerSign
)
==
0
{
log
.
Error
(
"Container signing failed................"
)
isSuccess
=
false
taskExecRes
.
TaskExecError
=
fmt
.
Sprintf
(
"%s-%s"
,
"Container sign failed"
,
taskExecRes
.
TaskExecError
)
containerSign
:=
make
([]
byte
,
0
)
if
taskExecRes
.
TaskRespBody
!=
nil
{
containerSign
=
taskMsgWorker
.
DockerOp
.
GetContainerSign
(
taskMsg
,
taskExecRes
.
TaskRespBody
)
if
containerSign
==
nil
||
len
(
containerSign
)
==
0
{
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
)
params
:=
buildParams
(
taskMsg
.
TaskId
,
containerSign
,
minerSign
,
taskExecRes
,
isSuccess
)
...
...
nm/task_msg.go
View file @
a002bc52
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment