Files
stinkinator/noteService.go

46 lines
1.2 KiB
Go

package main
import (
"fmt"
"time"
)
type NoteService struct {
db *DB
}
type NoteServiceQuery struct {
Timestamp *string `db:"timestamp"`
PlayerID *int64 `db:"player"`
ID *int64 `db:"id"`
}
func (ns *NoteService) Query(query NoteServiceQuery) (Note, error) {
note := Note{}
sqlQuery := "SELECT id, content, timestamp, player FROM note"
whereQuery := BuildWhereQuery(query)
if whereQuery != "" {
sqlQuery += " " + whereQuery
}
err := ns.db.readConn.QueryRow(sqlQuery).Scan(&note.ID, &note.Content, &note.Timestamp, &note.Player.ID)
if err != nil {
return note, fmt.Errorf("failed getting note for query %q: %v", sqlQuery, err)
}
return note, nil
}
func (ns *NoteService) Create(content string, timestamp time.Time, player Player) (Note, error) {
note := Note{}
res, err := ns.db.writeConn.Exec("insert into note (content, timestamp, player) values (?, ?, ?)", content, timestamp, player.ID)
if err != nil {
return note, fmt.Errorf("failed to insert note: %v", err)
}
id, err := res.LastInsertId()
if err != nil {
return note, fmt.Errorf("failed to get last insert id: %v", err)
}
return ns.Query(NoteServiceQuery{ID: &id})
}