From 6486e8799b0b36cd23c859fcd4b0fbbf368884bb Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Thu, 3 Dec 2020 18:10:10 +0800
Subject: [PATCH] Fix: user storage might be returned twice when canceling
 uploading request (#645)

---
 routers/controllers/file.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/routers/controllers/file.go b/routers/controllers/file.go
index 5fb7edb..c914ed5 100644
--- a/routers/controllers/file.go
+++ b/routers/controllers/file.go
@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"net/url"
 	"strconv"
+	"sync"
 
 	model "github.com/cloudreve/Cloudreve/v3/models"
 	"github.com/cloudreve/Cloudreve/v3/pkg/filesystem"
@@ -317,6 +318,7 @@ func FileUploadStream(c *gin.Context) {
 	fs.Use("AfterUploadFailed", filesystem.HookGiveBackCapacity)
 
 	// 执行上传
+	ctx = context.WithValue(ctx, fsctx.ValidateCapacityOnceCtx, &sync.Once{})
 	uploadCtx := context.WithValue(ctx, fsctx.GinCtx, c)
 	err = fs.Upload(uploadCtx, fileData)
 	if err != nil {