mirror of
https://github.com/drakkan/sftpgo.git
synced 2025-12-07 06:40:54 +03:00
upload: avoid a stat call if not strictly required
Signed-off-by: Nicola Murino <nicola.murino@gmail.com>
This commit is contained in:
@@ -329,6 +329,21 @@ func (t *BaseTransfer) getUploadFileSize() (int64, int, error) {
|
|||||||
var fileSize int64
|
var fileSize int64
|
||||||
var deletedFiles int
|
var deletedFiles int
|
||||||
|
|
||||||
|
switch dataprovider.GetQuotaTracking() {
|
||||||
|
case 0:
|
||||||
|
return fileSize, deletedFiles, errors.New("quota tracking disabled")
|
||||||
|
case 2:
|
||||||
|
if !t.Connection.User.HasQuotaRestrictions() {
|
||||||
|
vfolder, err := t.Connection.User.GetVirtualFolderForPath(path.Dir(t.requestPath))
|
||||||
|
if err != nil {
|
||||||
|
return fileSize, deletedFiles, errors.New("quota tracking disabled for this user")
|
||||||
|
}
|
||||||
|
if vfolder.IsIncludedInUserQuota() {
|
||||||
|
return fileSize, deletedFiles, errors.New("quota tracking disabled for this user and folder included in user quota")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
info, err := t.Fs.Stat(t.fsPath)
|
info, err := t.Fs.Stat(t.fsPath)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
fileSize = info.Size()
|
fileSize = info.Size()
|
||||||
|
|||||||
@@ -306,8 +306,9 @@ func TestRemovePartialCryptoFile(t *testing.T) {
|
|||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
u := dataprovider.User{
|
u := dataprovider.User{
|
||||||
BaseUser: sdk.BaseUser{
|
BaseUser: sdk.BaseUser{
|
||||||
Username: "test",
|
Username: "test",
|
||||||
HomeDir: os.TempDir(),
|
HomeDir: os.TempDir(),
|
||||||
|
QuotaFiles: 1000000,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
conn := NewBaseConnection(fs.ConnectionID(), ProtocolSFTP, "", "", u)
|
conn := NewBaseConnection(fs.ConnectionID(), ProtocolSFTP, "", "", u)
|
||||||
|
|||||||
Reference in New Issue
Block a user