Create data dir and move DB file into it for better organization
This commit is contained in:
@@ -2,6 +2,7 @@ package database
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"CatsOfMastodonBotGo/internal/domain"
|
"CatsOfMastodonBotGo/internal/domain"
|
||||||
|
"os"
|
||||||
|
|
||||||
"gorm.io/driver/sqlite"
|
"gorm.io/driver/sqlite"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
@@ -10,7 +11,14 @@ import (
|
|||||||
var Gorm *gorm.DB
|
var Gorm *gorm.DB
|
||||||
|
|
||||||
func Connect() (*gorm.DB, error) {
|
func Connect() (*gorm.DB, error) {
|
||||||
db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
|
_, err := os.ReadDir("data")
|
||||||
|
if err != nil {
|
||||||
|
err := os.Mkdir("data", 0755)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
db, err := gorm.Open(sqlite.Open("data/caom.db"), &gorm.Config{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -22,10 +30,10 @@ func Connect() (*gorm.DB, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// IDK if this is how it works or not, leave it as is for now
|
// IDK if this is how it works or not, leave it as is for now
|
||||||
func Init () {
|
func Init() {
|
||||||
var err error
|
var err error
|
||||||
Gorm, err = Connect()
|
Gorm, err = Connect()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user