refactored adding of differential privacy

1 files changed, 19 insertions, 0 deletions

diff --git a/config.go b/config.go
index fd80d87..c0951f4 100644
--- a/config.go
+++ b/config.go
@@ -2,8 +2,11 @@ package main
 
 import (
 	"encoding/json"
+	"errors"
+	"fmt"
 	"io/ioutil"
 	"log"
+	"time"
 )
 
 type Config struct {
@@ -12,6 +15,22 @@ type Config struct {
 	Epsilon  float64 `json:epsilon`
 }
 
+func (cfg *Config) getInterval() (interval time.Duration, err error) {
+	switch cfg.Interval {
+	case "5min":
+		interval = time.Minute * 5
+	case "10min":
+		interval = time.Minute * 10
+	case "hour":
+		interval = time.Hour
+	case "day":
+		interval = time.Hour * 24
+	default:
+		err = errors.New(fmt.Sprintf("Invalid interval: %s", cfg.Interval))
+	}
+	return
+}
+
 func readConfig() (conf Config, err error) {
 	content, err := ioutil.ReadFile("config.json")
 	if err != nil {