You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
990 B
Go
44 lines
990 B
Go
5 years ago
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Body struct {
|
||
|
|
ID int
|
||
|
|
PlayerID int
|
||
|
|
X float64
|
||
|
|
Y float64
|
||
|
|
Z float64
|
||
|
|
DiedAt time.Time
|
||
|
|
FoundAt *time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *SQLite) ListBodies() ([]Body, error) {
|
||
|
|
rows, err := db.db.Query(`select id, player, x, y, z, died_at from bodies where found_at is null`)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to fetch bodies from db: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
bodies := make([]Body, 0, 64)
|
||
|
|
for rows.Next() {
|
||
|
|
var body Body
|
||
|
|
if err := rows.Scan(&body.ID, &body.PlayerID, &body.X, &body.Y, &body.Z, &body.DiedAt); err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to read bodies response rows: %w", err)
|
||
|
|
}
|
||
|
|
bodies = append(bodies, body)
|
||
|
|
}
|
||
|
|
return bodies, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *SQLite) AddBody(body *Body) error {
|
||
|
|
_, err := db.db.Exec(`insert into bodies
|
||
|
|
(player, x, y, z)
|
||
|
|
values (?, ?, ?, ?);`, body.PlayerID, body.X, body.Y, body.Z)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to add body to db: %w", err)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|