70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
|
package main
|
||
|
|
||
|
import ()
|
||
|
|
||
|
type Theme struct {
|
||
|
id int64
|
||
|
Name string
|
||
|
}
|
||
|
|
||
|
func GetThemes() ([]Theme, error) {
|
||
|
if rows, err := DBQuery("SELECT id_theme, name FROM themes"); err != nil {
|
||
|
return nil, err
|
||
|
} else {
|
||
|
defer rows.Close()
|
||
|
|
||
|
var themes = make([]Theme, 0)
|
||
|
for rows.Next() {
|
||
|
var t Theme
|
||
|
if err := rows.Scan(&t.id, &t.Name); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
themes = append(themes, t)
|
||
|
}
|
||
|
if err := rows.Err(); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return themes, nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func GetTheme(id int) (Theme, error) {
|
||
|
var t Theme
|
||
|
if err := DBQueryRow("SELECT id_theme, name FROM themes WHERE id_theme=?", id).Scan(&t.id, &t.Name); err != nil {
|
||
|
return t, err
|
||
|
}
|
||
|
|
||
|
return t, nil
|
||
|
}
|
||
|
|
||
|
func CreateTheme(name string) (Theme, error) {
|
||
|
if res, err := DBExec("INSERT INTO themes (name) VALUES (?)", name); err != nil {
|
||
|
return Theme{}, err
|
||
|
} else if tid, err := res.LastInsertId(); err != nil {
|
||
|
return Theme{}, err
|
||
|
} else {
|
||
|
return Theme{tid, name}, nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (t Theme) Update() (int64, error) {
|
||
|
if res, err := DBExec("UPDATE themes 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 Theme) Delete() (int64, error) {
|
||
|
if res, err := DBExec("DELETE FROM themes WHERE id_theme = ?", t.id); err != nil {
|
||
|
return 0, err
|
||
|
} else if nb, err := res.RowsAffected(); err != nil {
|
||
|
return 0, err
|
||
|
} else {
|
||
|
return nb, err
|
||
|
}
|
||
|
}
|