apminsight 0.0.3

data/lib/agent/metrics/am_metricstore.rb

@@ -0,0 +1,29 @@
+require 'agent/am_objectholder'
+
+module ManageEngine
+	class APMMetricsStore
+		attr_accessor :metrics
+		def initialize
+			@metrics = Hash.new
+		end
+
+		def remove keys
+			if keys!=nil
+				keys.each {|key| @metrics.delete("#{key}")} 
+			end
+		end
+
+		def metrics_dup
+			@metrics.dup
+		end
+
+		def removeData key, strt_indx,end_indx
+			if @metrics.has_key?(key)
+				val = @metrics[key]
+				val = val.drop(end_indx)
+				@metrics[key]=val
+			end
+		end
+
+	end
+end

data/lib/agent/server/am_agent.rb

@@ -0,0 +1,108 @@
+require "agent/am_objectholder"
+require "agent/server/worker/am_worker"
+
+require 'socket'
+
+module ManageEngine
+	class APMAgent
+		def initialize
+			@obj = ManageEngine::APMObjectHolder.instance
+			@obj.log.debug "Agent Initialization - START"
+			doConnect
+
+			if !@obj.shutdown && @obj.agent_initialized
+				@obj.log.info "Agent Initialization - DONE"
+				@obj.instrumenter.doSubscribe
+				doDispatcherActions
+				doCollect
+				puts "ManageEngine APM Ruby Agent Started"
+			else
+				@obj.log.info "Agent Initialization Failed - Going to shutdown"
+				@obj.instrumenter.doSubscribe
+				@obj.shutdownagent
+			end
+           	
+
+       end
+
+		def doConnect
+			begin	
+				if @obj.shutdown
+					@obj.log.info "[ Problem in Agent Startup ]"
+				else
+					agentInfo = @obj.config.getAgentInfo
+					resp = nil
+					if @obj.config.alreadyconnected
+						@obj.log.debug "[doConnect] Already Connected - Make Contact - Instance id = #{@obj.config.instance_id}"
+						resp = startConnect "?instance_id="+@obj.config.instance_id,agentInfo
+					else
+						@obj.log.debug "[doConnect] Going to connect - New "
+						resp = startConnect "",agentInfo
+						if resp.has_key?("instance-info")
+							aData = resp["instance-info"]
+							aData["agent.id"]=aData.delete("instanceid")
+							aData["agent.enabled"]=true
+							@obj.config.updateAgentInfoFile(aData)
+							@obj.log.debug "[doConnect] Connected - InstanceID  : #{@obj.config.instance_id}"
+						else
+							@obj.log.info "[doConnect] [ Problem in connecting server] [ Going to shutdown ]"
+							@obj.shutdown=true
+						end
+					end
+
+					if resp==nil
+						@obj.log.info "[doConnect] [ Error in Response while connecting Server . [ Going to shutdown ]"
+						@obj.shutdown= true
+					end
+
+					if(!@obj.shutdown)
+						@obj.agent_initialized=true
+					end
+				end
+			rescue Exception=>e
+				@obj.shutdown = true
+				@obj.log.logException "[doConnect] Exception while connecting server. [ Going to shutdown ] ",e
+			end
+		end
+
+
+		def doCollect
+			@obj.log.info "[doCollect] Starts - Wait time : #{@obj.config.connect_interval} seconds "
+			begin
+				ManageEngine::APMWorker.getInstance.start
+			rescue Exception=>e
+				@obj.log.logException "[doCollect]  Exception during worker  startup  #{e.message}",e
+				@obj.shutdown=true
+			end
+		end
+
+
+		def startConnect uri,data
+			resp = @obj.connector.post @obj.constants.connect_uri+uri,data
+		end
+
+		def doDispatcherActions
+			case @obj.config.app_dispatcher
+				when 'passenger'
+					#starting a new process
+      				PhusionPassenger.on_event(:starting_worker_process) do |forked|
+        				if forked
+                			@obj.log.info "starting_worker_process : Process ID :#{Process.pid} : Creating new apm worker"
+                			doCollect
+        				else
+                			doCollect
+		                	@obj.log.info "starting_worker_process : Conservative Process ID :#{Process.pid} - No new worker"
+        				end
+     			 	end
+      				# shutting down a process.
+				    PhusionPassenger.on_event(:stopping_worker_process) do
+                		ManageEngine::APMWorker.getInstance.stop
+                		@obj.log.info "stopping_worker_process :Process ID :#{Process.pid} ---->  #$$ "
+     				end
+				else#case
+
+				end#case
+		end
+	end#c
+end#m
+

