Merge pull request #1494 from 0chain/fix/save-progress

add opt for save progress
0chain · May 18, 2024 · 912ef0c · 912ef0c
2 parents 9065d01 + 68e6d64
commit 912ef0c
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 9 deletions.
diff --git a/zboxcore/sdk/chunked_upload.go b/zboxcore/sdk/chunked_upload.go
@@ -48,6 +48,7 @@ var (
 	CancelOpCtx                      = make(map[string]context.CancelCauseFunc)
 	cancelLock                       sync.Mutex
 	CurrentMode                      = UploadModeMedium
+	shouldSaveProgress               = true
 )
 
 // DefaultChunkSize default chunk size for file and thumbnail
@@ -224,7 +225,7 @@ func CreateChunkedUpload(
 		opt(su)
 	}
 
-	if su.progressStorer == nil {
+	if su.progressStorer == nil && shouldSaveProgress {
 		su.progressStorer = createFsChunkedUploadProgress(context.Background())
 	}
 
@@ -352,27 +353,34 @@ func (su *ChunkedUpload) loadProgress() {
 	su.progress.ChunkIndex = -1
 
 	progressID := su.progressID()
+	if shouldSaveProgress {
+		progress := su.progressStorer.Load(progressID)
 
-	progress := su.progressStorer.Load(progressID)
-
-	if progress != nil {
-		su.progress = *progress
-		su.progress.ID = progressID
+		if progress != nil {
+			su.progress = *progress
+			su.progress.ID = progressID
+		}
 	}
 }
 
 // saveProgress save progress to ~/.zcn/upload/[progressID]
 func (su *ChunkedUpload) saveProgress() {
-	su.progressStorer.Save(su.progress)
+	if su.progressStorer != nil {
+		su.progressStorer.Save(su.progress)
+	}
 }
 
 // removeProgress remove progress info once it is done
 func (su *ChunkedUpload) removeProgress() {
-	su.progressStorer.Remove(su.progress.ID) //nolint
+	if su.progressStorer != nil {
+		su.progressStorer.Remove(su.progress.ID) //nolint
+	}
 }
 
 func (su *ChunkedUpload) updateProgress(chunkIndex int) {
-	su.progressStorer.Update(su.progress.ID, chunkIndex)
+	if su.progressStorer != nil {
+		su.progressStorer.Update(su.progress.ID, chunkIndex)
+	}
 }
 
 // createUploadProgress create a new UploadProgress

diff --git a/zboxcore/sdk/sdk.go b/zboxcore/sdk/sdk.go
@@ -71,6 +71,10 @@ func SetShouldVerifyHash(verify bool) {
 	shouldVerifyHash = verify
 }
 
+func SetSaveProgress(save bool) {
+	shouldSaveProgress = save
+}
+
 // GetVersion - returns version string
 func GetVersion() string {
 	return version.VERSIONSTR