2 files changed, 43 insertions, 0 deletions
diff --git a/api/database/database.go b/api/database/database.go
new file mode 100644
index 0000000..e5ecca8
--- /dev/null
+++ b/api/database/database.go
@@ -0,0 +1,32 @@
+package database
+
+import (
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+
+	"errors"
+)
+
+// Global variable but private
+var db *gorm.DB = nil
+
+// Init the database from a DSN string which must be a valid PostgreSQL dsn.
+// Also, auto migrate all the models.
+func InitDb(dsn string) (*gorm.DB, error) {
+	var err error
+	db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
+
+	if err == nil {
+		db.AutoMigrate(&User{})
+	}
+
+	return db, err
+}
+
+// Return the instance or error if the config is not laoded yet
+func GetDb() (*gorm.DB, error) {
+	if db == nil {
+		return nil, errors.New("You must call `InitDb()` first.")
+	}
+	return db, nil
+}
diff --git a/api/database/models.go b/api/database/models.go
new file mode 100644
index 0000000..e309a36
--- /dev/null
+++ b/api/database/models.go
@@ -0,0 +1,11 @@
+package database
+
+import "time"
+
+type User struct {
+	ID        int       `json:"id"`
+	Username  string    `json:"username"`
+	Password  string    `json:"password"`
+	CreatedAt time.Time `json:"created_at"`
+	UpdatedAt time.Time `json:"updated_at"`
+}