modules/urfs_client/urchin/schedule.go

@@ -71,8 +71,8 @@ func GetGPUDataBack(cloudbrainID int64, jobName, centerId string) error {
 			endpoint, bucket, objectKey, destPeerHost, grampus.GetGPUModelObjectKey(jobName), err)
 		_, err = models.InsertScheduleRecord(&models.ScheduleRecord{
 			CloudbrainID:    cloudbrainID,
-			EndPoint:        "",
-			Bucket:          "",
+			EndPoint:        endpoint,
+			Bucket:          bucket,
 			ObjectKey:       objectKey,
 			ProxyServer:     destPeerHost,
 			Status:          models.StorageScheduleFailed,
@@ -90,9 +90,9 @@ func GetGPUDataBack(cloudbrainID int64, jobName, centerId string) error {
 
 	_, err = models.InsertScheduleRecord(&models.ScheduleRecord{
 		CloudbrainID:    cloudbrainID,
-		EndPoint:        res.DataEndpoint,
-		Bucket:          res.DataRoot,
-		ObjectKey:       res.DataPath,
+		EndPoint:        endpoint,
+		Bucket:          bucket,
+		ObjectKey:       objectKey,
 		ProxyServer:     destPeerHost,
 		Status:          res.StatusCode,
 		IsDir:           true,
@@ -155,9 +155,9 @@ func GetNPUDataBack(cloudbrainID int64, jobName, centerId string) error {
 			endpoint, bucket, objectKey, destPeerHost, err)
 		_, err = models.InsertScheduleRecord(&models.ScheduleRecord{
 			CloudbrainID:  cloudbrainID,
-			EndPoint:      "",
-			Bucket:        "",
-			ObjectKey:     destBucket,
+			EndPoint:      endpoint,
+			Bucket:        bucket,
+			ObjectKey:     objectKey,
 			ProxyServer:   destPeerHost,
 			Status:        models.StorageScheduleFailed,
 			IsDir:         false,
@@ -173,9 +173,9 @@ func GetNPUDataBack(cloudbrainID int64, jobName, centerId string) error {
 
 	_, err = models.InsertScheduleRecord(&models.ScheduleRecord{
 		CloudbrainID:  cloudbrainID,
-		EndPoint:      res.DataEndpoint,
-		Bucket:        res.DataRoot,
-		ObjectKey:     res.DataPath,
+		EndPoint:      endpoint,
+		Bucket:        bucket,
+		ObjectKey:     objectKey,
 		ProxyServer:   destPeerHost,
 		Status:        res.StatusCode,
 		IsDir:         false,