23 lines
400 B
Go
23 lines
400 B
Go
package database
|
|
|
|
import (
|
|
"log"
|
|
"maot-shortner/internal/domain"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Database struct {
|
|
*gorm.DB
|
|
}
|
|
|
|
func InitDatabase() (database *Database, err error) {
|
|
log.Println("Starting database...")
|
|
|
|
db, err := gorm.Open(sqlite.Open("shortener.db"), &gorm.Config{})
|
|
db.AutoMigrate(domain.ImportModels()...)
|
|
|
|
database = &Database{db}
|
|
return database, err
|
|
}
|