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)) + } }