examples/with_middlewares.rb in upperkut-0.5.1 vs examples/with_middlewares.rb in upperkut-0.5.2

@@ -1,7 +1,18 @@
 require_relative '../lib/upperkut/worker'
+require_relative '../lib/upperkut/logging'
 
+class ClientMiddleware
+  def call(worker, items)
+    logger = Upperkut::Logging.logger
+
+    logger.info("inserting worker=#{worker} items=#{items.count}")
+    yield
+    logger.info("inserted worker=#{worker} items=#{items.count}")
+  end
+end
+
 class MyMiddleware
   def call(worker, items)
     logger = Upperkut::Logging.logger
 
     logger.info("performing worker=#{worker} items=#{items.count}")
@@ -12,11 +23,15 @@
 
 class WithMiddlewares
   include Upperkut::Worker
 
   setup_upperkut do |config|
-    config.middlewares do |chain|
+    config.server_middlewares do |chain|
       chain.add MyMiddleware
+    end
+
+    config.client_middlewares do |chain|
+      chain.add ClientMiddleware
     end
   end
 
   def perform(items)
     puts "executing........."