apminsight 1.0.1 → 1.8.6

data/lib/agent/configuration/am_configuration.rb

@@ -1,14 +1,15 @@
 require 'rubygems'
 require 'json'
 require 'socket'
-require 'rails'
+require 'net/http'
 require 'agent/am_objectholder'
+require 'agent/version'
 
 module ManageEngine
 	class APMConfig
-		attr_reader :agenthost,:agentport,:instance_id,:alreadyconnected,:apmhost,:apmport
-		attr_reader :appname,:proxyneeded, :apdex_t, :trans_trace, :trans_trace_t, :sql_capture, :sql_capture_params, :sql_trace_t,:proxy_user,:proxy_pass
-		attr_reader :proxy_host,:proxy_port ,:is_secured, :logs_dir ,:connection_retry,:agent_enabled,:connect_interval,:db_operations,:include_packages
+		attr_reader :agenthost,:agentport,:instance_id,:alreadyconnected,:apmhost,:apmport,:license_key,:site24x7, :site24x7url, :hostType
+		attr_reader :appname,:proxyneeded, :apdex_t, :trans_trace, :trans_trace_t, :sql_capture, :sql_capture_params, :sql_trace_t,:proxy_user,:proxy_pass, :metric_overflow_t, :trace_overflow_t, :dbmetric_overflow_t
+		attr_reader :proxy_host,:proxy_port ,:is_secured, :logs_dir ,:connection_retry,:agent_enabled,:connect_interval,:db_operations,:txn_skip_listen, :url_merge_pattern
 		attr_accessor :app_db,:app_dispatcher,:lastupdatedtime
 		def initialize
 			@obj = ManageEngine::APMObjectHolder.instance
@@ -21,19 +22,31 @@ module ManageEngine
 			@instance_id  = 0
 			@agent_enabled = false
 			@alreadyconnected = checkAgentInfo
+			@site24x7 = checkLicenseFile
+			if (@site24x7)
+			  @site24x7url = @license_key.start_with?('eu_') ? @obj.constants.site24x7EUurl : @license_key.start_with?('cn_') ? @obj.constants.site24x7CNurl : @license_key.start_with?('au_') ? @obj.constants.site24x7AUurl : @license_key.start_with?('in_') ? @obj.constants.site24x7INurl : @license_key.start_with?('gd_') ? @obj.constants.site24x7GDurl : @license_key.start_with?('jp_') ? @obj.constants.site24x7JPurl : @obj.constants.site24x7USurl
+			end
 			@db_operations =["select","insert","update","delete"]
-			@obj.log.debug "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
-			@obj.log.debug "APP HOME #{File.absolute_path(".")} "
-			@obj.log.debug "APP HOME #{Dir.pwd} "
-			@obj.log.debug "Configuration : "
-			@obj.log.debug "Agent Name : #{@agenthost}"
-			@obj.log.debug "Agent Already Connected : #{@alreadyconnected}"
-			@obj.log.debug "Agent Enabled : #{@agent_enabled}"
-			@obj.log.debug "Allowed DB Operations : #{@db_operations}"
-			@config.each do|key,val|
-			@obj.log.info "#{key} => #{val}"
-				end
-			@obj.log.debug "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
+			urlMergePattern
+			@hostType = getHostType
+			@obj.log.info "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
+			@obj.log.info "APP HOME #{File.absolute_path(".")} "
+			@obj.log.info "APP HOME #{Dir.pwd} "
+			@obj.log.info "Agent Version : #{ManageEngine::APMInsight::VERSION}"
+			#@obj.log.info "Configuration : "
+			#@obj.log.info "Hostname : #{@agenthost}"
+			@obj.log.info "Host Type: #{@hostType}"
+			@obj.log.info "Agent Already Connected : #{@alreadyconnected}"
+			@obj.log.info "Agent Enabled : #{@agent_enabled}"
+			@obj.log.info "Allowed DB Operations : #{@db_operations}"
+#			@config.each do|key,val|
+#			@obj.log.info "#{key} => #{val}"
+#				end
+			@obj.log.info "URL Merge Patterns"
+			@url_merge_pattern.each do |key, val|
+				@obj.log.info "#{key} => #{val}"
+			end
+			@obj.log.info "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
 			@app_db="dummydb"
 			@app_dispatcher = getDispatcher
 			@lastupdatedtime=File.mtime(@obj.constants.apm_conf).to_i
@@ -41,27 +54,33 @@ module ManageEngine
 		
 		def configureFile
 			begin
