kedaya_haitao/us-coach-spider
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix 修复计算价格没生效的bug
All checks were successful
Build image / build (push) Successful in 3m22s
Details

Browse Source
This commit is contained in:
timerzz 2025-03-28 12:31:15 +08:00
parent 0a5b5de491
commit dd5b127681
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
spider/controller.go
View File

@ -126,9 +126,15 @@ func (c *Controller) providerChange(ctx context.Context, message string) error {
c.provider.CalculateProcess = provider.CalculateProcess c.provider.CalculateProcess = provider.CalculateProcess
if needUpdate { if needUpdate {
logrus.Infof("供应商价格计算过程发生变化，重新计算所有商品价格")
provider.Status = v2.ProviderStatus_Calculating provider.Status = v2.ProviderStatus_Calculating
_ = c.storage.Provider().UpdateStatus(provider) if err = c.storage.Provider().UpdateStatus(provider); err != nil {
logrus.Errorf("更新供应商状态失败: %v", err)
}
var results = make([]v2.ProviderArticle, 0, 20) var results = make([]v2.ProviderArticle, 0, 20)
var processed = 0
err = c.storage.ProviderArticle().FindInBatches(storage.NewGetProviderArticleQuery().SetProviderId(c.providerId), &results, func(tx *gorm.DB, batch int) error { err = c.storage.ProviderArticle().FindInBatches(storage.NewGetProviderArticleQuery().SetProviderId(c.providerId), &results, func(tx *gorm.DB, batch int) error {
for idx := range results { for idx := range results {
results[idx].Cost = utils.CalculateProviderPrice(append(provider.CalculateProcess, results[idx].CalculateProcess...), map[string]float64{ results[idx].Cost = utils.CalculateProviderPrice(append(provider.CalculateProcess, results[idx].CalculateProcess...), map[string]float64{
@ -141,15 +147,19 @@ func (c *Controller) providerChange(ctx context.Context, message string) error {
return err return err
} }
ids := lo.Map(results, func(item v2.ProviderArticle, index int) string { ids := lo.Map(results, func(item v2.ProviderArticle, index int) string {
return fmt.Sprintf("%d", item.ID) return fmt.Sprintf("%d", item.ArticleID)
}) })
if err = c.subscribeClient.Publish(c.ctx, utils.ProfitRate_Channel, strings.Join(ids, ",")); err != nil { if err = c.subscribeClient.Publish(c.ctx, utils.ProfitRate_Channel, strings.Join(ids, ",")); err != nil {
logrus.Errorf("通知商品利润率失败: %v", err) logrus.Errorf("通知商品利润率失败: %v", err)
} }
processed += len(results)
return nil return nil
}) })
provider.Status = v2.ProviderStatus_Normal provider.Status = v2.ProviderStatus_Normal
_ = c.storage.Provider().UpdateStatus(provider) if err = c.storage.Provider().UpdateStatus(provider); err != nil {
logrus.Errorf("更新供应商状态失败: %v", err)
}
logrus.Infof("重新计算所有商品价格完成共%d个", processed)
} }
return err return err
} }