35f98ce162
-Added travis support -updated travis to analyze code beneath the root. -refactored upload.go to fix travis errors. -gocyclo should ignore test files. $GOFILES needed to be adjusted. -fix rows.Close() ignoring error. Created func to handle so defer can be used when needed also. -fixed ignored errors. -fixed unit test that was not passing correctly to anonymous function. -fixed govet error for passing param inside go func. -removed returned error, in favor of logging instead. -added error logging for ignored error. -fixed potential race conditions. -removed unused append -fixed time usage to align with go standards. -removed unused variables -made changes for code review. -code comments for exported functions. -Documented bitmap.go and insert into contact list. -Documented dht, message, bootstrap -Fixed comment typos -Documented message,node, routing_table, testing in DHT package. -Documented server, client, prism, server and shared in peer and reflector packages. -Documented the stores in Store package. -made defer adjustments inline and deleted the separate function. -adjusted method in upload to take the only parameter it requires.
97 lines
2 KiB
Go
97 lines
2 KiB
Go
package store
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/lbryio/lbry.go/errors"
|
|
)
|
|
|
|
// FileBlobStore is a local disk store.
|
|
type FileBlobStore struct {
|
|
dir string
|
|
|
|
initialized bool
|
|
}
|
|
|
|
// NewFileBlobStore returns an initialized file disk store pointer.
|
|
func NewFileBlobStore(dir string) *FileBlobStore {
|
|
return &FileBlobStore{dir: dir}
|
|
}
|
|
|
|
func (f *FileBlobStore) path(hash string) string {
|
|
return path.Join(f.dir, hash)
|
|
}
|
|
|
|
func (f *FileBlobStore) initOnce() error {
|
|
if f.initialized {
|
|
return nil
|
|
}
|
|
defer func() { f.initialized = true }()
|
|
|
|
if stat, err := os.Stat(f.dir); err != nil {
|
|
if os.IsNotExist(err) {
|
|
err2 := os.Mkdir(f.dir, 0755)
|
|
if err2 != nil {
|
|
return err2
|
|
}
|
|
} else {
|
|
return err
|
|
}
|
|
} else if !stat.IsDir() {
|
|
return errors.Err("blob dir exists but is not a dir")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Has returns T/F or Error if it the blob stored already. It will error with any IO disk error.
|
|
func (f *FileBlobStore) Has(hash string) (bool, error) {
|
|
err := f.initOnce()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
_, err = os.Stat(f.path(hash))
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// Get returns the byte slice of the blob stored or will error if the blob doesn't exist.
|
|
func (f *FileBlobStore) Get(hash string) ([]byte, error) {
|
|
err := f.initOnce()
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
|
|
file, err := os.Open(f.path(hash))
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return []byte{}, errors.Err(ErrBlobNotFound)
|
|
}
|
|
return []byte{}, err
|
|
}
|
|
|
|
return ioutil.ReadAll(file)
|
|
}
|
|
|
|
// Put stores the blob on disk or errors with any IO error.
|
|
func (f *FileBlobStore) Put(hash string, blob []byte) error {
|
|
err := f.initOnce()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ioutil.WriteFile(f.path(hash), blob, 0644)
|
|
}
|
|
|
|
// PutSD stores the sd blob on the disk or errors with any IO error.
|
|
func (f *FileBlobStore) PutSD(hash string, blob []byte) error {
|
|
//Todo - need to handle when streaming hash is not present.
|
|
return f.Put(hash, blob)
|
|
}
|