39 lines
780 B
Go
39 lines
780 B
Go
package infrastructure
|
|
|
|
import (
|
|
"bill-go-fiber/internal/bill"
|
|
"bill-go-fiber/internal/label"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"os"
|
|
)
|
|
|
|
func ConnectToSqlite() (*gorm.DB, error) {
|
|
dsn := "bill.db"
|
|
//dsn := "file::memory:?cache=shared"
|
|
db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
panic("failed to connect database.\n")
|
|
}
|
|
|
|
err = db.AutoMigrate(&label.Label{})
|
|
if err != nil {
|
|
panic("failed to migrate database.\n")
|
|
}
|
|
err = db.AutoMigrate(&bill.Bill{})
|
|
if err != nil {
|
|
panic("failed to migrate database.\n")
|
|
}
|
|
|
|
// migrate data
|
|
go func() {
|
|
sql, err := os.ReadFile("scripts/sqliteMigrations.sql")
|
|
if err != nil {
|
|
panic("failed to read migration sql file: " + err.Error())
|
|
}
|
|
db.Exec(string(sql))
|
|
}()
|
|
|
|
return db, nil
|
|
}
|