diff options
Diffstat (limited to 'cleaner.go')
| -rw-r--r-- | cleaner.go | 26 |
1 files changed, 17 insertions, 9 deletions
@@ -9,11 +9,19 @@ import ( "time" ) +var ( + logger *log.Logger +) + +func init() { + logger = log.New(os.Stdout, "[ Cleaner ]", log.LstdFlags) +} + func cleanFromDB(cfg *Config) (rowsLeft int, err error) { db, err := sql.Open("mysql", cfg.DBUser+":"+cfg.DBPass+"@"+cfg.DBConn+"/"+cfg.DBName) if err != nil { - log.Println("Failed to connect to db") + logger.Println("Failed to connect to db") return } defer db.Close() @@ -33,20 +41,20 @@ func cleanFromDB(cfg *Config) (rowsLeft int, err error) { //Fetch data that should be cleaned rDat, err := fetchRawData(db, cfg, cleanLimit) if err != nil { - log.Println("Faild to fetch raw data") + logger.Println("Faild to fetch raw data") return } cDat, err := clean(rDat, cfg) if err != nil { - log.Println("Failed to clean data from db:", err) + logger.Println("Failed to clean data from db:", err) return } //Begin transaction tx, err := db.Begin() if err != nil { - log.Println("Failed to initialize transaction") + logger.Println("Failed to initialize transaction") return } @@ -54,7 +62,7 @@ func cleanFromDB(cfg *Config) (rowsLeft int, err error) { err = insertCleanData(tx, cfg, cDat) if err != nil { tx.Rollback() - log.Println("Failed to save cleaned data") + logger.Println("Failed to save cleaned data") return } @@ -62,13 +70,13 @@ func cleanFromDB(cfg *Config) (rowsLeft int, err error) { err = purgeRawData(tx, cfg, rDat) if err != nil { tx.Rollback() - log.Println("Failed to remove old data") + logger.Println("Failed to remove old data") return } rowsLeft, err = availableRows(tx, cfg, cleanLimit) if err != nil { tx.Rollback() - log.Println("Failed to fetch available rows") + logger.Println("Failed to fetch available rows") return } @@ -80,12 +88,12 @@ func cleanFromStdin(rDatChan <-chan []RawData, cfg *Config) error { for rDat := range rDatChan { cDat, err := clean(rDat, cfg) if err != nil { - log.Println("Failed to clean data from stdin:", err) + logger.Println("Failed to clean data from stdin:", err) return err } err = insertCleanDataToDB(cfg, cDat) if err != nil { - log.Println("Failed to insert clean data from stdin:", err) + logger.Println("Failed to insert clean data from stdin:", err) return err } } |