Examples with WebsocketClientEndpoint com.github.jnidzwetzki.bitfinex.v2.WebsocketClientEndpoint used on opensource projects

Example 1 with WebsocketClientEndpoint

use of com.github.jnidzwetzki.bitfinex.v2.WebsocketClientEndpoint in project bitfinex-v2-wss-api-java by jnidzwetzki.

the class HeartbeatManagerTest method testConnectWhenDisconnected.

@Test(timeout = 30000)
public void testConnectWhenDisconnected() throws Exception {
    // Events
    // * Disconnect websocket from heartbeat
    // * Disconnect from bitfinex API reconnect method
    // * Reconnect on bitfinex api
    final CountDownLatch connectLatch = new CountDownLatch(3);
    // Count down the latch on method call
    final Answer<Void> answer = new Answer<Void>() {

        public Void answer(final InvocationOnMock invocation) throws Throwable {
            connectLatch.countDown();
            return null;
        }
    };
    final BitfinexApiBroker bitfinexApiBroker = Mockito.mock(BitfinexApiBroker.class);
    final HeartbeatThread heartbeatThreadRunnable = new HeartbeatThread(bitfinexApiBroker);
    final WebsocketClientEndpoint websocketClientEndpoint = Mockito.mock(WebsocketClientEndpoint.class);
    Mockito.when(websocketClientEndpoint.isConnected()).thenReturn(connectLatch.getCount() == 0);
    Mockito.when(bitfinexApiBroker.getWebsocketEndpoint()).thenReturn(websocketClientEndpoint);
    Mockito.doAnswer(answer).when(bitfinexApiBroker).reconnect();
    Mockito.doAnswer(answer).when(websocketClientEndpoint).close();
    final Thread heartbeatThread = new Thread(heartbeatThreadRunnable);
    try {
        heartbeatThread.start();
        connectLatch.await();
    } catch (Exception e) {
        // Should not happen
        throw e;
    } finally {
        heartbeatThread.interrupt();
    }
}

Example 2 with WebsocketClientEndpoint

use of com.github.jnidzwetzki.bitfinex.v2.WebsocketClientEndpoint in project bitfinex-v2-wss-api-java by jnidzwetzki.

the class BitfinexApiBroker method connect.

/**
 * Open the connection
 * @throws APIException
 */
public void connect() throws APIException {
    try {
        final URI bitfinexURI = new URI(BITFINEX_URI);
        websocketEndpoint = new WebsocketClientEndpoint(bitfinexURI);
        websocketEndpoint.addConsumer(apiCallback);
        websocketEndpoint.connect();
        updateConnectionHeartbeat();
        executeAuthentification();
        heartbeatThread = new Thread(new HeartbeatThread(this));
        heartbeatThread.start();
    } catch (Exception e) {
        throw new APIException(e);
    }
}