MTP20_IF/controllers/common/table.go

package common

import (
	"fmt"
	"mtp2_if/db"
	"mtp2_if/global/app"
	"mtp2_if/global/e"
	"mtp2_if/global/utils"
	"mtp2_if/logger"
	"mtp2_if/models"
	"net/http"
	"strconv"
	"strings"

	"github.com/gin-gonic/gin"
)

// QueryTraderMenuReq 查询交易端菜单请求参数
type QueryTraderMenuReq struct {
	LoginID int `form:"loginid" binding:"required"`
}

// QueryTraderMenuRsp 交易端菜单
type QueryTraderMenuRsp struct {
	QuoteMenu     []QuotePrimaryMenu     `json:"QuoteMenu"`     // 报价牌分类菜单
	OperationMenu []OperationPrimaryMenu `json:"OperationMenu"` // 功能菜单
}

// QuotePrimaryMenu 报价牌一级分类菜单
type QuotePrimaryMenu struct {
	Index        int                  `json:"Index"`        // 序号
	Key          string               `json:"Key"`          // 键名
	Name         string               `json:"Name"`         // 菜单名称
	SubTitleType int                  `json:"SubTitleType"` // 子菜单标题模式：0-市场名称；1-外部交易所名称
	TradeModes   string               `json:"TradeModes"`   // 包含市场交易类型
	SubMenus     []QuoteSecondaryMenu `json:"SubMenus"`     // 子菜单
}

// QuoteSecondaryMenu 报价牌二级分类菜单
type QuoteSecondaryMenu struct {
	Index          int    `json:"Index"`                                  // 序号
	MarketID       int    `json:"MarketID"`                               // 市场ID
	TradeMode      int    `json:"TradeMode"`                              // 交易模式
	MenuTitle      string `json:"MenuTitle" xorm:"'ExExchangeName'"`      // 菜单标题（市场名称或外部交易所名称）
	GoodsGroupIDs  []int  `json:"GoodsGroupIDs"`                          // 商品组ID列表
	ExExchangeID   int    `json:"ExExchangeID" xorm:"'ExExchangeID'"`     // 外部交易所ID
	ExExchangeCode string `json:"ExExchangeCode" xorm:"'ExExchangeCode'"` // 外部交易所代码
}

// OperationPrimaryMenu 一级功能菜单
type OperationPrimaryMenu struct {
	Key      string                   `json:"Key"`      // 菜单KEY
	Label    string                   `json:"Label"`    // 菜单标题
	Children []OperationSecondaryMenu `json:"Children"` // 二级功能菜单
}

// OperationSecondaryMenu 二级功能菜单
type OperationSecondaryMenu struct {
	Key     string             `json:"Key"`     // 菜单KEY
	Label   string             `json:"Label"`   // 菜单标题
	TabList []OperationTabMenu `json:"TabList"` // 三级功能菜单
}

// OperationTabMenu 三级功能菜单
type OperationTabMenu struct {
	Key   string `json:"Key"`   // 菜单KEY
	Label string `json:"Label"` // 菜单标题
}

// QueryTraderMenu 查询交易端菜单
// @Summary 查询交易端菜单
// @Produce json
// @Security ApiKeyAuth
// @Param loginid query int true "登录账号"
// @Success 200 {object} QueryTraderMenuRsp
// @Failure 500 {object} app.Response
// @Router /Common/QueryTraderMenu [get]
// @Tags 通用
func QueryTraderMenu(c *gin.Context) {
	appG := app.Gin{C: c}

	// 获取请求参数
	var req QueryTraderMenuReq
	if err := appG.C.ShouldBindQuery(&req); err != nil {
		logger.GetLogger().Errorf("QueryPreasleApply failed: %s", err.Error())
		appG.Response(http.StatusBadRequest, e.INVALID_PARAMS, nil)
		return
	}

	var queryTraderMenuRsp QueryTraderMenuRsp
	// 获取行情报价牌分类菜单
	if datas, err := getQuoteMenu(req.LoginID); err == nil {
		queryTraderMenuRsp.QuoteMenu = datas
	}
	// 获取功能菜单
	if datas, err := getOperationMenu(); err == nil {
		queryTraderMenuRsp.OperationMenu = datas
	}

	// 查询成功
	logger.GetLogger().Infof("QueryPreasleApply successed: %v", queryTraderMenuRsp)
	appG.Response(http.StatusOK, e.SUCCESS, queryTraderMenuRsp)
}

