gifs/api

Paddy 2014-08-27 Child:eb450538f079

gifs/api/usage.go

Spike out functionality and tests. Create our interfaces around storing data and retrieving it. Create an in-memory implementation of our interfaces, for testing and rapid dev purposes. Begin sketching out what our unit tests look like. Create our Google Cloud Storage datastore implementation. Sketch out an idea for a usage collection process to keep track of which users are actually using stuff.

Download raw file

View source Diff to previous Annotate

     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/usage.go	Wed Aug 27 22:34:02 2014 -0400
     1.3 @@ -0,0 +1,87 @@
     1.4 +package api
     1.5 +
     1.6 +import (
     1.7 +	"sync"
     1.8 +)
     1.9 +
    1.10 +func NewUsageTracker() *UsageTracker {
    1.11 +	return &UsageTracker{
    1.12 +		usages: make(map[string]*Usage),
    1.13 +	}
    1.14 +}
    1.15 +
    1.16 +type UsageTracker struct {
    1.17 +	usages map[string]*Usage
    1.18 +	sync.Mutex
    1.19 +}
    1.20 +
    1.21 +func (u *UsageTracker) TrackUploads(id string) (bytes, requests chan int64) {
    1.22 +	u.Lock()
    1.23 +	defer u.Unlock()
    1.24 +	if _, ok := u.usages[id]; !ok {
    1.25 +		u.usages[id] = &Usage{
    1.26 +			UploadBytesChan:      make(chan int64),
    1.27 +			DownloadBytesChan:    make(chan int64),
    1.28 +			UploadRequestsChan:   make(chan int64),
    1.29 +			DownloadRequestsChan: make(chan int64),
    1.30 +		}
    1.31 +		go u.usages[id].collect()
    1.32 +	}
    1.33 +	return u.usages[id].UploadBytesChan, u.usages[id].UploadRequestsChan
    1.34 +}
    1.35 +
    1.36 +func (u *UsageTracker) TrackDownloads(id string) (bytes, requests chan int64) {
    1.37 +	u.Lock()
    1.38 +	defer u.Unlock()
    1.39 +	if _, ok := u.usages[id]; !ok {
    1.40 +		u.usages[id] = &Usage{
    1.41 +			UploadBytesChan:      make(chan int64),
    1.42 +			DownloadBytesChan:    make(chan int64),
    1.43 +			UploadRequestsChan:   make(chan int64),
    1.44 +			DownloadRequestsChan: make(chan int64),
    1.45 +		}
    1.46 +		go u.usages[id].collect()
    1.47 +	}
    1.48 +	return u.usages[id].DownloadBytesChan, u.usages[id].DownloadRequestsChan
    1.49 +}
    1.50 +
    1.51 +type Usage struct {
    1.52 +	UploadedBytes        int64
    1.53 +	UploadBytesChan      chan int64
    1.54 +	DownloadedBytes      int64
    1.55 +	DownloadBytesChan    chan int64
    1.56 +	UploadRequests       int64
    1.57 +	UploadRequestsChan   chan int64
    1.58 +	DownloadRequests     int64
    1.59 +	DownloadRequestsChan chan int64
    1.60 +}
    1.61 +
    1.62 +func (u *Usage) collect() {
    1.63 +	for {
    1.64 +		select {
    1.65 +		case b, ok := <-u.UploadBytesChan:
    1.66 +			if !ok {
    1.67 +				u.UploadBytesChan = nil
    1.68 +			}
    1.69 +			u.UploadedBytes += b
    1.70 +		case r, ok := <-u.UploadRequestsChan:
    1.71 +			if !ok {
    1.72 +				u.UploadRequestsChan = nil
    1.73 +			}
    1.74 +			u.UploadRequests += r
    1.75 +		case b, ok := <-u.DownloadBytesChan:
    1.76 +			if !ok {
    1.77 +				u.DownloadBytesChan = nil
    1.78 +			}
    1.79 +			u.DownloadedBytes += b
    1.80 +		case r, ok := <-u.DownloadRequestsChan:
    1.81 +			if !ok {
    1.82 +				u.DownloadRequestsChan = nil
    1.83 +			}
    1.84 +			u.DownloadRequests += r
    1.85 +		}
    1.86 +		if u.UploadBytesChan == nil && u.UploadRequestsChan == nil && u.DownloadBytesChan == nil && u.DownloadRequestsChan == nil {
    1.87 +			break
    1.88 +		}
    1.89 +	}
    1.90 +}