enable client expiry for non-TCP sockets

It's conceivable a sub-optimally configured instance can have too
many Unix sockets connected to us.  This also implements expiry for
systems without sufficient "struct tcp_info" support.

1 files changed, 45 insertions, 0 deletions

diff --git a/lib/yahns/client_expire_generic.rb b/lib/yahns/client_expire_generic.rb
new file mode 100644
index 0000000..f2f5369
--- /dev/null
+++ b/lib/yahns/client_expire_generic.rb
@@ -0,0 +1,45 @@
+# Copyright (C) 2013, Eric Wong <normalperson@yhbt.net> and all contributors
+# License: GPLv3 or later (https://www.gnu.org/licenses/gpl-3.0.txt)
+module Yahns::ClientExpireGeneric # :nodoc:
+  def __timestamp
+    Time.now.to_f
+  end
+
+  def yahns_init
+    super # Yahns::HttpClient#yahns_init
+    @last_io_at = 0
+  end
+
+  def yahns_expire(timeout)
+    return 0 if closed?
+    if (__timestamp - @last_io_at) > timeout
+      shutdown
+      1
+    else
+      0
+    end
+  # shutdown may race with the shutdown in http_response_done
+  rescue
+    0
+  end
+
+  def kgio_write(*args)
+    @last_io_at = __timestamp
+    super
+  end
+
+  def kgio_trywrite(*args)
+    @last_io_at = __timestamp
+    super
+  end
+
+  def kgio_tryread(*args)
+    @last_io_at = __timestamp
+    super
+  end
+
+  def trysendfile(*args)
+    @last_io_at = __timestamp
+    super
+  end
+end