whee sqlite

cmd/server/config.go

@@ -2,10 +2,8 @@ package main
 
 import (
 	"context"
-	"database/sql"
 	"flag"
 	"os"
-	"time"
 )
 
 type Config struct {
@@ -17,7 +15,7 @@ type Config struct {
 	}
 }
 
-func NewConfig() (Config, error) {
+func NewConfig(ctx context.Context) (Config, error) {
 	var c Config
 	fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
 
@@ -26,39 +24,13 @@ func NewConfig() (Config, error) {
 	fs.StringVar(&c.DB.Scheme, "db-scheme", "sqlite", "sql scheme")
 	fs.StringVar(&c.DB.Conn, "db-conn", "/tmp/out.sql", "sql conn string")
 
-	err := fs.Parse(os.Args[1:])
-	return c, err
-}
-
-type DB struct {
-	scheme string
-	conn   string
-}
-
-func NewDB(ctx context.Context, scheme, conn string) (DB, error) {
-	ctx, can := context.WithTimeout(ctx, time.Second*10)
-	defer can()
-
-	db := DB{
-		scheme: scheme,
-		conn:   conn,
+	if err := fs.Parse(os.Args[1:]); err != nil {
+		return Config{}, err
 	}
 
-	sql, err := db.dial(ctx)
-	if sql != nil {
-		sql.Close()
+	if _, err := NewDB(ctx, c.DB.Scheme, c.DB.Conn); err != nil {
+		return Config{}, err
 	}
 
-	return db, err
-}
-
-func (db DB) dial(ctx context.Context) (*sql.DB, error) {
-	c, err := sql.Open(db.scheme, db.conn)
-	if err != nil {
-		return nil, err
-	}
-	if err := c.PingContext(ctx); err != nil {
-		return nil, err
-	}
 	return c, nil
 }

cmd/server/db.go

@@ -0,0 +1,67 @@
+package main
+
+import (
+	"context"
+	"database/sql"
+	"io"
+	"time"
+
+	_ "github.com/glebarez/sqlite"
+)
+
+type DB struct {
+	scheme string
+	conn   string
+}
+
+func NewDB(ctx context.Context, scheme, conn string) (DB, error) {
+	ctx, can := context.WithTimeout(ctx, time.Second*10)
+	defer can()
+
+	db := DB{
+		scheme: scheme,
+		conn:   conn,
+	}
+
+	sql, err := db.dial(ctx)
+	if err != nil {
+		return DB{}, err
+	}
+	defer sql.Close()
+
+	if _, err := sql.ExecContext(ctx, `
+		CREATE TABLE IF NOT EXISTS users (
+			uuid TEXT,
+			name TEXT
+		);
+		CREATE TABLE IF NOT EXISTS games (
+			uuid TEXT
+		);
+		CREATE TABLE IF NOT EXISTS players (
+			user_uuid TEXT,
+			game_uuid TEXT
+		);
+		CREATE TABLE IF NOT EXISTS events (
+			game_uuid TEXT
+		);
+	`); err != nil {
+		return DB{}, err
+	}
+
+	return db, err
+}
+
+func (db DB) dial(ctx context.Context) (*sql.DB, error) {
+	c, err := sql.Open(db.scheme, db.conn)
+	if err != nil {
+		return nil, err
+	}
+	if err := c.PingContext(ctx); err != nil {
+		return nil, err
+	}
+	return c, nil
+}
+
+func (db DB) GetParty(id string) (string, error) {
+	return "", io.EOF
+}

cmd/server/main.go

@@ -22,7 +22,7 @@ func main() {
 }
 
 func run(ctx context.Context) error {
-	config, err := NewConfig()
+	config, err := NewConfig(ctx)
 	if err != nil {
 		return err
 	}