add locker

Author: ExplodingDragon

routers/api/packages/arch/arch.go

@@ -14,6 +14,7 @@ import (
 	"strings"
 
 	packages_model "code.gitea.io/gitea/models/packages"
+	"code.gitea.io/gitea/modules/globallock"
 	packages_module "code.gitea.io/gitea/modules/packages"
 	arch_module "code.gitea.io/gitea/modules/packages/arch"
 	"code.gitea.io/gitea/modules/util"
@@ -47,9 +48,18 @@ func GetRepositoryKey(ctx *context.Context) {
 	})
 }
 
+func refreshLocker(ctx *context.Context, group string) (globallock.ReleaseFunc, error) {
+	return globallock.Lock(ctx, fmt.Sprintf("pkg_arch_pkg_%s", group))
+}
+
 func PushPackage(ctx *context.Context) {
 	group := ctx.PathParam("group")
-
+	releaser, err := refreshLocker(ctx, group)
+	if err != nil {
+		apiError(ctx, http.StatusInternalServerError, err)
+		return
+	}
+	defer releaser()
 	upload, needToClose, err := ctx.UploadStream()
 	if err != nil {
 		apiError(ctx, http.StatusInternalServerError, err)
@@ -211,6 +221,12 @@ func RemovePackage(ctx *context.Context) {
 		ver     = ctx.PathParam("version")
 		pkgArch = ctx.PathParam("arch")
 	)
+	releaser, err := refreshLocker(ctx, group)
+	if err != nil {
+		apiError(ctx, http.StatusInternalServerError, err)
+		return
+	}
+	defer releaser()
 	pv, err := packages_model.GetVersionByNameAndVersion(
 		ctx, ctx.Package.Owner.ID, packages_model.TypeArch, pkg, ver,
 	)

services/packages/arch/repository.go

@@ -18,6 +18,7 @@ import (
 
 	packages_model "code.gitea.io/gitea/models/packages"
 	user_model "code.gitea.io/gitea/models/user"
+	"code.gitea.io/gitea/modules/globallock"
 	"code.gitea.io/gitea/modules/httplib"
 	packages_module "code.gitea.io/gitea/modules/packages"
 	arch_module "code.gitea.io/gitea/modules/packages/arch"
@@ -103,6 +104,11 @@ func NewFileSign(ctx context.Context, ownerID int64, input io.Reader) (*packages
 
 // BuildPacmanDB Create db signature cache
 func BuildPacmanDB(ctx context.Context, ownerID int64, group, arch string) error {
+	lock, err := globallock.Lock(ctx, fmt.Sprintf("pkg_arch_db_%s", group))
+	if err != nil {
+		return err
+	}
+	defer lock()
 	pv, err := GetOrCreateRepositoryVersion(ctx, ownerID)
 	if err != nil {
 		return err