Search in sources :

Example 1 with DefaultRaftClientService

use of org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService in project ignite-3 by apache.

the class NodeImpl method init.

@Override
public boolean init(final NodeOptions opts) {
    Requires.requireNonNull(opts, "Null node options");
    Requires.requireNonNull(opts.getRaftOptions(), "Null raft options");
    Requires.requireNonNull(opts.getServiceFactory(), "Null jraft service factory");
    this.serviceFactory = opts.getServiceFactory();
    this.options = opts;
    this.raftOptions = opts.getRaftOptions();
    this.metrics = new NodeMetrics(opts.isEnableMetrics());
    this.serverId.setPriority(opts.getElectionPriority());
    this.electionTimeoutCounter = 0;
    if (opts.getReplicationStateListeners() != null)
        this.replicatorStateListeners.addAll(opts.getReplicationStateListeners());
    if (this.serverId.getIp().equals(Utils.IP_ANY)) {
        LOG.error("Node can't start from IP_ANY.");
        return false;
    }
    // Init timers.
    initTimers(opts);
    // Init pools.
    initPools(opts);
    this.configManager = new ConfigurationManager();
    applyDisruptor = opts.getNodeApplyDisruptor();
    applyQueue = applyDisruptor.subscribe(groupId, new LogEntryAndClosureHandler());
    if (this.metrics.getMetricRegistry() != null) {
        this.metrics.getMetricRegistry().register("jraft-node-impl-disruptor", new DisruptorMetricSet(this.applyQueue));
    }
    this.fsmCaller = new FSMCallerImpl();
    if (!initLogStorage()) {
        LOG.error("Node {} initLogStorage failed.", getNodeId());
        return false;
    }
    if (!initMetaStorage()) {
        LOG.error("Node {} initMetaStorage failed.", getNodeId());
        return false;
    }
    if (!initFSMCaller(new LogId(0, 0))) {
        LOG.error("Node {} initFSMCaller failed.", getNodeId());
        return false;
    }
    this.ballotBox = new BallotBox();
    final BallotBoxOptions ballotBoxOpts = new BallotBoxOptions();
    ballotBoxOpts.setWaiter(this.fsmCaller);
    ballotBoxOpts.setClosureQueue(this.closureQueue);
    if (!this.ballotBox.init(ballotBoxOpts)) {
        LOG.error("Node {} init ballotBox failed.", getNodeId());
        return false;
    }
    if (!initSnapshotStorage()) {
        LOG.error("Node {} initSnapshotStorage failed.", getNodeId());
        return false;
    }
    final Status st = this.logManager.checkConsistency();
    if (!st.isOk()) {
        LOG.error("Node {} is initialized with inconsistent log, status={}.", getNodeId(), st);
        return false;
    }
    this.conf = new ConfigurationEntry();
    this.conf.setId(new LogId());
    // if have log using conf in log, else using conf in options
    if (this.logManager.getLastLogIndex() > 0) {
        checkAndSetConfiguration(false);
    } else {
        this.conf.setConf(this.options.getInitialConf());
        // initially set to max(priority of all nodes)
        this.targetPriority = getMaxPriorityOfNodes(this.conf.getConf().getPeers());
    }
    if (!this.conf.isEmpty()) {
        Requires.requireTrue(this.conf.isValid(), "Invalid conf: %s", this.conf);
    } else {
        LOG.info("Init node {} with empty conf.", this.serverId);
    }
    this.replicatorGroup = new ReplicatorGroupImpl();
    this.rpcClientService = new DefaultRaftClientService();
    final ReplicatorGroupOptions rgOpts = new ReplicatorGroupOptions();
    rgOpts.setHeartbeatTimeoutMs(heartbeatTimeout(this.options.getElectionTimeoutMs()));
    rgOpts.setElectionTimeoutMs(this.options.getElectionTimeoutMs());
    rgOpts.setLogManager(this.logManager);
    rgOpts.setBallotBox(this.ballotBox);
    rgOpts.setNode(this);
    rgOpts.setRaftRpcClientService(this.rpcClientService);
    rgOpts.setSnapshotStorage(this.snapshotExecutor != null ? this.snapshotExecutor.getSnapshotStorage() : null);
    rgOpts.setRaftOptions(this.raftOptions);
    rgOpts.setTimerManager(this.options.getScheduler());
    // Adds metric registry to RPC service.
    this.options.setMetricRegistry(this.metrics.getMetricRegistry());
    if (!this.rpcClientService.init(this.options)) {
        LOG.error("Fail to init rpc service.");
        return false;
    }
    this.replicatorGroup.init(new NodeId(this.groupId, this.serverId), rgOpts);
    this.readOnlyService = new ReadOnlyServiceImpl();
    final ReadOnlyServiceOptions rosOpts = new ReadOnlyServiceOptions();
    rosOpts.setGroupId(groupId);
    rosOpts.setFsmCaller(this.fsmCaller);
    rosOpts.setNode(this);
    rosOpts.setRaftOptions(this.raftOptions);
    rosOpts.setReadOnlyServiceDisruptor(opts.getReadOnlyServiceDisruptor());
    if (!this.readOnlyService.init(rosOpts)) {
        LOG.error("Fail to init readOnlyService.");
        return false;
    }
    // set state to follower
    this.state = State.STATE_FOLLOWER;
    if (LOG.isInfoEnabled()) {
        LOG.info("Node {} init, term={}, lastLogId={}, conf={}, oldConf={}.", getNodeId(), this.currTerm, this.logManager.getLastLogId(false), this.conf.getConf(), this.conf.getOldConf());
    }
    if (this.snapshotExecutor != null && this.options.getSnapshotIntervalSecs() > 0) {
        LOG.debug("Node {} start snapshot timer, term={}.", getNodeId(), this.currTerm);
        this.snapshotTimer.start();
    }
    if (!this.conf.isEmpty()) {
        stepDown(this.currTerm, false, new Status());
    }
    // Now the raft node is started , have to acquire the writeLock to avoid race
    // conditions
    this.writeLock.lock();
    if (this.conf.isStable() && this.conf.getConf().size() == 1 && this.conf.getConf().contains(this.serverId)) {
        // The group contains only this server which must be the LEADER, trigger
        // the timer immediately.
        electSelf();
    } else {
        this.writeLock.unlock();
    }
    return true;
}
Also used : Status(org.apache.ignite.raft.jraft.Status) DisruptorMetricSet(org.apache.ignite.raft.jraft.util.DisruptorMetricSet) ReplicatorGroupOptions(org.apache.ignite.raft.jraft.option.ReplicatorGroupOptions) BallotBoxOptions(org.apache.ignite.raft.jraft.option.BallotBoxOptions) ReadOnlyServiceOptions(org.apache.ignite.raft.jraft.option.ReadOnlyServiceOptions) NodeId(org.apache.ignite.raft.jraft.entity.NodeId) DefaultRaftClientService(org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService) ConfigurationManager(org.apache.ignite.raft.jraft.conf.ConfigurationManager) LogId(org.apache.ignite.raft.jraft.entity.LogId) ConfigurationEntry(org.apache.ignite.raft.jraft.conf.ConfigurationEntry)

