Ran go formatter
This commit is contained in:
parent
766dabc238
commit
4c1f2b99a2
|
@ -1,8 +1,8 @@
|
||||||
package tusd
|
package tusd
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"errors"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Error indicating that the data store has locked the file for further edits.
|
// Error indicating that the data store has locked the file for further edits.
|
||||||
|
|
|
@ -7,7 +7,7 @@
|
||||||
package filestore
|
package filestore
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
@ -25,16 +25,16 @@ type FileStore struct {
|
||||||
// Relative or absolute path to store files in. FileStore does not check
|
// Relative or absolute path to store files in. FileStore does not check
|
||||||
// whether the path exists, you os.MkdirAll in this case on your own.
|
// whether the path exists, you os.MkdirAll in this case on your own.
|
||||||
Path string
|
Path string
|
||||||
locks map[string]bool
|
locks map[string]bool
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewFileStore creates a new FileStore instance.
|
// NewFileStore creates a new FileStore instance.
|
||||||
func NewFileStore(path string) (store *FileStore) {
|
func NewFileStore(path string) (store *FileStore) {
|
||||||
store = &FileStore{
|
store = &FileStore{
|
||||||
Path: path,
|
Path: path,
|
||||||
locks: make(map[string]bool),
|
locks: make(map[string]bool),
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (store *FileStore) NewUpload(info tusd.FileInfo) (id string, err error) {
|
func (store *FileStore) NewUpload(info tusd.FileInfo) (id string, err error) {
|
||||||
|
@ -55,11 +55,11 @@ func (store *FileStore) NewUpload(info tusd.FileInfo) (id string, err error) {
|
||||||
|
|
||||||
func (store *FileStore) WriteChunk(id string, offset int64, src io.Reader) (int64, error) {
|
func (store *FileStore) WriteChunk(id string, offset int64, src io.Reader) (int64, error) {
|
||||||
if !store.getLock(id) {
|
if !store.getLock(id) {
|
||||||
return 0, tusd.ErrFileLocked
|
return 0, tusd.ErrFileLocked
|
||||||
}
|
}
|
||||||
defer store.clearLock(id)
|
defer store.clearLock(id)
|
||||||
|
|
||||||
file, err := os.OpenFile(store.binPath(id), os.O_WRONLY|os.O_APPEND, defaultFilePerm)
|
file, err := os.OpenFile(store.binPath(id), os.O_WRONLY|os.O_APPEND, defaultFilePerm)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
@ -75,7 +75,7 @@ func (store *FileStore) WriteChunk(id string, offset int64, src io.Reader) (int6
|
||||||
}
|
}
|
||||||
|
|
||||||
func (store *FileStore) GetInfo(id string) (info tusd.FileInfo, err error) {
|
func (store *FileStore) GetInfo(id string) (info tusd.FileInfo, err error) {
|
||||||
data, err := ioutil.ReadFile(store.infoPath(id))
|
data, err := ioutil.ReadFile(store.infoPath(id))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return info, err
|
return info, err
|
||||||
}
|
}
|
||||||
|
@ -85,17 +85,17 @@ func (store *FileStore) GetInfo(id string) (info tusd.FileInfo, err error) {
|
||||||
|
|
||||||
func (store *FileStore) GetReader(id string) (io.Reader, error) {
|
func (store *FileStore) GetReader(id string) (io.Reader, error) {
|
||||||
if !store.getLock(id) {
|
if !store.getLock(id) {
|
||||||
return bytes.NewReader(make([]byte, 0)), tusd.ErrFileLocked
|
return bytes.NewReader(make([]byte, 0)), tusd.ErrFileLocked
|
||||||
}
|
}
|
||||||
defer store.clearLock(id)
|
defer store.clearLock(id)
|
||||||
return os.Open(store.binPath(id))
|
return os.Open(store.binPath(id))
|
||||||
}
|
}
|
||||||
|
|
||||||
func (store *FileStore) Terminate(id string) error {
|
func (store *FileStore) Terminate(id string) error {
|
||||||
if !store.getLock(id) {
|
if !store.getLock(id) {
|
||||||
return tusd.ErrFileLocked
|
return tusd.ErrFileLocked
|
||||||
}
|
}
|
||||||
defer store.clearLock(id)
|
defer store.clearLock(id)
|
||||||
if err := os.Remove(store.infoPath(id)); err != nil {
|
if err := os.Remove(store.infoPath(id)); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -142,16 +142,16 @@ func (store *FileStore) setOffset(id string, offset int64) error {
|
||||||
|
|
||||||
// getLock obtains a lock on reading/writing data for the given file ID.
|
// getLock obtains a lock on reading/writing data for the given file ID.
|
||||||
func (store *FileStore) getLock(id string) (hasLock bool) {
|
func (store *FileStore) getLock(id string) (hasLock bool) {
|
||||||
if _, locked := store.locks[id]; locked {
|
if _, locked := store.locks[id]; locked {
|
||||||
hasLock = false
|
hasLock = false
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
store.locks[id] = true
|
store.locks[id] = true
|
||||||
hasLock = true
|
hasLock = true
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// clearLock removes the lock for the given file ID.
|
// clearLock removes the lock for the given file ID.
|
||||||
func (store *FileStore) clearLock(id string) {
|
func (store *FileStore) clearLock(id string) {
|
||||||
delete(store.locks, id)
|
delete(store.locks, id)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue