88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package storage
|
|
|
|
import (
|
|
v2 "gitea.timerzz.com/kedaya_haitao/common/structs/v2"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// SellerArticleApi 管理供应商商品的接口
|
|
type SellerArticleApi interface {
|
|
Get(query *GetProviderArticleQuery) (article v2.ProviderArticle, err error)
|
|
Upsert(article v2.SellerArticle) error
|
|
AutoMigrate() error
|
|
}
|
|
|
|
type sellerArticleApi struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
type GetSellerArticleQuery struct {
|
|
ID uint `query:"id"`
|
|
Brand v2.Brand `query:"brand"`
|
|
Pid string `query:"pid"`
|
|
SellerId string `query:"sellerId"`
|
|
SkuId string `query:"skuId"`
|
|
}
|
|
|
|
func NewGetSellerArticleQuery() *GetSellerArticleQuery {
|
|
return &GetSellerArticleQuery{}
|
|
}
|
|
|
|
func (g *GetSellerArticleQuery) SetID(id uint) *GetSellerArticleQuery {
|
|
g.ID = id
|
|
return g
|
|
}
|
|
|
|
func (g *GetSellerArticleQuery) SetBrand(brand v2.Brand) *GetSellerArticleQuery {
|
|
g.Brand = brand
|
|
return g
|
|
}
|
|
func (g *GetSellerArticleQuery) SetPid(pid string) *GetSellerArticleQuery {
|
|
g.Pid = pid
|
|
return g
|
|
}
|
|
func (g *GetSellerArticleQuery) SetSellerId(sellerId string) *GetSellerArticleQuery {
|
|
g.SellerId = sellerId
|
|
return g
|
|
}
|
|
func (g *GetSellerArticleQuery) SetSkuId(skuId string) *GetSellerArticleQuery {
|
|
g.SkuId = skuId
|
|
return g
|
|
}
|
|
|
|
func (g *GetSellerArticleQuery) 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.SellerId != "" {
|
|
db = db.Where("seller_id=?", g.SellerId)
|
|
}
|
|
if g.SkuId != "" {
|
|
db = db.Where("sku_id=?", g.SkuId)
|
|
}
|
|
return db
|
|
}
|
|
|
|
func (p *sellerArticleApi) Get(query *GetSellerArticleQuery) (article v2.SellerArticle, err error) {
|
|
err = p.db.Scopes(query.Scope).First(&article).Error
|
|
return
|
|
}
|
|
|
|
func (p *sellerArticleApi) Upsert(article v2.SellerArticle) error {
|
|
return p.db.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "seller_id"}, {Name: "sku_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"sell", "exclude"}),
|
|
}).Create(&article).Error
|
|
}
|
|
|
|
func (p *sellerArticleApi) AutoMigrate() error {
|
|
return p.db.AutoMigrate(&v2.SellerArticle{})
|
|
}
|