From 7664144957b4916c76ec52132eaaa552a348de28 Mon Sep 17 00:00:00 2001
From: Eric Wong <normalperson@yhbt.net>
Date: Tue, 4 May 2010 17:01:37 -0700
Subject: use Unicorn.listener_names method if available

This allows easier configuration when used with Unicorn and
derived servers (Rainbows! and Zbatery).
---
 lib/raindrops/middleware.rb | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'lib/raindrops/middleware.rb')

diff --git a/lib/raindrops/middleware.rb b/lib/raindrops/middleware.rb
index 4ef6368..a1503bd 100644
--- a/lib/raindrops/middleware.rb
+++ b/lib/raindrops/middleware.rb
@@ -13,7 +13,12 @@ class Middleware < ::Struct.new(:app, :stats, :path, :tcp, :unix)
 
   def initialize(app, opts = {})
     super(app, opts[:stats] || Stats.new, opts[:path] || "/_raindrops")
-    if tmp = opts[:listeners]
+    tmp = opts[:listeners]
+    if tmp.nil? && defined?(Unicorn) && Unicorn.respond_to?(:listener_names)
+      tmp = Unicorn.listener_names
+    end
+
+    if tmp
       self.tcp = tmp.grep(/\A[^:]+:\d+\z/)
       self.unix = tmp.grep(%r{\A/})
       self.tcp = nil if tcp.empty?
-- 
cgit v1.2.3-24-ge0c7