diff options
| -rw-r--r-- | crawler/config.go | 34 |
1 files changed, 18 insertions, 16 deletions
diff --git a/crawler/config.go b/crawler/config.go index f05ad18..af02a6c 100644 --- a/crawler/config.go +++ b/crawler/config.go @@ -6,14 +6,16 @@ import ( ) type Config struct { - DBDriver string - DBDBName string - DBHost string - DBPort string - DBUser string - DBPassword string - DBOptions string - DBPath string // for sqlite + DBDriver string + DBDBName string + DBHost string + DBPort string + DBUser string + DBPassword string + DBOptions string + DBPath string // for sqlite + Polr_URL string + Polr_API_Key string Debug bool } @@ -64,13 +66,13 @@ func (c *Config) parseConfig(configFile string) { // Actually sets the config struct func (c *Config) setsConfig() { - c.DBDriver = viper.GetString("DBDriver") - c.DBHost = viper.GetString("DBHost") - c.DBPort = viper.GetString("DBPort") - c.DBUser = viper.GetString("DBUser") - c.DBPassword = viper.GetString("DBPassword") - c.DBDBName = viper.GetString("DBName") - c.DBOptions = viper.GetString("DBOptions") - c.DBPath = viper.GetString("DBPath") + c.DBDriver = viper.GetString("DB_Driver") + c.DBHost = viper.GetString("DB_Host") + c.DBPort = viper.GetString("DB_Port") + c.DBUser = viper.GetString("DB_User") + c.DBPassword = viper.GetString("DB_Password") + c.DBDBName = viper.GetString("DB_Name") + c.DBOptions = viper.GetString("DB_Options") + c.DBPath = viper.GetString("DB_Path") c.Debug = viper.GetBool("Debug") } |