grpc server added

internal/api/merch/provider.go

@@ -0,0 +1,93 @@
+package merch
+
+import (
+	"gorm.io/gorm"
+	"merch-parser-api/internal/shared"
+)
+
+type Link struct {
+	Surugaya  []Surugaya
+	Mandarake []Mandarake
+}
+
+type TaskProvider struct {
+	repo TaskRepository
+}
+
+type TaskRepository interface {
+	GetLinks() (*Link, error)
+	InsertPrices() error
+}
+
+type TaskRepo struct {
+	db *gorm.DB
+}
+
+func NewTaskProvider(repo TaskRepository) *TaskProvider {
+	return &TaskProvider{
+		repo: repo,
+	}
+}
+
+func NewTaskRepository(db *gorm.DB) TaskRepository {
+	return &TaskRepo{db: db}
+}
+
+func (p *TaskProvider) PrepareTasks() (map[string]shared.Task, error) {
+	getLinks, err := p.repo.GetLinks()
+	if err != nil {
+		return nil, err
+	}
+
+	taskMap := make(map[string]shared.Task)
+
+	for _, item := range getLinks.Surugaya {
+		if task, exists := taskMap[item.MerchUuid]; exists {
+			task.OriginSurugayaLink = item.Link
+			taskMap[item.MerchUuid] = task
+		} else {
+			taskMap[item.MerchUuid] = shared.Task{
+				MerchUuid:          item.MerchUuid,
+				OriginSurugayaLink: item.Link,
+			}
+		}
+	}
+
+	for _, item := range getLinks.Mandarake {
+		if task, exists := taskMap[item.MerchUuid]; exists {
+			task.OriginMandarakeLink = item.Link
+			taskMap[item.MerchUuid] = task
+		} else {
+			taskMap[item.MerchUuid] = shared.Task{
+				MerchUuid:           item.MerchUuid,
+				OriginMandarakeLink: item.Link,
+			}
+		}
+	}
+	return taskMap, nil
+}
+
+func (p *TaskProvider) InsertPrices([]shared.TaskResult) error {
+	return nil
+}
+
+func (r *TaskRepo) GetLinks() (*Link, error) {
+	var surugayaList []Surugaya
+	if err := r.db.Model(&Surugaya{}).Find(&surugayaList).Error; err != nil {
+		return nil, err
+	}
+
+	var mandarakeList []Mandarake
+	if err := r.db.Model(&Mandarake{}).Find(&mandarakeList).Error; err != nil {
+		return nil, err
+	}
+
+	return &Link{
+		Surugaya:  surugayaList,
+		Mandarake: mandarakeList,
+	}, nil
+}
+
+func (r *TaskRepo) InsertPrices() error {
+	return nil
+}