add toml config

conf/ad_server.toml

@@ -0,0 +1,4 @@
+
+GeoBlockFileName = "./data/GeoLiteCity-Blocks.csv"
+GeoLocationFileName = "./data/GeoLiteCity-Location.csv"
+AdFileName = "./data/ad_info.txt"

src/adserver/global_conf.go

@@ -0,0 +1,36 @@
+package adserver
+
+import (
+	"github.com/spf13/viper"
+	"fmt"
+)
+
+type GlobalConf struct {
+	GeoBlockFileName string
+	GeoLocationFileName string
+	AdFileName string
+}
+
+var GlobalConfObject *GlobalConf
+
+func init()  {
+	GlobalConfObject = new(GlobalConf)
+}
+
+func LoadGlobalConf(configPath, configFileName string)  {
+	viper.AddConfigPath(configPath)
+	viper.SetConfigName(configFileName)
+	if err := viper.ReadInConfig(); err != nil {
+		fmt.Printf("Error reading config file, %s\n", err)
+		panic(-1)
+	}
+	err := viper.Unmarshal(GlobalConfObject)
+	if err != nil {
+		fmt.Printf("unable to decode into struct, %v", err)
+		panic(-1)
+	}
+	fmt.Printf("GeoBlockFileName=%s GeoLocationFileName=%s AdFileName=%s\n",
+		GlobalConfObject.GeoBlockFileName,
+	    GlobalConfObject.GeoLocationFileName,
+	    GlobalConfObject.AdFileName)
+}

src/main/main.go

@@ -7,10 +7,11 @@ import (
 )
 
 func main() {
+	adserver.LoadGlobalConf("./conf", "ad_server")
 	adserver.LoadLocationDict(
-		"./data/GeoLiteCity-Blocks.csv",
-		"./data/GeoLiteCity-Location.csv")
-	adserver.ReadAdDict("./data/ad_info.txt")
+		adserver.GlobalConfObject.GeoBlockFileName,
+		adserver.GlobalConfObject.GeoLocationFileName)
+	adserver.ReadAdDict(adserver.GlobalConfObject.AdFileName)
 	http.HandleFunc("/ad/search", adhandler.SearchHandler)
 	http.HandleFunc("/ad/impression",adhandler.ImpressionHandler)
 	http.HandleFunc("/ad/click",adhandler.ClickHandler)