-				gem_conf = Gem.loaded_specs[@obj.constants.apm_gem].full_gem_path
-				#gem_conf = File.join(gem_conf, 'lib')
-				gem_conf = File.join(gem_conf, 'conf')
-				gem_conf = File.join(gem_conf, 'apminsight.conf')
 				if(FileTest.exists?(@obj.constants.apm_conf))
 					#conf file exists in APPlication Home
-						@obj.log.debug "Config File Read from #{@obj.constants.apm_conf}"
+						@obj.log.debug "Config File Exists. It is read from #{@obj.constants.apm_conf}"
 					@config = @obj.util.readProperties(@obj.constants.apm_conf)
+          secureConfFile "#{@obj.constants.apm_conf}"
 				else
+				  gemSpecs = Gem.loaded_specs[@obj.constants.s247_apm_gem]
+				  if (gemSpecs == nil)
+				    gemSpecs = Gem.loaded_specs[@obj.constants.apm_gem]
+				  end
+					gem_conf = gemSpecs.full_gem_path
+					#gem_conf = File.join(gem_conf, 'lib')
+					gem_conf = File.join(gem_conf, 'conf')
+					gem_conf = File.join(gem_conf, 'apminsight.conf')
 					#conf file not exists in APPlications Home. So 1. copy it for gem locations
 					if @obj.util.copyFiles gem_conf,@obj.constants.apm_conf
 						#copied sucessfully
-						@obj.log.debug "Config File Read from #{@obj.constants.apm_conf}"
+						@obj.log.info "Config File copied to application home directory. It is read from #{@obj.constants.apm_conf}"
 						@config = @obj.util.readProperties(@obj.constants.apm_conf)
+            secureConfFile "#{@obj.constants.apm_conf}"
 					else
 						#Problem in copying, so reading props from Conf file in Gem Location
-						@obj.log.debug "Config File Read from #{gem_conf}"
+						@obj.log.warn "Config File not copied. It is read from #{gem_conf}"
 						@config = @obj.util.readProperties(gem_conf)
+            secureConfFile gem_conf
 					end
 				end
-				
 			rescue Exception=>e
 				@obj.log.info "[Exception] Problem in Reading Configuration File : \n File : #{@obj.constants.apm_conf}"
 				@obj.log.logException "#{e.message}",e
@@ -70,11 +89,29 @@ module ManageEngine
 			end
 		end
 
+		def secureConfFile(file)
+		  begin
+		    File.chmod(0600, file)
+		  rescue Exception => e
+		    @obj.log.warn "Unable to secure the conf file #{file}"
+		  end
+		end
+		
 		def checkAgentInfo
 			if FileTest.exists?(@obj.constants.agent_conf)
 				@obj.log.debug "Status : Agent Already Connected"
 				props = @obj.util.readProperties(@obj.constants.agent_conf)
-				@instance_id = props["agent.id"]
+				instance_id = props["agent.id"]
+				
+				if (instance_id == nil || instance_id == "")
+				  # If instance id is not found or empty, it means the apminsight.info is being modified by user
+				  # Ignore all its entry
+				  @obj.log.warn "File: #{@obj.constants.agent_conf} is corrupted. Agent will continue ignoring these values."
+				  return false
+				else
+				  @instance_id = instance_id
+				end
+				
 				@agent_enabled= @obj.util.getBooleanValue props["agent.enabled"]
 				true
 			else
@@ -83,10 +120,35 @@ module ManageEngine
 			end
 		end
 
+		def checkLicenseFile
+      @obj.constants.setLicenseKey @license_key
+		  if(@license_key.start_with?('APMI_')) 
+		    @obj.log.info "Connecting to App Manager"
+		    false
+		  else
+		    @obj.log.info "Connecting to Site24x7"
+		    true
+		  end
+		end
+		
+		def urlMergePattern
+			@url_merge_pattern = Hash.new
+			begin
+				if (FileTest.exists?(@obj.constants.mergepattern_conf))
+					@url_merge_pattern=@obj.util.readProperties(@obj.constants.mergepattern_conf)
+				end
+			rescue Exception => e
+				@obj.log.info "[Exception] Problem in Reading Configuration File : \n File : #{@obj.constants.mergepattern_conf}"
+				@obj.log.logException "#{e.message}",e
+			end
+		end
+
 		def updateAgentInfoFile(props)
 			@instance_id = props["agent.id"]
 			@agent_enabled= @obj.util.getBooleanValue props["agent.enabled"]
 			@obj.util.writeProperties(@obj.constants.agent_conf,props)
