1 files changed, 21 insertions, 0 deletions
diff --git a/db/db.go b/db/db.go
new file mode 100644
index 0000000..128dcc8
--- /dev/null
+++ b/db/db.go
@@ -0,0 +1,21 @@
+package db
+
+import (
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+var (
+	DB  *gorm.DB
+	err error
+)
+
+func ConnectDb() {
+	DB, err = gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
+
+	if err != nil {
+		panic("failed to connect to the database.")
+	}
+
+	DB.AutoMigrate(&YdlEntry{})
+}