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

Fix bug

Browse Source
This commit is contained in:
Lunny Xiao 2024-04-11 20:37:53 +08:00
parent dd0ff150bf
commit e05ec3a7bb
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
2 changed files with 52 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models/packages/package.go
View File

@ -235,7 +235,6 @@ func TryInsertPackage(ctx context.Context, p *Package) (*Package, error) {
if _, err := db.GetEngine(ctx).Exec(sql); err != nil {
return nil, err
}
}
var existing Package

61
models/packages/package_file.go
View File

@ -5,11 +5,13 @@ package packages
import (
"context"
"fmt"
"strconv"
"strings"
"time"
"code.gitea.io/gitea/models/db"
"code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/timeutil"
"code.gitea.io/gitea/modules/util"
@ -44,11 +46,55 @@ type PackageFile struct {
// TryInsertFile inserts a file. If the file exists already ErrDuplicatePackageFile is returned
func TryInsertFile(ctx context.Context, pf *PackageFile) (*PackageFile, error) {
e := db.GetEngine(ctx)
switch {
case setting.Database.Type.IsMySQL():
if _, err := db.GetEngine(ctx).Exec("INSERT INTO package (version_id,blob_id,name,lower_name,composite_key,is_lead) VALUES (?,?,?,?,?,?) ON DUPLICATE KEY UPDATE 1=1",
pf.VersionID, pf.BlobID, pf.Name, pf.LowerName, pf.CompositeKey, pf.IsLead); err != nil {
return nil, err
}
case setting.Database.Type.IsPostgreSQL(), setting.Database.Type.IsSQLite3():
if _, err := db.GetEngine(ctx).Exec("INSERT INTO package (version_id,blob_id,name,lower_name,composite_key,is_lead) VALUES (?,?,?,?,?,?) ON CONFLICT (version_id,lower_name,composite_key) DO UPDATE SET lower_name=lower_name",
pf.VersionID, pf.BlobID, pf.Name, pf.LowerName, pf.CompositeKey, pf.IsLead); err != nil {
return nil, err
}
case setting.Database.Type.IsMSSQL():
r := func(s string) string {
return strings.ReplaceAll(s, "'", "''")
}
sql := fmt.Sprintf(`
MERGE INTO package WITH (HOLDLOCK) AS target USING (
SELECT
%d AS version_id,
%d AS blob_id,
'%s' AS name,
'%s' AS lower_name,
'%s' AS composite_key,
%s AS is_lead
) AS source (
version_id, blob_id, name, lower_name, composite_key, is_lead
) ON (
target.version_id = source.version_id
AND target.lower_name = source.lower_name
AND target.composite_key = source.composite_key
) WHEN MATCHED
THEN UPDATE SET 1 = 1
WHEN NOT MATCHED
THEN INSERT (
version_id, blob_id, name, lower_name, composite_key, is_lead
) VALUES (
%d, %d, '%s', '%s', '%s', %s
)`,
pf.VersionID, pf.BlobID, r(pf.Name), r(pf.LowerName), r(pf.CompositeKey), strconv.FormatBool(pf.IsLead),
pf.VersionID, pf.BlobID, r(pf.Name), r(pf.LowerName), r(pf.CompositeKey), strconv.FormatBool(pf.IsLead),
)
if _, err := db.GetEngine(ctx).Exec(sql); err != nil {
return nil, err
}
}
existing := &PackageFile{}
has, err := e.Where(builder.Eq{
has, err := db.GetEngine(ctx).Where(builder.Eq{
"version_id": pf.VersionID,
"lower_name": pf.LowerName,
"composite_key": pf.CompositeKey,
@ -56,13 +102,10 @@ func TryInsertFile(ctx context.Context, pf *PackageFile) (*PackageFile, error) {
if err != nil {
return nil, err
}
if has {
return existing, ErrDuplicatePackageFile
if !has {
return nil, util.ErrNotExist
}
if _, err = e.Insert(pf); err != nil {
return nil, err
}
return pf, nil
return existing, nil
}
// GetFilesByVersionID gets all files of a version