+			
+			secureConfFile(@obj.constants.agent_conf)
 		end
 
 		def initValues
@@ -102,18 +164,30 @@ module ManageEngine
 			@connection_retry = 0
 			@connect_interval = 60
 			@apdex_t=0.5
-			@include_packages=Array.new
+			@txn_skip_listen=Array.new
 			@trans_trace_t=2
 			@sql_trace_t=3
+			@metric_overflow_t=250
+      @dbmetric_overflow_t=500
+			@trace_overflow_t=30
+			@site24x7url = @obj.constants.site24x7USurl #default agent communication URL
 		end
 
 		def assignConfig
 		initValues
 		 @config.each do |key,value|
+		 	 value = checkAndGetEnvValue(value)
 			 case key
 			 when "application.name" then @appname=value
+			                              if (ENV.has_key?('APM_APPLICATION_NAME'))
+			                                @appname = ENV['APM_APPLICATION_NAME']
+			                              end
 			 when "apm.host" then @apmhost=value
 			 when "apm.port" then @apmport=isInteger(@apmport,value)
+			 when "license.key" then @license_key=value
+			                         if (@license_key.empty? && ENV.has_key?('S247_LICENSE_KEY'))
+			                           @license_key = ENV['S247_LICENSE_KEY']
+			                         end
 			 when "behind.proxy" then @proxyneeded=@obj.util.getBooleanValue value
 			 when "agent.server.port" then @agentport=isInteger(@agentport,value)
 			 when "apdex.threshold" then @apdex_t=isFloat(@apdex_t,value)
@@ -123,46 +197,175 @@ module ManageEngine
 			 when "transaction.trace.sql.parametrize" then @sql_capture_params=@obj.util.getBooleanValue value
 			 when "transaction.trace.sql.stacktrace.threshold" then @sql_trace_t=isFloat(@sql_trace_t,value)
 			 when "proxy.server.host" then @proxy_host=value
-			 when "proxy.server.port" then @proxy_port=value
+			 when "proxy.server.port" then @proxy_port=isInteger(@proxy_port,value)
 			 when "proxy.auth.username" then @proxy_user=value
-			 when "proxy.auth.password" then @proxy_pass=value
-			 when "apminsight.secure" then @is_secured=@obj.util.getBooleanValue value
+			 when "proxy.auth.password" then @proxy_pass=@obj.util.decrypt value, @license_key
+			                                    if (@proxy_pass == nil)
+			                                      @proxy_pass = value
+			                                    end
+			 when "apm.protocol.https" then @is_secured=@obj.util.getBooleanValue value
 			 when "apminsight.log.dir" then @logs_dir=value
+			 when "apminsight.log.level" then @obj.log.setLevel value
 			 when "agent.connection.retry" then @connection_retry=value #Not in Conf - yet to come
-			when "agent.connection.interval" then	 @connect_interval=value#Not in Conf - yet to come
-			when "include.packages" then @include_packages=@obj.util.getArray value,","
+			 when "agent.polling.interval" then	 @connect_interval=isInteger(@connect_interval, value)#Not in Conf - yet to come
+			 when "transaction.skip.listening" then @txn_skip_listen=@obj.util.getArray value.gsub("\s", ""),","
+			 when "metricstore.metric.bucket.size" then @metric_overflow_t = isInteger(@metric_overflow_t, value)
+       when "metricstore.dbmetric.bucket.size" then @dbmetric_overflow_t = isInteger(@dbmetric_overflow_t, value)
+       when "transaction.tracestore.size" then @trace_overflow_t = isInteger(@trace_overflow_t, value)
 			end
 		 end
+     store_encrypted_data(@config)
+		end
+		
+		#Checks whether the given value is Environment Variable
+		def checkAndGetEnvValue(data)
+		  begin
+  			value = "#{data}"[/\{(.*)\}/,1]
+  			if (value != nil && ENV.has_key?(value))
+  				return data.gsub(/\{.*\}/, ENV[value])
+			  end
+			rescue Exception=>e
+			end
+			return data
 		end
 
