/
game_team.go
53 lines (45 loc) · 1.54 KB
/
game_team.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package repository
import (
"github.com/elabosak233/cloudsdale/internal/model"
"gorm.io/gorm"
)
type IGameTeamRepository interface {
Create(gameTeam model.GameTeam) (err error)
Update(gameTeam model.GameTeam) (err error)
Delete(gameTeam model.GameTeam) (err error)
Find(gameTeam model.GameTeam) (gameTeams []model.GameTeam, err error)
}
type GameTeamRepository struct {
db *gorm.DB
}
func NewGameTeamRepository(db *gorm.DB) IGameTeamRepository {
return &GameTeamRepository{db: db}
}
func (g *GameTeamRepository) Create(gameTeam model.GameTeam) (err error) {
result := g.db.Table("game_teams").Create(&gameTeam)
return result.Error
}
func (g *GameTeamRepository) Delete(gameTeam model.GameTeam) (err error) {
result := g.db.Table("game_teams").Delete(&gameTeam)
return result.Error
}
func (g *GameTeamRepository) Update(gameTeam model.GameTeam) (err error) {
result := g.db.Table("game_teams").Where(
"game_id = ? AND team_id = ?", gameTeam.GameID, gameTeam.TeamID,
).Updates(&gameTeam)
return result.Error
}
func (g *GameTeamRepository) Find(gameTeam model.GameTeam) (gameTeams []model.GameTeam, err error) {
result := g.db.Table("game_teams").
Where(&gameTeam).
Preload("Team", func(db *gorm.DB) *gorm.DB {
return db.Preload("Captain", func(db *gorm.DB) *gorm.DB {
return db.Select([]string{"id", "nickname", "username", "email"})
}).Preload("Users", func(db *gorm.DB) *gorm.DB {
return db.Select([]string{"id", "nickname", "username", "email"})
})
}).
Preload("Game").
Find(&gameTeams)
return gameTeams, result.Error
}