log-server/service/rentComputer/rent_set_meal.go

package rentComputer

import (
	"context"
	"errors"
	"go.uber.org/zap"
	"gorm.io/gorm"
	"log-server/global"
	"log-server/model/rentComputer"
	"log-server/model/rentComputer/request"
	"log-server/model/rentComputer/response"
	"time"
)

type ServiceRentSetMeal struct {
}

func (s *ServiceRentSetMeal) RentSetMealList(ctx context.Context, api rentComputer.RentSetMeal, info request.PageInfo, order string, desc bool) (interface{}, int64, error) {
	//global.GVA_LOG.Info("service----GetComputerShopList----strat")
	//date := api.CreateTime
	//isCurrentDate := false
	//if date == "" {
	//	date = s.CurrentDate()
	//	isCurrentDate = true
	//}
	db := global.GVA_DB.Model(&rentComputer.RentSetMeal{})
	var total int64
	db = db.Select("rent_set_meal.id, rent_set_meal.update_time,rent_set_meal.create_time,rent_set_meal.name," +
		"rent_set_meal.remark,rent_set_meal.rent_price,rent_set_meal.price_type,rent_set_meal.rent_price_day,rent_set_meal.shop_id,s.name as shop_name")
	db = db.Joins("left join rent_computer_shop s on s.id = rent_set_meal.shop_id")
	//db = db.Distinct("pc_code", "id", "operator", "create_date", "GROUP_CONCAT(game_id) AS game_ids")
	//db = db.Where("create_date = ?", date)
	//if api.Operator != "" {
	//	db = db.Where("operator like ?", "%"+api.Operator+"%")
	//}
	//if api.PcCode != "" {
	//	db = db.Where("pc_code = ?", api.PcCode)
	//}
	//if api.GameId != 0 {
	//	db = db.Where("game_id = ?", api.GameId)
	//}
	//.Select("count(distinct(name))").Count(&count)
	//err := db.Count(&total).Error
	//err := db.Select("count(distinct(pc_code))").Count(&total).Error
	//db = db.Group("pc_code")
	//db2 := global.GVA_DB.Model(&log.LogComputer{})
	//db2 = db2.Select("count(distinct(pc_code))")
	//db2 = db2.Where("create_date = ?", date)
	//if api.Operator != "" {
	//	db2 = db2.Where("operator like ?", "%"+api.Operator+"%")
	//}
	//if api.PcCode != "" {
	//	db2 = db2.Where("pc_code = ?", api.PcCode)
	//}
	err := db.Count(&total).Error
	//if err != nil {
	//	return nil, 0, err
	//}
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	//var statisticsLogs []*log.LogComputer
	//var statisticsLogs []*rentComputer.RentSetMeal
	var statisticsLogs []*response.SetMealStatisticsReply1
	db = db.Limit(limit).Offset(offset)
	if order != "" {
		var OrderStr string
		// 设置有效排序key 防止sql注入
		// 感谢 Tom4t0 提交漏洞信息
		orderMap := make(map[string]bool, 3)
		orderMap["id"] = true
		if orderMap[order] {
			if desc {
				OrderStr = order + " desc"
			} else {
				OrderStr = order
			}
		} else { // didn't matched any order key in `orderMap`
			global.GVA_LOG.Error("获取失败!", zap.Error(err))
			return statisticsLogs, total, err
		}
		err = db.Order(OrderStr).Find(&statisticsLogs).Error
	} else {
		err = db.Order("id").Find(&statisticsLogs).Error
	}
	if err != nil {
		return nil, 0, err
	}
	var statisticsSetMeals []*response.SetMealStatisticsReply1

	for _, statisticsLog := range statisticsLogs {

		//gameIdStr:=strconv.Itoa(statisticsLog.GameId)
		//taskStatistics := s.GameTargetInfo(ctx,s.LogicalLog.CurrentDate(),gameIdStr)
		statisticsSetMeal := new(response.SetMealStatisticsReply1)
		statisticsSetMeal.Id = statisticsLog.Id
		//updateTime, _ := time.ParseInLocation("2006-01-02 15:04:05", statisticsLog.UpdateTime, time.Local)
		statisticsSetMeal.UpdateTime = statisticsLog.UpdateTime
		//global.GVA_LOG.Info(statisticsLog.CreateTime.String())
		//createTime, _ := time.ParseInLocation("2006-01-02 15:04:05", statisticsLog.CreateTime, time.Local)
		//global.GVA_LOG.Info(createTime.String())
		statisticsSetMeal.CreateTime = statisticsLog.CreateTime
		statisticsSetMeal.Name = statisticsLog.Name
		statisticsSetMeal.Remark = statisticsLog.Remark
		statisticsSetMeal.PriceType = statisticsLog.PriceType
		statisticsSetMeal.RentPriceDay = statisticsLog.RentPriceDay
		statisticsSetMeal.RentPrice = statisticsLog.RentPrice
		statisticsSetMeal.ShopId = statisticsLog.ShopId
		statisticsSetMeal.ShopName = statisticsLog.ShopName
		//if isCurrentDate {
		//	statisticsLogComputer.ComputerFeeRate = s.GetStatisticsPcFeeRate(ctx, statisticsLog.PcCode, statisticsLog.GameId)
		//}
		//statisticsLogComputer.ComputerHourAverageRate = statisticsLog.ComputerHourAverageRate
		statisticsSetMeals = append(statisticsSetMeals, statisticsSetMeal)
	}
	return statisticsSetMeals, total, err
}

