feat provider-article添加AutoMrigate

pkg/cron/cron.go

@@ -6,18 +6,19 @@ import (
 )
 
 type Cron struct {
-	time time.Time
-	ctx  context.Context
-	f    func()
+	time   time.Time
+	ctx    context.Context
+	cancel context.CancelFunc
+	f      func()
 }
 
-func NewCron(ctx context.Context) *Cron {
+func NewCron() *Cron {
 	now := time.Now()
 	l, _ := time.LoadLocation("Asia/Shanghai")
-	return &Cron{
-		ctx:  ctx,
+	c := &Cron{
 		time: time.Date(now.Year(), now.Month(), now.Day(), 01, 0, 0, 0, l),
 	}
+	return c
 }
 
 func (c *Cron) SetFunc(f func()) *Cron {
@@ -38,7 +39,8 @@ func (c *Cron) SetTimeHHmm(HHmm string) *Cron {
 	return c
 }
 
-func (c *Cron) Run() {
+func (c *Cron) Run(ctx context.Context) {
+	c.ctx, c.cancel = context.WithCancel(ctx)
 	for {
 		if time.Now().After(c.time) {
 			c.time = c.time.Add(time.Hour * 24)
@@ -53,3 +55,7 @@ func (c *Cron) Run() {
 		}
 	}
 }
+
+func (c *Cron) Stop() {
+	c.cancel()
+}

structs/storage/provider-article.go

@@ -10,6 +10,7 @@ import (
 type ProviderArticleApi interface {
 	Get(query GetProviderArticleQuery) (article v2.ProviderArticle, err error)
 	Upsert(article v2.ProviderArticle) error
+	AutoMigrate() error
 }
 
 type providerArticleApi struct {
@@ -58,3 +59,7 @@ func (p *providerArticleApi) Upsert(article v2.ProviderArticle) error {
 		DoUpdates: clause.AssignmentColumns([]string{"cost"}),
 	}).Create(&article).Error
 }
+
+func (p *providerArticleApi) AutoMigrate() error {
+	return p.db.AutoMigrate(&v2.ProviderArticle{})
+}

structs/storage/seller-article.go

@@ -10,6 +10,7 @@ import (
 type SellerArticleApi interface {
 	Get(query GetProviderArticleQuery) (article v2.ProviderArticle, err error)
 	Upsert(article v2.SellerArticle) error
+	AutoMigrate() error
 }
 
 type sellerArticleApi struct {
@@ -54,3 +55,7 @@ func (p *sellerArticleApi) Upsert(article v2.SellerArticle) error {
 		DoUpdates: clause.AssignmentColumns([]string{"sell"}),
 	}).Create(&article).Error
 }
+
+func (p *sellerArticleApi) AutoMigrate() error {
+	return p.db.AutoMigrate(&v2.SellerArticle{})
+}