registering works

models/setup.go

@@ -78,6 +78,10 @@ func ConnectDatabase() {
 	//defer db.Close()
 }
 
+func DisconnectDatabase() {
+	defer db.Close()
+}
+
 func CreateTables() {
 	var err error
 	// Create database tables if it doesn't exist