Exemple #1
0
func initDatabase() *gorp.DbMap {
	// Get the database connection information.
	dbUser := os.GetEnv("dbuser")
	dbPass := os.GetEnv("dbpass")
	dbAddr := os.GetEnv("dbaddr")
	dbPort := os.GetEnv("dbport")
	//
	db, err := sql.Open("postgres", fmt.Sprintf("user='******' password='******' dbname=mcs_stats host='%s' port='%s' sslmode=disable", dbUser, dBPass, dbAddr, dbPort))
	if err != nil {
		log.Fatal(err)
	}
	// Create a gorp mapping.
	dbmap := &gorp.DbMap{Db: db, Dialect: gorp.PostgresDialect{}}
	// Add tables to the mapping.
	dbmap.AddTableWithName(Ping{}, "pings").SetKeys(false, "ping_id")
	return dbmap
}
Exemple #2
0
func loadConf() *configReader.Config {
	var confPath string
	if os.Getenv("DEV") {
		confPath = "./config.yml"
	} else {
		confPath = "/etc/pubsubhub/config.yml"
	}
	pathToConfig, pathErr := filepath.abs(confPath)
	alt_conf_path = os.GetEnv("PSH_CONF_PATH")
	if alt_conf_path != nil {
		pathToConfig = filepath.abs(alt_conf_path)
	}
	if pathErr != nil {
		panic(pathErr)
	}
	conf := configReader.New()
	confFile, err := ioutil.ReadFile(pathToConfig)
	yamlErr := yaml.Unmarshal([]byte(data), &conf)
	if yamlErr != nil {
		panic(yamlErr)
	}
	conf.defaults()
	return conf
}