data/lib/agent/server/am_connector.rb

@@ -0,0 +1,195 @@
+require 'net/http'
+require 'net/https'
+require 'uri'
+require 'json'
+require "agent/server/instrument/am_instrumenter"
+
+module ManageEngine
+	class APMConnector
+
+		def initialize 
+			@obj = ManageEngine::APMObjectHolder.instance
+			@pretry =0
+			@gretry =0
+		end
+
+		def post uri,data
+			@pretry = @pretry +1
+			begin
+
+				u = url uri
+				#@obj.log.info "[connector] [ POST]  START"
+				@obj.log.debug "[connector] [ POST]  : \n\n#{u}\n\n#{data}\n\n"
+				con =  connection(u)
+				req = Net::HTTP::Post.new(u.request_uri,initheader = {'Content-Type' =>'application/json'})
+				req.body=data.to_json
+				resp = con.request(req)
+				@obj.log.debug "[connector] [POST ] \n Response : #{resp} \nResponse Code : #{resp.code}\nMessage : #{resp.message}\nBody : #{resp.body}"
+				rdata = responseParser resp
+				@pretry = 0
+				#@obj.log.info "[connector] [ POST]  END"
+				return rdata
+			rescue Exception=>e
+				@obj.log.logException "[connector] Exception while connecting server- Data not sent \n",e
+				if @pretry >=@obj.config.connection_retry
+					#@obj.shutdown= true
+					return nil
+				else
+					@obj.log.info "[connector] Exception found in Post request - Retrying - Count - #{@pretry}"
+					return post uri,data
+				end
+
+			end
+			@pretry = 0
+		end
+
+		def get uri
+			@gretry = @gretry +1
+			begin
+				u = url uri
+				#@obj.log.info "[connector] [ GET ]  START"
+				@obj.log.debug "[connector] [ GET]  : \n#{u}\n"
+				req = Net::HTTP::Get.new(u.request_uri)
+				resp = con.request(req)
+				#@obj.log.info "[connector] [ GET ]  END"
+			rescue Exception=>e
+				@obj.log.logException "[connector] [ GET]  Exception while connecting server  - Data not sent ",e
+				if @pretry >=@obj.config.connection_retry
+					#@obj.shutdown= true
+				else
+					@obj.log.info "[connector] Exception found in Get request - Retrying - Count - #{@gretry}"
+					return get uri
+				end
+			end
+			@gretry = 0
+		end
+
+		def url(uri)
+			ru=nil
+			p="http"
+			if(@obj.config.is_secured)
+				p="https"
+			end			 	
+			u = p+"://"+@obj.config.apmhost+":#{@obj.config.apmport}/"+uri
+			begin
+				ru = URI.parse(u)
+			rescue
+				raise URI::InvalidURIError, "Invalid url '#{ru}'"
+			end
+
+			if (ru.class != URI::HTTP && ru.class != URI::HTTPS)
+				raise URI::InvalidURIError, "Invalid url '#{u}'"
+			end
+			ru
+		end
+
+		def connection(url)
+
+			if (@obj.config.proxyneeded)
+				@obj.log.debug "[connect] Through Proxy"
+				con = Net::HTTP::Proxy(@obj.config.proxy_host, @obj.config.proxy_port,@obj.config.proxy_user,@obj.config.proxy_pass).new(url.host, url.port)		
+			else
+				#@obj.log.info "Proxy Not Needed"
+				con = Net::HTTP.new(url.host, url.port)
+			end
+			con=getScheme(con)
+			con.open_timeout = @obj.constants.connection_open_timeout
+			con.read_timeout = @obj.constants.connection_read_timeout
+			con
+		end
+
+		def getScheme(con)
+			if(@obj.config.is_secured)
+				#@obj.log.info "[connect] Secured"
+				con = Net::HTTP::Proxy(@obj.config.proxy_host, @obj.config.proxy_port,@obj.config.proxy_user,@obj.config.proxy_pass).new(url.host, url.port)		
+				con.use_ssl=true
+				con.verify_mode=OpenSSL::SSL::VERIFY_NONE
+			end
+			con
+		end
+
+		def responseParser resp
+			if resp == Net::HTTPSuccess || Net::HTTPOK
+				rawData = resp.body
+				if rawData.length>=2
+					rBody = JSON.parse(rawData)
+					result = rBody["result"]
+					data = rBody["data"]
+					if !@obj.util.getBooleanValue result
+						if data!=nil
+							if data.has_key?("exception")
+								raise Exception.new("Exception from server - "+data["exception"])
+							end
+						end
+
+					end
+					if data!=nil && data.has_key?(@obj.constants.response_code)
+
+						srCode = data[@obj.constants.response_code]
+						response_action srCode
+					end
+					return data
+				end
+				return rawData
+			else
+				raise Exception.new("Http Connection Response Error #{resp.to_s}")
+			end
+		end
+
+
+
+		def response_action rCode
+			case rCode
+			when @obj.constants.licence_expired then
+				@obj.log.info "License Expired. Going to shutdown"
+				raise Exception.new("License Expired. Going to shutdown")
+			when @obj.constants.licence_exceeds then
+				@obj.log.info "License Exceeds. Going to shutdown"
+				raise Exception.new("License Exceeds. Going to shutdown")
+			when @obj.constants.delete_agent then
+				@obj.log.info "Action from Server - Delete the Agent. Going to shutdown and remove the Agent"
+				deleteAgent
+				raise Exception.new("Action from Server - Delete the Agent. Going to shutdown and remove the Agent")
+			when @obj.constants.unmanage_agent then
+				@obj.log.info "Action from Server - Unmanage the Agent. Going to  Stop the DC - Disabling the Agent"
+				unManage
+			when @obj.constants.manage_agent then
+				@obj.log.info "Action from Server - Manage the Agent. Going to Sart the DC - Enabling the Agent"
+				manage	
+			end
+		end
+
+		def unManage
+			@obj.instrumenter.doUnSubscribe
+			@obj.instrumenter =nil
+			@obj.instrumenter = ManageEngine::APMInstrumenter.new
+			uManage = Hash.new
+			uManage["agent.id"]=@obj.config.instance_id
+			uManage["agent.enabled"]=false	
+			@obj.config.updateAgentInfoFile uManage
+		end
+
+		def manage
+			@obj.instrumenter.doSubscribe
+			uManage = Hash.new
+			uManage["agent.id"]=@obj.config.instance_id
+			uManage["agent.enabled"]=true	
+			@obj.config.updateAgentInfoFile uManage
+		end
+
+		def deleteAgent
+			@obj.instrumenter.doUnSubscribe
+			@obj.instrumenter =nil
+			uManage = Hash.new
+			uManage["agent.id"]=@obj.config.instance_id
+			uManage["agent.enabled"]=false	
+			@obj.config.updateAgentInfoFile uManage
+			begin
+				File.delete(@obj.constants.agent_conf)
+			rescue Exceptione=>e
+				@obj.log.logException "#{e.message}",e
+			end
+		end
+
+	end#c
+end#m

