Examples with Address org.infinispan.remoting.transport.Address used on opensource projects

Example 6 with Address

use of org.infinispan.remoting.transport.Address in project wildfly by wildfly.

the class CacheRegistry method topologyChanged.

@TopologyChanged
public void topologyChanged(TopologyChangedEvent<Node, Map.Entry<K, V>> event) {
    if (event.isPre())
        return;
    ConsistentHash previousHash = event.getConsistentHashAtStart();
    List<Address> previousMembers = previousHash.getMembers();
    ConsistentHash hash = event.getConsistentHashAtEnd();
    List<Address> members = hash.getMembers();
    Address localAddress = event.getCache().getCacheManager().getAddress();
    // Determine which nodes have left the cache view
    Set<Address> addresses = new HashSet<>(previousMembers);
    addresses.removeAll(members);
    try {
        this.topologyChangeExecutor.submit(() -> {
            if (!addresses.isEmpty()) {
                // We're only interested in the entries for which we are the primary owner
                List<Node> nodes = addresses.stream().filter(address -> hash.locatePrimaryOwner(address).equals(localAddress)).map(address -> this.factory.createNode(address)).collect(Collectors.toList());
                if (!nodes.isEmpty()) {
                    Cache<Node, Map.Entry<K, V>> cache = this.cache.getAdvancedCache().withFlags(Flag.FORCE_SYNCHRONOUS);
                    Map<K, V> removed = new HashMap<>();
                    try (Batch batch = this.batcher.createBatch()) {
                        for (Node node : nodes) {
                            Map.Entry<K, V> old = cache.remove(node);
                            if (old != null) {
                                removed.put(old.getKey(), old.getValue());
                            }
                        }
                    } catch (CacheException e) {
                        ClusteringServerLogger.ROOT_LOGGER.registryPurgeFailed(e, this.cache.getCacheManager().toString(), this.cache.getName(), nodes);
                    }
                    // Invoke listeners outside above tx context
                    if (!removed.isEmpty()) {
                        this.notifyListeners(Event.Type.CACHE_ENTRY_REMOVED, removed);
                    }
                }
            } else {
                // This is a merge after cluster split: re-populate the cache registry with lost registry entries
                if (!previousMembers.contains(localAddress)) {
                    // If this node is not a member at merge start, its mapping is lost and needs to be recreated and listeners notified
                    try {
                        this.populateRegistry();
                        // Local cache events do not trigger notifications
                        this.notifyListeners(Event.Type.CACHE_ENTRY_CREATED, this.entry);
                    } catch (CacheException e) {
                        ClusteringServerLogger.ROOT_LOGGER.failedToRestoreLocalRegistryEntry(e, this.cache.getCacheManager().toString(), this.cache.getName());
                    }
                }
            }
        });
    } catch (RejectedExecutionException e) {
    // Executor was shutdown
    }
}

Example 7 with Address

use of org.infinispan.remoting.transport.Address in project wildfly by wildfly.

the class AffinityIdentityFactoryTestCase method createIdentifier.

@Test
public void createIdentifier() {
    String expected = "id";
    Address address = mock(Address.class);
    when(this.manager.getAddress()).thenReturn(address);
    when(this.affinity.getKeyForAddress(address)).thenReturn(new Key<>(expected));
    String result = this.subject.createIdentifier();
    assertSame(expected, result);
}

Example 8 with Address

use of org.infinispan.remoting.transport.Address in project hibernate-orm by hibernate.

the class NonTxPutFromLoadInterceptor method endInvalidating.

public void endInvalidating(Object key, Object lockOwner, boolean successful) {
    assert lockOwner != null;
    if (!putFromLoadValidator.endInvalidatingKey(lockOwner, key, successful)) {
        log.failedEndInvalidating(key, cacheName);
    }
    EndInvalidationCommand endInvalidationCommand = commandInitializer.buildEndInvalidationCommand(cacheName, new Object[] { key }, lockOwner);
    List<Address> members = stateTransferManager.getCacheTopology().getMembers();
    rpcManager.invokeRemotely(members, endInvalidationCommand, asyncUnordered);
}

Example 9 with Address

use of org.infinispan.remoting.transport.Address in project hibernate-orm by hibernate.

the class TxPutFromLoadInterceptor method endInvalidationAndInvokeNextInterceptor.

protected Object endInvalidationAndInvokeNextInterceptor(TxInvocationContext<?> ctx, VisitableCommand command) throws Throwable {
    try {
        if (ctx.isOriginLocal()) {
            // During evictAll inside transaction this would cause unnecessary invalidate command
            if (!ctx.getModifications().isEmpty()) {
                Object[] keys = ctx.getModifications().stream().flatMap(mod -> mod.getAffectedKeys().stream()).distinct().toArray();
                if (log.isTraceEnabled()) {
                    log.tracef("Sending end invalidation for keys %s asynchronously, modifications are %s", Arrays.toString(keys), ctx.getCacheTransaction().getModifications());
                }
                GlobalTransaction globalTransaction = ctx.getGlobalTransaction();
                EndInvalidationCommand commitCommand = cacheCommandInitializer.buildEndInvalidationCommand(cacheName, keys, globalTransaction);
                List<Address> members = stateTransferManager.getCacheTopology().getMembers();
                rpcManager.invokeRemotely(members, commitCommand, asyncUnordered);
                // we have to end invalidation from here manually (in successful case as well)
                for (Object key : keys) {
                    putFromLoadValidator.endInvalidatingKey(globalTransaction, key);
                }
            }
        }
    } finally {
        return invokeNextInterceptor(ctx, command);
    }
}

Example 10 with Address

use of org.infinispan.remoting.transport.Address in project hibernate-orm by hibernate.

the class UnorderedDistributionInterceptor method visitPutKeyValueCommand.

@Override
public Object visitPutKeyValueCommand(InvocationContext ctx, PutKeyValueCommand command) throws Throwable {
    if (command.hasFlag(Flag.CACHE_MODE_LOCAL)) {
        // for state-transfer related writes
        return invokeNextInterceptor(ctx, command);
    }
    int commandTopologyId = command.getTopologyId();
    int currentTopologyId = stateTransferManager.getCacheTopology().getTopologyId();
    if (commandTopologyId != -1 && currentTopologyId != commandTopologyId) {
        throw new OutdatedTopologyException("Cache topology changed while the command was executing: expected " + commandTopologyId + ", got " + currentTopologyId);
    }
    ConsistentHash writeCH = distributionManager.getWriteConsistentHash();
    List<Address> owners = null;
    if (writeCH.isReplicated()) {
        // local result is always ignored
        invokeNextInterceptor(ctx, command);
    } else {
        owners = writeCH.locateOwners(command.getKey());
        if (owners.contains(rpcManager.getAddress())) {
            invokeNextInterceptor(ctx, command);
        } else {
            log.tracef("Not invoking %s on %s since it is not an owner", command, rpcManager.getAddress());
        }
    }
    if (ctx.isOriginLocal() && command.isSuccessful()) {
        // unlocking (and committing) the entry.
        return rpcManager.invokeRemotelyAsync(owners, command, isSynchronous(command) ? syncRpcOptions : asyncRpcOptions);
    }
    return null;
}