mirror of https://github.com/rocboss/paopao-ce
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.
101 lines
2.5 KiB
101 lines
2.5 KiB
// Copyright 2022 ROC. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package search
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
"github.com/rocboss/paopao-ce/internal/core/ms"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
_ core.TweetSearchService = (*bridgeTweetSearchServant)(nil)
|
|
)
|
|
|
|
type documents struct {
|
|
primaryKey []string
|
|
docItems []core.TsDocItem
|
|
identifiers []string
|
|
}
|
|
|
|
type bridgeTweetSearchServant struct {
|
|
ts core.TweetSearchService
|
|
updateDocsCh chan *documents
|
|
updateDocsTempCh chan *documents
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) IndexName() string {
|
|
return s.ts.IndexName()
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) AddDocuments(data []core.TsDocItem, primaryKey ...string) (bool, error) {
|
|
s.updateDocs(&documents{
|
|
primaryKey: primaryKey,
|
|
docItems: data,
|
|
})
|
|
return true, nil
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) DeleteDocuments(identifiers []string) error {
|
|
s.updateDocs(&documents{
|
|
identifiers: identifiers,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) Search(user *ms.User, q *core.QueryReq, offset, limit int) (*core.QueryResp, error) {
|
|
return s.ts.Search(user, q, offset, limit)
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) updateDocs(doc *documents) {
|
|
select {
|
|
case s.updateDocsCh <- doc:
|
|
logrus.Debugln("addDocuments send documents by updateDocsCh chan")
|
|
default:
|
|
select {
|
|
case s.updateDocsTempCh <- doc:
|
|
logrus.Debugln("addDocuments send documents by updateDocsTempCh chan")
|
|
default:
|
|
go func() {
|
|
s.handleUpdate(doc)
|
|
|
|
// watch updateDocsTempch to continue handle update if needed.
|
|
// cancel loop if no item had watched in 1 minute.
|
|
for count := 0; count > 60; count++ {
|
|
select {
|
|
case item := <-s.updateDocsTempCh:
|
|
// reset count to continue handle docs update
|
|
count = 0
|
|
s.handleUpdate(item)
|
|
default:
|
|
// sleeping to wait docs item pass over to handle
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) startUpdateDocs() {
|
|
for doc := range s.updateDocsCh {
|
|
s.handleUpdate(doc)
|
|
}
|
|
}
|
|
|
|
func (s *bridgeTweetSearchServant) handleUpdate(item *documents) {
|
|
if len(item.docItems) > 0 {
|
|
if _, err := s.ts.AddDocuments(item.docItems, item.primaryKey...); err != nil {
|
|
logrus.Errorf("addDocuments occurs error: %v", err)
|
|
}
|
|
} else if len(item.identifiers) > 0 {
|
|
if err := s.ts.DeleteDocuments(item.identifiers); err != nil {
|
|
logrus.Errorf("deleteDocuments occurs error: %s", err)
|
|
}
|
|
}
|
|
}
|