Use ObjectReader instance for processing JSON.

Author: joaopsilva

src/main/java/com/binance/api/client/impl/BinanceApiWebSocketListener.java

@@ -1,13 +1,15 @@
 package com.binance.api.client.impl;
 
-import java.io.IOException;
-import okhttp3.Response;
-import okhttp3.WebSocket;
-import okhttp3.WebSocketListener;
 import com.binance.api.client.BinanceApiCallback;
 import com.binance.api.client.exception.BinanceApiException;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.ObjectReader;
+import okhttp3.Response;
+import okhttp3.WebSocket;
+import okhttp3.WebSocketListener;
+
+import java.io.IOException;
 
 /**
  * Binance API WebSocket listener.
@@ -18,31 +20,24 @@ public class BinanceApiWebSocketListener<T> extends WebSocketListener {
 
   private static final ObjectMapper mapper = new ObjectMapper();
 
-  private Class<T> eventClass;
-
-  private TypeReference<T> eventTypeReference;
+  private final ObjectReader objectReader;
 
   private boolean closing = false;
 
   public BinanceApiWebSocketListener(BinanceApiCallback<T> callback, Class<T> eventClass) {
     this.callback = callback;
-    this.eventClass = eventClass;
+    this.objectReader = mapper.readerFor(eventClass);
   }
 
   public BinanceApiWebSocketListener(BinanceApiCallback<T> callback, TypeReference<T> eventTypeReference) {
     this.callback = callback;
-    this.eventTypeReference = eventTypeReference;
+    this.objectReader = mapper.readerFor(eventTypeReference);
   }
 
   @Override
   public void onMessage(WebSocket webSocket, String text) {
     try {
-      T event = null;
-      if (eventClass == null) {
-        event = mapper.readValue(text, eventTypeReference);
-      } else {
-        event = mapper.readValue(text, eventClass);
-      }
+      T event = objectReader.readValue(text);
       callback.onResponse(event);
     } catch (IOException e) {
       throw new BinanceApiException(e);