data/lib/agent/server/instrument/am_apm.rb

@@ -0,0 +1,99 @@
+require 'agent/am_objectholder'
+@obj = ManageEngine::APMObjectHolder.instance
+class Class
+   alias old_new new
+   def new(*args, &block)
+	 result =nil;
+	 begin
+		if(block==nil || block=="")
+     		result = old_new(*args)
+		elsif
+     		result = old_new(*args,&block)
+		end
+	 rescue Excetion=>exe
+        raise exe		 
+		result = self
+	 end
+	 me_apm_injector(self,result)
+	 
+		return result
+	 end
+ end
+
+def me_apm_injector(s,result)
+	begin
+		if(ManageEngine::APMObjectHolder.instance.config.include_packages.index(s.name)!=nil)
+		ms =s.instance_methods(false)
+		cms = s.methods(false)
+		begin
+			ms.each do |m|
+			if( m.to_s.index("APMTEST"))
+				return;
+			end
+			end
+			cms.each do |m|
+			if( m.to_s.index("APMTEST"))
+				return;
+			end
+			end
+		rescue Exception=>e
+			return;
+		end
+		ManageEngine::APMObjectHolder.instance.log.debug "Injection Method : #{ms} "
+		ManageEngine::APMObjectHolder.instance.log.debug "Injection Class Method : #{cms} "
+		ms.each do |m|
+		mn = m.to_s
+		#ManageEngine::APMObjectHolder.instance.log.info "ManageEngine Monitor Method : #{s.name} # #{m.to_s}"
+		omn = "APMTEST"+mn+"APMTEST"
+		s.class_eval  %{
+			alias_method :#{omn}, :#{mn}
+		def #{mn} *args, &block
+			begin
+			ActiveSupport::Notifications.instrument("apm.methodstart", {:method=>"#{mn}",:args=>args})
+			res = #{omn} *args, &block
+			ActiveSupport::Notifications.instrument("apm.methodend", {:method=>"#{mn}",:args=>args})
+			return res
+			rescue Exception => exe
+			puts "error in calling method"
+			raise exe
+			ensure
+			end
+			end
+		}
+		end#do
+		default_methods = Array.new
+		default_methods.push("_helpers");
+		default_methods.push("middleware_stack");
+		default_methods.push("helpers_path");
+		default_methods.push("_wrapper_options");
+		cms.each do |m|
+		if(default_methods.index(m.to_s)==nil)
+			mn = m.to_s
+			#ManageEngine::APMObjectHolder.instance.log.debug "ManageEngine Monitor Singleton Method : #{s.name} ---> #{m.to_s}"
+			omn = "APMTEST"+mn+"APMTEST"
+			s.instance_eval  %{
+				class << self
+					alias_method :#{omn}, :#{mn}
+				end
+				def self.#{mn} *args, &block
+				begin
+					ActiveSupport::Notifications.instrument("apm.methodstart", {:method=>"#{mn}",:args=>args})
+					res = #{omn} *args, &block
+					ActiveSupport::Notifications.instrument("apm.methodend", {:method=>"#{mn}",:args=>args})
+					return res
+				rescue Exception=>exe
+					puts "Instrument : error in calling class method"
+				raise exe
+				ensure
+				end
+				end
+			}
+		end
+		end#do
+	end#if
+	rescue Exception=>e
+		puts "Exception in instrument : #{e}"
+	ensure
+	end
+end
+

