From 51092fbb91cb941df0ef248d0618d528505f2ca0 Mon Sep 17 00:00:00 2001
From: Sdogruyol <dogruyolserdar@gmail.com>
Date: Sun, 17 Jan 2016 14:17:27 +0200
Subject: [PATCH] Move logger to config

---
 src/kemal.cr        | 10 ++++------
 src/kemal/config.cr |  9 +++++++++
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/kemal.cr b/src/kemal.cr
index eaab7a7..c4b524c 100644
--- a/src/kemal.cr
+++ b/src/kemal.cr
@@ -6,12 +6,8 @@ at_exit do
   config = Kemal.config
   if config.logging
     logger = Kemal::Logger.new
-    config.add_handler logger
-    logger.write "[#{config.env}] Kemal is ready to lead at #{config.scheme}://#{config.host_binding}:#{config.port}\n"
-    Signal::INT.trap {
-      logger.write "Kemal is going to take a rest!\n"
-      logger.handler.close
-    }
+    config.logger = logger
+    config.logger.write "[#{config.env}] Kemal is ready to lead at #{config.scheme}://#{config.host_binding}:#{config.port}\n"
   end
   config.add_handler Kemal::StaticFileHandler.new(config.public_folder)
   config.add_handler Kemal::Handler::INSTANCE
@@ -20,6 +16,8 @@ at_exit do
   server.ssl = config.ssl
 
   Signal::INT.trap {
+    config.logger.write "Kemal is going to take a rest!\n"
+    config.logger.handler.close
     server.close
     exit
   }
diff --git a/src/kemal/config.cr b/src/kemal/config.cr
index ea93802..6a62ae1 100644
--- a/src/kemal/config.cr
+++ b/src/kemal/config.cr
@@ -22,6 +22,15 @@ module Kemal
       HANDLERS
     end
 
+    def logger
+      @logger
+    end
+
+    def logger=(logger)
+      HANDLERS << logger
+      @logger = logger
+    end
+
     def add_handler(handler : HTTP::Handler)
       HANDLERS << handler
     end