+		def store_encrypted_data config
+		  data = config["proxy.auth.password"]
+		  if (data != nil && @obj.util.decrypt(data, @license_key) == nil) # checking whether data is already encrypted
+		    begin
+		      file_contents = ""
+          conf_file = File.open(@obj.constants.apm_conf, 'r')
+          conf_file.read.each_line do |line|
+            if line.start_with?("proxy.auth.password")
+              file_contents += "proxy.auth.password=" + @obj.util.encrypt(config["proxy.auth.password"], @license_key)
+            else
+              file_contents += line
+            end
+            #file_contents += "\n"
+          end #end of do read loop
+          conf_file.close
+          conf_file = File.open(@obj.constants.apm_conf, "w+")
+          conf_file.puts file_contents
+          conf_file.close
+          file_contents = nil # clearing memory
+		    rescue Exception=>e
+		      @obj.log.logException "Error while encrypting file", e
+		    end
+		  end # if already encrypted
+		end
+		
+		def getHostType
+		  begin
+  		  # Check for AWS environment
+		    url = URI.parse('http://169.254.169.254/latest/meta-data/') # AWS metadata url
+        request = Net::HTTP::Get.new(url.path)
+		    response = Net::HTTP.start(url.host, url.port, :read_timeout => 2) {|http| http.request(request)}
+  		  if (response.kind_of? Net::HTTPOK)
+  		    @hostType = "AWS"
+  		    return @hostType
+  		  end
+      rescue Exception => e
+      end
+      
+      begin
+  		  #Check for Azure environment
+        url = URI.parse('http://169.254.169.254/metadata/v1/maintenance') # Azure metadata url
+        request = Net::HTTP::Get.new(url.path)
+        response = Net::HTTP.start(url.host, url.port, :read_timeout => 2) {|http| http.request(request)}
+        if (response.kind_of? Net::HTTPOK)
+          @hostType = "AZURE"
+          return @hostType
+        end
+		  rescue Exception => e
+		  end
+		  
+      begin
+        # Check for Heroku env. In the backgroud it is using AWS EC2, hence sending as AWS
+        if (ENV.has_key?('DYNO') || ENV.has_key?('STACK'))
+          @hostType = "AWS"
+          return @hostType
+        end
+      rescue Exception => e
+      end
+		  
+      @hostType = nil
+		end
+		
 		def getAgentInfo
 			data =  Hash.new
 			agentdata = Hash.new
-			agentdata = {"application.type"=>"RUBY","application.name"=>@appname,"hostname"=>@agenthost,"port"=>@agentport,"agent.version"=>"1.0"}
+			agentdata = {"application.type"=>"RUBY","application.name"=>@appname,"hostname"=>@agenthost,"port"=>@agentport,"agent.version"=>ManageEngine::APMInsight::MAJOR_VERSION}
+		  if (@hostType != nil)
+		    agentdata["host.type"]=@hostType
+		  end
 			data["agent_info"]=agentdata
 			data["environment"]=getEnvData
+			data["custom_config_info"]=getAgentConfigData
 			data
 		end
 
 		def getEnvData
 			env =  Hash.new
 			begin
-			ENV.to_hash.each do |key, value|
-				env[key] = value
-			end
-			env["Application Path"]="#{Rails.root}"
+			  env["OS"] = Gem::Platform.local.os
+			  env["OS Version"] = Gem::Platform.local.version
+			  env["OS Arch"] = Gem::Platform.local.cpu
+			  env["Ruby Version"] = "#{RUBY_VERSION}"
+        gemSpecs = Gem.loaded_specs[@obj.constants.s247_apm_gem]
+        if (gemSpecs == nil)
+          gemSpecs = Gem.loaded_specs[@obj.constants.apm_gem]
+        end
+        if (gemSpecs != nil)
+          env["Agent Installed Path"] = gemSpecs.full_gem_path
+        end
+			  
+#			ENV.to_hash.each do |key, value|
+#				env[key] = value
+#			end
+			env["Application Path"] = "#{Dir.pwd}"
 			rescue Exception=>e
+			  @obj.log.warn "Error in capturing env data. #{e.message}"
 			end
 			env
 		end
 			
+		def getAgentConfigData
+			agentconfig = Hash.new
+			agentconfig["last.modified.time"]=@lastupdatedtime*1000
+			agentconfig["apdex.threshold"]=@apdex_t
+			agentconfig["sql.capture.enabled"]=0
+			if @sql_capture
+				agentconfig["sql.capture.enabled"]=1
+			end
+			agentconfig["transaction.trace.enabled"]=0
+			if @trans_trace
+				agentconfig["transaction.trace.enabled"]=1
+			end
+			agentconfig["transaction.trace.threshold"]=@trans_trace_t
+			agentconfig["transaction.trace.sql.parametrize"]=0
+			if @sql_capture_params
+				agentconfig["transaction.trace.sql.parametrize"]=1
+			end
+			agentconfig["transaction.trace.sql.stacktrace.threshold"]=@sql_trace_t
+			agentconfig["transaction.tracking.request.interval"]=1
+			agentconfig
+		end
 
 		def getDispatcher
