Backup_Repos/filestash
1
0
Fork 0
mirror of https://github.com/mickael-kerjean/filestash synced 2026-01-04 14:52:36 +01:00
Code Issues Projects Releases Wiki Activity

feature (zip): configurable timeout on zip download

Browse source
This commit is contained in:
Mickael 2021-03-29 23:44:04 +11:00
parent 9f67900902
commit cddbcfc6d1
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
server/ctrl/files.go
View file

@ -23,9 +23,10 @@ type FileInfo struct {
Time int64 `json:"time"`
}
const ZIP_TIMEOUT = 100
var FileCache AppCache
var (
FileCache AppCache
ZipTimeout int
)
func init() {
FileCache = NewAppCache()
@ -33,6 +34,17 @@ func init() {
FileCache.OnEvict(func(key string, value interface{}) {
os.RemoveAll(filepath.Join(cachePath, key))
})
ZipTimeout = Config.Get("features.protection.zip_timeout").Schema(func(f *FormElement) *FormElement{
if f == nil {
f = &FormElement{}
}
f.Default = 60
f.Name = "zip_timeout"
f.Type = "number"
f.Description = "Timeout when user wants to download archive as a zip"
f.Placeholder = "Default: 60seconds"
return f
}).Int()
}
func FileLs(ctx App, res http.ResponseWriter, req *http.Request) {
@ -426,7 +438,7 @@ func FileDownloader(ctx App, res http.ResponseWriter, req *http.Request) {
start := time.Now()
var addToZipRecursive func(App, *zip.Writer, string, string) error
addToZipRecursive = func(c App, zw *zip.Writer, backendPath string, zipRoot string) (err error) {
if time.Now().Sub(start) > ZIP_TIMEOUT * time.Second {
if time.Now().Sub(start) > time.Duration(ZipTimeout) * time.Second {
return ErrTimeout
}
if strings.HasSuffix(backendPath, "/") == false {