package repo

import (
	"github.com/zhilv666/navsite/pkg/config"
	"github.com/zhilv666/navsite/pkg/db"
	"github.com/zhilv666/navsite/pkg/logger"
	"gorm.io/gorm"
)

func DB() *gorm.DB {
	_db := db.GetDB()
	if _db == nil {
		db.InitDB(config.GetConfig().Database, logger.GetLogger())
		return db.GetDB()
	}

	return _db
}