c0aa51902c
Co-authored-by: haraldnilsen <harald_998@hotmail.com> Signed-off-by: Sindre Kjelsrud <kjelsrudsindre@gmail.com>
107 lines
No EOL
2.4 KiB
Go
107 lines
No EOL
2.4 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
const (
|
|
host = "postgres"
|
|
port = 5432 // This is the default port for PostgreSQL
|
|
user = "admin"
|
|
password = "helse123"
|
|
dbname = "helseveileder"
|
|
)
|
|
|
|
func SetupDb() {
|
|
// Connection string
|
|
psqlInfo := fmt.Sprintf("host=%s port=%d user=%s "+
|
|
"password=%s dbname=%s sslmode=disable",
|
|
host, port, user, password, dbname)
|
|
|
|
// Connect to the database
|
|
db, err := sql.Open("postgres", psqlInfo)
|
|
if err != nil {
|
|
log.Fatalf("Error opening database: %v\n", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// Check the connection
|
|
err = db.Ping()
|
|
if err != nil {
|
|
log.Fatalf("Error connecting to the database: %v\n", err)
|
|
}
|
|
|
|
// SQL statements to create tables
|
|
createTableStatements := []string{
|
|
`CREATE TABLE IF NOT EXISTS Spørsmål (
|
|
spørsmålID SERIAL PRIMARY KEY,
|
|
tekst TEXT NOT NULL
|
|
);`,
|
|
`CREATE TABLE IF NOT EXISTS SpørsmålSvar (
|
|
svarID SERIAL PRIMARY KEY,
|
|
spørsmålID INT REFERENCES Spørsmål(spørsmålID),
|
|
chatgpt BOOL NOT NULL,
|
|
svartekst TEXT NOT NULL
|
|
);`,
|
|
`CREATE TABLE IF NOT EXISTS Respondent (
|
|
respondentID SERIAL PRIMARY KEY,
|
|
alder TEXT NOT NULL,
|
|
utdanningsgrad TEXT NOT NULL,
|
|
helsepersonell BOOL NOT NULL,
|
|
kjønn TEXT NOT NULL,
|
|
svartfør BOOL NOT NULL,
|
|
fylke TEXT NOT NULL,
|
|
dato TEXT NOT NULL
|
|
);`,
|
|
`CREATE TABLE IF NOT EXISTS SvarVurdering (
|
|
vurderingID SERIAL PRIMARY KEY,
|
|
respondentID INT REFERENCES Respondent(respondentID),
|
|
svarID INT REFERENCES SpørsmålSvar(svarID),
|
|
kunnskap INT,
|
|
empati INT,
|
|
hjelpsomhet INT
|
|
);`,
|
|
`CREATE TABLE IF NOT EXISTS Evaluering (
|
|
evalueringtekst TEXT
|
|
);`,
|
|
`CREATE TABLE IF NOT EXISTS FeilRapport (
|
|
feiltekst TEXT
|
|
);`,
|
|
}
|
|
|
|
// Execute SQL statements
|
|
for _, stmt := range createTableStatements {
|
|
_, err := db.Exec(stmt)
|
|
if err != nil {
|
|
log.Fatalf("Error creating table: %v\n", err)
|
|
}
|
|
}
|
|
|
|
insertDb(db, "cmd/db/queries/insert_questions_sql_query.sql")
|
|
insertDb(db, "cmd/db/queries/insert_answers_sql_query.sql")
|
|
|
|
fmt.Println("Tables created successfully.")
|
|
}
|
|
|
|
func insertDb(db *sql.DB, filepath string) {
|
|
file, err := os.ReadFile(filepath)
|
|
if err != nil {
|
|
// handle error
|
|
log.Fatalf("Error reading file: %v\n", err)
|
|
}
|
|
|
|
requests := strings.Split(string(file), ";")
|
|
for _, request := range requests {
|
|
_, err := db.Exec(request)
|
|
// do whatever you need with result and error
|
|
if err != nil {
|
|
log.Fatalf("Error creating table: %v\n", err)
|
|
}
|
|
}
|
|
} |