go_dreamfactory/modules/troll/api_buyorsell.go

package troll

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/pb"
	"math"
	"time"

	"google.golang.org/protobuf/proto"
)

//参数校验
func (this *apiComp) BuyOrSellCheck(session comm.IUserSession, req *pb.TrollBuyOrSellReq) (code pb.ErrorCode) {
	if len(req.Items) == 0 {
		code = pb.ErrorCode_ReqParameterError
	}
	return
}

func (this *apiComp) BuyOrSell(session comm.IUserSession, req *pb.TrollBuyOrSellReq) (code pb.ErrorCode, data proto.Message) {
	var (
		bSell   bool  // 是否有出售
		gold    int32 // 当次交易 获得的金币
		earn    int32 // 只统计赚的金币
		update  map[string]interface{}
		gridNum int32 // 格子数量
	)
	update = make(map[string]interface{})
	if code = this.BuyOrSellCheck(session, req); code != pb.ErrorCode_Success {
		return // 参数校验失败直接返回
	}
	trolltrain, err := this.module.modelTroll.getTrollList(session.GetUserId())
	if err != nil {
		code = pb.ErrorCode_DBError
		return
	}
	for k, v := range req.Items {
		if v < 0 {
			if !bSell {
				bSell = true
				trolltrain.SellCount += 1 // 交易次数+1
				if trolltrain.SellCount > 5 {
					code = pb.ErrorCode_TrollMaxSellCount // 达到最大交易次数 直接返回
					return
				}
				update["sellCount"] = trolltrain.SellCount
			}
		}
		if _, ok := trolltrain.Items[k]; !ok {
			trolltrain.Items[k] = v
		} else {
			trolltrain.Items[k] += v
		}
		if _, ok := trolltrain.Shop[k]; !ok {
			trolltrain.Shop[k] = v
		} else {
			trolltrain.Shop[k] += v
		}
		// 校验 是否大于买入最大限制
		goods := this.configure.GetTrollGoods(k)
		if goods == nil {
			return
		}
		if trolltrain.Shop[k] > goods.Max { // 判断是否有效交易
			//  买入上限  直接返回
			code = pb.ErrorCode_TrollBuyMax
			return
		} else if trolltrain.Items[k] < 0 { //卖出数量不足
			code = pb.ErrorCode_TrollSellMax
			return
		}

		// 第一次购买商品
		if trolltrain.TarinPos == 0 {
			if _, ok := trolltrain.Price[k]; !ok {
				trolltrain.Price[k] = 0
			}
			trolltrain.Price[k] = goods.Goodsprice * goods.StarMoney / 1000
			trolltrain.RefreshTime = time.Now().Unix()
			//消耗的金币
			gold -= trolltrain.Price[k] * trolltrain.Items[k]
		} else {
			p := this.configure.GetTrollCoefficient(trolltrain.TarinPos)
			if p == nil {
				return
			}
			var sellPrice int32 // 交易价格
			if trolltrain.TarinPos == 0 {
				sellPrice = goods.Goodsprice * goods.StarMoney / 1000
			} else {
				sellPrice = int32(p.Coefficient) * goods.Goodsprice / 1000
			}
			if v < 0 { // 卖出
				if sellPrice > trolltrain.Price[k] { // 赚了
					earn += (sellPrice - trolltrain.Price[k]) * v
				}
				gold += sellPrice * v
			} else { // 买入 计算平均价格
				totalGold := (trolltrain.Items[k] - v) * trolltrain.Price[k]
				totalGold += v * sellPrice
				trolltrain.Price[k] = totalGold / (trolltrain.Items[k])
				gold -= v * sellPrice
			}
		}
	}
	// 重新计算格子数量
	grid := this.configure.GetTrollRule(comm.TrollItemCount)
	for _, v := range trolltrain.Items {
		if v > 0 {
			gridNum += int32(math.Ceil(float64(v) / float64(grid)))
		}
	}

	trolltrain.GridNum = gridNum
	if gridNum > this.configure.GetTrollRule(comm.TrollGridCount) { // 背包格子上限
		code = pb.ErrorCode_TrollMaxItemCount
		return
	}
	code = this.module.ModuleUser.AddAttributeValue(session, comm.ResGold, int32(gold), true)
	if code != pb.ErrorCode_Success { // 金币不足
		code = pb.ErrorCode_GoldNoEnough
		return
	}
	// check npc level
	if confLv := this.configure.GetTrollLv(trolltrain.GetNpcLv() + 1); confLv != nil {
		if earn >= confLv.Money {
			trolltrain.NpcLv += 1 // npc levelUp
			update["npcLv"] = trolltrain.NpcLv
		}
	}
	// 清除数量为0 的
	for k, v := range trolltrain.Items {
		if v == 0 {
			delete(trolltrain.Items, k)
		}
	}
	trolltrain.TotalEarn += -int64(earn) // 累计获得的金币
	update["items"] = trolltrain.Items
	update["price"] = trolltrain.Price
	update["totalEarn"] = trolltrain.TotalEarn

	update["gridNum"] = trolltrain.GridNum
	this.module.ModifyTrollData(session.GetUserId(), update)
	session.SendMsg(string(this.module.GetType()), TrollBuyOrSellResp, &pb.TrollBuyOrSellResp{Data: trolltrain})
	return
}