testing new api request analytics in dashboard

1 files changed, 120 insertions, 3 deletions

diff --git a/system/api/analytics/init.go b/system/api/analytics/init.go
index 79cace0..e8416ea 100644
--- a/system/api/analytics/init.go
+++ b/system/api/analytics/init.go
@@ -11,6 +11,8 @@ import (
 
 	"runtime"
 
+	"encoding/json"
+
 	"github.com/boltdb/bolt"
 )
 
@@ -106,8 +108,123 @@ func serve() {
 	}
 }
 
-// Week returns the []byte of javascript needed to chart a week of data by day
-func Week() {
-	// get request analytics for today and the 6 days preceeding
+// Week returns the map containing decoded javascript needed to chart a week of data by day
+func Week() (map[string]interface{}, error) {
+	// set thresholds for today and the 6 days preceeding
+	times := [7]time.Time{}
+	now := time.Now()
+	today := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, time.UTC)
+
+	for i := range times {
+		// subtract 24 * i hours to make days prior
+		dur := time.Duration(24 * i * -1)
+		day := today.Add(time.Hour * dur)
+
+		// day threshold is [...n-1-i, n-1, n]
+		times[len(times)-1-i] = day
+	}
+
+	// get api request analytics from db
+	var requests = []apiRequest{}
+	err := store.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("requests"))
+
+		err := b.ForEach(func(k, v []byte) error {
+			var r apiRequest
+			err := json.Unmarshal(v, r)
+			if err != nil {
+				log.Println("Error decoding json from analytics db:", err)
+				return nil
+			}
+
+			requests = append(requests, r)
+
+			return nil
+		})
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	dates := [7]string{}
+	ips := [7]map[string]struct{}{}
+	total := [7]int{}
+	unique := [7]int{}
+
+CHECK_REQUEST:
+	for i := range requests {
+		ts := time.Unix(requests[i].Timestamp/1000, 0)
+
+		for j := range times {
+			// format times[j] (time.Time) into a MM/DD format for dates
+			dates[j] = times[j].Format("_1/_2")
+
+			// if on today, there will be no next iteration to set values for
+			// day prior so all valid requests belong to today
+			if j == len(times) {
+				if ts.After(times[j]) || ts.Equal(times[j]) {
+					// do all record keeping
+					total[j]++
+
+					if _, ok := ips[j][requests[i].RemoteAddr]; !ok {
+						unique[j-1]++
+						ips[j][requests[i].RemoteAddr] = struct{}{}
+					}
+				}
+			}
+
+			if ts.Equal(times[j]) {
+				// increment total count for current time threshold (day)
+				total[j]++
+
+				// if no IP found for current threshold, increment unique and record IP
+				if _, ok := ips[j][requests[i].RemoteAddr]; !ok {
+					unique[j]++
+					ips[j][requests[i].RemoteAddr] = struct{}{}
+				}
+			}
+
+			if ts.Before(times[j]) {
+				// check if older than earliest threshold
+				if j == 0 {
+					continue CHECK_REQUEST
+				}
+
+				// increment total count for previous time threshold (day)
+				total[j-1]++
+
+				// if no IP found for day prior, increment unique and record IP
+				if _, ok := ips[j-1][requests[i].RemoteAddr]; !ok {
+					unique[j-1]++
+					ips[j-1][requests[i].RemoteAddr] = struct{}{}
+				}
+			}
+		}
+	}
+
+	jsUnique, err := json.Marshal(unique)
+	if err != nil {
+		return nil, err
+	}
+
+	jsTotal, err := json.Marshal(total)
+	if err != nil {
+		return nil, err
+	}
+
+	jsDates, err := json.Marshal(dates)
+	if err != nil {
+		return nil, err
+	}
 
+	return map[string]interface{}{
+		"dates":  jsDates,
+		"unique": jsUnique,
+		"total":  jsTotal,
+	}, nil
 }