在上一篇博客中,我们为我们的搜索框架实现了建立索引schema以及初始化索引的工作。对于一个搜索引擎来说,应该可以跟踪搜索对象的更新,确保永远将最新的内容保存在索引中,因此我们需要一个更新算法以确保我们搜索的内容永远是最新的。
我们向engine.py文件中添加如下代码,实现对搜索内容的增量更新:
# blogsearchengine/searchengine.py
class searchengine:
# ...
def updateindex(self):
print('updateindex')
storage = FileStorage(self.indexpath)
ix = storage.open_index(indexname=self.indexname)
index_id = set()
to_index_id = set()
objlist = self.model.objects.all()
with ix.searcher() as searcher:
writer = AsyncWriter(ix)
for indexfield in searcher.all_stored_fields():
if len(indexfield) > 0:
indexId = indexfield['id']
print(indexId)
index_id.add(indexId)
# 数据库未找到此篇,则可能已被删除,故从index中删除此篇
if not self.model.objects.filter(id=indexId):
writer.delete_by_term('id', indexId)
for key in indexfield:
# 根据updatefield进行更新
if key == self.updatefield:
objfromdb = self.model.objects.get(id=indexId)
contentofobj = getattr(objfromdb,self.updatefield)
if contentofobj != indexfield[key]:
writer.delete_by_term('id',indexId)
to_index_id.add(indexId)
print('update id is %s, title is %s' % (indexId,objfromdb.title))
for obj