storj/pkg/ranger/file.go
2018-04-23 20:16:34 -06:00

41 lines
825 B
Go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.
package ranger
import (
"io"
"os"
)
// FileHandleRanger returns a RangerCloser from a file handle. The
// RangerCloser's Close method will call fh.Close().
// Footgun: If FileHandleRanger fails, fh.Close will not have been called.
func FileHandleRanger(fh *os.File) (RangerCloser, error) {
stat, err := fh.Stat()
if err != nil {
return nil, Error.Wrap(err)
}
return struct {
Ranger
io.Closer
}{
Ranger: ReaderAtRanger(fh, stat.Size()),
Closer: fh,
}, nil
}
// FileRanger returns a RangerCloser from a path.
func FileRanger(path string) (RangerCloser, error) {
fh, err := os.Open(path)
if err != nil {
return nil, err
}
r, err := FileHandleRanger(fh)
if err != nil {
fh.Close()
return nil, err
}
return r, nil
}