mirror of
https://github.com/dutchcoders/transfer.sh.git
synced 2024-12-25 20:00:19 +01:00
CR Changes
This commit is contained in:
parent
b3f2319cba
commit
9f9e1b66b2
1 changed files with 10 additions and 3 deletions
|
@ -590,6 +590,8 @@ func NewStorjStorage(access, bucket string, logger *log.Logger) (*StorjStorage,
|
|||
|
||||
instance.bucket, err = instance.project.EnsureBucket(ctx, bucket)
|
||||
if err != nil {
|
||||
//Ignoring the error to return the one that occurred first, but try to clean up.
|
||||
_ = instance.project.Close()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
|
@ -628,9 +630,10 @@ func (s *StorjStorage) Get(token string, filename string) (reader io.ReadCloser,
|
|||
if err != nil {
|
||||
return nil, "", 0, err
|
||||
}
|
||||
|
||||
contentType = download.Info().Standard.ContentType
|
||||
contentLength = uint64(download.Info().Standard.ContentLength)
|
||||
//return the download as a reader
|
||||
|
||||
reader = download
|
||||
return
|
||||
}
|
||||
|
@ -661,18 +664,22 @@ func (s *StorjStorage) Put(token string, filename string, reader io.Reader, cont
|
|||
|
||||
n, err := io.Copy(writer, reader)
|
||||
if err != nil {
|
||||
//Ignoring the error to return the one that occurred first, but try to clean up.
|
||||
_ = writer.Abort()
|
||||
return err
|
||||
}
|
||||
|
||||
err = writer.SetMetadata(ctx, &uplink.StandardMetadata{ContentType: contentType, ContentLength: n}, nil)
|
||||
if err != nil {
|
||||
//Ignoring the error to return the one that occurred first, but try to clean up.
|
||||
_ = writer.Abort()
|
||||
return err
|
||||
}
|
||||
//Commit the object!
|
||||
|
||||
err = writer.Commit()
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *StorjStorage) IsNotExist(err error) bool {
|
||||
return storj.ErrObjectNotFound.Has(err)
|
||||
return uplink.ErrObjectNotFound.Has(err)
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue