Implement upload updates via channel.

Author: alexandrevicenzi

Diff for: .travis.yml

@@ -2,6 +2,8 @@ language: go
 go:
   - 1.6
   - 1.7
+  - 1.8
+  - 1.9
   - tip
 install:
   - go get github.com/stretchr/testify

Diff for: Dockerfile

@@ -1,8 +1,9 @@
-FROM golang:1.7
+FROM golang:1.9
 
 RUN mkdir -p /go/src/github.com/eventials/go-tus
 
 WORKDIR /go/src/github.com/eventials/go-tus
 
 RUN go get github.com/stretchr/testify
 RUN go get github.com/tus/tusd
+RUN go get github.com/syndtr/goleveldb/leveldb

Diff for: README.md

@@ -49,8 +49,8 @@ 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) |
+| MemoryStore  | In-Memory | None |
+| LeveldbStore | LevelDB   | [goleveldb](https://github.com/syndtr/goleveldb) |
 
 ## Future Work
 

Diff for: upload.go

@@ -13,12 +13,33 @@ type Metadata map[string]string
 type Upload struct {
 	stream io.ReadSeeker
 	size   int64
+	offset int64
 
 	Fingerprint string
 	Metadata    Metadata
 }
 
-// Size returns the size of the upload body.
+// Updates the Upload information based on offset.
+func (u *Upload) updateProgress(offset int64) {
+	u.offset = offset
+}
+
+// Returns whether this upload is finished or not.
+func (u *Upload) Finished() bool {
+	return u.offset >= u.size
+}
+
+// Returns the progress in a percentage.
+func (u *Upload) Progress() int64 {
+	return (u.offset * 100) / u.size
+}
+
+// Returns the current upload offset.
+func (u *Upload) Offset() int64 {
+	return u.offset
+}
+
+// Returns the size of the upload body.
 func (u *Upload) Size() int64 {
 	return u.size
 }

Diff for: uploader.go

@@ -5,11 +5,18 @@ import (
 )
 
 type Uploader struct {
-	client  *Client
-	url     string
-	upload  *Upload
-	offset  int64
-	aborted bool
+	client     *Client
+	url        string
+	upload     *Upload
+	offset     int64
+	aborted    bool
+	uploadSubs []chan Upload
+	notifyChan chan bool
+}
+
+// Subscribes to progress updates.
+func (u *Uploader) NotifyUploadProgress(c chan Upload) {
+	u.uploadSubs = append(u.uploadSubs, c)
 }
 
 // Abort aborts the upload process.
@@ -72,16 +79,37 @@ func (u *Uploader) UploadChunck() error {
 
 	u.offset = newOffset
 
+	u.upload.updateProgress(u.offset)
+
+	u.notifyChan <- true
+
 	return nil
 }
 
+// Waits for a signal to broadcast to all subscribers
+func (u *Uploader) broadcastProgress() {
+	for _ = range u.notifyChan {
+		for _, c := range u.uploadSubs {
+			c <- *u.upload
+		}
+	}
+}
+
 // NewUploader creates a new Uploader.
 func NewUploader(client *Client, url string, upload *Upload, offset int64) *Uploader {
-	return &Uploader{
+	notifyChan := make(chan bool)
+
+	uploader := &Uploader{
 		client,
 		url,
 		upload,
 		offset,
 		false,
+		nil,
+		notifyChan,
 	}
+
+	go uploader.broadcastProgress()
+
+	return uploader
 }