lib/memolog.rb in memolog-0.3.3 vs lib/memolog.rb in memolog-0.3.4

@@ -4,13 +4,13 @@
 require "securerandom"
 require "stringio"
 
 require "memolog/version"
 require "memolog/config"
-require "memolog/extension"
 require "memolog/formatter"
 require "memolog/init"
+require "memolog/logger_extension"
 require "memolog/rails_middleware"
 require "memolog/sentry_extension"
 require "memolog/sidekiq_middleware"
 
 module Memolog
@@ -22,15 +22,15 @@
 
   def configure
     yield(config) if block_given?
   end
 
-  def init!
-    Memolog::Init.new.call
+  def init_middlewares!
+    Memolog::Init.init_middlewares!
   end
 
   def extend_logger(other_logger)
-    other_logger.extend(Memolog::Extension)
+    other_logger.extend(Memolog::LoggerExtension)
     other_logger.formatter = config.formatter
   end
 
   def uuid
     Thread.current[:memolog_uuid]