9 Sinatra logger and production env

Author: id774

app.rb

@@ -1,7 +1,6 @@
 # -*- coding: utf-8 -*-
 require 'rubygems'
 require 'sinatra/base'
-require 'sinatra/reloader'
 require 'fluent-logger'
 require 'haml'
 require 'json'
@@ -16,6 +15,11 @@ def initialize(app = nil, params = {})
     @fluentd = Fluent::Logger::FluentLogger.open(nil,
       host = 'localhost',
       port = '29999')
+    @root = Sinatra::Application.environment == :production ? '/post/' : '/'
+  end
+
+  def logger
+    env['app.logger'] || env['rack.logger']
   end
 
   helpers do

config.ru

@@ -1,6 +1,26 @@
 require 'rubygems'
 require 'sinatra'
+require 'logger'
+require './lib/applog'
 require './app'
 
-run FluentdJsonReceiver
+logdir = File.dirname(__FILE__) + "/log"
+logger = ::Logger.new(logdir + '/app.log')
+
+def logger.write(msg)
+  self << msg
+end
+
+configure :development do
+  require 'sinatra/reloader'
+  register Sinatra::Reloader
+end
 
+configure :production do
+end
+
+set :public_dir, File.dirname(__FILE__) + '/public'
+use AppLog, logger
+use Rack::CommonLogger, logger
+
+run FluentdJsonReceiver

lib/applog.rb

@@ -0,0 +1,10 @@
+class AppLog
+  def initialize(app, logger)
+    @app, @logger = app, logger
+  end
+
+  def call(env)
+    env["app.logger"] = @logger
+    @app.call(env)
+  end
+end

log/.gitkeep

@@ -1,9 +1,9 @@
 !!! 5
-%html{html_attrs}
+%html
   %head
     %meta{:'http-equiv' => "Content-Type", :content => "text/html; charset=utf-8"}
     %title Fluentd JSON Receiver
-    %link{:href=>'/stylesheets/application.css', :media => "screen", :rel => 'stylesheet', :type => "text/css"}
+    %link{:href=> "#{@root}stylesheets/application.css", :media => "screen", :rel => 'stylesheet', :type => "text/css"}
     %script{:type => "text/javascript", :src => "//ajax.googleapis.com/ajax/libs/jquery/1.10.1/jquery.min.js"}
     %script{:type => "text/javascript", :src => "/javascripts/application.js"}
   %body