archives/trantor
Archived
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update to use pg/v10

Browse source
This commit is contained in:
Las Zenow 2020-11-30 19:03:31 +00:00
parent 94c75b1d2a
commit 1536d97d8d
13 changed files with 449 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lib/database/database.go
View file

@ -3,8 +3,8 @@ package database
import (
"time"
"github.com/go-pg/pg/v9"
"github.com/go-pg/pg/v9/orm"
"github.com/go-pg/pg/v10"
"github.com/go-pg/pg/v10/orm"
)
type DB interface {
@ -108,6 +108,7 @@ func (db pgDB) Close() error {
}
func (db pgDB) create() error {
orm.RegisterTable(&BookListEntry{})
models := []interface{}{&Book{}, &New{}, &User{}, &Visit{}, &Submission{}, &BookList{}, &BookListEntry{}}
for _, model := range models {
options := &orm.CreateTableOptions{
@ -119,7 +120,7 @@ func (db pgDB) create() error {
// XXX: ON DELETE SET NULL contraint is not supported
options.FKConstraints = false
}
err := db.sql.CreateTable(model, options)
err := db.sql.Model(model).CreateTable(options)
if err != nil {
return err
}