修改“获取企业风管终端权限”接口

controllers/ermcp/qryUser.go

@@ -55,6 +55,7 @@ func QueryUserInfo(c *gin.Context) {
 // GetErmcpRolefuncMenuReq 获取企业风管终端权限请求参数
 type GetErmcpRolefuncMenuReq struct {
 	LoginID int `form:"loginID" binding:"required"`
+	Mode    int `form:"mode"`
 }
 
 // GetErmcpRolefuncMenu 获取企业风管终端权限
@@ -62,6 +63,7 @@ type GetErmcpRolefuncMenuReq struct {
 // @Produce json
 // @Security ApiKeyAuth
 // @Param loginID query int true "登录账号"
+// @Param mode query int false "模式，0：普通（交易所版本）；1：企业风管"
 // @Success 200 {object} models.Funcmenulist
 // @Failure 500 {object} app.Response
 // @Router /Ermcp/GetErmcpRoleFuncMenuLists [get]
@@ -78,12 +80,28 @@ func GetErmcpRoleFuncMenuLists(c *gin.Context) {
 	}
 
 	// 获取数据
-	rst, err := models.GetRoleFuncMenuLists(req.LoginID, "")
-	if err != nil {
-		// 查询失败
-		logger.GetLogger().Errorf("GetErmcpRolefuncMenu failed: %s", err.Error())
-		appG.Response(http.StatusBadRequest, e.ERROR_QUERY_FAIL, nil)
-		return
+	rst := make([]models.Funcmenulist, 0)
+	if req.Mode == 0 {
+		// 普通（交易所版本）
+		var err error
+		rst, err = models.GetRoleFuncMenuLists(req.LoginID, "")
+		if err != nil {
+			// 查询失败
+			logger.GetLogger().Errorf("GetErmcpRolefuncMenu failed: %s", err.Error())
+			appG.Response(http.StatusBadRequest, e.ERROR_QUERY_FAIL, nil)
+			return
+		}
+	} else if req.Mode == 1 {
+		// 企业风管
+		// 普通（交易所版本）
+		var err error
+		rst, err = models.GetErmcpRoleFuncMenuLists(req.LoginID, "")
+		if err != nil {
+			// 查询失败
+			logger.GetLogger().Errorf("GetErmcpRolefuncMenu failed: %s", err.Error())
+			appG.Response(http.StatusBadRequest, e.ERROR_QUERY_FAIL, nil)
+			return
+		}
 	}
 
 	// 查询成功返回