func (s *ServiceRentSetMeal) GetRentSetMealNum() int64 {
	var total int64

	db := global.GVA_DB.Model(&rentComputer.RentSetMeal{})
	db = db.Distinct("id")
	_ = db.Count(&total).Error
	return total
}

func (s *ServiceRentSetMeal) GetRentSetMealById(id int) (computerShop response.SetMealStatisticsReply1, err error) {
	var api rentComputer.RentSetMeal
	err = global.GVA_DB.Where("id = ?", id).First(&api).Error
	if err != nil {
		return
	}
	computerShop.Id = api.Id
	computerShop.Name = api.Name
	computerShop.Remark = api.Remark
	computerShop.RentPrice = api.RentPrice
	computerShop.PriceType = api.PriceType
	computerShop.RentPriceDay = api.RentPriceDay
	computerShop.ShopId = api.ShopId
	computerShop.UpdateTime = api.UpdateTime
	computerShop.CreateTime = api.CreateTime
	return
}

func (s *ServiceRentSetMeal) AddRentSetMeal(requestCoding request.SetMealRequest) (err error) {
	//fmt.Printf("%#v\n", &requestCoding)
	if !errors.Is(global.GVA_DB.Where("name = ? and price_type = ? and shop_id = ?", requestCoding.Name, requestCoding.PriceType, requestCoding.ShopId).First(&rentComputer.RentSetMeal{}).Error, gorm.ErrRecordNotFound) {
		return errors.New("存在相同套餐")
	}
	computerShop := new(rentComputer.RentSetMeal)

	priceType := requestCoding.PriceType
	rentPrice := requestCoding.RentPrice
	rentPriceDay := float64(0)
	computerShop.Name = requestCoding.Name
	computerShop.Remark = requestCoding.Remark

	if priceType == 0 {
		rentPriceDay = float64(int(rentPrice))
	} else if priceType == 1 {
		rentPriceDay = rentPrice / 7
	} else if priceType == 2 {
		rentPriceDay = rentPrice / 30
	}
	computerShop.PriceType = priceType
	computerShop.RentPrice = rentPrice
	computerShop.RentPriceDay = rentPriceDay
	computerShop.ShopId = requestCoding.ShopId
	computerShop.CreateTime = time.Now().Format("2006-01-02 15:04:05")
	computerShop.UpdateTime = time.Now().Format("2006-01-02 15:04:05")
	//key := codeListCacheKey
	//_ = s.cache.DelBatheHsCache(context.Background(), key)
	return global.GVA_DB.Create(&computerShop).Error
}

func (s *ServiceRentSetMeal) EditRentSetMeal(setMeal rentComputer.RentSetMeal) (err error) {
	priceType := setMeal.PriceType
	rentPrice := setMeal.RentPrice
	rentPriceDay := float64(0)
	if priceType == 0 {
		rentPriceDay = float64(int(rentPrice))
	} else if priceType == 1 {
		rentPriceDay = rentPrice / 7
	} else if priceType == 2 {
		rentPriceDay = rentPrice / 30
	}
	setMeal.RentPriceDay = rentPriceDay
	err = global.GVA_DB.Save(setMeal).Error
	return err
}

func (s *ServiceRentSetMeal) DeleteRentSetMealByIds(ids request.IdsReq) (err error) {
	err = global.GVA_DB.Delete(&[]rentComputer.RentSetMeal{}, "id in ?", ids.Ids).Error
	return err
}

func (s *ServiceRentSetMeal) DeleteRentSetMealById(api rentComputer.RentSetMeal) (err error) {
	var entity rentComputer.RentSetMeal
	err = global.GVA_DB.Where("id = ?", api.Id).First(&entity).Error // 根据id查询api记录
	if errors.Is(err, gorm.ErrRecordNotFound) {                      // api记录不存在
		return err
	}
	return global.GVA_DB.Delete(&entity).Error
}