-		dispatcher = "unknown"
-		if defined?(PhusionPassenger) then
-			dispatcher = "passenger"
-		end
-		dispatcher
+		  
+		  @obj.log.info "Server: #{ENV['SERVER_SOFTWARE']}"
+		  
+  		dispatcher = "unknown"
+  		if defined?(PhusionPassenger) then
+  			dispatcher = "passenger"
+  		end
+  		if defined?(Unicorn) then
+  		  dispatcher = "unicorn"
+  		end
+  		if defined?(Rainbows) then
+  		  dispatcher = "rainbows"
+  		end
+  		dispatcher
 		end
 		
 		def isInteger default,value
@@ -184,6 +387,46 @@ module ManageEngine
 			end
 		end
 
+		def update_config configInfo
+			filepath = @obj.constants.apm_conf
+			f = "apminsight.conf.new"
+			begin
+				propsFile=File.open(filepath, 'r')
+				file = File.new(f,"w+")
+	      			propsFile.read.each_line do |line|
+        			line.strip!
+        			if (line[0] != ?# and line[0] != ?=)
+          				i = line.index('=')
+        	  			if (i)
+						key1 = line[0..i - 1].strip
+						if configInfo.has_key?(key1)
+							file.puts "#{key1}=#{configInfo[key1]}\n"
+						else
+							file.puts "#{line}\n"
+						end
+	          			else
+						file.puts "#{line}\n"
+         				end
+				else
+					file.puts "#{line}\n"
+        			end
+      	  		end
+			rescue Exception=>e
+				@obj.log.info "Problem in Reading / Writing Property File :  #{e.message} "
+				@obj.log.error "#{e.backtrace}"
+			ensure
+					propsFile.close
+					file.close
+			end
+			res = @obj.util.copyFiles f, filepath
+			if res
+				@obj.log.info "copyFiles result = #{res}"
+				#delete apminsight.conf.new has to be done
+			end
+			configureFile
+			assignConfig
+		end
+
 
 	end#c
 end#m

data/lib/agent/handler/custom_api_handler.rb

@@ -0,0 +1,40 @@
+require 'agent/handler/sequence_book'
+
+module APMInsight
+  module API
+    class CustomAPIHandler
+      
+      ## Create tracker for custom instrumented methods and send them to tracker handler
+      def self.invokeTracker name
+        begin
+#          @obj = ManageEngine::APMObjectHolder.instance
+                  
+          if Thread.current[:apminsight] != nil
+            tracker = ManageEngine::Tracker::DefaultTracker.new(name)
+            tracker = ManageEngine::Agent::TrackerHandler.invokeTracker(tracker)
+            return tracker
+          end
+          
+          return nil
+        rescue Exception=>e
+          return nil
+        end
+      end
+      
+      def self.exitTracker tracker
+        if tracker != nil
+          tracker.finish
+          ManageEngine::Agent::TrackerHandler.exitTracker(tracker)
+        end
+      end
+      
+      def self.track_exception exception
+        seqBook = Thread.current[:apminsight]
+        if seqBook != nil
+          seqBook.addExceptionInfo exception
+        end
+      end
+      
+    end #class CustomAPIhandler
+  end #module API
+end

data/lib/agent/handler/sequence_book.rb

@@ -0,0 +1,125 @@
+require 'agent/metrics/exception_record'
+
+module APMInsight
+  module Agent
+    class SequenceBook
+      attr_reader :openTracker, :closedTracker, :rootTracker, :trackerCount, :closedTrackerCount, :exceptionBag, :listenFlag
+      
+      def initialize
+        @rootTracker = createDummyTracker()
+        @closedTracker = @rootTracker
+        @openTracker = nil
+        
+        @trackerCount = 0
+        @closedTrackerCount = 0
+        @listenFlag = -1
+#        @exceptionBag = Array.new
+      end
+      
+      def attachTracker tracker
+        if tracker == nil
+          return nil
+        end
+        
+        # If RootTracker is not set, check type and set
+        if @rootTracker == @closedTracker
+          if !tracker.is_a?(ManageEngine::Tracker::RootTracker)
+            closeSequence()
+            return nil
+          end
+          @rootTracker = tracker
+          
+          updateListenFlag()
+        end
+        
+        
+        # Attach tracker as Sibling or Child and set nominee
+        if @closedTracker != nil
+          tracker.sibling = @closedTracker
+          @closedTracker.sibling = tracker # Nominee - if dropped/corrupted, defaults to this tracker
+          @openTracker = tracker
+          @closedTracker = nil
+        else
+          if tracker.equal?(@openTracker)
+            return nil
+          end
+          
+          @openTracker.child = tracker
+          tracker.sibling = @openTracker
+          @openTracker = tracker
+        end
+        
+        checkAndArrestSequence()
+        
+        return tracker
+      end
+      
+      def closeTracker tracker
+        @closedTrackerCount += 1
+        @closedTracker = tracker
+        tracker.sibling = nil
+        @openTracker = nil
+        
+        # Marks end of transaction
+        if @rootTracker == tracker
+          if @listenFlag < 1 || (@listenFlag >= 1 && @trackerCount > 1)
+            # if some trackers are not closed, while processing the metrics, it may go into infinite loop
+            if (@closedTrackerCount - @trackerCount) == 0
+              sequenceBag = Hash.new
+              sequenceBag["roottracker"] = @rootTracker
+              sequenceBag["exceptions"] = @exceptionBag
+              
+              ManageEngine::APMObjectHolder.instance.collector.updateTransaction(@rootTracker.url, sequenceBag)
+            else
+              ManageEngine::APMObjectHolder.instance.log.warn "Some trackers are not closed, dropping the metrics for #{@rootTracker.url}"
+            end
+          end
+          closeSequence()
+        end
+      end
+      
+      def closeSequence
+        @rootTracker = nil
+        @openTracker = @closedTracker = nil
+        @trackerCount = 0
+        @closedTrackerCount = 0
+        Thread.current[:apminsight] = nil
+      end
+      
+      def addExceptionInfo(exception)
+        begin
+          if @exceptionBag == nil
+            @exceptionBag = Set.new
+          end
+          if (@exceptionBag.size() < 10)
+            exceptionRecord = ::APMInsight::Errors::ExceptionRecord.new(exception)
+            @exceptionBag.add(exceptionRecord)
+          end
+        rescue Exception=>e
+        end
+      end
+      
+      def updateListenFlag
+        if !ManageEngine::APMObjectHolder.instance.txn_util.listen?(@rootTracker.url())
+          @listenFlag = 1
+        end
+        
+        ## Check for sampling factor & for bg txn chk if enabled
+      end
+      
+      def checkAndArrestSequence
+        @trackerCount += 1
+        if @trackerCount == 1000
+          @listenFlag = 1
+        end
+        
+        ## Can check for timeout
+      end
+      
+      def createDummyTracker
+        return ManageEngine::Tracker::DefaultTracker.new("dummy")
+      end
+      
+    end
+  end
+end

data/lib/agent/handler/tracker_handler.rb

@@ -0,0 +1,58 @@
+require 'agent/handler/sequence_book'
+
+module ManageEngine
+  module Agent
+    class TrackerHandler
+      
+      def self.invokeTracker tracker
+        begin
+          @obj = ManageEngine::APMObjectHolder.instance
+          
+          if !@obj.config.agent_enabled || tracker == nil
+            return nil
+          end
+          
+          seqBook = Thread.current[:apminsight]
+          if seqBook != nil
+            if seqBook.listenFlag == 1
+              return nil
+            end
+          else
+            seqBook = ::APMInsight::Agent::SequenceBook.new
+            Thread.current[:apminsight] = seqBook
+          end
+          
+          tracker = seqBook.attachTracker(tracker)
+          
+          return tracker
+        rescue Exception=>ex
+          # Logging to be done here, Not sure whether its safe to do
+          if (@obj != nil)
+            @obj.log.logException "[TrackerHandler] Exception occurred at invoketracker.", ex
+          end
+          return nil
+        end
+      end
+      
+      
+      # Closes tracker properly and set everything ready to process next tracker
+      # If roottracker closes, sequence book is cleaned and data are push to store
+      def self.exitTracker tracker
+        begin
+          if tracker != nil
+            seqBook = Thread.current[:apminsight]
+            if seqBook != nil
+              seqBook.closeTracker tracker
+            end
+          end
+        rescue Exception=>ex
+          if (@obj != nil)
+            @obj.log.logException "[TrackerHandler] Exception occurred at exittracker.", ex
+          end
+        end
+      end
+      
+    end # Class TrackerHandler
+    
+  end # module Agent
+end