duoduo/apis/shanghu/account.go

package shanghu

import (
	"duoduo/apis/shanghu/models"
	"duoduo/models/shanghu"
	"duoduo/tools/app"
	"github.com/gin-gonic/gin"
)

func GetAccountAmount(c *gin.Context) {
	var inData models.GetAccountAmountRequest
	var outData models.GetAccountAmountReply
	var sqlMerchantData shanghu.MerchantAccount
	var sqlClientData shanghu.MerchantClientAccount

	err := c.ShouldBindJSON(&inData)
	if err != nil {
		app.Error(c, 400, err, err.Error())
		return
	}
	if inData.Type == "merchant" {
		sqlMerchantData.MerchantOpenID = inData.OpenId
		merchantAmount, err := sqlMerchantData.GetMerchantAccount()
		if err != nil {
			app.Error(c, 500, err, err.Error())
			return
		}
		outData.Amount = merchantAmount.Amount
		app.OK(c, outData, app.Success)
		return

	} else if inData.Type == "client" {
		sqlClientData.ClientOpenID = inData.OpenId
		clientAccount, err := sqlClientData.GetClientAccount()
		if err != nil {
			app.Error(c, 500, err, err.Error())
			return
		}
		outData.Amount = clientAccount.Amount
		app.OK(c, outData, app.Success)
		return
	} else {
		app.Error(c, 400, err, err.Error())
		return
	}

}