types

type EngineOptions struct {
	NumShards           int //channel分片
	AddBuffer           int //add channel长度
	RemoveBuffer        int //remove channel长度
	SearchBuffer        int //channel长度
	SearchWorkerThreads int //每个搜索channel worker数量

	DefaultSearchOption *SearchOptions  //默认搜索配置
	IndexerOption       *IndexerOptions //索引器配置项
}

type IndexedDocument struct {
	DocId     uint64      `msg:"id"`
	Latitude  float64     `msg:"lat"`
	Longitude float64     `msg:"long"`
	Fields    interface{} `msg:"f"`
}

type IndexerOptions struct {
	RedisHost       string
	RedisPassword   string
	RedisDb         int
	HashSize        uint64  //hash分片大小
	GeoShard        uint64  //GEOHASH分片大小
	GeoPrecious     uint    //GEOHASH位数
	CenterLatitude  float64 //城市中心纬度
	CenterLongitude float64 //城市中心经度
	Location        string  //城市
}

type ScoredDocument struct {
	Distance float64
	DocId    uint64
	Model    interface{} //为外围组装数据提供
}

type ScoredDocuments []*ScoredDocument

type SearchOptions struct {
	Refresh   bool
	OrderDesc bool
	Timeout   time.Duration
	Accuracy  int             //计算进度
	Circles   int             //圈数，默认1，不扩散
	Excepts   map[uint64]bool //排除指定ID
	Filter    func(doc IndexedDocument) bool
}

type SearchRequest struct {
	Latitude     float64
	Longitude    float64
	CountOnly    bool
	Offset       int
	Limit        int
	SearchOption *SearchOptions //可留空，使用引擎默认参数
}

type SearchResponse struct {
	Docs    ScoredDocuments //排序好的结果
	Timeout bool            //是否超时
	Count   int             //数量
}