log-server/service/rentComputer/rent_computer_shop.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 ServiceRentComputerShop struct {
}

func (s *ServiceRentComputerShop) RentComputerShopList(ctx context.Context, api rentComputer.RentComputerShop, 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.RentComputerShop{})
	var total int64
	//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.RentComputerShop
	db = db.Limit(limit).Offset(offset)
	if order != "" {
		var OrderStr string
		// 设置有效排序key 防止sql注入
		// 感谢 Tom4t0 提交漏洞信息
		orderMap := make(map[string]bool, 3)
		orderMap["pc_code"] = true
		orderMap["game_id"] = true
		orderMap["operator"] = 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 statisticsLogsComputer []*response.ComputerStatisticsReply1

	for _, statisticsLog := range statisticsLogs {
		/*var gameInfo []*response.GameInfo
		err = json.Unmarshal([]byte(statisticsLog.GameInfo), &gameInfo)
		for i, _ := range gameInfo {
			statisticsLogComputer := new(response.ComputerStatisticsReply1)
			statisticsLogComputer.GameInfo = gameInfo
			statisticsLogComputer.PcCode = statisticsLog.PcCode
			statisticsLogComputer.Operator = statisticsLog.Operator
			statisticsLogComputer.CreateDate = statisticsLog.CreateDate[:10]
			statisticsLogComputer.GameId = gameInfo[i].GameId
			statisticsLogComputer.PullAccountNum = gameInfo[i].PullAccountNum
			statisticsLogComputer.TaskSuccessNum = gameInfo[i].TaskSuccessNum
			statisticsLogsComputer = append(statisticsLogsComputer, statisticsLogComputer)
		}*/
		//gameIdStr:=strconv.Itoa(statisticsLog.GameId)
		//taskStatistics := s.GameTargetInfo(ctx,s.LogicalLog.CurrentDate(),gameIdStr)
		statisticsLogComputer := new(response.ComputerStatisticsReply1)
		statisticsLogComputer.Id = statisticsLog.Id
		//updateTime, _ := time.ParseInLocation("2006-01-02 15:04:05", statisticsLog.UpdateTime, time.Local)
		statisticsLogComputer.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())
		statisticsLogComputer.PayMethod = statisticsLog.PayMethod
		statisticsLogComputer.CreateTime = statisticsLog.CreateTime

		statisticsLogComputer.Name = statisticsLog.Name
		statisticsLogComputer.Description = statisticsLog.Description

		//if isCurrentDate {
		//	statisticsLogComputer.ComputerFeeRate = s.GetStatisticsPcFeeRate(ctx, statisticsLog.PcCode, statisticsLog.GameId)
		//}
		//statisticsLogComputer.ComputerHourAverageRate = statisticsLog.ComputerHourAverageRate
		statisticsLogsComputer = append(statisticsLogsComputer, statisticsLogComputer)
	}
	return statisticsLogsComputer, total, err
}

func (s *ServiceRentComputerShop) GetRentComputerShopNum() int64 {
	var total int64

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

func (s *ServiceRentComputerShop) GetRentComputerShopById(id int) (computerShop response.ComputerStatisticsReply1, err error) {
	var api rentComputer.RentComputerShop
	err = global.GVA_DB.Where("id = ?", id).First(&api).Error
	if err != nil {
		return
	}
	computerShop.Id = api.Id
	computerShop.Name = api.Name
	computerShop.PayMethod = api.PayMethod
	computerShop.Description = api.Description
	computerShop.UpdateTime = api.UpdateTime
	computerShop.CreateTime = api.CreateTime
	return
}

func (s *ServiceRentComputerShop) CurrentDate() (current string) {
	current = time.Now().Format("2006-01-02")
	return
}

func (s *ServiceRentComputerShop) AddRentComputerShop(requestCoding request.ComputerShopRequest) (err error) {
	if !errors.Is(global.GVA_DB.Where("name = ?", requestCoding.Name).First(&rentComputer.RentComputerShop{}).Error, gorm.ErrRecordNotFound) {
		return errors.New("存在相同供应商名")
	}
	computerShop := new(rentComputer.RentComputerShop)

	computerShop.Name = requestCoding.Name
	computerShop.PayMethod = requestCoding.PayMethod
	computerShop.Description = requestCoding.Description
	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 *ServiceRentComputerShop) EditRentComputerShop(computerShop rentComputer.RentComputerShop) (err error) {
	err = global.GVA_DB.Save(computerShop).Error
	return err
}

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

func (s *ServiceRentComputerShop) DeleteRentComputerShopById(api rentComputer.RentComputerShop) (err error) {
	var entity rentComputer.RentComputerShop
	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
}