128 lines
2.8 KiB
Go
128 lines
2.8 KiB
Go
package merch
|
|
|
|
import (
|
|
"database/sql"
|
|
log "github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
"merch-parser-api/internal/shared"
|
|
"time"
|
|
)
|
|
|
|
type Link struct {
|
|
Surugaya []Surugaya
|
|
Mandarake []Mandarake
|
|
}
|
|
|
|
type TaskProvider struct {
|
|
repo TaskRepository
|
|
}
|
|
|
|
type TaskRepository interface {
|
|
getLinks() (*Link, error)
|
|
insertPrices(prices []Price) 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
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"surugaya links": len(getLinks.Surugaya),
|
|
"mandarake links": len(getLinks.Mandarake),
|
|
}).Info("gRPC Server | Prepare tasks")
|
|
|
|
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,
|
|
}
|
|
}
|
|
}
|
|
log.WithField("data", taskMap).Debug("Prepare tasks")
|
|
return taskMap, nil
|
|
}
|
|
|
|
func (p *TaskProvider) InsertPrices(prices []shared.TaskResult) error {
|
|
if len(prices) == 0 {
|
|
log.WithField("msg", "no prices received").Debug("Merch provider | Insert prices")
|
|
return nil
|
|
}
|
|
|
|
var insertPrices []Price
|
|
for _, item := range prices {
|
|
origin, ok := parseOrigin(item.Origin)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
insertPrices = append(insertPrices, Price{
|
|
CreatedAt: time.Now().UTC(),
|
|
UpdatedAt: sql.NullTime{Time: time.Time{}, Valid: false},
|
|
DeletedAt: sql.NullTime{Time: time.Time{}, Valid: false},
|
|
MerchUuid: item.MerchUuid,
|
|
Price: int(item.Price),
|
|
Origin: origin,
|
|
})
|
|
}
|
|
|
|
if err := p.repo.insertPrices(insertPrices); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *TaskRepo) getLinks() (*Link, error) {
|
|
var surugayaList []Surugaya
|
|
if err := r.db.Model(&Surugaya{}).Where("deleted_at IS NULL").Find(&surugayaList).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var mandarakeList []Mandarake
|
|
if err := r.db.Model(&Mandarake{}).Where("deleted_at IS NULL").Find(&mandarakeList).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Link{
|
|
Surugaya: surugayaList,
|
|
Mandarake: mandarakeList,
|
|
}, nil
|
|
}
|
|
|
|
func (r *TaskRepo) insertPrices(prices []Price) error {
|
|
return r.db.Model(&Price{}).Create(&prices).Error
|
|
}
|