// getQuoteMenu 获取行情报价牌分类菜单
func getQuoteMenu(loginID int) ([]QuotePrimaryMenu, error) {
	engine := db.GetEngine()
	rst := make([]QuotePrimaryMenu, 0)

	// ********************* 准备数据 *********************
	var taAccounts []string // 账户下所有资金账户
	var marketIDs []int     // 资金账户下有权限的市场ID列表
	// 先要获取当前登录账户对应的资金账户
	if err := engine.SQL(fmt.Sprintf(`select to_char(AccountID) from LoginTAAccount where LoginID = %d`, loginID)).Find(&taAccounts); err != nil {
		return nil, err
	}
	// 如果一条记录都没有则直接通过资金账户表获取
	if len(taAccounts) == 0 {
		sql := fmt.Sprintf(`select to_char(ta.accountid) from taaccount ta
							inner join loginaccount l on ta.userid = l.userid
							where l.loginid = %d`, loginID)
		if err := engine.SQL(sql).Find(&taAccounts); err != nil {
			return nil, err
		}
	}
	// 获取资金账户对应的市场权限（有权限的市场ID）
	if len(taAccounts) > 0 {
		taAccountStr := strings.Join(taAccounts, ",")
		if err := engine.SQL(fmt.Sprintf(`select MarketID from TAAccountMarket where AccountID in (%s)`, taAccountStr)).Find(&marketIDs); err != nil {
			return nil, err
		}
	}

	// ********************* 构建行情报价牌菜单 *********************
	// 获取一级菜单列表
	datas := make([]models.Tablecolumnconfig, 0)
	if err := engine.Where("TableKey = 'trader_master_menu' and IsShow = 1").Find(&datas); err != nil {
		return nil, err
	}
	// 构建一级菜单对象
	for i, v := range datas {
		quotePrimaryMenu := QuotePrimaryMenu{
			Index: i,
			Key:   v.Columnfield,
			Name:  v.Columntitle,
		}
		// 跳过自选
		if v.Columnfield == "optional" {
			rst = append(rst, quotePrimaryMenu)
			continue
		}
		// FormatterString：模式，0-使用市场名称，1-使用外部交易所名称；Remark：包含市场交易模式
		quotePrimaryMenu.SubTitleType, _ = strconv.Atoi(v.Formatterstring)
		quotePrimaryMenu.TradeModes = v.Remark

		// 构建二级子菜单对象
		marketIDsStr := utils.JoinItoString(marketIDs, ",")
		if quotePrimaryMenu.SubTitleType == 0 {
			// 获取目标交易模式的市场信息
			markets := make([]models.Market, 0)
			if err := engine.Where(fmt.Sprintf(`TradeMode in (%s) and MarketID in (%s)`, quotePrimaryMenu.TradeModes, marketIDsStr)).Find(&markets); err != nil {
				return nil, err
			}
			// 使用市场名称
			for mi, mv := range markets {
				quoteSecondaryMenu := QuoteSecondaryMenu{
					Index:     mi,
					MarketID:  int(mv.Marketid),
					TradeMode: int(mv.Trademode),
					MenuTitle: mv.Marketname,
				}
				quotePrimaryMenu.SubMenus = append(quotePrimaryMenu.SubMenus, quoteSecondaryMenu)
			}
		} else {
			// 使用外部交易所名称
			quoteSecondaryMenus := make([]QuoteSecondaryMenu, 0)
			sql := fmt.Sprintf(`select distinct
									e.autoid ExExchangeID,
									e.ExExchangeName,
									e.ExExchangeCode 
								from ExternalExchange e
								inner join goodsgroup g  on g.exexchangeid = e.autoid
								inner join Market m on g.marketid = m.marketid
								where m.trademode in (%s) and m.marketid in (%s)`, quotePrimaryMenu.TradeModes, marketIDsStr)
			if err := engine.SQL(sql).Find(&quoteSecondaryMenus); err != nil {
				return nil, err
			}
			// 获取外部交易所对应的商品组信息
			for ei, ev := range quoteSecondaryMenus {
				q := &quoteSecondaryMenus[ei]
				q.Index = ei
				// 商品组列表
				goodsgroups := make([]models.Goodsgroup, 0)
				if err := engine.Where("Exexchangeid = ?", ev.ExExchangeID).Find(&goodsgroups); err != nil {
					return nil, err
				}
				marketID := 0
				var goodsGroupIDs []int
				for _, gv := range goodsgroups {
					marketID = int(gv.Marketid)
					goodsGroupIDs = append(goodsGroupIDs, int(gv.Goodsgroupid))
				}
				q.MarketID = marketID
				q.GoodsGroupIDs = goodsGroupIDs
			}
			quotePrimaryMenu.SubMenus = quoteSecondaryMenus
		}

		rst = append(rst, quotePrimaryMenu)
	}

	return rst, nil
}

// getOperationMenu 获取功能菜单
func getOperationMenu() ([]OperationPrimaryMenu, error) {
	engine := db.GetEngine()
	rst := make([]OperationPrimaryMenu, 0)

	// 获取一级功能菜单
	opm := make([]models.Tablecolumnconfig, 0)
	if err := engine.Where("IsShow = 1 and TableKey = ?", "trader_operation_master_menu").Asc("ORDERINDEX").Find(&opm); err != nil {
		return nil, err
	}
	for _, pv := range opm {
		var operationPrimaryMenu = OperationPrimaryMenu{
			Key:      pv.Columnfield,
			Label:    pv.Columntitle,
			Children: make([]OperationSecondaryMenu, 0),
		}

		// 获取二级功能菜单
		osm := make([]models.Tablecolumnconfig, 0)
		if err := engine.Where("IsShow = 1 and Remark = ?", operationPrimaryMenu.Key).Asc("ORDERINDEX").Find(&osm); err != nil {
			return nil, err
		}
		for _, sv := range osm {
			var operationSecondaryMenu = OperationSecondaryMenu{
				Key:     sv.Columnfield,
				Label:   sv.Columntitle,
				TabList: make([]OperationTabMenu, 0),
			}

			// 获取三级功能菜单
			otm := make([]models.Tablecolumnconfig, 0)
			if err := engine.Where("IsShow = 1 and Remark = ?", operationSecondaryMenu.Key).Asc("ORDERINDEX").Find(&otm); err != nil {
				return nil, err
			}
			for _, tv := range otm {
				var operationTabMenu = OperationTabMenu{
					Key:   tv.Columnfield,
					Label: tv.Columntitle,
				}

				operationSecondaryMenu.TabList = append(operationSecondaryMenu.TabList, operationTabMenu)
			}

			operationPrimaryMenu.Children = append(operationPrimaryMenu.Children, operationSecondaryMenu)
		}

		rst = append(rst, operationPrimaryMenu)
	}

	return rst, nil
}