Refactoring + adds a more granular log level setting. (crawler)

1 files changed, 34 insertions, 0 deletions

diff --git a/crawler/init.go b/crawler/init.go
new file mode 100644
index 0000000..abf212b
--- /dev/null
+++ b/crawler/init.go
@@ -0,0 +1,34 @@
+package main
+
+import (
+	"flag"
+	"strings"
+
+	log "github.com/Sirupsen/logrus"
+	//"github.com/spf13/viper"
+	//flag "github.com/spf13/pflag"
+)
+
+// global config, gets overwritten by main
+var _conf Config
+
+func init() {
+	// we need to parse the config because of log level setting
+	configFile := flag.String("config", "", "path to config file")
+	debug := flag.Bool("debug", false, "debug outputs")
+	loglevel_f := flag.String("loglevel", "Warn", `sets log level, can be "Warn", "Info" or "Debug"`)
+	loglevel := strings.ToLower(*loglevel_f)
+
+	flag.Parse()
+
+	if *debug || loglevel == "debug" {
+		log.SetLevel(log.DebugLevel)
+	} else if loglevel == "Info" {
+		log.SetLevel(log.InfoLevel)
+	} else {
+		log.SetLevel(log.WarnLevel)
+	}
+
+	_conf.parseConfig(*configFile)
+
+}