完成路线图功能

Author: flycash

README.md

@@ -49,3 +49,4 @@
 - credit - 10
 - project - 11
 - marketing - 12
+- roadmap - 13 # 路线图

internal/question/internal/repository/dao/question_set.go

@@ -16,17 +16,12 @@ package dao
 
 import (
 	"context"
-	"errors"
 	"time"
 
 	"github.com/ego-component/egorm"
 	"gorm.io/gorm"
 )
 
-var (
-	ErrInvalidQuestionID = errors.New("问题ID非法")
-)
-
 type QuestionSetDAO interface {
 	Create(ctx context.Context, qs QuestionSet) (int64, error)
 	GetByID(ctx context.Context, id int64) (QuestionSet, error)
@@ -37,12 +32,19 @@ type QuestionSetDAO interface {
 	Count(ctx context.Context) (int64, error)
 	List(ctx context.Context, offset, limit int) ([]QuestionSet, error)
 	UpdateNonZero(ctx context.Context, set QuestionSet) error
+	GetByIDs(ctx context.Context, ids []int64) ([]QuestionSet, error)
 }
 
 type GORMQuestionSetDAO struct {
 	db *egorm.Component
 }
 
+func (g *GORMQuestionSetDAO) GetByIDs(ctx context.Context, ids []int64) ([]QuestionSet, error) {
+	var res []QuestionSet
+	err := g.db.WithContext(ctx).Where("id IN ?", ids).Find(&res).Error
+	return res, err
+}
+
 func (g *GORMQuestionSetDAO) UpdateNonZero(ctx context.Context, set QuestionSet) error {
 	set.Utime = time.Now().UnixMilli()
 	return g.db.WithContext(ctx).Where("id = ?", set.Id).Updates(set).Error

internal/question/internal/repository/question_set.go

@@ -31,13 +31,21 @@ type QuestionSetRepository interface {
 	Total(ctx context.Context) (int64, error)
 	List(ctx context.Context, offset int, limit int) ([]domain.QuestionSet, error)
 	UpdateNonZero(ctx context.Context, set domain.QuestionSet) error
+	GetByIDs(ctx context.Context, ids []int64) ([]domain.QuestionSet, error)
 }
 
 type questionSetRepository struct {
 	dao    dao.QuestionSetDAO
 	logger *elog.Component
 }
 
+func (q *questionSetRepository) GetByIDs(ctx context.Context, ids []int64) ([]domain.QuestionSet, error) {
+	qs, err := q.dao.GetByIDs(ctx, ids)
+	return slice.Map(qs, func(idx int, src dao.QuestionSet) domain.QuestionSet {
+		return q.toDomainQuestionSet(src)
+	}), err
+}
+
 func (q *questionSetRepository) UpdateNonZero(ctx context.Context, set domain.QuestionSet) error {
 	return q.dao.UpdateNonZero(ctx, q.toEntityQuestionSet(set))
 }

internal/question/internal/service/question_set.go

@@ -26,11 +26,13 @@ import (
 	"golang.org/x/sync/errgroup"
 )
 
+//go:generate mockgen -source=./question_set.go -destination=../../mocks/quetion_set.mock.go -package=quemocks -typed=true QuestionSetService
 type QuestionSetService interface {
 	Save(ctx context.Context, set domain.QuestionSet) (int64, error)
 	UpdateQuestions(ctx context.Context, set domain.QuestionSet) error
 	List(ctx context.Context, offset, limit int) ([]domain.QuestionSet, int64, error)
 	Detail(ctx context.Context, id int64) (domain.QuestionSet, error)
+	GetByIds(ctx context.Context, ids []int64) ([]domain.QuestionSet, error)
 }
 
 type questionSetService struct {
@@ -41,6 +43,10 @@ type questionSetService struct {
 	syncTimeout  time.Duration
 }
 
+func (q *questionSetService) GetByIds(ctx context.Context, ids []int64) ([]domain.QuestionSet, error) {
+	return q.repo.GetByIDs(ctx, ids)
+}
+
 func NewQuestionSetService(repo repository.QuestionSetRepository,
 	intrProducer event.InteractiveEventProducer,
 	producer event.SyncDataToSearchEventProducer) QuestionSetService {

internal/question/mocks/quetion_set.mock.go

@@ -15,7 +15,8 @@
 package baguwen
 
 type Module struct {
-	Svc   Service
-	Hdl   *Handler
-	QsHdl *QuestionSetHandler
+	Svc    Service
+	SetSvc QuestionSetService
+	Hdl    *Handler
+	QsHdl  *QuestionSetHandler
 }

internal/question/module.go

@@ -24,4 +24,6 @@ type Handler = web.Handler
 type QuestionSetHandler = web.QuestionSetHandler
 
 type Service = service.Service
+type QuestionSetService = service.QuestionSetService
 type Question = domain.Question
+type QuestionSet = domain.QuestionSet