From 78ee15cf4b5d3a55209a9c7559700a2c5b227f87 Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Thu, 3 Aug 2023 08:48:49 -0400
Subject: [PATCH] fix: close db on shutdown

---
 db/db.go | 6 ++++++
 main.go  | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/db/db.go b/db/db.go
index 715a50f..b773229 100644
--- a/db/db.go
+++ b/db/db.go
@@ -63,3 +63,9 @@ func Init() {
 func Get() *gorm.DB {
 	return db
 }
+func Close() error {
+
+	instance, _ := db.DB()
+
+	return instance.Close()
+}
diff --git a/main.go b/main.go
index 6ca5dc7..dc9e465 100644
--- a/main.go
+++ b/main.go
@@ -103,4 +103,10 @@ func main() {
 	if err != nil {
 		logger.Get().Error("Failed starting webserver proof", zap.Error(err))
 	}
+
+	err = db.Close()
+
+	if err != nil {
+		logger.Get().Error("Failed to close db connection", zap.Error(err))
+	}
 }