data/lib/agent/server/instrument/am_instrumenter.rb

@@ -0,0 +1,43 @@
+require 'agent/am_objectholder'
+require 'socket'
+module ManageEngine
+	class APMInstrumenter
+	@t =nil;
+	def initialize
+		@obj=ManageEngine::APMObjectHolder.instance
+	end
+
+	def doSubscribe
+		@obj=ManageEngine::APMObjectHolder.instance
+		@obj.log.debug "[ instrumenter ] [ Subscriber for Agent ]"
+		@subscriber = ActiveSupport::Notifications.subscribe  do |name, start, finish, id, payload|
+		if(ManageEngine::APMObjectHolder.instance.config.agent_enabled)
+			rt = (finish-start).to_i
+			 ManageEngine::APMWorker.getInstance.start
+			ManageEngine::APMObjectHolder.instance.log.debug "[ Notifications for Agent ] #{Thread.current} #{id} #{name} - #{rt} - #{payload}"
+			trace= caller;
+			id = "#{Thread.current}"
+			stats = Hash.new
+			stats["name"] = name;
+			stats["start"] = start.to_f * 1000;
+			stats["end"] = finish.to_f * 1000;
+			stats["id"] = id;
+			stats["payload"] = payload;
+			if (name=="sql.active_record" && (finish-start)>=(ManageEngine::APMObjectHolder.instance.config.sql_trace_t * 1000 ).to_i)
+				stats["trace"] = trace;
+			end
+			stats["ctime"] =ManageEngine::APMObjectHolder.instance.util.currenttimemillis;
+			ManageEngine::APMObjectHolder.instance.collector.updateTransaction(id,stats);
+		else
+			ActiveSupport::Notifications.unsubscribe @subscriber
+				@obj.log.info "[ instrumenter ] [ RETURNING NO METRICS] "
+		end
+		end
+	end
+
+	def doUnSubscribe
+		ActiveSupport::Notifications.unsubscribe @subscriber
+	end
+
+end #class
+end#module