update commit file cache

nm/task_msg.go

@@ -2,6 +2,7 @@ package nm
 
 import (
 	"bytes"
+	"encoding/base64"
 	"encoding/json"
 	"example.com/m/conf"
 	"example.com/m/log"
@@ -239,10 +240,10 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 				}
 				if len(containerResp.Output) == 1 {
 					if utils.IsBase64ImageStr(containerResp.Output[0]) {
-						//imageStr := strings.SplitN(containerResp.Output[0], ",", 2)[1]
+						imageStr := strings.SplitN(containerResp.Output[0], ",", 2)[1]
 						queryString := utils.MatchFileCacheQueryString(taskParam.Headers, taskCmd.ImageName, t.DockerOp.ModelsInfo)
-						ossUri, err := t.uploadOSS(taskMsg.TaskId, queryString, containerResp.Output[0])
-						//ossUri, err := t.uploadOSS(taskMsg.TaskId, queryString, imageStr)
+						//ossUri, err := t.uploadOSS(taskMsg.TaskId, queryString, containerResp.Output[0])
+						ossUri, err := t.uploadOSS(taskMsg.TaskId, queryString, imageStr)
 						if err != nil {
 							log.WithError(err).Error("upload image into file cache failed")
 							return
@@ -342,13 +343,13 @@ func (t *TaskHandler) uploadOSS(taskId string, queries string, base64Image strin
 		return "", err
 	}
 	// 将 base64 解码后的内容复制到表单字段
-	//decodedImage, err := base64.StdEncoding.DecodeString(base64Image)
-	//if err != nil {
-	//	log.WithError(err).Error("Error decoding base64 image")
-	//	return "", err
-	//}
-	//_, err = io.Copy(fileField, bytes.NewReader(decodedImage))
-	_, err = io.Copy(fileField, strings.NewReader(base64Image))
+	decodedImage, err := base64.StdEncoding.DecodeString(base64Image)
+	if err != nil {
+		log.WithError(err).Error("Error decoding base64 image")
+		return "", err
+	}
+	_, err = io.Copy(fileField, bytes.NewReader(decodedImage))
+	//_, err = io.Copy(fileField, strings.NewReader(base64Image))
 	if err != nil {
 		log.WithError(err).Error("Error copying file contents")
 		return "", err