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
96e90c07
Commit
96e90c07
authored
Mar 16, 2024
by
duanjinfei
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update use file cache
parent
68b1d9d1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
10 deletions
+10
-10
task_msg.go
nm/task_msg.go
+10
-10
No files found.
nm/task_msg.go
View file @
96e90c07
...
...
@@ -232,8 +232,8 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
}
}
log
.
WithField
(
"isUseFileCache"
,
isUseFileCache
)
.
Info
(
"is use file cache"
)
if
isUseFileCache
&&
readBody
!=
nil
{
base64Image
StrArr
:=
make
([]
string
,
0
)
if
readBody
!=
nil
{
containerResp
StrArr
:=
make
([]
string
,
0
)
data
:=
parseData
(
readBody
)
if
data
!=
nil
{
switch
v
:=
data
.
(
type
)
{
...
...
@@ -242,7 +242,7 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
res
:=
data
.
([][]
string
)
for
_
,
innerSlice
:=
range
res
{
for
_
,
str
:=
range
innerSlice
{
base64ImageStrArr
=
append
(
base64Image
StrArr
,
str
)
containerRespStrArr
=
append
(
containerResp
StrArr
,
str
)
}
}
log
.
Info
(
"data is [][]string type"
)
...
...
@@ -251,23 +251,23 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
{
res
:=
data
.
([]
string
)
for
_
,
slice
:=
range
res
{
base64ImageStrArr
=
append
(
base64Image
StrArr
,
slice
)
containerRespStrArr
=
append
(
containerResp
StrArr
,
slice
)
}
log
.
Info
(
"data is []string type"
)
}
case
string
:
{
res
:=
data
.
(
string
)
base64ImageStrArr
=
append
(
base64Image
StrArr
,
res
)
containerRespStrArr
=
append
(
containerResp
StrArr
,
res
)
log
.
Info
(
"data is string type"
)
}
default
:
log
.
Error
(
"data is unknown type"
,
v
)
}
apiRes
:=
make
([]
string
,
0
)
for
_
,
base64ImageStr
:=
range
base64Image
StrArr
{
isBase64
,
decodeByte
,
respFormat
,
suffix
:=
utils
.
IsBase64ImageStr
(
base64Image
Str
)
if
isBase64
{
for
_
,
respStr
:=
range
containerResp
StrArr
{
isBase64
,
decodeByte
,
respFormat
,
suffix
:=
utils
.
IsBase64ImageStr
(
resp
Str
)
if
isBase64
&&
isUseFileCache
{
log
.
WithField
(
"taskId"
,
taskMsg
.
TaskId
)
.
WithField
(
"format"
,
respFormat
)
.
WithField
(
"suffix"
,
suffix
)
.
Info
(
"Parse container resp"
)
queryString
:=
utils
.
MatchFileCacheQueryString
(
taskParam
.
Headers
,
taskCmd
.
ImageName
,
t
.
DockerOp
.
ModelsInfo
,
respFormat
)
ossUri
,
err
:=
t
.
uploadOSS
(
taskMsg
.
TaskId
,
queryString
,
decodeByte
,
suffix
)
...
...
@@ -276,7 +276,7 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
return
}
log
.
WithField
(
"uri"
,
ossUri
)
.
Info
(
"upload image OSS successful"
)
if
ossUri
!=
""
&&
len
(
base64Image
StrArr
)
==
1
{
if
ossUri
!=
""
&&
len
(
containerResp
StrArr
)
==
1
{
taskExecResult
.
TaskHttpStatusCode
=
models
.
RedirectCode
post
.
Header
.
Set
(
"Location"
,
ossUri
)
}
...
...
@@ -284,7 +284,7 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
apiRes
=
append
(
apiRes
,
ossUri
)
}
}
else
{
apiRes
=
append
(
apiRes
,
base64Image
Str
)
apiRes
=
append
(
apiRes
,
resp
Str
)
}
}
if
len
(
apiRes
)
>
1
{
...
...
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