57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package storage
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
func CreateTables() error {
|
|
// Requests DB
|
|
db_r, err := sql.Open("sqlite",
|
|
"/var/lib/banforge/requests.db?"+
|
|
"mode=rwc&"+
|
|
"_pragma=journal_mode(WAL)&"+
|
|
"_pragma=busy_timeout(30000)&"+
|
|
"_pragma=synchronous(NORMAL)")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open requests db: %w", err)
|
|
}
|
|
defer func() {
|
|
err = db_r.Close()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}()
|
|
|
|
_, err = db_r.Exec(CreateRequestsTable)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create requests table: %w", err)
|
|
}
|
|
|
|
// Bans DB
|
|
db_b, err := sql.Open("sqlite",
|
|
"/var/lib/banforge/bans.db?"+
|
|
"mode=rwc&"+
|
|
"_pragma=journal_mode(WAL)&"+
|
|
"_pragma=busy_timeout(30000)&"+
|
|
"_pragma=synchronous(FULL)")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open bans db: %w", err)
|
|
}
|
|
defer func() {
|
|
err = db_b.Close()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}()
|
|
|
|
_, err = db_b.Exec(CreateBansTable)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create bans table: %w", err)
|
|
}
|
|
fmt.Println("Tables created successfully!")
|
|
return nil
|
|
}
|