···
+
// heavily inspired by gitea's model (basically copy-pasted)
+
"github.com/blevesearch/bleve/v2"
+
"github.com/blevesearch/bleve/v2/index/upsidedown"
+
"github.com/blevesearch/bleve/v2/search/query"
+
"tangled.org/core/appview/db"
+
"tangled.org/core/appview/indexer/base36"
+
"tangled.org/core/appview/indexer/bleve"
+
"tangled.org/core/appview/models"
+
"tangled.org/core/appview/pagination"
+
tlog "tangled.org/core/log"
+
func NewIndexer(indexDir string) *Indexer {
+
// Init initializes the indexer
+
func (ix *Indexer) Init(ctx context.Context, e db.Execer) {
+
l := tlog.FromContext(ctx)
+
existed, err := ix.intialize(ctx)
+
log.Fatalln("failed to initialize issue indexer", err)
+
l.Debug("Populating the issue indexer")
+
err := PopulateIndexer(ctx, ix, e)
+
log.Fatalln("failed to populate issue indexer", err)
+
l.Info("Initialized the issue indexer")
+
func (ix *Indexer) intialize(ctx context.Context) (bool, error) {
+
return false, errors.New("indexer is already initialized")
+
indexer, err := openIndexer(ctx, ix.path)
+
mapping := bleve.NewIndexMapping()
+
indexer, err = bleve.New(ix.path, mapping)
+
func openIndexer(ctx context.Context, path string) (bleve.Index, error) {
+
l := tlog.FromContext(ctx)
+
indexer, err := bleve.Open(path)
+
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)
+
func PopulateIndexer(ctx context.Context, ix *Indexer, e db.Execer) error {
+
l := tlog.FromContext(ctx)
+
err := pagination.IterateAll(
+
func(page pagination.Page) ([]models.Issue, error) {
+
return db.GetIssuesPaginated(e, page)
+
func(issues []models.Issue) error {
+
return ix.Index(ctx, issues...)
+
l.Info("issues indexed", "count", count)
+
// issueData data stored and will be indexed
+
type issueData struct {
+
RepoAt string `json:"repo_at"`
+
IssueID int `json:"issue_id"`
+
Title string `json:"title"`
+
Body string `json:"body"`
+
IsOpen bool `json:"is_open"`
+
Comments []IssueCommentData `json:"comments"`
+
func makeIssueData(issue *models.Issue) *issueData {
+
RepoAt: issue.RepoAt.String(),
+
IssueID: issue.IssueId,
+
type IssueCommentData struct {
+
Body string `json:"body"`
+
type SearchResult struct {
+
const maxBatchSize = 20
+
func (ix *Indexer) Index(ctx context.Context, issues ...models.Issue) error {
+
batch := bleveutil.NewFlushingBatch(ix.indexer, maxBatchSize)
+
for _, issue := range issues {
+
issueData := makeIssueData(&issue)
+
if err := batch.Index(base36.Encode(issue.Id), issueData); err != nil {
+
// Search searches for issues
+
func (ix *Indexer) Search(ctx context.Context, opts models.IssueSearchOptions) (*SearchResult, error) {
+
var queries []query.Query
+
if opts.Keyword != "" {
+
queries = append(queries, bleve.NewDisjunctionQuery(
+
bleveutil.MatchAndQuery("title", opts.Keyword),
+
bleveutil.MatchAndQuery("body", opts.Keyword),
+
queries = append(queries, bleveutil.KeywordFieldQuery("repo_at", opts.RepoAt))
+
queries = append(queries, bleveutil.BoolFieldQuery("is_open", opts.IsOpen))
+
// TODO: append more queries
+
var indexerQuery query.Query = bleve.NewConjunctionQuery(queries...)
+
searchReq := bleve.NewSearchRequestOptions(indexerQuery, opts.Page.Limit, opts.Page.Offset, false)
+
res, err := ix.indexer.SearchInContext(ctx, searchReq)
+
Hits: make([]int64, len(res.Hits)),
+
for i, hit := range res.Hits {
+
id, err := base36.Decode(hit.ID)