lib/popcorn/examples/middlewares/example_advanced_logger.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Copyright 2016 Alexandre Terrasa <alexandre@moz-code.org>
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #     http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16
  17 module example_advanced_logger is example
  18
  19 import popcorn
  20 import realtime
  21
  22 redef class HttpRequest
  23         # Time that request was received by the Popcorn app.
  24         var timer: nullable Clock = null
  25 end
  26
  27 class RequestTimeHandler
  28         super Handler
  29
  30         redef fun all(req, res) do req.timer = new Clock
  31 end
  32
  33 class AdvancedLoggerHandler
  34         super Handler
  35
  36         redef fun all(req, res) do
  37                 var timer = req.timer
  38                 if timer != null then
  39                         print "{req.method} {req.uri} {res.color_status} ({timer.total}s)"
  40                 else
  41                         print "{req.method} {req.uri} {res.color_status}"
  42                 end
  43         end
  44 end
  45
  46 class AnotherHandler
  47         super Handler
  48
  49         redef fun get(req, res) do res.send "Hello World!"
  50 end
  51
  52 var app = new App
  53 app.use_before("/*", new RequestTimeHandler)
  54 app.use("/", new AnotherHandler)
  55 app.use_after("/*", new AdvancedLoggerHandler)
  56 app.listen("localhost", 3000)