cloudpi-appender 0.1.0

data/lib/cloudpi-appender.rb

@@ -0,0 +1,116 @@
+require 'rubygems'
+require 'socket'
+require 'json'
+require 'logger'
+require 'timeout'
+
+module CloudPI
+
+class Appender
+  
+  CONNECT_TIMEOUT =  0.1  # sec
+  
+  def initialize(bridge_ip = 'localhost', bridge_port = 5999, logger = nil, policy = {})
+    @bridge_ip     = bridge_ip
+    @bridge_port   = bridge_port
+
+    @logger = logger
+    @logger = Logger.new(STDOUT) unless @logger
+    
+    begin
+      json         = JSON.parse(ENV['VCAP_APPLICATION'], {:symbolize_names => true})
+      @app_ip      = json[:host]
+      @app_port    = json[:port]
+      @app_id      = json[:app_id]
+      @instance_id = json[:instance_id]
+    rescue JSON::ParseError, TypeError
+      @logger.error "parsing error: ENV['VCAP_APPLICATION']"
+    end
+
+    @policy = policy
+    @policy = {} unless @policy.is_a? Hash
+
+    connect_to_bridge
+  end
+
+  def connect_to_bridge
+    Timeout::timeout(CONNECT_TIMEOUT) do
+      @bridge = TCPSocket.new(@bridge_ip, @bridge_port)
+    end
+    send_policy
+    @logger.info("connected to bridge: #{@bridge_ip}:#{@bridge_port}")
+  rescue => e
+    @logger.error("bridge connection failed: #{e.message}")
+  end
+
+  def disconnect_bridge
+    @bridge.close if @bridge
+    @logger.info("bridge connection closed.")
+  end
+
+  def connected_to_bridge?
+    return false unless @bridge && !@bridge.closed?
+    true
+  end
+
+  def check_and_reconnect_to_bridge
+    connect_to_bridge unless connected_to_bridge?
+  end
+
+  def parse_json(json)
+    body = JSON.parse(json, {:symbolize_names => true})
+    raise JSON::ParseError unless body.is_a?(Hash)  # not allowed array json
+    body
+  rescue JSON::ParseError
+    @logger.error("wrong metric json message recieved.")
+    {}
+  end
+
+  def send_to_bridge(msg)
+    return unless msg
+    check_and_reconnect_to_bridge
+    @bridge.write("#{msg}\0")
+    @logger.debug("sent message to bridge: #{msg}")
+  rescue => e
+    @logger.error("bridge write error: #{e.message}")
+    @bridge.close
+  end
+
+  def metric_key
+    # instance_ip:instance_port:app_id:instance_id
+    :"#{@app_ip}:#{@app_port}:#{@app_id}:#{@instance_id}"
+  end
+
+  def set_policy(policy)
+    @policy = policy
+    @policy = {} unless @policy.is_a? Hash
+    @logger.info("set metric policy: #{@policy.to_json}")
+    send_policy
+  end
+
+  def send_heartbeat
+    msg = { metric_key => {:alive => Time.now.to_i} }.to_json
+    send_to_bridge(msg)
+    @logger.info("heartbeat sent: #{msg}")
+  end
+
+  def msg_with_when(msg)
+    msg.merge({ :when => Time.now.to_i })
+  end
+
+  def send_policy
+    msg = { metric_key => msg_with_when(@policy) }.to_json
+    send_to_bridge(msg)
+    @logger.info("policy sent: #{msg}")
+  end
+
+  def send(msg)
+    return unless msg.is_a? Hash
+    msg = { metric_key => msg_with_when(msg) }.to_json
+    send_to_bridge(msg)
+    @logger.info("value sent: #{msg}")
+  end
+
+end
+
+end

metadata

@@ -0,0 +1,45 @@
+--- !ruby/object:Gem::Specification
+name: cloudpi-appender
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Junsang Yang
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-01 00:00:00.000000000 Z
+dependencies: []
+description: CloudPI PaaS - Application Custom Metric Client for Ruby
+email: zuns00@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/cloudpi-appender.rb
+homepage: http://www.not.yet/cloudpi-appender
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: CloudPI Bridge Appender for Ruby
+test_files: []