fix for mismatch values of eventStats

Author: IshikaDawda

newrelic-security-agent/src/main/java/com/newrelic/agent/security/instrumentator/dispatcher/DispatcherPool.java

@@ -239,6 +239,7 @@ public void dispatchExitEvent(ExitEventBean exitEventBean) {
         securityMetaData.addCustomAttribute(NR_APM_TRACE_ID, traceMetadata.getTraceId());
         securityMetaData.addCustomAttribute(NR_APM_SPAN_ID, traceMetadata.getSpanId());
         this.executor.submit(new Dispatcher(exitEventBean));
+        AgentInfo.getInstance().getJaHealthCheck().getEventStats().getDispatcher().incrementSubmitted();
         AgentInfo.getInstance().getJaHealthCheck().getEventStats().getExitEvents().incrementSubmitted();
     }
 

newrelic-security-agent/src/main/java/com/newrelic/agent/security/intcodeagent/websocket/EventSendPool.java

@@ -46,7 +46,7 @@ private EventSendPool() {
             protected void afterExecute(Runnable r, Throwable t) {
                 try {
                     if (r instanceof CustomFutureTask<?> && ((CustomFutureTask<?>) r).getTask() instanceof EventSender) {
-                        EventSender task = (EventSender) ((CustomFutureTask<?>) r).get();
+                        EventSender task = (EventSender) ((CustomFutureTask<?>) r).getTask();
                         if(task.getEvent() instanceof JavaAgentEventBean){
                             if (t != null) {
                                 AgentInfo.getInstance().getJaHealthCheck().getEventStats().getEventSender().incrementError();

newrelic-security-agent/src/main/java/com/newrelic/agent/security/intcodeagent/websocket/WSClient.java

@@ -262,7 +262,7 @@ private static void cleanIASTState() {
     public void onMessage(String message) {
         // Receive communication from IC side.
         try {
-            AgentInfo.getInstance().getJaHealthCheck().getWebSocketConnectionStats().incrementMessagesSent();
+            AgentInfo.getInstance().getJaHealthCheck().getWebSocketConnectionStats().incrementMessagesReceived();
             if (logger.isLogLevelEnabled(LogLevel.FINEST)) {
                 logger.log(LogLevel.FINEST, String.format(INCOMING_CONTROL_COMMAND_S, message),
                         this.getClass().getName());