update ReplicateImageNameSuffix

conf/config.go

@@ -9,25 +9,26 @@ import (
 )
 
 type Config struct {
-	SignPrv              string
-	SignPub              string
-	DockerServer         string
-	BenefitAddress       string
-	HeartRespTimeMillis  int64
-	ExternalIp           string
-	SignPublicAddress    common.Address
-	SignPrivateKey       *ecdsa.PrivateKey
-	NmSeed               string  `json:"nm_seed"`
-	HeartRespTimeSecond  int64   `json:"heart_response"`
-	TaskValidatorTime    float64 `json:"task_validator_time"`
-	ContainerNum         int64   `json:"container_num"`
-	NodeManagerNum       int64   `json:"node_manager_num"`
-	ChainID              int64   `json:"chain_id"`
-	ApiUrl               string  `json:"api_url"`
-	ValidatorUrl         string  `json:"validator_url"`
-	OssUrl               string  `json:"oss_url"`
-	WaitLastTaskExecTime int64   `json:"wait_last_task_exec_time"`
-	OpSys                string  `json:"op_sys"`
+	SignPrv                  string
+	SignPub                  string
+	DockerServer             string
+	BenefitAddress           string
+	HeartRespTimeMillis      int64
+	ExternalIp               string
+	SignPublicAddress        common.Address
+	SignPrivateKey           *ecdsa.PrivateKey
+	NmSeed                   string  `json:"nm_seed"`
+	HeartRespTimeSecond      int64   `json:"heart_response"`
+	TaskValidatorTime        float64 `json:"task_validator_time"`
+	ContainerNum             int64   `json:"container_num"`
+	NodeManagerNum           int64   `json:"node_manager_num"`
+	ChainID                  int64   `json:"chain_id"`
+	ApiUrl                   string  `json:"api_url"`
+	ValidatorUrl             string  `json:"validator_url"`
+	OssUrl                   string  `json:"oss_url"`
+	WaitLastTaskExecTime     int64   `json:"wait_last_task_exec_time"`
+	OpSys                    string  `json:"op_sys"`
+	ReplicateImageNameSuffix string  `json:"replicate_image_name_suffix"`
 }
 
 var _cfg *Config = nil

config.json

@@ -9,5 +9,6 @@
   "validator_url": "172.31.35.130:20011",
   "oss_url": "https://tmp-file.aigic.ai/api/v1/upload",
   "wait_last_task_exec_time": 60,
-  "op_sys": "linux"
+  "op_sys": "linux",
+  "replicate_image_name_suffix": "docker.aigic.ai/ai"
 }
\ No newline at end of file

models/const.go

 package models
 
 const (
-	TaskType                 = "taskType"
-	ContainerSign            = "container"
-	MinerSign                = "miner"
-	ReqHash                  = "reqHash"
-	RespHash                 = "respHash"
-	ResultFileExpiresDB      = "expires"
-	ContentType              = "type"
-	RedirectCode             = 303
-	UseFileCache             = "Use-File-Cache"
-	UseRedirect              = "Use-Redirect"
-	Prefer                   = "Prefer"
-	Async                    = "respond-async"
-	MaxExecTime              = "MaxExecTime"
-	HealthCheckAPI           = "/health-check"
-	ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin"
-	READY                    = "READY"
-	ZeroHost                 = "0.0.0.0"
-	ModelPublishStatusYes    = 1
-	ModelPublishStatusNo     = 2
-	DefaultMaxExecTime       = 300
-	DefaultTaskTimer         = 2
+	TaskType              = "taskType"
+	ContainerSign         = "container"
+	MinerSign             = "miner"
+	ReqHash               = "reqHash"
+	RespHash              = "respHash"
+	ResultFileExpiresDB   = "expires"
+	ContentType           = "type"
+	RedirectCode          = 303
+	UseFileCache          = "Use-File-Cache"
+	UseRedirect           = "Use-Redirect"
+	Prefer                = "Prefer"
+	Async                 = "respond-async"
+	MaxExecTime           = "MaxExecTime"
+	HealthCheckAPI        = "/health-check"
+	READY                 = "READY"
+	ZeroHost              = "0.0.0.0"
+	ModelPublishStatusYes = 1
+	ModelPublishStatusNo  = 2
+	DefaultMaxExecTime    = 300
+	DefaultTaskTimer      = 2
 )

nm/task_handler.go

@@ -476,7 +476,7 @@ func (op *TaskOp) waitContainerRunning(handler *TaskWorker, imageId string) erro
 			if !running {
 				continue
 			}
-			if isMatch := strings.HasPrefix(op.taskCmd.ImageName, models.ReplicateImageNameSuffix); isMatch {
+			if isMatch := strings.HasPrefix(op.taskCmd.ImageName, conf.GetConfig().ReplicateImageNameSuffix); isMatch {
 				if isReqSuccess, err := op.checkContainerHealthy(internalIp, internalPort); err != nil {
 					log.WithField("err", err).Errorf("check container healthy failed")
 					return fmt.Errorf("%s-%s", "check container healthy failed", err.Error())