feat: initial commit
This commit is contained in:
@ -7,16 +7,29 @@ import (
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func (s *Store) UpsertPlayer(ctx context.Context, player *Player) error {
|
||||
return errors.WithStack(s.Do(ctx, func(db *gorm.DB) error {
|
||||
func (s *Store) UpsertPlayer(ctx context.Context, name string, userEmail string, userProvider string) (*Player, error) {
|
||||
var player *Player
|
||||
err := s.Do(ctx, func(db *gorm.DB) error {
|
||||
var existing *Player
|
||||
err := db.Find(&existing, "user_id = ? and user_provider = ?", player.UserID, player.UserProvider).Error
|
||||
err := db.First(&existing, "user_email = ? and user_provider = ?", userEmail, userProvider).Error
|
||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
if existing != nil {
|
||||
player.Model = existing.Model
|
||||
if existing != nil && existing.ID != 0 {
|
||||
err := db.Model(&existing).UpdateColumn("name", name).Error
|
||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
player = existing
|
||||
return nil
|
||||
}
|
||||
|
||||
player = &Player{
|
||||
Name: name,
|
||||
UserEmail: userEmail,
|
||||
UserProvider: userProvider,
|
||||
}
|
||||
|
||||
if err := db.Save(player).Error; err != nil {
|
||||
@ -24,5 +37,26 @@ func (s *Store) UpsertPlayer(ctx context.Context, player *Player) error {
|
||||
}
|
||||
|
||||
return nil
|
||||
}))
|
||||
})
|
||||
if err != nil {
|
||||
return nil, errors.WithStack(err)
|
||||
}
|
||||
|
||||
return player, nil
|
||||
}
|
||||
|
||||
func (s *Store) GetPlayerRank(ctx context.Context, playerID uint) (int, error) {
|
||||
var rank int
|
||||
err := s.Tx(ctx, func(db *gorm.DB) error {
|
||||
err := db.Model(&Player{}).Select("rank() over (order by score desc) player_rank").First(&rank).Error
|
||||
if err != nil {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return 0, errors.WithStack(err)
|
||||
}
|
||||
return rank, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user