60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"strings"
|
|
|
|
"github.com/aldy505/bob"
|
|
)
|
|
|
|
// Set up the table connection, create table if not exists
|
|
func Setup() error {
|
|
db := New()
|
|
|
|
// Jokesbapak2 table & data
|
|
sql, _, err := bob.CreateTableIfNotExists("jokesbapak2").
|
|
Columns("id", "link", "key").
|
|
Types("SERIAL", "TEXT", "VARCHAR(255)").
|
|
Primary("id").ToSql()
|
|
if err != nil {
|
|
log.Fatalln("10 - failed on table creation: ", err)
|
|
}
|
|
|
|
splitSql := strings.Split(sql, ";")
|
|
for i := range splitSql {
|
|
_, err = db.Query(context.Background(), splitSql[i])
|
|
if err != nil {
|
|
log.Fatalln("11 - failed on table creation: ", err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Authorization
|
|
sql, _, err = bob.CreateTableIfNotExists("authorization").
|
|
Columns("id", "token", "key").
|
|
Types("SERIAL", "TEXT", "VARCHAR(255)").
|
|
Primary("id").
|
|
Unique("token").
|
|
ToSql()
|
|
if err != nil {
|
|
log.Fatalln("14 - failed on table creation: ", err)
|
|
return err
|
|
}
|
|
|
|
splitSql = strings.Split(sql, ";")
|
|
for i := range splitSql {
|
|
_, err = db.Query(context.Background(), splitSql[i])
|
|
if err != nil {
|
|
log.Fatalln("15 - failed on table creation: ", err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
_, err = db.Query(context.Background(), "ALTER TABLE jokesbapak2 ADD CONSTRAINT fk_jokesbapak2_key FOREIGN KEY (key) REFERENCES authorization (id)")
|
|
if err != nil {
|
|
log.Fatalln("16 - failed on foreign key iteration: ", err)
|
|
}
|
|
return nil
|
|
}
|