You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.0 KiB
82 lines
2.0 KiB
// Copyright 2013 Hui Chen |
|
// Copyright 2016 ego authors |
|
// |
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may |
|
// not use this file except in compliance with the License. You may obtain |
|
// a copy of the License at |
|
// |
|
// http://www.apache.org/licenses/LICENSE-2.0 |
|
// |
|
// Unless required by applicable law or agreed to in writing, software |
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
|
// License for the specific language governing permissions and limitations |
|
// under the License. |
|
|
|
package riot |
|
|
|
import ( |
|
"github.com/go-ego/riot/types" |
|
) |
|
|
|
type rankerAddDocReq struct { |
|
docId uint64 |
|
fields interface{} |
|
// new |
|
content string |
|
// new 属性 |
|
attri interface{} |
|
} |
|
|
|
type rankerRankReq struct { |
|
docs []types.IndexedDoc |
|
options types.RankOpts |
|
rankerReturnChan chan rankerReturnReq |
|
countDocsOnly bool |
|
} |
|
|
|
type rankerReturnReq struct { |
|
// docs types.ScoredDocs |
|
docs interface{} |
|
numDocs int |
|
} |
|
|
|
type rankerRemoveDocReq struct { |
|
docId uint64 |
|
} |
|
|
|
func (engine *Engine) rankerAddDocWorker(shard int) { |
|
for { |
|
request := <-engine.rankerAddDocChans[shard] |
|
if engine.initOptions.IDOnly { |
|
engine.rankers[shard].AddDoc(request.docId, request.fields) |
|
return |
|
} |
|
// } else { |
|
engine.rankers[shard].AddDoc(request.docId, request.fields, |
|
request.content, request.attri) |
|
// } |
|
} |
|
} |
|
|
|
func (engine *Engine) rankerRankWorker(shard int) { |
|
for { |
|
request := <-engine.rankerRankChans[shard] |
|
if request.options.MaxOutputs != 0 { |
|
request.options.MaxOutputs += request.options.OutputOffset |
|
} |
|
request.options.OutputOffset = 0 |
|
outputDocs, numDocs := engine.rankers[shard].Rank(request.docs, |
|
request.options, request.countDocsOnly) |
|
|
|
request.rankerReturnChan <- rankerReturnReq{ |
|
docs: outputDocs, numDocs: numDocs} |
|
} |
|
} |
|
|
|
func (engine *Engine) rankerRemoveDocWorker(shard int) { |
|
for { |
|
request := <-engine.rankerRemoveDocChans[shard] |
|
engine.rankers[shard].RemoveDoc(request.docId) |
|
} |
|
}
|
|
|