Add LevelDB support to store upload state between program runs

Author: Maran

.travis.yml

@@ -6,6 +6,7 @@ go:
 install:
   - go get github.com/stretchr/testify
   - go get github.com/tus/tusd
+  - go get github.com/syndtr/goleveldb/leveldb
 script: go test -v ./...
 notifications:
   email:

README.md

@@ -50,6 +50,7 @@ Store is used to map an upload's fingerprint with the corresponding upload URL.
 | Name | Backend | Dependencies |
 |:----:|:-------:|:------------:|
 | MemoryStore | In-Memory | None |
+| LeveldbStore | LevelDB | [goleveldb](https://github.com/syndtr/goleveldb) |
 
 ## Future Work
 

leveldbstore/store.go

@@ -0,0 +1,41 @@
+package leveldbstore
+
+import (
+	"github.com/eventials/go-tus"
+	"github.com/syndtr/goleveldb/leveldb"
+)
+
+type LeveldbStore struct {
+	db *leveldb.DB
+}
+
+func NewLeveldbStore(path string) (tus.Store, error) {
+	db, err := leveldb.OpenFile(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	store := &LeveldbStore{db: db}
+	return store, err
+}
+
+func (s *LeveldbStore) Get(fingerprint string) (string, bool) {
+	url, err := s.db.Get([]byte(fingerprint), nil)
+	ok := true
+	if err != nil {
+		ok = false
+	}
+	return string(url), ok
+}
+
+func (s *LeveldbStore) Set(fingerprint, url string) {
+	s.db.Put([]byte(fingerprint), []byte(url), nil)
+}
+
+func (s *LeveldbStore) Delete(fingerprint string) {
+	s.db.Delete([]byte(fingerprint), nil)
+}
+
+func (s *LeveldbStore) Close() {
+	s.Close()
+}