コード例 #1
0
ファイル: main.go プロジェクト: uronce-cc/weat
func newRecommender() recommend.Recommender {
	var categoryList data.CategoryList
	if flagCategoryFile == "" {
		list, err := crawler.GetCategoryList()
		if err != nil {
			glog.Fatalf("Failed to crawl category list: %v", err)
		}
		categoryList = list
	} else {
		categoryList = loadCategoryFile()
	}
	return recommend.NewRandomRecommender(categoryList, ImgDir)
}
コード例 #2
0
ファイル: main.go プロジェクト: uronce-cc/weat
func main() {
	if flagCategory {
		categoryList, err := crawler.GetCategoryList()
		if err != nil {
			glog.Fatalf("Failed to crawl category list: %v", err)
		}
		printJson(categoryList)
		return
	}

	if flagRecommend {
		printJson(newRecommender().Do())
		return
	}

	// CLI code paths must be at above.
	if flagWeb {
		http.Handle("/img/", app.ImgHandler(path.Dir(ImgDir), http.FileServer(http.Dir(path.Dir(ImgDir)))))
		http.HandleFunc("/", app.Handler(newRecommender()))
		glog.Infof(fmt.Sprintf("Starting a web server at localhost:%d", flagPort))
		glog.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", flagPort), http.DefaultServeMux))
	}
}