mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v1_26
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
|
|
"github.com/hanzoai/xorm/schemas"
|
|
)
|
|
|
|
type UserBadge struct { //revive:disable-line:exported
|
|
ID int64 `xorm:"pk autoincr"`
|
|
BadgeID int64
|
|
UserID int64
|
|
}
|
|
|
|
// TableIndices implements xorm's TableIndices interface
|
|
func (n *UserBadge) TableIndices() []*schemas.Index {
|
|
indices := make([]*schemas.Index, 0, 1)
|
|
ubUnique := schemas.NewIndex("unique_user_badge", schemas.UniqueType)
|
|
ubUnique.AddColumn("user_id", "badge_id")
|
|
indices = append(indices, ubUnique)
|
|
return indices
|
|
}
|
|
|
|
// AddUniqueIndexForUserBadge adds a compound unique indexes for user badge table
|
|
// and it replaces an old index on user_id
|
|
func AddUniqueIndexForUserBadge(x db.EngineMigration) error {
|
|
// remove possible duplicated records in table user_badge
|
|
type result struct {
|
|
UserID int64
|
|
BadgeID int64
|
|
Cnt int
|
|
}
|
|
var results []result
|
|
if err := x.Select("user_id, badge_id, count(*) as cnt").
|
|
Table("user_badge").
|
|
GroupBy("user_id, badge_id").
|
|
Having("count(*) > 1").
|
|
Find(&results); err != nil {
|
|
return err
|
|
}
|
|
for _, r := range results {
|
|
if x.Dialect().URI().DBType == schemas.MSSQL {
|
|
if _, err := x.Exec(fmt.Sprintf("delete from user_badge where id in (SELECT top %d id FROM user_badge WHERE user_id = ? and badge_id = ?)", r.Cnt-1), r.UserID, r.BadgeID); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
var ids []int64
|
|
if err := x.SQL("SELECT id FROM user_badge WHERE user_id = ? and badge_id = ? limit ?", r.UserID, r.BadgeID, r.Cnt-1).Find(&ids); err != nil {
|
|
return err
|
|
}
|
|
if _, err := x.Table("user_badge").In("id", ids).Delete(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return x.Sync(new(UserBadge))
|
|
}
|