How to use

registerConnection

method

in

org.jboss.ejb.client.EJBClientContext

@Override
public void associate(final Connection connection) {
  this.ejbClientContext.registerConnection(connection);
}

  private EJBClientContext getContext(final Connection connection) {
    synchronized (contextByConnection) {
      EJBClientContext ret = contextByConnection.get(connection);
      if(ret == null) {
        ret = EJBClientContext.create();
        ret.registerConnection(connection);
        contextByConnection.put(connection, ret);
        connection.addCloseHandler(new CloseHandler<Connection>() {
          @Override
          public void handleClose(final Connection connection, final IOException e) {
            synchronized (contextByConnection) {
              contextByConnection.remove(connection);
            }
          }
        });
      }
      return ret;
    }
  }
}

private synchronized void createConnection() {
  try {
    endpoint = Remoting.createEndpoint("endpoint", OptionMap.EMPTY);
    endpoint.addConnectionProvider("remote", new RemoteConnectionProviderFactory(), OptionMap.create(Options.SSL_ENABLED, Boolean.FALSE));
    // open a connection
    final IoFuture<Connection> futureConnection = endpoint.connect(new URI(hostUrl), OptionMap.create(Options.SASL_POLICY_NOANONYMOUS, Boolean.FALSE, Options.SASL_POLICY_NOPLAINTEXT, Boolean.FALSE), callbackHandler);
    connection = IoFutureHelper.get(futureConnection, 30L, TimeUnit.SECONDS);
    final EJBClientContext ejbClientContext = EJBClientContext.create();
    ejbClientContext.registerConnection(connection);
    this.clientContext = ejbClientContext;
  } catch (IOException e) {
    throw new RuntimeException(e);
  } catch (URISyntaxException e) {
    throw new RuntimeException(e);
  }
}