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
|
// Eko: A terminal based social media platform
// Copyright (C) 2025 Kyren223
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package api
import (
"database/sql"
"embed"
"log/slog"
_ "github.com/mattn/go-sqlite3"
"github.com/pressly/goose/v3"
"github.com/kyren223/eko/pkg/assert"
)
//go:embed migrations/*.sql
var embedMigrations embed.FS
var db *sql.DB
func ConnectToDatabase() {
var err error
db, err = sql.Open("sqlite3", "file:server.db?cache=shared")
if err != nil {
slog.Error("unable to open database", "error", err)
assert.Abort("see logs")
}
assert.AddFlush(db)
slog.Info("established connection with the database")
pragmas := []string{
"PRAGMA journal_mode = WAL;",
"PRAGMA synchronous = NORMAL;",
"PRAGMA temp_store = MEMORY;",
"PRAGMA mmap_size = 30000000000;",
}
for _, pragma := range pragmas {
_, err := db.Exec(pragma)
assert.NoError(err, "DB pragmas should always execute with no errors")
}
slog.Info("opened database, running migrations...")
goose.SetBaseFS(embedMigrations)
if err := goose.SetDialect("sqlite3"); err != nil {
assert.NoError(err, "sqlite3 is a valid dialect that shouldn't error")
}
if err := goose.Up(db, "migrations"); err != nil {
_ = db.Close()
slog.Error("error running migrations", "error", err)
assert.Abort("see logs")
}
slog.Info("database connection ready to be used")
}
func DB() *sql.DB {
return db
}
|
