aniani/gitea
1
0
Fork 0
mirror of https://github.com/go-gitea/gitea.git synced 2024-12-04 14:46:57 -05:00
Code

Use insert on duplicate to try insert package

Browse Source
This commit is contained in:
Lunny Xiao 2024-04-08 23:13:04 +08:00
parent ba342cb58e
commit 142fe05115
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 46 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
models/packages/package.go
View File

@ -6,9 +6,11 @@ package packages
import ( import (
"context" "context"
"fmt" "fmt"
"strconv"
"strings" "strings"
"code.gitea.io/gitea/models/db" "code.gitea.io/gitea/models/db"
"code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/util" "code.gitea.io/gitea/modules/util"
"xorm.io/builder" "xorm.io/builder"
@ -189,23 +191,59 @@ type Package struct {
// TryInsertPackage inserts a package. If a package exists already, ErrDuplicatePackage is returned // TryInsertPackage inserts a package. If a package exists already, ErrDuplicatePackage is returned
func TryInsertPackage(ctx context.Context, p *Package) (*Package, error) { func TryInsertPackage(ctx context.Context, p *Package) (*Package, error) {
e := db.GetEngine(ctx) switch {
case setting.Database.Type.IsMySQL():
if _, err := db.GetEngine(ctx).Exec("INSERT INTO package (owner_id,`type`,lower_name,name,semver_compatible) VALUES (?,?,?,?,?) ON DUPLICATE KEY UPDATE 1=1",
p.OwnerID, p.Type, p.LowerName, p.Name, p.SemverCompatible); err != nil {
return nil, err
}
case setting.Database.Type.IsPostgreSQL(), setting.Database.Type.IsSQLite3():
if _, err := db.GetEngine(ctx).Exec("INSERT INTO package (owner_id,`type`,lower_name,name,semver_compatible) VALUES (?,?,?,?,?) ON CONFLICT (owner_id,`type`,lower_name) DO UPDATE SET lower_name=lower_name",
p.OwnerID, p.Type, p.LowerName, p.Name, p.SemverCompatible); err != nil {
return nil, err
}
case setting.Database.Type.IsMSSQL():
r := func(s string) string {
return strings.Replace(s, "'", "''", -1)
}
sql := fmt.Sprintf(`MERGE INTO package WITH (HOLDLOCK) AS target
USING (SELECT
%d AS owner_id
,'%s' AS [type]
,'%s' AS lower_name
,'%s' AS name
, %s AS semver_compatible) AS source
(owner_id, [type], lower_name, name, semver_compatible)
ON (target.owner_id = source.owner_id
AND target.[type] = source.[type]
AND target.lower_name = source.lower_name)
WHEN MATCHED
THEN UPDATE
SET 1 = 1
WHEN NOT MATCHED
THEN INSERT (owner_id, [type], lower_name, name, semver_compatible)
VALUES (%d, '%s', '%s', '%s', %s)`,
p.OwnerID, r(string(p.Type)), r(p.LowerName), r(p.Name), strconv.FormatBool(p.SemverCompatible),
p.OwnerID, r(string(p.Type)), r(p.LowerName), r(p.Name), strconv.FormatBool(p.SemverCompatible),
)
existing := &Package{} if _, err := db.GetEngine(ctx).Exec(sql); err != nil {
return nil, err
}
has, err := e.Where(builder.Eq{ }
var existing Package
has, err := db.GetEngine(ctx).Where(builder.Eq{
"owner_id": p.OwnerID, "owner_id": p.OwnerID,
"type": p.Type, "type": p.Type,
"lower_name": p.LowerName, "lower_name": p.LowerName,
}).Get(existing) }).Get(&existing)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if has { if has {
return existing, ErrDuplicatePackage return &existing, ErrDuplicatePackage
}
if _, err = e.Insert(p); err != nil {
return nil, err
} }
return p, nil return p, nil
} }