131 lines
3.9 KiB
Go
131 lines
3.9 KiB
Go
package storage
|
|
|
|
import (
|
|
v2 "gitea.timerzz.com/kedaya_haitao/common/structs/v2"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// ProviderArticleApi 管理供应商商品的接口
|
|
type ProviderArticleApi interface {
|
|
Get(query *GetProviderArticleQuery) (article v2.ProviderArticle, err error)
|
|
Upsert(article v2.ProviderArticle) error
|
|
Update(providerArticle v2.ProviderArticle, selects ...string) error
|
|
BatchUpdate(articles []v2.ProviderArticle) error
|
|
AutoMigrate() error
|
|
FindInBatches(query *GetProviderArticleQuery, results *[]v2.ProviderArticle, f func(tx *gorm.DB, batch int) error) error
|
|
ProviderPrice(providerArticleID uint) (history []v2.ProviderPrice, err error)
|
|
UpdateStatus(article v2.ProviderArticle) error
|
|
}
|
|
|
|
type providerArticleApi struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewProviderArticleApi(db *gorm.DB) ProviderArticleApi {
|
|
return &providerArticleApi{db: db}
|
|
}
|
|
|
|
// ******************GET
|
|
|
|
type GetProviderArticleQuery struct {
|
|
ID uint `query:"id"`
|
|
Brand v2.Brand `query:"brand"`
|
|
Pid string `query:"pid"`
|
|
ProviderId v2.ProviderId `query:"providerId"`
|
|
SkuId string `query:"skuId"`
|
|
}
|
|
|
|
func NewGetProviderArticleQuery() *GetProviderArticleQuery {
|
|
return &GetProviderArticleQuery{}
|
|
}
|
|
|
|
func (g *GetProviderArticleQuery) SetID(id uint) *GetProviderArticleQuery {
|
|
g.ID = id
|
|
return g
|
|
}
|
|
func (g *GetProviderArticleQuery) SetBrand(brand v2.Brand) *GetProviderArticleQuery {
|
|
g.Brand = brand
|
|
return g
|
|
}
|
|
func (g *GetProviderArticleQuery) SetPid(pid string) *GetProviderArticleQuery {
|
|
g.Pid = pid
|
|
return g
|
|
}
|
|
|
|
func (g *GetProviderArticleQuery) SetProviderId(providerId v2.ProviderId) *GetProviderArticleQuery {
|
|
g.ProviderId = providerId
|
|
return g
|
|
}
|
|
|
|
func (g *GetProviderArticleQuery) SetSkuId(skuId string) *GetProviderArticleQuery {
|
|
g.SkuId = skuId
|
|
return g
|
|
}
|
|
|
|
func (g *GetProviderArticleQuery) Scope(db *gorm.DB) *gorm.DB {
|
|
if g.ID > 0 {
|
|
db = db.Where("id=?", g.ID)
|
|
}
|
|
if g.Brand != "" {
|
|
db = db.Where("brand=?", g.Brand)
|
|
}
|
|
if g.Pid != "" {
|
|
db = db.Where("pid=?", g.Pid)
|
|
}
|
|
if g.ProviderId != "" {
|
|
db = db.Where("provider_id=?", g.ProviderId)
|
|
}
|
|
if g.SkuId != "" {
|
|
db = db.Where("sku_id=?", g.SkuId)
|
|
}
|
|
return db
|
|
}
|
|
|
|
func (p *providerArticleApi) Get(query *GetProviderArticleQuery) (article v2.ProviderArticle, err error) {
|
|
err = p.db.Scopes(query.Scope).Preload("CalculateProcess").First(&article).Error
|
|
return
|
|
}
|
|
|
|
func (p *providerArticleApi) Upsert(article v2.ProviderArticle) error {
|
|
if err := p.db.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "provider_id"}, {Name: "sku_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"cost", "available", "updated_at", "ast"}),
|
|
}).Create(&article).Error; err != nil {
|
|
return err
|
|
}
|
|
if len(article.HistoryPrice) > 0 {
|
|
return p.db.Save(&article.HistoryPrice).Error
|
|
}
|
|
return nil
|
|
}
|
|
func (p *providerArticleApi) UpdateStatus(article v2.ProviderArticle) error {
|
|
return p.db.Where("id=?", article.ID).Select("status").Updates(&article).Error
|
|
}
|
|
|
|
func (p *providerArticleApi) FindInBatches(query *GetProviderArticleQuery, results *[]v2.ProviderArticle, f func(tx *gorm.DB, batch int) error) error {
|
|
err := p.db.Scopes(query.Scope).Preload("CalculateProcess").FindInBatches(results, 20, f).Error
|
|
return err
|
|
}
|
|
|
|
// 批量更新，更新价格时用到
|
|
func (p *providerArticleApi) BatchUpdate(articles []v2.ProviderArticle) error {
|
|
return p.db.Select("id", "cost").Save(&articles).Error
|
|
}
|
|
|
|
func (p *providerArticleApi) AutoMigrate() error {
|
|
return p.db.AutoMigrate(&v2.ProviderArticle{}, &v2.ProviderPrice{})
|
|
}
|
|
|
|
func (p *providerArticleApi) ProviderPrice(providerArticleID uint) (history []v2.ProviderPrice, err error) {
|
|
err = p.db.Find(&history, "provider_article_id = ?", providerArticleID).Error
|
|
return
|
|
}
|
|
|
|
func (p *providerArticleApi) Update(providerArticle v2.ProviderArticle, selects ...string) error {
|
|
if len(selects) == 0 {
|
|
selects = []string{"Exclude"}
|
|
}
|
|
return p.db.Model(&providerArticle).Select(selects).Omit(clause.Associations).Updates(&providerArticle).Error
|
|
}
|