Commit 44642bba authored by duanjinfei's avatar duanjinfei

update task handler

parent f2771c8b
...@@ -13,9 +13,13 @@ const ( ...@@ -13,9 +13,13 @@ const (
UseRedirect = "Use-Redirect" UseRedirect = "Use-Redirect"
Prefer = "Prefer" Prefer = "Prefer"
Async = "respond-async" Async = "respond-async"
MaxExecTime = "MaxExecTime"
HealthCheckAPI = "/health-check" HealthCheckAPI = "/health-check"
ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin" ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin"
READY = "READY" READY = "READY"
ZeroHost = "0.0.0.0"
ModelPublishStatusYes = 1 ModelPublishStatusYes = 1
ModelPublishStatusNo = 2 ModelPublishStatusNo = 2
DefaultMaxExecTime = 300
DefaultTaskTimer = 2
) )
This diff is collapsed.
...@@ -111,12 +111,11 @@ func TestTaskHandler_computeTaskHandler(t1 *testing.T) { ...@@ -111,12 +111,11 @@ func TestTaskHandler_computeTaskHandler(t1 *testing.T) {
} }
for _, tt := range tests { for _, tt := range tests {
t1.Run(tt.name, func(t1 *testing.T) { t1.Run(tt.name, func(t1 *testing.T) {
t := &nm.TaskHandler{ t := &nm.TaskWorker{
Wg: tt.fields.wg, Wg: tt.fields.wg,
LruCache: tt.fields.lruCache, LruCache: tt.fields.lruCache,
DockerOp: tt.fields.DockerOp, DockerOp: tt.fields.DockerOp,
TaskMsg: tt.fields.TaskMsg, TaskMsg: tt.fields.TaskMsg,
HttpClient: tt.fields.HttpClient,
} }
tt.fields.wg.Add(1) tt.fields.wg.Add(1)
t.ComputeTaskHandler(tt.args.taskMsg) t.ComputeTaskHandler(tt.args.taskMsg)
......
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