MTP20_IF/models/wallet.go

package models

import (
	"mtp2_if/db"
	"mtp2_if/utils"
)

type DigitalWalletAddress struct {
	ID               int64  `json:"id" xorm:"ID"`                                                                        // 地址ID
	DIGITALACCOUNTID int64  `json:"digitalaccountid" xorm:"DIGITALACCOUNTID" form:"digitalaccountid" binding:"required"` // 数字账户ID
	CHANNEL_CODE     string `json:"channelcode" xorm:"CHANNEL_CODE"`                                                     // 渠道代码
	ADDR_TYPE        int32  `json:"addrtype" xorm:"ADDR_TYPE"`                                                           // 地址类型：1-充值；2-提现
	CHAIN_CODE       string `json:"chaincode" xorm:"CHAIN_CODE"`                                                         // 链代码(如BTC,ETH)
	ADDRESS          string `json:"address" xorm:"ADDRESS"`                                                              // 区块链地址
	MEMO             string `json:"memo" xorm:"MEMO"`                                                                    // 地址备注(某些链需要)
}

func (r *DigitalWalletAddress) calc() {

}

func (r *DigitalWalletAddress) buildSql() string {
	var sqlId utils.SQLVal = `
	SELECT 
		t.ID,
		t.DIGITALACCOUNTID,
		t.CHANNEL_CODE,
		t.ADDR_TYPE,
		t.CHAIN_CODE,
		t.ADDRESS,
		t.MEMO,
		t.STATUS,
		t.CREATE_TIME 
	FROM WALLET_ADDRESS t 
	WHERE t.DIGITALACCOUNTID = %v  
	`
	sqlId.FormatParam(r.DIGITALACCOUNTID)

	return sqlId.String()
}

// GetDataEx 从数据库中查询数据
func (r *DigitalWalletAddress) GetDataEx() (interface{}, error) {
	sData := make([]DigitalWalletAddress, 0)
	err := db.GetEngine().SQL(r.buildSql()).Find(&sData)
	for i := range sData {
		sData[i].calc()
	}
	return sData, err
}

type WalletChainRsp struct {
	Walletchains
	Tokens []Wallettokens `json:"tokens"` // 链支持的代币列表
}

func GetWalletChains(channelCode string) ([]WalletChainRsp, error) {
	rsp := make([]WalletChainRsp, 0)
	datas := make([]Walletchains, 0)
	err := db.GetEngine().Where("CHANNEL_CODE = ?", channelCode).Find(&datas)
	if err != nil {
		return nil, err
	}
	for i := range datas {
		tokens := make([]Wallettokens, 0)
		err = db.GetEngine().Where("CHANNEL_CODE = ? AND CHAIN_ID = ?", channelCode, datas[i].CHAIN_ID).Find(&tokens)
		if err != nil {
			return nil, err
		}
		rsp = append(rsp, WalletChainRsp{
			Walletchains: datas[i],
			Tokens:       tokens,
		})
	}

	return rsp, nil
}