mirror of
https://github.com/Damillora/Shioriko.git
synced 2024-11-21 20:07:33 +00:00
49 lines
864 B
Go
49 lines
864 B
Go
package database
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/Damillora/Shioriko/pkg/config"
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
|
|
"gorm.io/driver/postgres"
|
|
)
|
|
|
|
var DB *gorm.DB
|
|
|
|
func Initialize() {
|
|
dsn := config.CurrentConfig.PostgresDatabase
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
|
|
|
db.AutoMigrate(&User{})
|
|
db.AutoMigrate(&TagType{})
|
|
db.AutoMigrate(&Tag{})
|
|
db.AutoMigrate(&Blob{})
|
|
db.AutoMigrate(&Post{})
|
|
|
|
var general TagType
|
|
result := db.Where("name = ?", "general").First(&general)
|
|
if result.Error != nil {
|
|
db.Create(&TagType{
|
|
Name: "general",
|
|
})
|
|
}
|
|
|
|
var tagme Tag
|
|
result = db.Where("name = ?", "tagme").First(&tagme)
|
|
if result.Error != nil {
|
|
db.Create(&Tag{
|
|
ID: uuid.NewString(),
|
|
Name: "tagme",
|
|
TagType: general,
|
|
})
|
|
}
|
|
|
|
if err != nil {
|
|
log.Fatal("Unable to connect to database" + err.Error())
|
|
}
|
|
|
|
DB = db
|
|
}
|