X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-akka-raft%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fraft%2FRaftActorContextImpl.java;h=4644a2d79807bf36d069fa010ce561e0232cbcf9;hp=926e9748d477d8c54fcbde9265197e97a1369063;hb=866cdf141262dd7b57e95b3e3aa75709b7d0c586;hpb=a7740542c8ce1985c0a35767966c781805dfad84 diff --git a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/RaftActorContextImpl.java b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/RaftActorContextImpl.java index 926e9748d4..4644a2d798 100644 --- a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/RaftActorContextImpl.java +++ b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/RaftActorContextImpl.java @@ -24,14 +24,25 @@ import java.util.List; import java.util.Map; import java.util.Optional; import java.util.Set; +import java.util.function.Consumer; import java.util.function.LongSupplier; +import javax.annotation.Nonnull; +import javax.annotation.Nullable; import org.opendaylight.controller.cluster.DataPersistenceProvider; +import org.opendaylight.controller.cluster.io.FileBackedOutputStreamFactory; +import org.opendaylight.controller.cluster.raft.base.messages.ApplyState; import org.opendaylight.controller.cluster.raft.behaviors.RaftActorBehavior; import org.opendaylight.controller.cluster.raft.persisted.ServerConfigurationPayload; import org.opendaylight.controller.cluster.raft.persisted.ServerInfo; import org.opendaylight.controller.cluster.raft.policy.RaftPolicy; import org.slf4j.Logger; +/** + * Implementation of the RaftActorContext interface. + * + * @author Moiz Raja + * @author Thomas Pantelis + */ public class RaftActorContextImpl implements RaftActorContext { private static final LongSupplier JVM_MEMORY_RETRIEVER = () -> Runtime.getRuntime().maxMemory(); @@ -51,7 +62,7 @@ public class RaftActorContextImpl implements RaftActorContext { private final Map peerInfoMap = new HashMap<>(); - private final Logger LOG; + private final Logger log; private ConfigParams configParams; @@ -76,20 +87,32 @@ public class RaftActorContextImpl implements RaftActorContext { private Optional cluster; + private final Consumer applyStateConsumer; + + private final FileBackedOutputStreamFactory fileBackedOutputStreamFactory; + + private RaftActorLeadershipTransferCohort leadershipTransferCohort; + public RaftActorContextImpl(ActorRef actor, ActorContext context, String id, - ElectionTerm termInformation, long commitIndex, long lastApplied, Map peerAddresses, - ConfigParams configParams, DataPersistenceProvider persistenceProvider, Logger logger) { + @Nonnull ElectionTerm termInformation, long commitIndex, long lastApplied, + @Nonnull Map peerAddresses, + @Nonnull ConfigParams configParams, @Nonnull DataPersistenceProvider persistenceProvider, + @Nonnull Consumer applyStateConsumer, @Nonnull Logger logger) { this.actor = actor; this.context = context; this.id = id; - this.termInformation = termInformation; + this.termInformation = Preconditions.checkNotNull(termInformation); this.commitIndex = commitIndex; this.lastApplied = lastApplied; - this.configParams = configParams; - this.persistenceProvider = persistenceProvider; - this.LOG = logger; + this.configParams = Preconditions.checkNotNull(configParams); + this.persistenceProvider = Preconditions.checkNotNull(persistenceProvider); + this.log = Preconditions.checkNotNull(logger); + this.applyStateConsumer = Preconditions.checkNotNull(applyStateConsumer); + + fileBackedOutputStreamFactory = new FileBackedOutputStreamFactory( + configParams.getFileBackedStreamingThreshold(), configParams.getTempFileDirectory()); - for(Map.Entry e: peerAddresses.entrySet()) { + for (Map.Entry e: Preconditions.checkNotNull(peerAddresses).entrySet()) { peerInfoMap.put(e.getKey(), new PeerInfo(e.getKey(), e.getValue(), VotingState.VOTING)); } } @@ -109,12 +132,12 @@ public class RaftActorContextImpl implements RaftActorContext { } @Override - public ActorRef actorOf(Props props){ + public ActorRef actorOf(Props props) { return context.actorOf(props); } @Override - public ActorSelection actorSelection(String path){ + public ActorSelection actorSelection(String path) { return context.actorSelection(path); } @@ -129,13 +152,14 @@ public class RaftActorContextImpl implements RaftActorContext { } @Override + @SuppressWarnings("checkstyle:IllegalCatch") public Optional getCluster() { - if(cluster == null) { + if (cluster == null) { try { cluster = Optional.of(Cluster.get(getActorSystem())); - } catch(Exception e) { + } catch (Exception e) { // An exception means there's no cluster configured. This will only happen in unit tests. - LOG.debug("{}: Could not obtain Cluster: {}", getId(), e); + log.debug("{}: Could not obtain Cluster: {}", getId(), e); cluster = Optional.empty(); } } @@ -164,6 +188,7 @@ public class RaftActorContextImpl implements RaftActorContext { @Override public void setLastApplied(long lastApplied) { + log.debug("{}: Moving last applied index from {} to {}", id, this.lastApplied, lastApplied); this.lastApplied = lastApplied; } @@ -182,7 +207,7 @@ public class RaftActorContextImpl implements RaftActorContext { } @Override public Logger getLogger() { - return this.LOG; + return this.log; } @Override @@ -202,11 +227,11 @@ public class RaftActorContextImpl implements RaftActorContext { @Override public String getPeerAddress(String peerId) { - String peerAddress = null; + String peerAddress; PeerInfo peerInfo = peerInfoMap.get(peerId); - if(peerInfo != null) { + if (peerInfo != null) { peerAddress = peerInfo.getAddress(); - if(peerAddress == null) { + if (peerAddress == null) { peerAddress = configParams.getPeerAddressResolver().resolve(peerId); peerInfo.setAddress(peerAddress); } @@ -218,19 +243,19 @@ public class RaftActorContextImpl implements RaftActorContext { } @Override - public void updatePeerIds(ServerConfigurationPayload serverConfig){ + public void updatePeerIds(ServerConfigurationPayload serverConfig) { votingMember = true; boolean foundSelf = false; Set currentPeers = new HashSet<>(this.getPeerIds()); - for(ServerInfo server: serverConfig.getServerConfig()) { - if(getId().equals(server.getId())) { + for (ServerInfo server : serverConfig.getServerConfig()) { + if (getId().equals(server.getId())) { foundSelf = true; - if(!server.isVoting()) { + if (!server.isVoting()) { votingMember = false; } } else { - VotingState votingState = server.isVoting() ? VotingState.VOTING: VotingState.NON_VOTING; - if(!currentPeers.contains(server.getId())) { + VotingState votingState = server.isVoting() ? VotingState.VOTING : VotingState.NON_VOTING; + if (!currentPeers.contains(server.getId())) { this.addToPeers(server.getId(), null, votingState); } else { this.getPeerInfo(server.getId()).setVotingState(votingState); @@ -239,15 +264,15 @@ public class RaftActorContextImpl implements RaftActorContext { } } - for(String peerIdToRemove: currentPeers) { + for (String peerIdToRemove : currentPeers) { this.removePeer(peerIdToRemove); } - if(!foundSelf) { + if (!foundSelf) { votingMember = false; } - LOG.debug("{}: Updated server config: isVoting: {}, peers: {}", id, votingMember, peerInfoMap.values()); + log.debug("{}: Updated server config: isVoting: {}, peers: {}", id, votingMember, peerInfoMap.values()); setDynamicServerConfigurationInUse(); } @@ -257,14 +282,14 @@ public class RaftActorContextImpl implements RaftActorContext { } @Override - public void addToPeers(String id, String address, VotingState votingState) { - peerInfoMap.put(id, new PeerInfo(id, address, votingState)); + public void addToPeers(String peerId, String address, VotingState votingState) { + peerInfoMap.put(peerId, new PeerInfo(peerId, address, votingState)); numVotingPeers = -1; } @Override public void removePeer(String name) { - if(getId().equals(name)) { + if (getId().equals(name)) { votingMember = false; } else { peerInfoMap.remove(name); @@ -274,7 +299,7 @@ public class RaftActorContextImpl implements RaftActorContext { @Override public ActorSelection getPeerActorSelection(String peerId) { String peerAddress = getPeerAddress(peerId); - if(peerAddress != null){ + if (peerAddress != null) { return actorSelection(peerAddress); } return null; @@ -283,16 +308,16 @@ public class RaftActorContextImpl implements RaftActorContext { @Override public void setPeerAddress(String peerId, String peerAddress) { PeerInfo peerInfo = peerInfoMap.get(peerId); - if(peerInfo != null) { - LOG.info("Peer address for peer {} set to {}", peerId, peerAddress); + if (peerInfo != null) { + log.info("Peer address for peer {} set to {}", peerId, peerAddress); peerInfo.setAddress(peerAddress); } } @Override public SnapshotManager getSnapshotManager() { - if(snapshotManager == null){ - snapshotManager = new SnapshotManager(this, LOG); + if (snapshotManager == null) { + snapshotManager = new SnapshotManager(this, log); } return snapshotManager; } @@ -309,7 +334,7 @@ public class RaftActorContextImpl implements RaftActorContext { @Override public boolean hasFollowers() { - return getPeerIds().size() > 0; + return !getPeerIds().isEmpty(); } @Override @@ -340,15 +365,15 @@ public class RaftActorContextImpl implements RaftActorContext { } Collection peers = getPeers(); List newConfig = new ArrayList<>(peers.size() + 1); - for(PeerInfo peer: peers) { + for (PeerInfo peer: peers) { newConfig.add(new ServerInfo(peer.getId(), peer.isVoting())); } - if(includeSelf) { + if (includeSelf) { newConfig.add(new ServerInfo(getId(), votingMember)); } - return (new ServerConfigurationPayload(newConfig)); + return new ServerConfigurationPayload(newConfig); } @Override @@ -358,10 +383,10 @@ public class RaftActorContextImpl implements RaftActorContext { @Override public boolean anyVotingPeers() { - if(numVotingPeers < 0) { + if (numVotingPeers < 0) { numVotingPeers = 0; - for(PeerInfo info: getPeers()) { - if(info.isVoting()) { + for (PeerInfo info: getPeers()) { + if (info.isVoting()) { numVotingPeers++; } } @@ -379,13 +404,36 @@ public class RaftActorContextImpl implements RaftActorContext { this.currentBehavior = Preconditions.checkNotNull(behavior); } + @Override + public Consumer getApplyStateConsumer() { + return applyStateConsumer; + } + + @Override + public FileBackedOutputStreamFactory getFileBackedOutputStreamFactory() { + return fileBackedOutputStreamFactory; + } + + @SuppressWarnings("checkstyle:IllegalCatch") void close() { if (currentBehavior != null) { try { currentBehavior.close(); } catch (Exception e) { - LOG.debug("{}: Error closing behavior {}", getId(), currentBehavior.state()); + log.debug("{}: Error closing behavior {}", getId(), currentBehavior.state(), e); } } } + + @Override + @Nullable + public RaftActorLeadershipTransferCohort getRaftActorLeadershipTransferCohort() { + return leadershipTransferCohort; + } + + @Override + public void setRaftActorLeadershipTransferCohort( + @Nullable RaftActorLeadershipTransferCohort leadershipTransferCohort) { + this.leadershipTransferCohort = leadershipTransferCohort; + } }