LumeWeb
/
portal
1
0
Fork 0
Code Issues 5 Pull Requests Packages Projects Releases Wiki Activity

refactor: add terminateUpload method

This commit is contained in:
Derrick Hammer 2023-05-17 13:34:27 -04:00
parent 80561f89e9
commit 0a90ff6439
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tus/tus.go
View File

@ -140,6 +140,10 @@ func tusWorker(upload *tusd.Upload) error {
_, err = files.Upload(file.(io.ReadSeeker), info.Size) _, err = files.Upload(file.(io.ReadSeeker), info.Size)
if err != nil { if err != nil {
log.Print(err) log.Print(err)
err1 := terminateUpload(*upload)
if err1 != nil {
return err1
}
return err return err
} }
@ -147,14 +151,29 @@ func tusWorker(upload *tusd.Upload) error {
var tusUpload model.Tus var tusUpload model.Tus
ret := db.Get().Where(&model.Tus{Hash: hash}).First(&tusUpload) ret := db.Get().Where(&model.Tus{Hash: hash}).First(&tusUpload)
if ret.Error != nil && ret.Error.Error() != "record not found" { if ret.Error != nil && ret.Error.Error() != "record not found" {
log.Print(ret.Error) log.Print(ret.Error)
err1 := terminateUpload(*upload)
if err1 != nil {
return err1
}
return err return err
} }
ret = db.Get().Delete(&tusUpload) ret = db.Get().Delete(&tusUpload)
err = shared.GetTusComposer().Terminater.AsTerminatableUpload(*upload).Terminate(context.Background()) err = terminateUpload(*upload)
if err != nil {
return err
}
return nil
}
func terminateUpload(upload tusd.Upload) error {
err := shared.GetTusComposer().Terminater.AsTerminatableUpload(upload).Terminate(context.Background())
if err != nil { if err != nil {
log.Print(err) log.Print(err)