generated from kedaya_haitao/template
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
|
package rate
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"strconv"
|
|||
|
|
|
|||
|
|
"gitea.timerzz.com/kedaya_haitao/common/pkg/subscribe"
|
|||
|
|
"gitea.timerzz.com/kedaya_haitao/common/structs/storage"
|
|||
|
|
"gitea.timerzz.com/kedaya_haitao/common/structs/utils"
|
|||
|
|
"github.com/golang/glog"
|
|||
|
|
"github.com/redis/go-redis/v9"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Controller struct {
|
|||
|
|
ctx context.Context
|
|||
|
|
storage *storage.Storage
|
|||
|
|
subscribe *subscribe.Server
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewController(ctx context.Context, storage *storage.Storage, rdb *redis.Client) *Controller {
|
|||
|
|
return &Controller{
|
|||
|
|
storage: storage,
|
|||
|
|
subscribe: subscribe.NewServer(ctx, rdb),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Controller) Run() error {
|
|||
|
|
c.subscribe.SetErrorHandle(func(err error) {
|
|||
|
|
glog.Error(err)
|
|||
|
|
})
|
|||
|
|
if err := c.subscribe.Subscribe(utils.ProfitRate_Channel, c.Rate); err != nil {
|
|||
|
|
return fmt.Errorf("订阅失败：%v", err)
|
|||
|
|
}
|
|||
|
|
c.subscribe.Run()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理接受到的信息
|
|||
|
|
func (c *Controller) Rate(ctx context.Context, idString string) error {
|
|||
|
|
i, _ := strconv.Atoi(idString)
|
|||
|
|
if i <= 0 {
|
|||
|
|
return fmt.Errorf("接收到的id不正确：%s", idString)
|
|||
|
|
}
|
|||
|
|
id := uint(i)
|
|||
|
|
article, err := c.storage.Article().Get(storage.NewGetArticleQuery().SetID(id))
|
|||
|
|
if err != nil {
|
|||
|
|
return fmt.Errorf("获取商品 id: %d， 失败：%v", id, err)
|
|||
|
|
}
|
|||
|
|
// 拿到最低成本价和最低售价
|
|||
|
|
var cost, sell float64
|
|||
|
|
//检查能不能购买，只要有一个供应商能买，就认为能购买
|
|||
|
|
var available bool
|
|||
|
|
|
|||
|
|
for _, provider := range article.Providers {
|
|||
|
|
final := provider.Cost.FinalPrice
|
|||
|
|
if final > 0 && (final < cost || cost == 0) && !provider.Exclude {
|
|||
|
|
cost = final
|
|||
|
|
}
|
|||
|
|
available = available || provider.Available
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, seller := range article.Sellers {
|
|||
|
|
final := seller.Sell.FinalPrice
|
|||
|
|
if final > 0 && (final < sell || sell == 0) && !seller.Exclude {
|
|||
|
|
sell = final
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 成本和售价没有改动，不保存，直接返回
|
|||
|
|
if cost == article.CostPrice && sell == article.SellPrice {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
// 到这里说明成本和售价有变动
|
|||
|
|
article.CostPrice = Decimal(cost)
|
|||
|
|
article.SellPrice = Decimal(sell)
|
|||
|
|
if cost > 0 {
|
|||
|
|
article.Rate = Decimal((article.SellPrice - article.CostPrice) * 100 / article.CostPrice)
|
|||
|
|
}
|
|||
|
|
// 保存更新
|
|||
|
|
return c.storage.Article().Update(article, "cost_price", "sell_price", "available", "updated_at", "rate")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Decimal(num float64) float64 {
|
|||
|
|
num, _ = strconv.ParseFloat(fmt.Sprintf("%.2f", num), 64)
|
|||
|
|
return num
|
|||
|
|
}
|