initial commit

models/book.go

@@ -0,0 +1,9 @@
+package models
+
+type Book struct {
+  ID uint `json:"id" gorm:"primary_key"`
+  Title string `json:"title"`
+  Author string `json:"author"`
+}
+
+

models/setup.go

@@ -0,0 +1,20 @@
+package models
+
+import (
+  "gorm.io/gorm"
+  _ "gorm.io/driver/sqlite"
+)
+
+var DB *gorm.DB
+
+func ConnectDatabase() {
+        database, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
+        if err != nil {
+                panic("Failed to connect to database!")
+        }
+        err = database.AutoMigrate(&Book{})
+        if err != nil {
+                return
+        }
+        DB = database
+}