Example 2 with DefaultRaftClientService

use of org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService in project ignite-3 by apache.

the class ItNodeTest method testLeaseReadAfterSegmentation.

/**
 * Tests if a read using leader leases works correctly after previous leader segmentation.
 */
@Test
public void testLeaseReadAfterSegmentation() throws Exception {
    List<PeerId> peers = TestUtils.generatePeers(3);
    cluster = new TestCluster("unittest", dataPath, peers, 3_000, testInfo);
    for (PeerId peer : peers) {
        RaftOptions opts = new RaftOptions();
        // Election timeout divisor.
        opts.setElectionHeartbeatFactor(2);
        opts.setReadOnlyOptions(ReadOnlyOption.ReadOnlyLeaseBased);
        assertTrue(cluster.start(peer.getEndpoint(), false, 300, false, null, opts));
    }
    cluster.waitLeader();
    NodeImpl leader = (NodeImpl) cluster.getLeader();
    assertNotNull(leader);
    cluster.ensureLeader(leader);
    sendTestTaskAndWait(leader);
    cluster.ensureSame();
    DefaultRaftClientService rpcService = (DefaultRaftClientService) leader.getRpcClientService();
    RpcClientEx rpcClientEx = (RpcClientEx) rpcService.getRpcClient();
    AtomicInteger cnt = new AtomicInteger();
    rpcClientEx.blockMessages((msg, nodeId) -> {
        assertTrue(msg instanceof RpcRequests.AppendEntriesRequest);
        if (cnt.get() >= 2)
            return true;
        LOG.info("Send heartbeat: " + msg + " to " + nodeId);
        cnt.incrementAndGet();
        return false;
    });
    assertTrue(waitForCondition(() -> cluster.getLeader() != null && !leader.getNodeId().equals(cluster.getLeader().getNodeId()), 10_000));
    CompletableFuture<Status> res = new CompletableFuture<>();
    cluster.getLeader().readIndex(null, new ReadIndexClosure() {

        @Override
        public void run(Status status, long index, byte[] reqCtx) {
            res.complete(status);
        }
    });
    assertTrue(res.get().isOk());
}
Also used : Status(org.apache.ignite.raft.jraft.Status) RaftOptions(org.apache.ignite.raft.jraft.option.RaftOptions) RpcRequests(org.apache.ignite.raft.jraft.rpc.RpcRequests) CompletableFuture(java.util.concurrent.CompletableFuture) ReadIndexClosure(org.apache.ignite.raft.jraft.closure.ReadIndexClosure) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) DefaultRaftClientService(org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService) RpcClientEx(org.apache.ignite.raft.jraft.rpc.RpcClientEx) PeerId(org.apache.ignite.raft.jraft.entity.PeerId) Test(org.junit.jupiter.api.Test)

