38 lines
452 B
Go
38 lines
452 B
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"io/ioutil"
|
|
|
|
"go-sjles-pta-vote/config"
|
|
|
|
_ "github.com/glebarez/go-sqlite"
|
|
)
|
|
|
|
var db *sql.DB
|
|
|
|
func Connect() error {
|
|
var err error
|
|
db, err = sql.Open("sqlite", dbConfig.DBPath)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sql_create, err := ioutil.ReadFile("./db_format.sql")
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = db.Exec(sql_create)
|
|
|
|
return err
|
|
}
|
|
|
|
func Close() {
|
|
if db != nil {
|
|
_ = db.Close()
|
|
}
|
|
} |