|
|
@@ -1,7 +1,6 @@
|
|
|
package user
|
|
|
|
|
|
import (
|
|
|
- "fmt"
|
|
|
"mtp2_if/db"
|
|
|
"mtp2_if/global/app"
|
|
|
"mtp2_if/global/e"
|
|
|
@@ -20,8 +19,7 @@ type QueryUserReferNumReq struct {
|
|
|
// QueryUserReferNum 获取用户邀请码
|
|
|
// @Summary 获取用户邀请码
|
|
|
// @Produce json
|
|
|
-// @Security ApiKeyAuth
|
|
|
-// @Param username query string true "登录代码"
|
|
|
+// @Param userID query string true "用户ID"
|
|
|
// @Success 200 {object} app.Response
|
|
|
// @Failure 500 {object} app.Response
|
|
|
// @Router /User/QueryUserReferNum [get]
|
|
|
@@ -30,30 +28,26 @@ func QueryUserReferNum(c *gin.Context) {
|
|
|
appG := app.Gin{C: c}
|
|
|
|
|
|
// 获取请求参数
|
|
|
- var req GetLoginIDReq
|
|
|
+ var req QueryUserReferNumReq
|
|
|
if err := appG.C.ShouldBindQuery(&req); err != nil {
|
|
|
- logger.GetLogger().Errorf("QueryPreasleApply failed: %s", err.Error())
|
|
|
+ logger.GetLogger().Errorf("QueryUserReferNum failed: %s", err.Error())
|
|
|
appG.Response(http.StatusBadRequest, e.INVALID_PARAMS, nil)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 查询数据
|
|
|
engine := db.GetEngine()
|
|
|
- var loginaccount models.Loginaccount
|
|
|
+ var userAccount models.Useraccount
|
|
|
|
|
|
- loginid := req.UserName // 如果找不到,则直接把登录代码返回去
|
|
|
- has, err := engine.Where("Logincode = ?", req.UserName).Get(&loginaccount)
|
|
|
+ _, err := engine.Where("USERID = ?", req.UserID).Get(&userAccount)
|
|
|
if err != nil {
|
|
|
// 查询失败
|
|
|
- logger.GetLogger().Errorf("GetLoginID failed: %s", err.Error())
|
|
|
+ logger.GetLogger().Errorf("QueryUserReferNum failed: %s", err.Error())
|
|
|
appG.Response(http.StatusBadRequest, e.ERROR_QUERY_FAIL, nil)
|
|
|
return
|
|
|
}
|
|
|
- if has {
|
|
|
- loginid = fmt.Sprintf("%v", loginaccount.Loginid)
|
|
|
- }
|
|
|
|
|
|
// 查询成功
|
|
|
- logger.GetLogger().Infof("GetLoginID successed: %v", loginaccount)
|
|
|
- appG.Response(http.StatusOK, e.SUCCESS, loginid)
|
|
|
+ logger.GetLogger().Infof("QueryUserReferNum successed: %v", userAccount.Refernum)
|
|
|
+ appG.Response(http.StatusOK, e.SUCCESS, userAccount.Refernum)
|
|
|
}
|