refactor: config

acoshift · Apr 12, 2019 · 5a44624 · 5a44624
1 parent 7e2eae3
commit 5a44624
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/internal/pkg/config/config.go b/internal/pkg/config/config.go
@@ -113,3 +113,10 @@ func RedisClient() *redis.Client {
 func DBClient() *sql.DB {
 	return dbClient
 }
+
+func Close() {
+	dbClient.Close()
+	redisClient.Close()
+	storageClient.Close()
+	errorClient.Close()
+}
diff --git a/main.go b/main.go
@@ -12,8 +12,7 @@ import (
 )
 
 func main() {
-	defer config.DBClient().Close()
-	defer config.RedisClient().Close()
+	defer config.Close()
 
 	app := app.New()