blob: 18fa2e627498af644058113471b35a50ae0b3fab (
plain)
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
|
// Contains utility functions around the database
package database
import (
"archives/pkg/config"
"archives/pkg/models"
"context"
"github.com/go-pg/pg/v10"
"github.com/go-pg/pg/v10/orm"
)
// DBCon is the connection handle
// for the database
var (
DBCon *pg.DB
)
// CreateSchema creates the tables in the database
// in case they don't alreay exist
func CreateSchema() error {
if !tableExists("messages") {
err := DBCon.CreateTable((*models.Message)(nil), &orm.CreateTableOptions{
IfNotExists: true,
})
// Add tsvector column for subjects
DBCon.Exec("ALTER TABLE messages ADD COLUMN tsv_subject tsvector;")
DBCon.Exec("CREATE INDEX subject_idx ON messages USING gin(tsv_subject);")
// Add tsvector column for bodies
DBCon.Exec("ALTER TABLE messages ADD COLUMN tsv_body tsvector;")
DBCon.Exec("CREATE INDEX body_idx ON messages USING gin(tsv_body);")
return err
}
return nil
}
type dbLogger struct{}
func (d dbLogger) BeforeQuery(c context.Context, q *pg.QueryEvent) (context.Context, error) {
return c, nil
}
// AfterQuery is used to log SQL queries
func (d dbLogger) AfterQuery(c context.Context, q *pg.QueryEvent) error {
// logger.Debug.Println(q.FormattedQuery())
return nil
}
// Connect is used to connect to the database
// and turn on logging if desired
func Connect() {
DBCon = pg.Connect(&pg.Options{
User: config.PostgresUser(),
Password: config.PostgresPass(),
Database: config.PostgresDb(),
Addr: config.PostgresHost() + ":" + config.PostgresPort(),
})
DBCon.AddQueryHook(dbLogger{})
err := CreateSchema()
if err != nil {
// logger.Error.Println("ERROR: Could not create database schema")
// logger.Error.Println(err)
}
}
// utility methods
func tableExists(tableName string) bool {
_, err := DBCon.Exec("select * from " + tableName + ";")
return err == nil
}
|