diff options
Diffstat (limited to 'main.go')
| -rw-r--r-- | main.go | 19 |
1 files changed, 6 insertions, 13 deletions
@@ -10,16 +10,9 @@ import ( //"strings" ) -const ( - DATABASE_USER = "flowcleaner" - DATABASE_PASS = "nil" - DATABASE_CONNECTION = "" //e.g. "tcp(localhost:55555) - DATABASE_NAME = "pmacct" -) - func main() { log.Println("Reading config...") - conf, err := readConfig() + cfg, err := readConfig() if err != nil { log.Println("Could not read config") return @@ -37,7 +30,7 @@ func main() { */ log.Print("Cleaning data...") starttime := time.Now() - numOfRowsNotCleaned, err := cleanData(conf, DATABASE_USER, DATABASE_PASS, DATABASE_CONNECTION, DATABASE_NAME) + numOfRowsNotCleaned, err := cleanData(cfg) if err != nil { log.Println(err) log.Println("Exiting...") @@ -47,22 +40,22 @@ func main() { // If either all rows are processed or if there is no limit for the processing // we can safely add noise to the cleaned data - if (numOfRowsNotCleaned == 0 || conf.Limit == 0) && conf.Epsilon >= 0 { + if (numOfRowsNotCleaned == 0 || cfg.Limit == 0) && cfg.Epsilon >= 0 { log.Println("Adding differential privacy noise to processed data...") - db, err := sql.Open("mysql", DATABASE_USER+":"+DATABASE_PASS+"@"+DATABASE_CONNECTION+"/"+DATABASE_NAME) + db, err := sql.Open("mysql", cfg.DBUser+":"+cfg.DBPass+"@"+cfg.DBConn+"/"+cfg.DBName) if err != nil { log.Println("Failed to connect to db:", err) return } defer db.Close() - ival, err := conf.getInterval() + ival, err := cfg.getInterval() if err != nil { log.Println("erronous interval in conf prevents the privatization of data:", err) return } - err = privatizeCleaned(db, starttime.Add(-2*ival), conf) + err = privatizeCleaned(db, starttime.Add(-2*ival), cfg) if err != nil { log.Println("Failed to privatize data:", err) } |