Example 3 with DefaultRaftClientService

use of org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService in project ignite-3 by apache.

the class TestUtils method sender.

/**
 * Returns a message sender for this node.
 *
 * @param node The node.
 * @return The message sender.
 */
public static RpcClientEx sender(Node node) {
    NodeImpl node0 = (NodeImpl) node;
    DefaultRaftClientService rpcService = (DefaultRaftClientService) node0.getRpcClientService();
    return (RpcClientEx) rpcService.getRpcClient();
}
Also used : NodeImpl(org.apache.ignite.raft.jraft.core.NodeImpl) DefaultRaftClientService(org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService) RpcClientEx(org.apache.ignite.raft.jraft.rpc.RpcClientEx)

Aggregations

DefaultRaftClientService (org.apache.ignite.raft.jraft.rpc.impl.core.DefaultRaftClientService)3 Status (org.apache.ignite.raft.jraft.Status)2 RpcClientEx (org.apache.ignite.raft.jraft.rpc.RpcClientEx)2 CompletableFuture (java.util.concurrent.CompletableFuture)1 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)1 ReadIndexClosure (org.apache.ignite.raft.jraft.closure.ReadIndexClosure)1 ConfigurationEntry (org.apache.ignite.raft.jraft.conf.ConfigurationEntry)1 ConfigurationManager (org.apache.ignite.raft.jraft.conf.ConfigurationManager)1 NodeImpl (org.apache.ignite.raft.jraft.core.NodeImpl)1 LogId (org.apache.ignite.raft.jraft.entity.LogId)1 NodeId (org.apache.ignite.raft.jraft.entity.NodeId)1 PeerId (org.apache.ignite.raft.jraft.entity.PeerId)1 BallotBoxOptions (org.apache.ignite.raft.jraft.option.BallotBoxOptions)1 RaftOptions (org.apache.ignite.raft.jraft.option.RaftOptions)1 ReadOnlyServiceOptions (org.apache.ignite.raft.jraft.option.ReadOnlyServiceOptions)1 ReplicatorGroupOptions (org.apache.ignite.raft.jraft.option.ReplicatorGroupOptions)1 RpcRequests (org.apache.ignite.raft.jraft.rpc.RpcRequests)1 DisruptorMetricSet (org.apache.ignite.raft.jraft.util.DisruptorMetricSet)1 Test (org.junit.jupiter.api.Test)1