Add basic file upload retry with a hardcoded limit

This commit is contained in:
Gabe Kangas
2020-06-17 22:01:53 -07:00
parent 728b5fe3d7
commit e63d0656ba
4 changed files with 10 additions and 6 deletions

View File

@@ -36,14 +36,14 @@ func (s *S3Storage) Setup(configuration Config) {
s.sess = s.connectAWS()
}
func (s *S3Storage) Save(filePath string) string {
func (s *S3Storage) Save(filePath string, retryCount int) string {
// fmt.Println("Saving", filePath)
file, err := os.Open(filePath)
defer file.Close()
if err != nil {
log.Fatal(err)
log.Errorln(err)
}
uploader := s3manager.NewUploader(s.sess)
@@ -55,7 +55,11 @@ func (s *S3Storage) Save(filePath string) string {
})
if err != nil {
panic(err)
log.Errorln(err)
if retryCount < 4 {
log.Println("Retrying...")
s.Save(filePath, retryCount+1)
}
}
// fmt.Println("Uploaded", filePath, "to", response.Location)