116 lines
2.2 KiB
Go
116 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"time"
|
|
|
|
"git.ohea.xyz/cursorius/server/config"
|
|
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/op/go-logging"
|
|
)
|
|
|
|
var log = logging.MustGetLogger("cursorius-server")
|
|
|
|
func LaunchDB(conf config.DBConfig) error {
|
|
|
|
dbURL := fmt.Sprintf(
|
|
"postgres://%v:%v@%v:%v/%v",
|
|
conf.Username,
|
|
conf.Password,
|
|
conf.Address,
|
|
conf.Port,
|
|
conf.Name,
|
|
)
|
|
|
|
dbURLNoPasswd := fmt.Sprintf(
|
|
"postgres://%v:********@%v:%v/%v",
|
|
conf.Username,
|
|
conf.Address,
|
|
conf.Port,
|
|
conf.Name,
|
|
)
|
|
|
|
var conn *pgx.Conn
|
|
var err error
|
|
for i := 0; i < 10; i++ {
|
|
log.Infof("Connecting to database with URL \"%v\" (attempt %v)", dbURLNoPasswd, i)
|
|
conn, err = pgx.Connect(context.Background(), dbURL)
|
|
if err == nil {
|
|
defer conn.Close(context.Background())
|
|
break
|
|
}
|
|
time.Sleep(2 * time.Second)
|
|
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("Could not open database: %w", err)
|
|
}
|
|
|
|
log.Infof("Database connected sucessfully!")
|
|
|
|
versionTableExistsQuery := `
|
|
SELECT EXISTS (
|
|
SELECT FROM pg_tables
|
|
WHERE tablename = 'version'
|
|
);`
|
|
|
|
var versionTableExists bool
|
|
err = conn.QueryRow(context.Background(), versionTableExistsQuery).Scan(&versionTableExists)
|
|
if err != nil {
|
|
return fmt.Errorf("Could not check if database was initalized: %w", err)
|
|
}
|
|
|
|
if versionTableExists {
|
|
// TODO: migrations
|
|
} else {
|
|
log.Info("New database found, initializing....")
|
|
err = initDB(conn)
|
|
if err != nil {
|
|
return fmt.Errorf("Could not initalize database: %w", err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func initDB(conn *pgx.Conn) error {
|
|
createTablesQuery := `
|
|
CREATE TABLE version (
|
|
version INT NOT NULL
|
|
);
|
|
|
|
CREATE TABLE pipelines (
|
|
id SERIAL PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
url TEXT NOT NULL,
|
|
poll_interval INTEGER
|
|
);
|
|
|
|
CREATE TABLE webhooks (
|
|
id SERIAL PRIMARY KEY,
|
|
type TEXT,
|
|
secret TEXT,
|
|
pipeline INTEGER,
|
|
|
|
CONSTRAINT fk_pipeline
|
|
FOREIGN KEY(pipeline)
|
|
REFERENCES pipelines(id)
|
|
);
|
|
|
|
CREATE TABLE runners (
|
|
id SERIAL PRIMARY KEY,
|
|
name TEXT,
|
|
secret TEXT
|
|
);`
|
|
|
|
_, err := conn.Exec(context.Background(), createTablesQuery)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|