Add DB objects
This commit is contained in:
parent
abd5e2025e
commit
6ec37b83ce
7 changed files with 450 additions and 2 deletions
69
admin/team.go
Normal file
69
admin/team.go
Normal file
|
@ -0,0 +1,69 @@
|
|||
package main
|
||||
|
||||
import ()
|
||||
|
||||
type Team struct {
|
||||
id int64
|
||||
Name string
|
||||
}
|
||||
|
||||
func GetTeams() ([]Team, error) {
|
||||
if rows, err := DBQuery("SELECT id_team, name FROM teams"); err != nil {
|
||||
return nil, err
|
||||
} else {
|
||||
defer rows.Close()
|
||||
|
||||
var teams = make([]Team, 0)
|
||||
for rows.Next() {
|
||||
var t Team
|
||||
if err := rows.Scan(&t.id, &t.Name); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
teams = append(teams, t)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return teams, nil
|
||||
}
|
||||
}
|
||||
|
||||
func GetTeam(id int) (Team, error) {
|
||||
var t Team
|
||||
if err := DBQueryRow("SELECT id_team, name FROM teams WHERE id_team = ?", id).Scan(&t.id, &t.Name); err != nil {
|
||||
return t, err
|
||||
}
|
||||
|
||||
return t, nil
|
||||
}
|
||||
|
||||
func CreateTeam(name string) (Team, error) {
|
||||
if res, err := DBExec("INSERT INTO teams (name) VALUES (?)", name); err != nil {
|
||||
return Team{}, err
|
||||
} else if tid, err := res.LastInsertId(); err != nil {
|
||||
return Team{}, err
|
||||
} else {
|
||||
return Team{tid, name}, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (t Team) Update() (int64, error) {
|
||||
if res, err := DBExec("UPDATE teams SET name = ? WHERE id_team = ?", t.Name, t.id); err != nil {
|
||||
return 0, err
|
||||
} else if nb, err := res.RowsAffected(); err != nil {
|
||||
return 0, err
|
||||
} else {
|
||||
return nb, err
|
||||
}
|
||||
}
|
||||
|
||||
func (t Team) Delete() (int64, error) {
|
||||
if res, err := DBExec("DELETE FROM teams WHERE id_teams = ?", t.id); err != nil {
|
||||
return 0, err
|
||||
} else if nb, err := res.RowsAffected(); err != nil {
|
||||
return 0, err
|
||||
} else {
|
||||
return nb, err
|
||||
}
|
||||
}
|
Reference in a new issue