my_stuff-fb303 0.0.3 → 0.0.4

data/lib/my_stuff/fb303/proxy_handler.rb

@@ -0,0 +1,24 @@
+# Copyright 2011-present Fred Emmott. All Rights Reserved.
+
+module MyStuff
+  module Fb303
+    class ProxyHandler
+      def initialize server
+        @s = server
+      end
+
+      def method_missing meth, *args
+        @s.increment_counter "#{meth}.called"
+        begin
+          result = @s.send(meth, *args)
+          @s.increment_counter "#{meth}.success"
+          result
+        rescue => e
+          @s.increment_counter "#{meth}.exception"
+          @s.increment_counter "#{meth}.exception.#{e.class.name}"
+          raise
+        end
+      end
+    end
+  end
+end

data/lib/my_stuff/fb303/server.rb

@@ -0,0 +1,82 @@
+# Copyright 2011-present Fred Emmott. All Rights Reserved.
+
+require 'my_stuff/fb303/proxy_handler'
+
+# Code used for MyStuff.ws
+module MyStuff
+  module Fb303
+    class Server
+      def initialize interface, name, version = ''
+        @name      = name
+        @version   = version
+        @interface = interface
+
+        @status         = Fb_status::STARTING
+        @status_details = ''
+        @started_at     = Time.new.to_i
+        @counters       = Hash.new(0)
+
+        start_fb303_server
+      end
+
+      def run!
+        @status  = Fb_status::ALIVE
+        @server_thread.join
+      end
+
+      ##### fb303 implementation #####
+
+      def getName; @name; end
+      def getVersion; @version; end
+      def getStatus; @status; end
+      def getStatusDetails; @status_details; end
+      def aliveSince; @started_at; end
+      def getCounters; @counters; end
+      def getCounter key; @counters[key]; end
+      def getOption key; ''; end
+      def getOptions; {}; end
+      def getCpuProfile duration; ''; end
+      def setOption key, value; end
+      def reinitialize; raise NotImplementedError.new; end
+
+      def shutdown
+        @status = Fb_status::STOPPING
+        @status_details = 'Shutdown requested via fb303'
+        @fb303_thread.kill
+        @status = Fb_status::STOPPED
+      end
+
+      def increment_counter name
+        @counters[name] += 1
+      end
+
+      def method_missing *args
+        if self.handler != self
+          self.handler.send *args
+        else
+          super *args
+        end
+      end
+
+      def handler
+        self
+      end
+
+      protected
+
+      def server_for processor
+        raise NotImplementedError.new
+      end
+
+      def start_fb303_server
+        handler   = MyStuff::Fb303::ProxyHandler.new(self)
+        processor = @interface.const_get(:Processor).new(handler)
+        @server   = server_for(processor)
+
+        @server_thread = Thread.new do
+          @server.serve
+        end
+      end
+    end
+  end
+end

metadata

@@ -2,7 +2,7 @@
 name: my_stuff-fb303
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Fred Emmott
@@ -24,6 +24,8 @@ extra_rdoc_files: []
 
 files: 
 - lib/my_stuff/fb303_server.rb
+- lib/my_stuff/fb303/proxy_handler.rb
+- lib/my_stuff/fb303/server.rb
 homepage: https://github.com/fredemmott/my_stuff-fb303
 licenses: []