代码提交。

Signed-off-by: zouap <zouap@pcl.ac.cn>

models/user_invitation.go

@@ -3,6 +3,7 @@ package models
 import (
 	"fmt"
 
+	"code.gitea.io/gitea/modules/log"
 	"code.gitea.io/gitea/modules/timeutil"
 )
 
@@ -49,16 +50,20 @@ func InsertInvitaion(invitationUser *Invitation) error {
 	return err
 }
 
-func QueryInvitaionBySrcUserId(srcUserId int64) []*Invitation {
+func QueryInvitaionBySrcUserId(srcUserId int64, start int, pageSize int) ([]*Invitation, int64) {
 	statictisSess := xStatistic.NewSession()
 	defer statictisSess.Close()
 	cond := "src_user_id =" + fmt.Sprint(srcUserId)
-
+	allCount, err := statictisSess.Where(cond).Count(new(Invitation))
+	if err != nil {
+		log.Info("query error." + err.Error())
+		return nil, 0
+	}
 	invitationList := make([]*Invitation, 0)
 
-	if err := statictisSess.Table(new(Invitation)).Where(cond).OrderBy("created_unix desc").
+	if err := statictisSess.Table(new(Invitation)).Where(cond).OrderBy("created_unix desc").Limit(pageSize, start).
 		Find(&invitationList); err != nil {
-		return nil
+		return nil, 0
 	}
-	return invitationList
+	return invitationList, allCount
 }

routers/user/Invitation.go

@@ -18,6 +18,14 @@ const (
 )
 
 func GetInvitaionCode(ctx *context.Context) {
+	page := ctx.QueryInt("page")
+	if page <= 0 {
+		page = 1
+	}
+	pageSize := ctx.QueryInt("pageSize")
+	if pageSize <= 0 {
+		pageSize = setting.UI.IssuePagingNum
+	}
 
 	url := setting.RecommentRepoAddr + "invitaion_page"
 	result, err := repository.RecommendFromPromote(url)
@@ -35,7 +43,9 @@ func GetInvitaionCode(ctx *context.Context) {
 
 	if ctx.IsSigned {
 		resultJsonMap["invitation_code"] = getInvitaionCode(ctx)
-		resultJsonMap["invitation_users"] = models.QueryInvitaionBySrcUserId(ctx.User.ID)
+		re, count := models.QueryInvitaionBySrcUserId(ctx.User.ID, (page-1)*pageSize, pageSize)
+		resultJsonMap["invitation_users"] = re
+		resultJsonMap["invitation_users_count"] = count
 	}
 
 	ctx.JSON(200, resultJsonMap)