From bb11f2d9a35a9f97b692440ea0fc85b52d1735b8 Mon Sep 17 00:00:00 2001
From: "Namhyeon, Go" <gnh1201@gmail.com>
Date: Sun, 29 Apr 2018 00:58:22 +0900
Subject: [PATCH] Update database.php

---
 system/database.php | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/system/database.php b/system/database.php
index bbd3af2..79993a7 100644
--- a/system/database.php
+++ b/system/database.php
@@ -10,17 +10,21 @@ if(!function_exists("get_db_connect")) {
 	function get_db_connect() {
 		$config = get_config();
 
-		$conn = new PDO(
-			sprintf(
-				"mysql:host=%s;dbname=%s;charset=utf8",
-				$config['db_host'],
-				$config['db_name']
-			),
-			$config['db_username'],
-			$config['db_password'],
-			array(PDO::MYSQL_ATTR_INIT_COMMAND => "SET NAMES utf8")
-		);
-		$conn->query("SET NAMES 'utf8'");
+		try {
+			$conn = new PDO(
+				sprintf(
+					"mysql:host=%s;dbname=%s;charset=utf8",
+					$config['db_host'],
+					$config['db_name']
+				),
+				$config['db_username'],
+				$config['db_password'],
+				array(PDO::MYSQL_ATTR_INIT_COMMAND => "SET NAMES utf8")
+			);
+			$conn->query("SET NAMES 'utf8'");
+		} catch(Exception $e) {
+			set_error($e->getMessage());
+		}
 
 		return $conn;
 	}