-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
113 lines (99 loc) · 2.58 KB
/
db.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package main
import (
"database/sql"
"encoding/json"
"log"
"strings"
_ "github.com/mattn/go-sqlite3"
)
type gamesList struct {
Games []string `json:"games"`
}
//each spot on the board will be either empty, or filled with a player string to indicate they've placed a token there.
//The format will be columns, then rows. For instance index 0,0 would be the bottom leftmost spot
type game struct {
GameID string `json:"gameID"`
Players []string `json:"players"`
Columns int `json:"columns"`
Rows int `json:"rows"`
BoardState [][]string `json:"boardState"`
GameState string `json:"gameState"`
Winner string `json:"winner"`
}
type gameID struct {
GameID int64 `json:"gameId"`
}
func openGamesDB() (*sql.DB, error) {
gamesDB, err := sql.Open("sqlite3", "./local.db")
if err != nil {
return nil, err
}
return gamesDB, err
}
func getGames(gamesDB *sql.DB) (gamesList, error) {
var games gamesList
const query = `SELECT game_id FROM game_table WHERE game_state = 'IN_PROGRESS'`
rows, err := gamesDB.Query(query)
if err != nil {
return games, err
}
defer rows.Close()
var game string
for rows.Next() {
err := rows.Scan(&game)
if err != nil {
return games, err
}
log.Println(game)
games.Games = append(games.Games, game)
}
err = rows.Err()
if err != nil {
return games, err
}
return games, err
}
func createGame(players []string, columns, rows int, gamesDB *sql.DB) (gameID, error) {
var gameID gameID
var boardState [4][4]string
playersString := strings.Join(players, ",")
boardJSON, err := json.Marshal(boardState)
const insertStmt = `INSERT INTO game_table (
players,
columns,
rows,
board_state,
game_state
) VALUES (?,?,?,?,'IN_PROGRESS')`
_, err = gamesDB.Exec(insertStmt, playersString, columns, rows, boardJSON)
if err != nil {
return gameID, err
}
const getIDStmt = `select MAX(game_id) FROM game_table`
var id int64
err = gamesDB.QueryRow(getIDStmt).Scan(&id)
if err != nil {
return gameID, err
}
gameID.GameID = id
return gameID, err
}
func getGamesState(gameID int, gamesDB *sql.DB) (map[string]string, error) {
const query = `SELECT
players,
game_state,
winner
FROM
game_table
WHERE game_id = ?`
var players, gameState string
var winner sql.NullString
err := gamesDB.QueryRow(query, gameID).Scan(&players, &gameState, &winner)
if err != nil {
return nil, err
}
if winner.Valid {
return map[string]string{"players": players, "state": gameState, "winner": winner.String}, err
}
return map[string]string{"players": players, "state": gameState}, err
}