commit
b6d67debee
|
@ -0,0 +1,3 @@
|
||||||
|
[submodule "vendor/github.com/nightlyone/lockfile"]
|
||||||
|
path = vendor/github.com/nightlyone/lockfile
|
||||||
|
url = https://github.com/nightlyone/lockfile.git
|
|
@ -3,8 +3,14 @@ go:
|
||||||
- 1.2
|
- 1.2
|
||||||
- 1.3
|
- 1.3
|
||||||
- 1.4
|
- 1.4
|
||||||
|
- 1.5
|
||||||
|
- 1.6
|
||||||
- tip
|
- tip
|
||||||
|
|
||||||
|
env:
|
||||||
|
global:
|
||||||
|
- GO15VENDOREXPERIMENT=1
|
||||||
|
|
||||||
matrix:
|
matrix:
|
||||||
allow_failures:
|
allow_failures:
|
||||||
- go: tip
|
- go: tip
|
||||||
|
|
|
@ -43,10 +43,8 @@ func main() {
|
||||||
stderr.Fatalf("Unable to ensure directory exists: %s", err)
|
stderr.Fatalf("Unable to ensure directory exists: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
var store tusd.DataStore
|
var store tusd.TerminaterDataStore
|
||||||
store = filestore.FileStore{
|
store = filestore.New(dir)
|
||||||
Path: dir,
|
|
||||||
}
|
|
||||||
|
|
||||||
if storeSize > 0 {
|
if storeSize > 0 {
|
||||||
store = limitedstore.New(storeSize, store)
|
store = limitedstore.New(storeSize, store)
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
|
@ -7,6 +7,8 @@ import (
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type concatPartialStore struct {
|
type concatPartialStore struct {
|
||||||
|
|
|
@ -1,8 +1,10 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestCORS(t *testing.T) {
|
func TestCORS(t *testing.T) {
|
||||||
|
|
29
datastore.go
29
datastore.go
|
@ -51,6 +51,14 @@ type DataStore interface {
|
||||||
// If the returned reader also implements the io.Closer interface, the
|
// If the returned reader also implements the io.Closer interface, the
|
||||||
// Close() method will be invoked once everything has been read.
|
// Close() method will be invoked once everything has been read.
|
||||||
GetReader(id string) (io.Reader, error)
|
GetReader(id string) (io.Reader, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TerminaterDataStore is the interface which must be implemented by DataStores
|
||||||
|
// if they want to receive DELETE requests using the Handler. If this interface
|
||||||
|
// is not implemented, no request handler for this method is attached.
|
||||||
|
type TerminaterDataStore interface {
|
||||||
|
DataStore
|
||||||
|
|
||||||
// Terminate an upload so any further requests to the resource, both reading
|
// Terminate an upload so any further requests to the resource, both reading
|
||||||
// and writing, must return os.ErrNotExist or similar.
|
// and writing, must return os.ErrNotExist or similar.
|
||||||
Terminate(id string) error
|
Terminate(id string) error
|
||||||
|
@ -61,3 +69,24 @@ type FinisherDataStore interface {
|
||||||
|
|
||||||
FinishUpload(id string) error
|
FinishUpload(id string) error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// LockerDataStore is the interface required for custom lock persisting mechanisms.
|
||||||
|
// Common ways to store this information is in memory, on disk or using an
|
||||||
|
// external service, such as ZooKeeper.
|
||||||
|
// When multiple processes are attempting to access an upload, whether it be
|
||||||
|
// by reading or writing, a syncronization mechanism is required to prevent
|
||||||
|
// data corruption, especially to ensure correct offset values and the proper
|
||||||
|
// order of chunks inside a single upload.
|
||||||
|
type LockerDataStore interface {
|
||||||
|
DataStore
|
||||||
|
|
||||||
|
// LockUpload attempts to obtain an exclusive lock for the upload specified
|
||||||
|
// by its id.
|
||||||
|
// If this operation fails because the resource is already locked, the
|
||||||
|
// tusd.ErrFileLocked must be returned. If no error is returned, the attempt
|
||||||
|
// is consider to be successful and the upload to be locked until UnlockUpload
|
||||||
|
// is invoked for the same upload.
|
||||||
|
LockUpload(id string) error
|
||||||
|
// UnlockUpload releases an existing lock for the given upload.
|
||||||
|
UnlockUpload(id string) error
|
||||||
|
}
|
||||||
|
|
|
@ -1,9 +1,18 @@
|
||||||
|
// Package filestore provide a storage backend based on the local file system.
|
||||||
|
//
|
||||||
// FileStore is a storage backend used as a tusd.DataStore in tusd.NewHandler.
|
// FileStore is a storage backend used as a tusd.DataStore in tusd.NewHandler.
|
||||||
// It stores the uploads in a directory specified in two different files: The
|
// It stores the uploads in a directory specified in two different files: The
|
||||||
// `[id].info` files are used to store the fileinfo in JSON format. The
|
// `[id].info` files are used to store the fileinfo in JSON format. The
|
||||||
// `[id].bin` files contain the raw binary data uploaded.
|
// `[id].bin` files contain the raw binary data uploaded.
|
||||||
// No cleanup is performed so you may want to run a cronjob to ensure your disk
|
// No cleanup is performed so you may want to run a cronjob to ensure your disk
|
||||||
// is not filled up with old and finished uploads.
|
// is not filled up with old and finished uploads.
|
||||||
|
//
|
||||||
|
// In addition, it provides an exclusive upload locking mechansim using lock files
|
||||||
|
// which are stored on disk. Each of them stores the PID of the process which
|
||||||
|
// aquired the lock. This allows locks to be automatically freed when a process
|
||||||
|
// is unable to release it on its own because the process is not alive anymore.
|
||||||
|
// For more information, consult the documentation for tusd.LockerDataStore
|
||||||
|
// interface, which is implemented by FileStore
|
||||||
package filestore
|
package filestore
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
@ -11,9 +20,12 @@ import (
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
"github.com/tus/tusd"
|
"github.com/tus/tusd"
|
||||||
"github.com/tus/tusd/uid"
|
"github.com/tus/tusd/uid"
|
||||||
|
|
||||||
|
"github.com/nightlyone/lockfile"
|
||||||
)
|
)
|
||||||
|
|
||||||
var defaultFilePerm = os.FileMode(0775)
|
var defaultFilePerm = os.FileMode(0775)
|
||||||
|
@ -22,10 +34,18 @@ var defaultFilePerm = os.FileMode(0775)
|
||||||
// methods.
|
// methods.
|
||||||
type FileStore struct {
|
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, use os.MkdirAll in this case on your own.
|
||||||
Path string
|
Path string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// New creates a new file based storage backend. The directory specified will
|
||||||
|
// be used as the only storage entry. This method does not check
|
||||||
|
// whether the path exists, use os.MkdirAll to ensure.
|
||||||
|
// In addition, a locking mechanism is provided.
|
||||||
|
func New(path string) FileStore {
|
||||||
|
return FileStore{path}
|
||||||
|
}
|
||||||
|
|
||||||
func (store FileStore) NewUpload(info tusd.FileInfo) (id string, err error) {
|
func (store FileStore) NewUpload(info tusd.FileInfo) (id string, err error) {
|
||||||
id = uid.Uid()
|
id = uid.Uid()
|
||||||
info.ID = id
|
info.ID = id
|
||||||
|
@ -82,17 +102,62 @@ func (store FileStore) Terminate(id string) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return the path to the .bin storing the binary data
|
func (store FileStore) LockUpload(id string) error {
|
||||||
|
lock, err := store.newLock(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = lock.TryLock()
|
||||||
|
if err == lockfile.ErrBusy {
|
||||||
|
return tusd.ErrFileLocked
|
||||||
|
}
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (store FileStore) UnlockUpload(id string) error {
|
||||||
|
lock, err := store.newLock(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = lock.Unlock()
|
||||||
|
|
||||||
|
// A "no such file or directory" will be returned if no lockfile was found.
|
||||||
|
// Since this means that the file has never been locked, we drop the error
|
||||||
|
// and continue as if nothing happend.
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
err = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// newLock contructs a new Lockfile instance.
|
||||||
|
func (store FileStore) newLock(id string) (lockfile.Lockfile, error) {
|
||||||
|
path, err := filepath.Abs(store.Path + "/" + id + ".lock")
|
||||||
|
if err != nil {
|
||||||
|
return lockfile.Lockfile(""), err
|
||||||
|
}
|
||||||
|
|
||||||
|
// We use Lockfile directly instead of lockfile.New to bypass the unnecessary
|
||||||
|
// check whether the provided path is absolute since we just resolved it
|
||||||
|
// on our own.
|
||||||
|
return lockfile.Lockfile(path), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// binPath returns the path to the .bin storing the binary data.
|
||||||
func (store FileStore) binPath(id string) string {
|
func (store FileStore) binPath(id string) string {
|
||||||
return store.Path + "/" + id + ".bin"
|
return store.Path + "/" + id + ".bin"
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return the path to the .info file storing the file's info
|
// infoPath returns the path to the .info file storing the file's info.
|
||||||
func (store FileStore) infoPath(id string) string {
|
func (store FileStore) infoPath(id string) string {
|
||||||
return store.Path + "/" + id + ".info"
|
return store.Path + "/" + id + ".info"
|
||||||
}
|
}
|
||||||
|
|
||||||
// Update the entire information. Everything will be overwritten.
|
// writeInfo updates the entire information. Everything will be overwritten.
|
||||||
func (store FileStore) writeInfo(id string, info tusd.FileInfo) error {
|
func (store FileStore) writeInfo(id string, info tusd.FileInfo) error {
|
||||||
data, err := json.Marshal(info)
|
data, err := json.Marshal(info)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -101,7 +166,7 @@ func (store FileStore) writeInfo(id string, info tusd.FileInfo) error {
|
||||||
return ioutil.WriteFile(store.infoPath(id), data, defaultFilePerm)
|
return ioutil.WriteFile(store.infoPath(id), data, defaultFilePerm)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Update the .info file using the new upload.
|
// setOffset updates the .info file to match the new offset.
|
||||||
func (store FileStore) setOffset(id string, offset int64) error {
|
func (store FileStore) setOffset(id string, offset int64) error {
|
||||||
info, err := store.GetInfo(id)
|
info, err := store.GetInfo(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
|
@ -95,3 +95,29 @@ func TestFilestore(t *testing.T) {
|
||||||
t.Fatal("expected os.ErrIsNotExist")
|
t.Fatal("expected os.ErrIsNotExist")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestFileLocker(t *testing.T) {
|
||||||
|
dir, err := ioutil.TempDir("", "tusd-file-locker")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var locker tusd.LockerDataStore
|
||||||
|
locker = FileStore{dir}
|
||||||
|
|
||||||
|
if err := locker.LockUpload("one"); err != nil {
|
||||||
|
t.Errorf("unexpected error when locking file: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := locker.LockUpload("one"); err != tusd.ErrFileLocked {
|
||||||
|
t.Errorf("expected error when locking locked file: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := locker.UnlockUpload("one"); err != nil {
|
||||||
|
t.Errorf("unexpected error when unlocking file: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := locker.UnlockUpload("one"); err != nil {
|
||||||
|
t.Errorf("unexpected error when unlocking file again: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
|
@ -6,6 +6,8 @@ import (
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type getStore struct {
|
type getStore struct {
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
// Package tusd provides ways to accept tusd calls using HTTP.
|
||||||
package tusd
|
package tusd
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
@ -38,9 +39,13 @@ func NewHandler(config Config) (*Handler, error) {
|
||||||
mux.Post("", http.HandlerFunc(handler.PostFile))
|
mux.Post("", http.HandlerFunc(handler.PostFile))
|
||||||
mux.Head(":id", http.HandlerFunc(handler.HeadFile))
|
mux.Head(":id", http.HandlerFunc(handler.HeadFile))
|
||||||
mux.Get(":id", http.HandlerFunc(handler.GetFile))
|
mux.Get(":id", http.HandlerFunc(handler.GetFile))
|
||||||
mux.Del(":id", http.HandlerFunc(handler.DelFile))
|
|
||||||
mux.Add("PATCH", ":id", http.HandlerFunc(handler.PatchFile))
|
mux.Add("PATCH", ":id", http.HandlerFunc(handler.PatchFile))
|
||||||
|
|
||||||
|
// Only attach the DELETE handler if the Terminate() method is provided
|
||||||
|
if _, ok := config.DataStore.(TerminaterDataStore); ok {
|
||||||
|
mux.Del(":id", http.HandlerFunc(handler.DelFile))
|
||||||
|
}
|
||||||
|
|
||||||
return routedHandler, nil
|
return routedHandler, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
|
@ -7,6 +7,8 @@ import (
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type zeroStore struct{}
|
type zeroStore struct{}
|
||||||
|
@ -26,10 +28,6 @@ func (store zeroStore) GetReader(id string) (io.Reader, error) {
|
||||||
return nil, ErrNotImplemented
|
return nil, ErrNotImplemented
|
||||||
}
|
}
|
||||||
|
|
||||||
func (store zeroStore) Terminate(id string) error {
|
|
||||||
return ErrNotImplemented
|
|
||||||
}
|
|
||||||
|
|
||||||
type httpTest struct {
|
type httpTest struct {
|
||||||
Name string
|
Name string
|
||||||
|
|
||||||
|
|
|
@ -1,9 +1,11 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type headStore struct {
|
type headStore struct {
|
||||||
|
|
|
@ -1,4 +1,6 @@
|
||||||
// Package limitedstore implements a simple wrapper around existing
|
// Package limitedstore provides a storage with a limited space.
|
||||||
|
//
|
||||||
|
// This goal is achieved by using a simple wrapper around existing
|
||||||
// datastores (tusd.DataStore) while limiting the used storage size.
|
// datastores (tusd.DataStore) while limiting the used storage size.
|
||||||
// It will start terminating existing uploads if not enough space is left in
|
// It will start terminating existing uploads if not enough space is left in
|
||||||
// order to create a new upload.
|
// order to create a new upload.
|
||||||
|
@ -20,7 +22,7 @@ import (
|
||||||
|
|
||||||
type LimitedStore struct {
|
type LimitedStore struct {
|
||||||
StoreSize int64
|
StoreSize int64
|
||||||
tusd.DataStore
|
tusd.TerminaterDataStore
|
||||||
|
|
||||||
uploads map[string]int64
|
uploads map[string]int64
|
||||||
usedSize int64
|
usedSize int64
|
||||||
|
@ -40,11 +42,13 @@ func (p pairlist) Len() int { return len(p) }
|
||||||
func (p pairlist) Swap(i, j int) { p[i], p[j] = p[j], p[i] }
|
func (p pairlist) Swap(i, j int) { p[i], p[j] = p[j], p[i] }
|
||||||
func (p pairlist) Less(i, j int) bool { return p[i].value < p[j].value }
|
func (p pairlist) Less(i, j int) bool { return p[i].value < p[j].value }
|
||||||
|
|
||||||
// Create a new limited store with the given size as the maximum storage size
|
// New creates a new limited store with the given size as the maximum storage
|
||||||
func New(storeSize int64, dataStore tusd.DataStore) *LimitedStore {
|
// size. The wrapped data store needs to implement the TerminaterDataStore
|
||||||
|
// interface, in order to provide the required Terminate method.
|
||||||
|
func New(storeSize int64, dataStore tusd.TerminaterDataStore) *LimitedStore {
|
||||||
return &LimitedStore{
|
return &LimitedStore{
|
||||||
StoreSize: storeSize,
|
StoreSize: storeSize,
|
||||||
DataStore: dataStore,
|
TerminaterDataStore: dataStore,
|
||||||
uploads: make(map[string]int64),
|
uploads: make(map[string]int64),
|
||||||
mutex: new(sync.Mutex),
|
mutex: new(sync.Mutex),
|
||||||
}
|
}
|
||||||
|
@ -58,7 +62,7 @@ func (store *LimitedStore) NewUpload(info tusd.FileInfo) (string, error) {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
id, err := store.DataStore.NewUpload(info)
|
id, err := store.TerminaterDataStore.NewUpload(info)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
@ -77,7 +81,7 @@ func (store *LimitedStore) Terminate(id string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (store *LimitedStore) terminate(id string) error {
|
func (store *LimitedStore) terminate(id string) error {
|
||||||
err := store.DataStore.Terminate(id)
|
err := store.TerminaterDataStore.Terminate(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,53 @@
|
||||||
|
// Package memorylocker provides an in-memory locking mechanism
|
||||||
|
//
|
||||||
|
// When multiple processes are attempting to access an upload, whether it be
|
||||||
|
// by reading or writing, a syncronization mechanism is required to prevent
|
||||||
|
// data corruption, especially to ensure correct offset values and the proper
|
||||||
|
// order of chunks inside a single upload.
|
||||||
|
//
|
||||||
|
// MemoryLocker persists locks using memory and therefore allowing a simple and
|
||||||
|
// cheap mechansim. Locks will only exist as long as this object is kept in
|
||||||
|
// reference and will be erased if the program exits.
|
||||||
|
package memorylocker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/tus/tusd"
|
||||||
|
)
|
||||||
|
|
||||||
|
// MemoryLocker persists locks using memory and therefore allowing a simple and
|
||||||
|
// cheap mechansim. Locks will only exist as long as this object is kept in
|
||||||
|
// reference and will be erased if the program exits.
|
||||||
|
type MemoryLocker struct {
|
||||||
|
tusd.DataStore
|
||||||
|
locks map[string]bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new lock memory wrapper around the provided storage.
|
||||||
|
func NewMemoryLocker(store tusd.DataStore) *MemoryLocker {
|
||||||
|
return &MemoryLocker{
|
||||||
|
DataStore: store,
|
||||||
|
locks: make(map[string]bool),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// LockUpload tries to obtain the exclusive lock.
|
||||||
|
func (locker *MemoryLocker) LockUpload(id string) error {
|
||||||
|
|
||||||
|
// Ensure file is not locked
|
||||||
|
if _, ok := locker.locks[id]; ok {
|
||||||
|
return tusd.ErrFileLocked
|
||||||
|
}
|
||||||
|
|
||||||
|
locker.locks[id] = true
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnlockUpload releases a lock. If no such lock exists, no error will be returned.
|
||||||
|
func (locker *MemoryLocker) UnlockUpload(id string) error {
|
||||||
|
// Deleting a non-existing key does not end in unexpected errors or panic
|
||||||
|
// since this operation results in a no-op
|
||||||
|
delete(locker.locks, id)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -0,0 +1,46 @@
|
||||||
|
package memorylocker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/tus/tusd"
|
||||||
|
)
|
||||||
|
|
||||||
|
type zeroStore struct{}
|
||||||
|
|
||||||
|
func (store zeroStore) NewUpload(info tusd.FileInfo) (string, error) {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
func (store zeroStore) WriteChunk(id string, offset int64, src io.Reader) (int64, error) {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (store zeroStore) GetInfo(id string) (tusd.FileInfo, error) {
|
||||||
|
return tusd.FileInfo{}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (store zeroStore) GetReader(id string) (io.Reader, error) {
|
||||||
|
return nil, tusd.ErrNotImplemented
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMemoryLocker(t *testing.T) {
|
||||||
|
var locker tusd.LockerDataStore
|
||||||
|
locker = NewMemoryLocker(&zeroStore{})
|
||||||
|
|
||||||
|
if err := locker.LockUpload("one"); err != nil {
|
||||||
|
t.Errorf("unexpected error when locking file: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := locker.LockUpload("one"); err != tusd.ErrFileLocked {
|
||||||
|
t.Errorf("expected error when locking locked file: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := locker.UnlockUpload("one"); err != nil {
|
||||||
|
t.Errorf("unexpected error when unlocking file: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := locker.UnlockUpload("one"); err != nil {
|
||||||
|
t.Errorf("unexpected error when unlocking file again: %s", err)
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,8 +1,10 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestOptions(t *testing.T) {
|
func TestOptions(t *testing.T) {
|
||||||
|
@ -15,7 +17,7 @@ func TestOptions(t *testing.T) {
|
||||||
Method: "OPTIONS",
|
Method: "OPTIONS",
|
||||||
Code: http.StatusNoContent,
|
Code: http.StatusNoContent,
|
||||||
ResHeader: map[string]string{
|
ResHeader: map[string]string{
|
||||||
"Tus-Extension": "creation,concatenation,termination",
|
"Tus-Extension": "creation,concatenation",
|
||||||
"Tus-Version": "1.0.0",
|
"Tus-Version": "1.0.0",
|
||||||
"Tus-Resumable": "1.0.0",
|
"Tus-Resumable": "1.0.0",
|
||||||
"Tus-Max-Size": "400",
|
"Tus-Max-Size": "400",
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
|
@ -7,6 +7,8 @@ import (
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type patchStore struct {
|
type patchStore struct {
|
||||||
|
@ -204,3 +206,87 @@ func TestPatchOverflow(t *testing.T) {
|
||||||
Code: http.StatusNoContent,
|
Code: http.StatusNoContent,
|
||||||
}).Run(handler, t)
|
}).Run(handler, t)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
LOCK = iota
|
||||||
|
INFO
|
||||||
|
WRITE
|
||||||
|
UNLOCK
|
||||||
|
END
|
||||||
|
)
|
||||||
|
|
||||||
|
type lockingPatchStore struct {
|
||||||
|
zeroStore
|
||||||
|
callOrder chan int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s lockingPatchStore) GetInfo(id string) (FileInfo, error) {
|
||||||
|
s.callOrder <- INFO
|
||||||
|
|
||||||
|
return FileInfo{
|
||||||
|
Offset: 0,
|
||||||
|
Size: 20,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s lockingPatchStore) WriteChunk(id string, offset int64, src io.Reader) (int64, error) {
|
||||||
|
s.callOrder <- WRITE
|
||||||
|
|
||||||
|
return 5, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s lockingPatchStore) LockUpload(id string) error {
|
||||||
|
s.callOrder <- LOCK
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s lockingPatchStore) UnlockUpload(id string) error {
|
||||||
|
s.callOrder <- UNLOCK
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLockingPatch(t *testing.T) {
|
||||||
|
callOrder := make(chan int, 10)
|
||||||
|
|
||||||
|
handler, _ := NewHandler(Config{
|
||||||
|
DataStore: lockingPatchStore{
|
||||||
|
callOrder: callOrder,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
(&httpTest{
|
||||||
|
Name: "Uploading to locking store",
|
||||||
|
Method: "PATCH",
|
||||||
|
URL: "yes",
|
||||||
|
ReqHeader: map[string]string{
|
||||||
|
"Tus-Resumable": "1.0.0",
|
||||||
|
"Content-Type": "application/offset+octet-stream",
|
||||||
|
"Upload-Offset": "0",
|
||||||
|
},
|
||||||
|
ReqBody: strings.NewReader("hello"),
|
||||||
|
Code: http.StatusNoContent,
|
||||||
|
}).Run(handler, t)
|
||||||
|
|
||||||
|
callOrder <- END
|
||||||
|
close(callOrder)
|
||||||
|
|
||||||
|
if <-callOrder != LOCK {
|
||||||
|
t.Error("expected call to LockUpload")
|
||||||
|
}
|
||||||
|
|
||||||
|
if <-callOrder != INFO {
|
||||||
|
t.Error("expected call to GetInfo")
|
||||||
|
}
|
||||||
|
|
||||||
|
if <-callOrder != WRITE {
|
||||||
|
t.Error("expected call to WriteChunk")
|
||||||
|
}
|
||||||
|
|
||||||
|
if <-callOrder != UNLOCK {
|
||||||
|
t.Error("expected call to UnlockUpload")
|
||||||
|
}
|
||||||
|
|
||||||
|
if <-callOrder != END {
|
||||||
|
t.Error("expected no more calls to happen")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -1,8 +1,10 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type postStore struct {
|
type postStore struct {
|
||||||
|
|
|
@ -1,8 +1,10 @@
|
||||||
package tusd
|
package tusd_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
. "github.com/tus/tusd"
|
||||||
)
|
)
|
||||||
|
|
||||||
type terminateStore struct {
|
type terminateStore struct {
|
||||||
|
@ -24,6 +26,16 @@ func TestTerminate(t *testing.T) {
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
|
(&httpTest{
|
||||||
|
Name: "Successful OPTIONS request",
|
||||||
|
Method: "OPTIONS",
|
||||||
|
URL: "",
|
||||||
|
ResHeader: map[string]string{
|
||||||
|
"Tus-Extension": "creation,concatenation,termination",
|
||||||
|
},
|
||||||
|
Code: http.StatusNoContent,
|
||||||
|
}).Run(handler, t)
|
||||||
|
|
||||||
(&httpTest{
|
(&httpTest{
|
||||||
Name: "Successful request",
|
Name: "Successful request",
|
||||||
Method: "DELETE",
|
Method: "DELETE",
|
||||||
|
@ -34,3 +46,19 @@ func TestTerminate(t *testing.T) {
|
||||||
Code: http.StatusNoContent,
|
Code: http.StatusNoContent,
|
||||||
}).Run(handler, t)
|
}).Run(handler, t)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestTerminateNotImplemented(t *testing.T) {
|
||||||
|
handler, _ := NewHandler(Config{
|
||||||
|
DataStore: zeroStore{},
|
||||||
|
})
|
||||||
|
|
||||||
|
(&httpTest{
|
||||||
|
Name: "TerminaterDataStore not implemented",
|
||||||
|
Method: "DELETE",
|
||||||
|
URL: "foo",
|
||||||
|
ReqHeader: map[string]string{
|
||||||
|
"Tus-Resumable": "1.0.0",
|
||||||
|
},
|
||||||
|
Code: http.StatusMethodNotAllowed,
|
||||||
|
}).Run(handler, t)
|
||||||
|
}
|
||||||
|
|
|
@ -75,8 +75,8 @@ type UnroutedHandler struct {
|
||||||
dataStore DataStore
|
dataStore DataStore
|
||||||
isBasePathAbs bool
|
isBasePathAbs bool
|
||||||
basePath string
|
basePath string
|
||||||
locks map[string]bool
|
|
||||||
logger *log.Logger
|
logger *log.Logger
|
||||||
|
extensions string
|
||||||
|
|
||||||
// For each finished upload the corresponding info object will be sent using
|
// For each finished upload the corresponding info object will be sent using
|
||||||
// this unbuffered channel. The NotifyCompleteUploads property in the Config
|
// this unbuffered channel. The NotifyCompleteUploads property in the Config
|
||||||
|
@ -109,14 +109,20 @@ func NewUnroutedHandler(config Config) (*UnroutedHandler, error) {
|
||||||
base = "/" + base
|
base = "/" + base
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Only promote extesions using the Tus-Extension header which are implemented
|
||||||
|
extensions := "creation,concatenation"
|
||||||
|
if _, ok := config.DataStore.(TerminaterDataStore); ok {
|
||||||
|
extensions += ",termination"
|
||||||
|
}
|
||||||
|
|
||||||
handler := &UnroutedHandler{
|
handler := &UnroutedHandler{
|
||||||
config: config,
|
config: config,
|
||||||
dataStore: config.DataStore,
|
dataStore: config.DataStore,
|
||||||
basePath: base,
|
basePath: base,
|
||||||
isBasePathAbs: uri.IsAbs(),
|
isBasePathAbs: uri.IsAbs(),
|
||||||
locks: make(map[string]bool),
|
|
||||||
CompleteUploads: make(chan FileInfo),
|
CompleteUploads: make(chan FileInfo),
|
||||||
logger: logger,
|
logger: logger,
|
||||||
|
extensions: extensions,
|
||||||
}
|
}
|
||||||
|
|
||||||
return handler, nil
|
return handler, nil
|
||||||
|
@ -169,7 +175,7 @@ func (handler *UnroutedHandler) Middleware(h http.Handler) http.Handler {
|
||||||
}
|
}
|
||||||
|
|
||||||
header.Set("Tus-Version", "1.0.0")
|
header.Set("Tus-Version", "1.0.0")
|
||||||
header.Set("Tus-Extension", "creation,concatenation,termination")
|
header.Set("Tus-Extension", handler.extensions)
|
||||||
|
|
||||||
w.WriteHeader(http.StatusNoContent)
|
w.WriteHeader(http.StatusNoContent)
|
||||||
return
|
return
|
||||||
|
@ -259,6 +265,16 @@ func (handler *UnroutedHandler) HeadFile(w http.ResponseWriter, r *http.Request)
|
||||||
handler.sendError(w, r, err)
|
handler.sendError(w, r, err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if locker, ok := handler.dataStore.(LockerDataStore); ok {
|
||||||
|
if err := locker.LockUpload(id); err != nil {
|
||||||
|
handler.sendError(w, r, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer locker.UnlockUpload(id)
|
||||||
|
}
|
||||||
|
|
||||||
info, err := handler.dataStore.GetInfo(id)
|
info, err := handler.dataStore.GetInfo(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
handler.sendError(w, r, err)
|
handler.sendError(w, r, err)
|
||||||
|
@ -288,8 +304,7 @@ func (handler *UnroutedHandler) HeadFile(w http.ResponseWriter, r *http.Request)
|
||||||
w.WriteHeader(http.StatusNoContent)
|
w.WriteHeader(http.StatusNoContent)
|
||||||
}
|
}
|
||||||
|
|
||||||
// PatchFile adds a chunk to an upload. Only allowed if the upload is not
|
// PatchFile adds a chunk to an upload. Only allowed enough space is left.
|
||||||
// locked and enough space is left.
|
|
||||||
func (handler *UnroutedHandler) PatchFile(w http.ResponseWriter, r *http.Request) {
|
func (handler *UnroutedHandler) PatchFile(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|
||||||
// Check for presence of application/offset+octet-stream
|
// Check for presence of application/offset+octet-stream
|
||||||
|
@ -311,19 +326,14 @@ func (handler *UnroutedHandler) PatchFile(w http.ResponseWriter, r *http.Request
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ensure file is not locked
|
if locker, ok := handler.dataStore.(LockerDataStore); ok {
|
||||||
if _, ok := handler.locks[id]; ok {
|
if err := locker.LockUpload(id); err != nil {
|
||||||
handler.sendError(w, r, ErrFileLocked)
|
handler.sendError(w, r, err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lock file for further writes (heads are allowed)
|
defer locker.UnlockUpload(id)
|
||||||
handler.locks[id] = true
|
}
|
||||||
|
|
||||||
// File will be unlocked regardless of an error or success
|
|
||||||
defer func() {
|
|
||||||
delete(handler.locks, id)
|
|
||||||
}()
|
|
||||||
|
|
||||||
info, err := handler.dataStore.GetInfo(id)
|
info, err := handler.dataStore.GetInfo(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -398,19 +408,14 @@ func (handler *UnroutedHandler) GetFile(w http.ResponseWriter, r *http.Request)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ensure file is not locked
|
if locker, ok := handler.dataStore.(LockerDataStore); ok {
|
||||||
if _, ok := handler.locks[id]; ok {
|
if err := locker.LockUpload(id); err != nil {
|
||||||
handler.sendError(w, r, ErrFileLocked)
|
handler.sendError(w, r, err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lock file for further writes (heads are allowed)
|
defer locker.UnlockUpload(id)
|
||||||
handler.locks[id] = true
|
}
|
||||||
|
|
||||||
// File will be unlocked regardless of an error or success
|
|
||||||
defer func() {
|
|
||||||
delete(handler.locks, id)
|
|
||||||
}()
|
|
||||||
|
|
||||||
info, err := handler.dataStore.GetInfo(id)
|
info, err := handler.dataStore.GetInfo(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -443,27 +448,29 @@ func (handler *UnroutedHandler) GetFile(w http.ResponseWriter, r *http.Request)
|
||||||
|
|
||||||
// DelFile terminates an upload permanently.
|
// DelFile terminates an upload permanently.
|
||||||
func (handler *UnroutedHandler) DelFile(w http.ResponseWriter, r *http.Request) {
|
func (handler *UnroutedHandler) DelFile(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// Abort the request handling if the required interface is not implemented
|
||||||
|
tstore, ok := handler.config.DataStore.(TerminaterDataStore)
|
||||||
|
if !ok {
|
||||||
|
handler.sendError(w, r, ErrNotImplemented)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
id, err := extractIDFromPath(r.URL.Path)
|
id, err := extractIDFromPath(r.URL.Path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
handler.sendError(w, r, err)
|
handler.sendError(w, r, err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ensure file is not locked
|
if locker, ok := handler.dataStore.(LockerDataStore); ok {
|
||||||
if _, ok := handler.locks[id]; ok {
|
if err := locker.LockUpload(id); err != nil {
|
||||||
handler.sendError(w, r, ErrFileLocked)
|
handler.sendError(w, r, err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lock file for further writes (heads are allowed)
|
defer locker.UnlockUpload(id)
|
||||||
handler.locks[id] = true
|
}
|
||||||
|
|
||||||
// File will be unlocked regardless of an error or success
|
err = tstore.Terminate(id)
|
||||||
defer func() {
|
|
||||||
delete(handler.locks, id)
|
|
||||||
}()
|
|
||||||
|
|
||||||
err = handler.dataStore.Terminate(id)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
handler.sendError(w, r, err)
|
handler.sendError(w, r, err)
|
||||||
return
|
return
|
||||||
|
|
|
@ -0,0 +1 @@
|
||||||
|
Subproject commit 22754258d2b05a18f75f228588041de6fe9fdcc8
|
Loading…
Reference in New Issue