chore: cleanup and fixes

Author: surajkumar

annotations/src/main/java/com/javadiscord/jdi/core/processor/ClassFileUtil.java

@@ -25,17 +25,6 @@ public static List<File> getClassesInClassPath() {
         String[] classpathEntries = classpath.split(File.pathSeparator);
 
         for (String entry : classpathEntries) {
-            if (
-                entry.contains("io.netty")
-                    || entry.contains("org.apache")
-                    || entry.contains("io.vertx")
-                    || entry.contains("com.fasterxml")
-                    || entry.contains("org.javassist")
-                    || entry.contains("com.github.mizosoft.methanol")
-            ) {
-                continue;
-            }
-
             File file = new File(entry);
             try {
                 classesInPath.addAll(getClasses(file));

gateway/src/main/java/com/javadiscord/jdi/internal/gateway/WebSocketHandler.java

@@ -28,18 +28,19 @@ public class WebSocketHandler implements Handler<WebSocket> {
     private static final Map<Integer, GatewayOperationHandler> OPERATION_HANDLER = new HashMap<>();
     private final ConnectionMediator connectionMediator;
     private final Cache cache;
+    private final HeartbeatService heartbeatService;
 
     public WebSocketHandler(
         ConnectionMediator connectionMediator,
-        Cache cache
+        Cache cache, HeartbeatService heartbeatService
     ) {
         this.connectionMediator = connectionMediator;
         this.cache = cache;
+        this.heartbeatService = heartbeatService;
         registerHandlers();
     }
 
     private void registerHandlers() {
-        HeartbeatService heartbeatService = new HeartbeatService(connectionMediator);
         OPERATION_HANDLER.put(GatewayOpcode.HELLO, new HelloOperationHandler(heartbeatService));
         OPERATION_HANDLER.put(
             GatewayOpcode.HEARTBEAT_ACK, new HeartbeatAckOperationHandler(heartbeatService)

gateway/src/main/java/com/javadiscord/jdi/internal/gateway/WebSocketManager.java

@@ -1,6 +1,7 @@
 package com.javadiscord.jdi.internal.gateway;
 
 import com.javadiscord.jdi.internal.cache.Cache;
+import com.javadiscord.jdi.internal.gateway.handlers.heartbeat.HeartbeatService;
 import com.javadiscord.jdi.internal.gateway.identify.IdentifyRequest;
 
 import com.fasterxml.jackson.core.JsonProcessingException;
@@ -23,6 +24,7 @@ public class WebSocketManager {
     private final Cache cache;
     private WebSocket webSocket;
     private WebSocketClient webSocketClient;
+    private HeartbeatService heartbeatService;
     private boolean retryAllowed;
 
     public WebSocketManager(
@@ -36,6 +38,8 @@ public WebSocketManager(
     }
 
     public void start(ConnectionMediator connectionMediator) {
+        heartbeatService = new HeartbeatService(connectionMediator);
+
         String gatewayURL = connectionMediator.getConnectionDetails().getGatewayURL();
 
         WebSocketConnectOptions webSocketConnectOptions =
@@ -61,7 +65,7 @@ public void start(ConnectionMediator connectionMediator) {
                     this.webSocket = webSocket;
 
                     WebSocketHandler webSocketHandler =
-                        new WebSocketHandler(connectionMediator, cache);
+                        new WebSocketHandler(connectionMediator, cache, heartbeatService);
 
                     webSocketHandler.handle(webSocket);
 
@@ -101,6 +105,9 @@ public void stop() {
         if (webSocket != null && !webSocket.isClosed()) {
             webSocket.close();
         }
+        if (heartbeatService != null) {
+            heartbeatService.stop();
+        }
         webSocketClient.close()
             .onSuccess(res -> LOGGER.info("Web socket client has been shutdown"))
             .onFailure(err -> LOGGER.error("Failed to shutdown web socket client", err));

gateway/src/main/java/com/javadiscord/jdi/internal/gateway/handlers/heartbeat/HeartbeatService.java

@@ -55,6 +55,10 @@ private void checkHeartbeatAckReceived(WebSocket webSocket) {
         }
     }
 
+    public void stop() {
+        EXECUTOR_SERVICE.shutdown();
+    }
+
     public void sendHeartbeat(WebSocket webSocket) {
         webSocket.write(
             Buffer.buffer()