reward_strategy.go 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. package db
  2. import (
  3. "context"
  4. "fmt"
  5. "github.com/sirupsen/logrus"
  6. "youngee_m_api/model/gorm_model"
  7. )
  8. func CreateRewardStrategy(ctx context.Context, RewardStrategys []gorm_model.RewardStrategy) error {
  9. db := GetReadDB(ctx)
  10. fmt.Println("RewardStrategys: ", RewardStrategys)
  11. err := db.Create(&RewardStrategys).Error
  12. if err != nil {
  13. return err
  14. }
  15. return nil
  16. }
  17. func DeleteRewardStrategyBySelectionId(ctx context.Context, SelectionId string) error {
  18. db := GetReadDB(ctx)
  19. err := db.Where("selection_id = ?", SelectionId).Delete(&gorm_model.RewardStrategy{}).Error
  20. if err != nil {
  21. return err
  22. }
  23. return nil
  24. }
  25. func GetRewardStrategyBySelectionId(ctx context.Context, SelectionId string) ([]*gorm_model.RewardStrategy, error) {
  26. db := GetReadDB(ctx)
  27. var RewardStrategys []*gorm_model.RewardStrategy
  28. err := db.Model(gorm_model.RewardStrategy{}).Where("selection_id = ?", SelectionId).Find(&RewardStrategys).Error
  29. if err != nil {
  30. logrus.WithContext(ctx).Errorf("[GetRewardStrategyBySelectionId] error query, err:%+v", err)
  31. return nil, err
  32. }
  33. return RewardStrategys, nil
  34. }