internal/repo/search_sync/search_sync.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package search_sync

import (
	"context"
	"github.com/apache/answer/internal/base/constant"
	"github.com/apache/answer/internal/base/data"
	"github.com/apache/answer/internal/entity"
	"github.com/apache/answer/internal/schema"
	"github.com/apache/answer/pkg/uid"
	"github.com/apache/answer/plugin"
	"github.com/segmentfault/pacman/log"
)

func NewPluginSyncer(data *data.Data) plugin.SearchSyncer {
	return &PluginSyncer{data: data}
}

type PluginSyncer struct {
	data *data.Data
}

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(&answers)
	if err != nil {
		return nil, err
	}
	return p.convertAnswers(ctx, answers)
}

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(&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
}