From 77bd3bb48c86a5378ff85ec804215b96358da1d9 Mon Sep 17 00:00:00 2001 From: Dmitry Lyukov Date: Sun, 27 Mar 2016 02:14:30 +0300 Subject: [PATCH] Added export to csv --- src/main.go | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/src/main.go b/src/main.go index d51de68..14adeb3 100644 --- a/src/main.go +++ b/src/main.go @@ -241,21 +241,23 @@ func main() { outputPath := flag.String("o", "", "Path to export file") flag.Parse() log.Println("keyboardId =", *keyboardId, "outputPath =", *outputPath) + + // Opening database + db, err := sql.Open("sqlite3", DATABASE_NAME) + if err != nil { + log.Fatal(err) + } + db.SetMaxIdleConns(5) + db.SetMaxOpenConns(5) + defer db.Close() + + keyMap := GetKeymap() + switch { case *keyboardId == -1 && *outputPath == "": flag.PrintDefaults() return case *keyboardId != -1: - // Opening database - db, err := sql.Open("sqlite3", DATABASE_NAME) - if err != nil { - log.Fatal(err) - } - db.SetMaxIdleConns(5) - db.SetMaxOpenConns(5) - defer db.Close() - - keyMap := GetKeymap() InitDb(db, keyMap) @@ -295,6 +297,7 @@ func main() { time.Sleep(SLEEP_TIME) } case *outputPath != "": - //exporting here + exportingData := GetStatTimesFromDb(db, 0, keyMap) //exporting here + SaveToCsvFile(exportingData, keyMap, *outputPath, false) } }