Examples with OperationsFactory org.infinispan.client.hotrod.impl.operations.OperationsFactory used on opensource projects

Example 1 with OperationsFactory

use of org.infinispan.client.hotrod.impl.operations.OperationsFactory in project keycloak by keycloak.

the class JDGPutTest method getIspnSegmentsCount.

protected static int getIspnSegmentsCount(RemoteCache remoteCache) {
    OperationsFactory operationsFactory = ((RemoteCacheImpl) remoteCache).getOperationsFactory();
    Map<SocketAddress, Set<Integer>> segmentsByAddress = operationsFactory.getPrimarySegmentsByAddress();
    for (Map.Entry<SocketAddress, Set<Integer>> entry : segmentsByAddress.entrySet()) {
        SocketAddress targetAddress = entry.getKey();
        // Same like RemoteCloseableIterator.startInternal
        IterationStartOperation iterationStartOperation = operationsFactory.newIterationStartOperation(null, null, null, 64, false, null, targetAddress);
        IterationStartResponse startResponse = await(iterationStartOperation.execute());
        try {
            // Could happen for non-clustered caches
            if (startResponse.getSegmentConsistentHash() == null) {
                return -1;
            } else {
                return startResponse.getSegmentConsistentHash().getNumSegments();
            }
        } finally {
            startResponse.getChannel().close();
        }
    }
    // Handle the case when primary segments owned by the address are not known
    return -1;
}

Example 2 with OperationsFactory

use of org.infinispan.client.hotrod.impl.operations.OperationsFactory in project keycloak by keycloak.

the class RemoteCacheSessionsLoader method getIspnSegmentsCount.

protected int getIspnSegmentsCount(RemoteCache remoteCache) {
    OperationsFactory operationsFactory = ((RemoteCacheImpl) remoteCache).getOperationsFactory();
    Map<SocketAddress, Set<Integer>> segmentsByAddress = operationsFactory.getPrimarySegmentsByAddress();
    for (Map.Entry<SocketAddress, Set<Integer>> entry : segmentsByAddress.entrySet()) {
        SocketAddress targetAddress = entry.getKey();
        // Same like RemoteCloseableIterator.startInternal
        IterationStartOperation iterationStartOperation = operationsFactory.newIterationStartOperation(null, null, null, sessionsPerSegment, false, null, targetAddress);
        IterationStartResponse startResponse = await(iterationStartOperation.execute());
        try {
            // Could happen for non-clustered caches
            if (startResponse.getSegmentConsistentHash() == null) {
                return -1;
            } else {
                return startResponse.getSegmentConsistentHash().getNumSegments();
            }
        } finally {
            startResponse.getChannel().close();
        }
    }
    // Handle the case when primary segments owned by the address are not known
    return -1;
}

Example 3 with OperationsFactory

use of org.infinispan.client.hotrod.impl.operations.OperationsFactory in project wildfly by wildfly.

the class HotRodStore method setRemoteCache.

synchronized void setRemoteCache(RemoteCacheContainer container, String cacheName) {
    this.cache = container.getCache(cacheName);
    if (this.cache == null) {
        // Administration support was introduced in protocol version 2.7
        throw InfinispanLogger.ROOT_LOGGER.remoteCacheMustBeDefined(container.getConfiguration().version().toString(), cacheName);
    }
    RemoteCacheManager manager = this.cache.getRemoteCacheManager();
    this.operationsFactory = new OperationsFactory(manager.getChannelFactory(), manager.getCodec(), null, manager.getConfiguration());
    this.cache.start();
}