Example #1
0
func main() {
	port, err := kernel.GetSetting(settings, "PORT")
	checkError(err)
	host, err := kernel.GetSetting(settings, "HOST")
	checkError(err)
	log.Printf("server run on %s:%s", host, port)
	http.Handle("/", routers.Register(&g))
	err = http.ListenAndServe(fmt.Sprintf("%s:%s", host, port), nil)
	checkError(err)
}
Example #2
0
func mountDB() *sql.DB {
	dbinfo, err := kernel.GetSetting(settings, "DBINFO")
	checkError(err)
	db, _ = sql.Open("mysql", dbinfo)
	db.SetMaxOpenConns(30)
	db.SetMaxIdleConns(30)
	checkError(db.Ping())
	return db
}
Example #3
0
func mountRedis() *redis.Client {
	addr, err := kernel.GetSetting(settings, "REDISADDR")
	checkError(err)
	passwd, err := kernel.GetSetting(settings, "REDISPASSWORD")
	checkError(err)
	redisdb, err := kernel.GetSetting(settings, "REDISDB")
	checkError(err)
	db, err := strconv.ParseInt(redisdb, 10, 64)
	checkError(err)
	cc = redis.NewClient(&redis.Options{
		Addr:     addr,
		Password: passwd, // no password set
		DB:       db,     // use default DB
	})

	_, e := cc.Ping().Result()
	checkError(e)
	return cc
}
Example #4
0
func init() {
	runtime.GOMAXPROCS(runtime.NumCPU())
	//加载配置文件
	loadConf()
	debug, e := kernel.GetSetting(settings, "DEBUG")
	checkError(e)
	//挂载数据库
	mountDB()
	//挂载redis
	mountRedis()
	dict, e1 := kernel.GetSetting(settings, "DICT")
	checkError(e1)
	stop, e2 := kernel.GetSetting(settings, "STOP")
	checkError(e2)
	//初始化分词, 采用引擎自带分词器
	seg := segmenter.InitChinaCut(dict)
	//初始化搜索引擎
	var searcher search.Engine
	//获取mongodb集合数
	indextmp, e3 := kernel.GetSetting(settings, "INDEXSTORENUM")
	checkError(e3)
	indexstorenum, err := strconv.Atoi(indextmp)
	checkError(err)
	//获取mongodb库名
	mongodbname, e4 := kernel.GetSetting(settings, "MONGODBNAME")
	checkError(e4)
	//获取mongodb连接url
	mongodburl, e5 := kernel.GetSetting(settings, "MONGODBURL")
	checkError(e5)
	//获取mongodb集合前缀
	collectionprefix, e5 := kernel.GetSetting(settings, "COLLECTIONPREFIX")
	checkError(e5)
	searcher.Init(search.EngineInitOptions{
		//分词器采用引擎自带的分词器
		Segmenter:     seg,
		StopTokenFile: stop,
		UsePersistentStorage: func() bool {
			if debug == "True" {
				return false
			} else {
				return true
			}
		}(),
		IndexerInitOptions: &search.IndexerInitOptions{
			IndexType: search.LocationsIndex,
			BM25Parameters: &search.BM25Parameters{
				K1: 2.0,
				B:  0.75,
			},
		},
		//pipline采用引擎自带的mongo pipline
		SearchPipline: pipeline.InitMongo(
			mongodbname,
			indexstorenum,
			mongodburl,
			collectionprefix),
		//索引器接口实现,采用自带的wukong索引器
		CreateIndexer: func() search.SearchIndexer {
			return indexer.NewWuKongIndexer()
		},
		//排序器生成方法
		CreateRanker: func() search.SearchRanker {
			return ranker.NewWuKongRanker()
		},
	})
	g = kernel.G{
		//可以处理的http方法字典
		Ml: map[string]string{
			"GET":     "Get",
			"POST":    "Post",
			"OPTIONS": "Options",
			"HEAD":    "Head",
			"PUT":     "Put",
			"DELETE":  "Delete",
			"CONNECT": "Connect",
		},
		Init:           initHandler,
		DefaultHandler: defaultHandler,
		End:            endHandler,
		DB:             db,
		CC:             cc,
		Settings:       settings,
		DIY: map[string]interface{}{
			"seg":      seg,
			"searcher": searcher,
		},
	}
}