close (opened) database connection

muharihar · Aug 19, 2020 · b1cf390 · b1cf390
1 parent b72e446
commit b1cf390
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/interface/restapi/server.go b/interface/restapi/server.go
@@ -68,6 +68,7 @@ func StartRestAPIServer() error {
 		// fmt.Errorf("StartRestAPIServer.initializeSystems: %s", err.Error())
 		return err
 	}
+	defer initialize.CloseDBConnections(superHandler)
 
 	// init echo server
 	e := echo.New()

diff --git a/system/handler/handler.go b/system/handler/handler.go
@@ -57,6 +57,11 @@ func (h *Handler) GetGormDB(conName string) (*gorm.DB, error) {
 	return db, nil
 }
 
+// GetGormDBs get Gorm Databases
+func (h *Handler) GetGormDBs() map[string]*gorm.DB {
+	return h.dbGorms
+}
+
 // SetCasbinEnforcer set CasbinEnforcer
 func (h *Handler) SetCasbinEnforcer(ceName string, ce *casbin.Enforcer) {
 	if h.casbinEnforcers == nil {

diff --git a/system/initialize/database.go b/system/initialize/database.go
@@ -2,6 +2,7 @@ package initialize
 
 import (
 	"errors"
+	"fmt"
 	"reflect"
 
 	// "github.com/jinzhu/gorm"
@@ -101,3 +102,15 @@ func openDBConnection(driverName, dataSourceName string) (*gorm.DB, error) {
 	}
 	return db, err
 }
+
+// CloseDBConnections close DB Connections
+func CloseDBConnections(h *handler.Handler) {
+	gorms := h.GetGormDBs()
+	for key, db := range gorms {
+		dbCon, _ := db.DB()
+		fmt.Printf("Closing DB Connection `%s`\n", key)
+		if err := dbCon.Close(); err != nil {
+			fmt.Printf("Error while closing DB Connection `%s`: %s\n", key, err.Error())
+		}
+	}
+}