From 188e55b0e0e12710bd2b01774e2fbe987b658828 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Thu, 2 May 2024 13:56:31 +0800
Subject: [PATCH] Use context cache for loading pusher

---
 routers/private/hook_post_receive.go | 47 ++++++++++++++--------------
 1 file changed, 24 insertions(+), 23 deletions(-)

diff --git a/routers/private/hook_post_receive.go b/routers/private/hook_post_receive.go
index 973b5e1a1a..0f4d28c832 100644
--- a/routers/private/hook_post_receive.go
+++ b/routers/private/hook_post_receive.go
@@ -15,6 +15,7 @@ import (
 	pull_model "code.gitea.io/gitea/models/pull"
 	repo_model "code.gitea.io/gitea/models/repo"
 	user_model "code.gitea.io/gitea/models/user"
+	"code.gitea.io/gitea/modules/cache"
 	"code.gitea.io/gitea/modules/git"
 	"code.gitea.io/gitea/modules/gitrepo"
 	"code.gitea.io/gitea/modules/log"
@@ -162,10 +163,8 @@ func HookPostReceive(ctx *gitea_context.PrivateContext) {
 		}
 	}
 
-	var pusher *user_model.User
-
 	// handle pull request merging, a pull request action should push at least 1 commit
-	handlePullRequestMerging(ctx, opts, pusher, ownerName, repoName, updates)
+	handlePullRequestMerging(ctx, opts, ownerName, repoName, updates)
 	if ctx.Written() {
 		return
 	}
@@ -184,17 +183,17 @@ func HookPostReceive(ctx *gitea_context.PrivateContext) {
 			wasEmpty = repo.IsEmpty
 		}
 
-		if pusher == nil {
-			var err error
-			pusher, err = user_model.GetUserByID(ctx, opts.UserID)
-			if err != nil {
-				log.Error("Failed to Update: %s/%s Error: %v", ownerName, repoName, err)
-				ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{
-					Err: fmt.Sprintf("Failed to Update: %s/%s Error: %v", ownerName, repoName, err),
-				})
-				return
-			}
+		pusher, err := cache.GetWithContextCache(ctx, contextCachePusherKey, opts.UserID, func() (*user_model.User, error) {
+			return user_model.GetUserByID(ctx, opts.UserID)
+		})
+		if err != nil {
+			log.Error("Failed to Update: %s/%s Error: %v", ownerName, repoName, err)
+			ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{
+				Err: fmt.Sprintf("Failed to Update: %s/%s Error: %v", ownerName, repoName, err),
+			})
+			return
 		}
+
 		perm, err := access_model.GetUserRepoPermission(ctx, repo, pusher)
 		if err != nil {
 			log.Error("Failed to Update: %s/%s Error: %v", ownerName, repoName, err)
@@ -323,7 +322,9 @@ func HookPostReceive(ctx *gitea_context.PrivateContext) {
 	})
 }
 
-func handlePullRequestMerging(ctx *gitea_context.PrivateContext, opts *private.HookOptions, pusher *user_model.User, ownerName, repoName string, updates []*repo_module.PushUpdateOptions) {
+const contextCachePusherKey = "hook_post_receive_pusher"
+
+func handlePullRequestMerging(ctx *gitea_context.PrivateContext, opts *private.HookOptions, ownerName, repoName string, updates []*repo_module.PushUpdateOptions) {
 	// handle pull request merging, a pull request action should only push 1 commit
 	if opts.PullRequestAction == repo_module.PullRequestActionMerge && len(updates) >= 1 {
 		// Get the pull request
@@ -336,15 +337,15 @@ func handlePullRequestMerging(ctx *gitea_context.PrivateContext, opts *private.H
 			return
 		}
 
-		if pusher == nil {
-			pusher, err = user_model.GetUserByID(ctx, opts.UserID)
-			if err != nil {
-				log.Error("Failed to Update: %s/%s Error: %v", ownerName, repoName, err)
-				ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{
-					Err: fmt.Sprintf("Failed to Update: %s/%s Error: %v", ownerName, repoName, err),
-				})
-				return
-			}
+		pusher, err := cache.GetWithContextCache(ctx, contextCachePusherKey, opts.UserID, func() (*user_model.User, error) {
+			return user_model.GetUserByID(ctx, opts.UserID)
+		})
+		if err != nil {
+			log.Error("Failed to Update: %s/%s Error: %v", ownerName, repoName, err)
+			ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{
+				Err: fmt.Sprintf("Failed to Update: %s/%s Error: %v", ownerName, repoName, err),
+			})
+			return
 		}
 
 		pr.MergedCommitID = updates[len(updates)-1].NewCommitID