Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329// heavily inspired by gitea's model (basically copy-pasted)package pulls_indexer
import ( "context" "errors" "log" "os"
"github.com/blevesearch/bleve/v2" "github.com/blevesearch/bleve/v2/analysis/analyzer/custom" "github.com/blevesearch/bleve/v2/analysis/token/camelcase" "github.com/blevesearch/bleve/v2/analysis/token/lowercase" "github.com/blevesearch/bleve/v2/analysis/token/unicodenorm" "github.com/blevesearch/bleve/v2/analysis/tokenizer/unicode" "github.com/blevesearch/bleve/v2/index/upsidedown" "github.com/blevesearch/bleve/v2/mapping" "github.com/blevesearch/bleve/v2/search/query" "tangled.org/core/appview/db" "tangled.org/core/appview/indexer/base36" bleveutil "tangled.org/core/appview/indexer/bleve" "tangled.org/core/appview/models" "tangled.org/core/appview/pagination" tlog "tangled.org/core/log")
const ( pullIndexerAnalyzer = "pullIndexer" pullIndexerDocType = "pullIndexerDocType"
unicodeNormalizeName = "uicodeNormalize"
// Bump this when the index mapping changes to trigger a rebuild. pullIndexerVersion = 4)
type Indexer struct { indexer bleve.Index path string}
func NewIndexer(indexDir string) *Indexer { return &Indexer{ path: indexDir, }}
// Init initializes the indexerfunc (ix *Indexer) Init(ctx context.Context, e db.Execer) { l := tlog.FromContext(ctx) existed, err := ix.initialize(ctx) if err != nil { log.Fatalln("failed to initialize pull indexer", err) } if !existed { l.Debug("Populating the pull indexer") err := PopulateIndexer(ctx, ix, e) if err != nil { log.Fatalln("failed to populate pull indexer", err) } }
count, _ := ix.indexer.DocCount() l.Info("Initialized the pull indexer", "docCount", count)}
func generatePullIndexMapping() (mapping.IndexMapping, error) { mapping := bleve.NewIndexMapping() docMapping := bleve.NewDocumentMapping()
textFieldMapping := bleve.NewTextFieldMapping() textFieldMapping.Store = false textFieldMapping.IncludeInAll = false
keywordFieldMapping := bleve.NewKeywordFieldMapping() keywordFieldMapping.Store = false keywordFieldMapping.IncludeInAll = false
// numericFieldMapping := bleve.NewNumericFieldMapping()
docMapping.AddFieldMappingsAt("title", textFieldMapping) docMapping.AddFieldMappingsAt("body", textFieldMapping)
docMapping.AddFieldMappingsAt("repo_did", keywordFieldMapping) docMapping.AddFieldMappingsAt("state", keywordFieldMapping) docMapping.AddFieldMappingsAt("author_did", keywordFieldMapping) docMapping.AddFieldMappingsAt("labels", keywordFieldMapping) docMapping.AddFieldMappingsAt("label_values", keywordFieldMapping)
err := mapping.AddCustomTokenFilter(unicodeNormalizeName, map[string]any{ "type": unicodenorm.Name, "form": unicodenorm.NFC, }) if err != nil { return nil, err }
err = mapping.AddCustomAnalyzer(pullIndexerAnalyzer, map[string]any{ "type": custom.Name, "char_filters": []string{}, "tokenizer": unicode.Name, "token_filters": []string{unicodeNormalizeName, camelcase.Name, lowercase.Name}, }) if err != nil { return nil, err }
mapping.DefaultAnalyzer = pullIndexerAnalyzer mapping.AddDocumentMapping(pullIndexerDocType, docMapping) mapping.AddDocumentMapping("_all", bleve.NewDocumentDisabledMapping()) mapping.DefaultMapping = bleve.NewDocumentDisabledMapping()
return mapping, nil}
func (ix *Indexer) initialize(ctx context.Context) (bool, error) { if ix.indexer != nil { return false, errors.New("indexer is already initialized") }
indexer, err := openIndexer(ctx, ix.path, pullIndexerVersion) if err != nil { return false, err } if indexer != nil { ix.indexer = indexer return true, nil }
mapping, err := generatePullIndexMapping() if err != nil { return false, err } indexer, err = bleve.New(ix.path, mapping) if err != nil { return false, err } indexer.SetInternal([]byte("mapping_version"), []byte{byte(pullIndexerVersion)})
ix.indexer = indexer
return false, nil}
func openIndexer(ctx context.Context, path string, version int) (bleve.Index, error) { l := tlog.FromContext(ctx) indexer, err := bleve.Open(path) if err != nil { if errors.Is(err, upsidedown.IncompatibleVersion) { l.Info("Indexer was built with a previous version of bleve, deleting and rebuilding") return nil, os.RemoveAll(path) } return nil, nil }
storedVersion, _ := indexer.GetInternal([]byte("mapping_version")) if storedVersion == nil || int(storedVersion[0]) != version { l.Info("Indexer mapping version changed, deleting and rebuilding") indexer.Close() return nil, os.RemoveAll(path) }
return indexer, nil}
func PopulateIndexer(ctx context.Context, ix *Indexer, e db.Execer) error { l := tlog.FromContext(ctx) count := 0 err := pagination.IterateAll( func(page pagination.Page) ([]*models.Pull, error) { return db.GetPullsPaginated(ctx, e, page) }, func(pulls []*models.Pull) error { count += len(pulls) return ix.Index(ctx, pulls...) }, ) l.Info("pulls indexed", "count", count) return err}
type pullData struct { ID int64 `json:"id"` RepoDid string `json:"repo_did"` PullID int `json:"pull_id"` Title string `json:"title"` Body string `json:"body"` State string `json:"state"` AuthorDid string `json:"author_did"` Labels []string `json:"labels"` LabelValues []string `json:"label_values"`
Comments []pullCommentData `json:"comments"`}
func makePullData(pull *models.Pull) *pullData { return &pullData{ ID: pull.ID, RepoDid: pull.RepoDid.String(), PullID: int(pull.PullId), Title: pull.Title, Body: pull.Body, State: pull.State.String(), AuthorDid: pull.OwnerDid.String(), Labels: pull.Labels.LabelNames(), LabelValues: pull.Labels.LabelNameValues(), }}
// Type returns the document type, for bleve's mapping.Classifier interface.func (i *pullData) Type() string { return pullIndexerDocType}
type pullCommentData struct { Body string `json:"body"`}
type searchResult struct { Hits []int64 Total uint64}
const maxBatchSize = 20
func (ix *Indexer) Index(ctx context.Context, pulls ...*models.Pull) error { batch := bleveutil.NewFlushingBatch(ix.indexer, maxBatchSize) for _, pull := range pulls { pullData := makePullData(pull) if err := batch.Index(base36.Encode(pullData.ID), pullData); err != nil { return err } } return batch.Flush()}
func (ix *Indexer) Delete(ctx context.Context, pullID int64) error { return ix.indexer.Delete(base36.Encode(pullID))}
func (ix *Indexer) Search(ctx context.Context, opts models.PullSearchOptions) (*searchResult, error) { var musts []query.Query var mustNots []query.Query
// TODO(boltless): remove this after implementing pulls page pagination limit := opts.Page.Limit if limit == 0 { limit = 500 }
for _, keyword := range opts.Keywords { musts = append(musts, bleve.NewDisjunctionQuery( bleveutil.MatchAndQuery("title", keyword, pullIndexerAnalyzer, 0), bleveutil.MatchAndQuery("body", keyword, pullIndexerAnalyzer, 0), )) }
for _, phrase := range opts.Phrases { musts = append(musts, bleve.NewDisjunctionQuery( bleveutil.MatchPhraseQuery("title", phrase, pullIndexerAnalyzer), bleveutil.MatchPhraseQuery("body", phrase, pullIndexerAnalyzer), )) }
for _, keyword := range opts.NegatedKeywords { mustNots = append(mustNots, bleve.NewDisjunctionQuery( bleveutil.MatchAndQuery("title", keyword, pullIndexerAnalyzer, 0), bleveutil.MatchAndQuery("body", keyword, pullIndexerAnalyzer, 0), )) }
for _, phrase := range opts.NegatedPhrases { mustNots = append(mustNots, bleve.NewDisjunctionQuery( bleveutil.MatchPhraseQuery("title", phrase, pullIndexerAnalyzer), bleveutil.MatchPhraseQuery("body", phrase, pullIndexerAnalyzer), )) }
musts = append(musts, bleveutil.KeywordFieldQuery("repo_did", opts.RepoDid)) if opts.State != nil { musts = append(musts, bleveutil.KeywordFieldQuery("state", opts.State.String())) }
if opts.AuthorDid != "" { musts = append(musts, bleveutil.KeywordFieldQuery("author_did", opts.AuthorDid)) }
for _, label := range opts.Labels { musts = append(musts, bleveutil.KeywordFieldQuery("labels", label)) }
for _, did := range opts.NegatedAuthorDids { mustNots = append(mustNots, bleveutil.KeywordFieldQuery("author_did", did)) }
for _, label := range opts.NegatedLabels { mustNots = append(mustNots, bleveutil.KeywordFieldQuery("labels", label)) }
for _, lv := range opts.LabelValues { musts = append(musts, bleveutil.KeywordFieldQuery("label_values", lv)) }
for _, lv := range opts.NegatedLabelValues { mustNots = append(mustNots, bleveutil.KeywordFieldQuery("label_values", lv)) }
indexerQuery := bleve.NewBooleanQuery() indexerQuery.AddMust(musts...) indexerQuery.AddMustNot(mustNots...) searchReq := bleve.NewSearchRequestOptions(indexerQuery, limit, opts.Page.Offset, false) res, err := ix.indexer.SearchInContext(ctx, searchReq) if err != nil { return nil, err } ret := &searchResult{ Total: res.Total, Hits: make([]int64, len(res.Hits)), } for i, hit := range res.Hits { id, err := base36.Decode(hit.ID) if err != nil { return nil, err } ret.Hits[i] = id } return ret, nil}