feat(search): convert answer and question to search content

This commit is contained in:
LinkinStars 2023-08-17 15:25:14 +08:00
parent a5f74a656f
commit 0d2c9cfd45
4 changed files with 105 additions and 15 deletions

View File

@ -391,6 +391,6 @@ func (ar *answerRepo) updateSearch(ctx context.Context, answerID string) (err er
Score: int64(answer.VoteCount),
HasAccepted: answer.Accepted == schema.AnswerAcceptedEnable,
}
err = s.UpdateContent(ctx, answerID, content)
err = s.UpdateContent(ctx, content)
return
}

View File

@ -467,6 +467,6 @@ func (qr *questionRepo) updateSearch(ctx context.Context, questionID string) (er
Score: int64(question.VoteCount),
HasAccepted: question.AcceptedAnswerID != "" && question.AcceptedAnswerID != "0",
}
err = s.UpdateContent(ctx, questionID, content)
err = s.UpdateContent(ctx, content)
return
}

View File

@ -2,9 +2,13 @@ package search_sync
import (
"context"
"github.com/answerdev/answer/internal/base/constant"
"github.com/answerdev/answer/internal/base/data"
"github.com/answerdev/answer/internal/entity"
"github.com/answerdev/answer/internal/schema"
"github.com/answerdev/answer/pkg/uid"
"github.com/answerdev/answer/plugin"
"github.com/segmentfault/pacman/log"
)
func NewPluginSyncer(data *data.Data) plugin.SearchSyncer {
@ -15,16 +19,103 @@ type PluginSyncer struct {
data *data.Data
}
func (p *PluginSyncer) GetAnswersPage(ctx context.Context, page, pageSize int) (answerList []*entity.Answer, err error) {
answerList = make([]*entity.Answer, 0)
func (p *PluginSyncer) GetAnswersPage(ctx context.Context, page, pageSize int) (
answerList []*plugin.SearchContent, err error) {
answers := make([]*entity.Answer, 0)
startNum := (page - 1) * pageSize
err = p.data.DB.Context(ctx).Limit(pageSize, startNum).Find(&answerList)
return answerList, err
err = p.data.DB.Context(ctx).Limit(pageSize, startNum).Find(&answers)
if err != nil {
return nil, err
}
return p.convertAnswers(ctx, answers)
}
func (p *PluginSyncer) GetQuestionsPage(ctx context.Context, page, pageSize int) (questionList []*entity.Question, err error) {
questionList = make([]*entity.Question, 0)
func (p *PluginSyncer) GetQuestionsPage(ctx context.Context, page, pageSize int) (
questionList []*plugin.SearchContent, err error) {
questions := make([]*entity.Question, 0)
startNum := (page - 1) * pageSize
err = p.data.DB.Context(ctx).Limit(pageSize, startNum).Find(&questionList)
return questionList, err
err = p.data.DB.Context(ctx).Limit(pageSize, startNum).Find(&questions)
if err != nil {
return nil, err
}
return p.convertQuestions(ctx, questions)
}
func (p *PluginSyncer) convertAnswers(ctx context.Context, answers []*entity.Answer) (
answerList []*plugin.SearchContent, err error) {
for _, answer := range answers {
question := &entity.Question{}
exist, err := p.data.DB.Context(ctx).Where("id = ?", answer.QuestionID).Get(question)
if err != nil {
log.Errorf("get question failed %s", err)
continue
}
if !exist {
continue
}
tagListList := make([]*entity.TagRel, 0)
tags := make([]string, 0)
err = p.data.DB.Context(ctx).Where("object_id = ?", uid.DeShortID(question.ID)).
Where("status = ?", entity.TagRelStatusAvailable).Find(&tagListList)
if err != nil {
log.Errorf("get tag list failed %s", err)
}
for _, tag := range tagListList {
tags = append(tags, tag.TagID)
}
content := &plugin.SearchContent{
ObjectID: answer.ID,
Title: question.Title,
Type: constant.AnswerObjectType,
Content: answer.ParsedText,
Answers: 0,
Status: plugin.SearchContentStatus(answer.Status),
Tags: tags,
QuestionID: answer.QuestionID,
UserID: answer.UserID,
Views: int64(question.ViewCount),
Created: answer.CreatedAt.Unix(),
Active: answer.UpdatedAt.Unix(),
Score: int64(answer.VoteCount),
HasAccepted: answer.Accepted == schema.AnswerAcceptedEnable,
}
answerList = append(answerList, content)
}
return answerList, nil
}
func (p *PluginSyncer) convertQuestions(ctx context.Context, questions []*entity.Question) (
questionList []*plugin.SearchContent, err error) {
for _, question := range questions {
tagListList := make([]*entity.TagRel, 0)
tags := make([]string, 0)
err := p.data.DB.Context(ctx).Where("object_id = ?", question.ID).
Where("status = ?", entity.TagRelStatusAvailable).Find(&tagListList)
if err != nil {
log.Errorf("get tag list failed %s", err)
}
for _, tag := range tagListList {
tags = append(tags, tag.TagID)
}
content := &plugin.SearchContent{
ObjectID: question.ID,
Title: question.Title,
Type: constant.QuestionObjectType,
Content: question.ParsedText,
Answers: int64(question.AnswerCount),
Status: plugin.SearchContentStatus(question.Status),
Tags: tags,
QuestionID: question.ID,
UserID: question.UserID,
Views: int64(question.ViewCount),
Created: question.CreatedAt.Unix(),
Active: question.UpdatedAt.Unix(),
Score: int64(question.VoteCount),
HasAccepted: question.AcceptedAnswerID != "" && question.AcceptedAnswerID != "0",
}
questionList = append(questionList, content)
}
return questionList, nil
}

View File

@ -2,7 +2,6 @@ package plugin
import (
"context"
"github.com/answerdev/answer/internal/entity"
)
type SearchResult struct {
@ -89,8 +88,8 @@ type Search interface {
SearchContents(ctx context.Context, cond *SearchBasicCond) (res []SearchResult, total int64, err error)
SearchQuestions(ctx context.Context, cond *SearchBasicCond) (res []SearchResult, total int64, err error)
SearchAnswers(ctx context.Context, cond *SearchBasicCond) (res []SearchResult, total int64, err error)
UpdateContent(ctx context.Context, contentID string, content *SearchContent) error
DeleteContent(ctx context.Context, contentID string) error
UpdateContent(ctx context.Context, content *SearchContent) (err error)
DeleteContent(ctx context.Context, objectID string) (err error)
}
type SearchDesc struct {
@ -99,8 +98,8 @@ type SearchDesc struct {
}
type SearchSyncer interface {
GetAnswersPage(ctx context.Context, page, pageSize int) (answerList []*entity.Answer, err error)
GetQuestionsPage(ctx context.Context, page, pageSize int) (questionList []*entity.Question, err error)
GetAnswersPage(ctx context.Context, page, pageSize int) (answerList []*SearchContent, err error)
GetQuestionsPage(ctx context.Context, page, pageSize int) (questionList []*SearchContent, err error)
}
var (