Examples with TokenFactory com.datastax.oss.driver.internal.core.metadata.token.TokenFactory used on opensource projects

Example 1 with TokenFactory

use of com.datastax.oss.driver.internal.core.metadata.token.TokenFactory in project java-driver by datastax.

the class PreparedStatementIT method should_infer_routing_information_when_partition_key_is_bound.

private void should_infer_routing_information_when_partition_key_is_bound(String queryString) {
    CqlSession session = sessionRule.session();
    TokenFactory tokenFactory = ((DefaultTokenMap) session.getMetadata().getTokenMap().orElseThrow(AssertionError::new)).getTokenFactory();
    // We'll bind a=1 in the query, check what token this is supposed to produce
    Token expectedToken = session.execute("SELECT token(a) FROM prepared_statement_test WHERE a = 1").one().getToken(0);
    BoundStatement boundStatement = session.prepare(queryString).bind().setInt("a", 1);
    assertThat(boundStatement.getRoutingKeyspace()).isEqualTo(sessionRule.keyspace());
    assertThat(tokenFactory.hash(boundStatement.getRoutingKey())).isEqualTo(expectedToken);
}

Example 2 with TokenFactory

use of com.datastax.oss.driver.internal.core.metadata.token.TokenFactory in project java-driver by datastax.

the class FullNodeListRefresh method compute.

@Override
public Result compute(DefaultMetadata oldMetadata, boolean tokenMapEnabled, InternalDriverContext context) {
    String logPrefix = context.getSessionName();
    TokenFactoryRegistry tokenFactoryRegistry = context.getTokenFactoryRegistry();
    Map<UUID, Node> oldNodes = oldMetadata.getNodes();
    Map<UUID, Node> added = new HashMap<>();
    Set<UUID> seen = new HashSet<>();
    TokenFactory tokenFactory = oldMetadata.getTokenMap().map(m -> ((DefaultTokenMap) m).getTokenFactory()).orElse(null);
    boolean tokensChanged = false;
    for (NodeInfo nodeInfo : nodeInfos) {
        UUID id = nodeInfo.getHostId();
        if (seen.contains(id)) {
            LOG.warn("[{}] Found duplicate entries with host_id {} in system.peers, " + "keeping only the first one", logPrefix, id);
        } else {
            seen.add(id);
            DefaultNode node = (DefaultNode) oldNodes.get(id);
            if (node == null) {
                node = new DefaultNode(nodeInfo.getEndPoint(), context);
                LOG.debug("[{}] Adding new node {}", logPrefix, node);
                added.put(id, node);
            }
            if (tokenFactory == null && nodeInfo.getPartitioner() != null) {
                tokenFactory = tokenFactoryRegistry.tokenFactoryFor(nodeInfo.getPartitioner());
            }
            tokensChanged |= copyInfos(nodeInfo, node, context);
        }
    }
    Set<UUID> removed = Sets.difference(oldNodes.keySet(), seen);
    if (added.isEmpty() && removed.isEmpty()) {
        // The list didn't change
        if (!oldMetadata.getTokenMap().isPresent() && tokenFactory != null) {
            // token map rebuild:
            return new Result(oldMetadata.withNodes(oldMetadata.getNodes(), tokenMapEnabled, true, tokenFactory, context));
        } else {
            // No need to create a new metadata instance
            return new Result(oldMetadata);
        }
    } else {
        ImmutableMap.Builder<UUID, Node> newNodesBuilder = ImmutableMap.builder();
        ImmutableList.Builder<Object> eventsBuilder = ImmutableList.builder();
        newNodesBuilder.putAll(added);
        for (Map.Entry<UUID, Node> entry : oldNodes.entrySet()) {
            if (!removed.contains(entry.getKey())) {
                newNodesBuilder.put(entry.getKey(), entry.getValue());
            }
        }
        for (Node node : added.values()) {
            eventsBuilder.add(NodeStateEvent.added((DefaultNode) node));
        }
        for (UUID id : removed) {
            Node node = oldNodes.get(id);
            eventsBuilder.add(NodeStateEvent.removed((DefaultNode) node));
        }
        return new Result(oldMetadata.withNodes(newNodesBuilder.build(), tokenMapEnabled, tokensChanged, tokenFactory, context), eventsBuilder.build());
    }
}

Example 3 with TokenFactory

use of com.datastax.oss.driver.internal.core.metadata.token.TokenFactory in project java-driver by datastax.

the class InitialNodeListRefresh method compute.

@Override
public Result compute(DefaultMetadata oldMetadata, boolean tokenMapEnabled, InternalDriverContext context) {
    String logPrefix = context.getSessionName();
    TokenFactoryRegistry tokenFactoryRegistry = context.getTokenFactoryRegistry();
    // metadata is empty.
    assert oldMetadata == DefaultMetadata.EMPTY;
    TokenFactory tokenFactory = null;
    Map<UUID, DefaultNode> newNodes = new HashMap<>();
    for (NodeInfo nodeInfo : nodeInfos) {
        UUID hostId = nodeInfo.getHostId();
        if (newNodes.containsKey(hostId)) {
            LOG.warn("[{}] Found duplicate entries with host_id {} in system.peers, " + "keeping only the first one", logPrefix, hostId);
        } else {
            EndPoint endPoint = nodeInfo.getEndPoint();
            DefaultNode node = findIn(contactPoints, endPoint);
            if (node == null) {
                node = new DefaultNode(endPoint, context);
                LOG.debug("[{}] Adding new node {}", logPrefix, node);
            } else {
                LOG.debug("[{}] Copying contact point {}", logPrefix, node);
            }
            if (tokenMapEnabled && tokenFactory == null && nodeInfo.getPartitioner() != null) {
                tokenFactory = tokenFactoryRegistry.tokenFactoryFor(nodeInfo.getPartitioner());
            }
            copyInfos(nodeInfo, node, context);
            newNodes.put(hostId, node);
        }
    }
    ImmutableList.Builder<Object> eventsBuilder = ImmutableList.builder();
    for (DefaultNode newNode : newNodes.values()) {
        if (findIn(contactPoints, newNode.getEndPoint()) == null) {
            eventsBuilder.add(NodeStateEvent.added(newNode));
        }
    }
    for (DefaultNode contactPoint : contactPoints) {
        if (findIn(newNodes.values(), contactPoint.getEndPoint()) == null) {
            eventsBuilder.add(NodeStateEvent.removed(contactPoint));
        }
    }
    return new Result(oldMetadata.withNodes(ImmutableMap.copyOf(newNodes), tokenMapEnabled, true, tokenFactory, context), eventsBuilder.build());
}