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=a96d5026639c693c7164616040c332c72b640b8d;hp=833c8a9e8ad1ca7b8fe170aadffc25b3c12b6b24;hb=8882e6077db69d22bcc57fcf12dd4a02a81a4967;hpb=8c6fac2a853a107c49983fb17934958eadaee416 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 833c8a9e8a..a96d502663 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 @@ -8,22 +8,30 @@ package org.opendaylight.controller.cluster.raft; +import akka.actor.ActorContext; import akka.actor.ActorRef; import akka.actor.ActorSelection; import akka.actor.ActorSystem; import akka.actor.Props; -import akka.actor.UntypedActorContext; -import akka.event.LoggingAdapter; - +import com.google.common.annotations.VisibleForTesting; +import com.google.common.base.Supplier; +import java.util.ArrayList; +import java.util.Collection; +import java.util.HashMap; +import java.util.HashSet; +import java.util.List; import java.util.Map; +import java.util.Set; +import org.opendaylight.controller.cluster.DataPersistenceProvider; +import org.opendaylight.controller.cluster.raft.ServerConfigurationPayload.ServerInfo; +import org.opendaylight.controller.cluster.raft.policy.RaftPolicy; +import org.slf4j.Logger; -import static com.google.common.base.Preconditions.checkState; - -public class RaftActorContextImpl implements RaftActorContext{ +public class RaftActorContextImpl implements RaftActorContext { private final ActorRef actor; - private final UntypedActorContext context; + private final ActorContext context; private final String id; @@ -33,47 +41,84 @@ public class RaftActorContextImpl implements RaftActorContext{ private long lastApplied; - private final ReplicatedLog replicatedLog; + private ReplicatedLog replicatedLog; + + private final Map peerInfoMap = new HashMap<>(); + + private final Logger LOG; + + private ConfigParams configParams; + + private boolean dynamicServerConfiguration = false; + + @VisibleForTesting + private Supplier totalMemoryRetriever; - private final Map peerAddresses; + // Snapshot manager will need to be created on demand as it needs raft actor context which cannot + // be passed to it in the constructor + private SnapshotManager snapshotManager; - private final LoggingAdapter LOG; + private final DataPersistenceProvider persistenceProvider; - public RaftActorContextImpl(ActorRef actor, UntypedActorContext context, - String id, - ElectionTerm termInformation, long commitIndex, - long lastApplied, ReplicatedLog replicatedLog, Map peerAddresses, LoggingAdapter logger) { + private short payloadVersion; + + public RaftActorContextImpl(ActorRef actor, ActorContext context, String id, + ElectionTerm termInformation, long commitIndex, long lastApplied, Map peerAddresses, + ConfigParams configParams, DataPersistenceProvider persistenceProvider, Logger logger) { this.actor = actor; this.context = context; this.id = id; this.termInformation = termInformation; this.commitIndex = commitIndex; this.lastApplied = lastApplied; - this.replicatedLog = replicatedLog; - this.peerAddresses = peerAddresses; + this.configParams = configParams; + this.persistenceProvider = persistenceProvider; this.LOG = logger; + + for(Map.Entry e: peerAddresses.entrySet()) { + peerInfoMap.put(e.getKey(), new PeerInfo(e.getKey(), e.getValue(), VotingState.VOTING)); + } + } + + public void setPayloadVersion(short payloadVersion) { + this.payloadVersion = payloadVersion; + } + + @Override + public short getPayloadVersion() { + return payloadVersion; + } + + public void setConfigParams(ConfigParams configParams) { + this.configParams = configParams; } + @Override public ActorRef actorOf(Props props){ return context.actorOf(props); } + @Override public ActorSelection actorSelection(String path){ return context.actorSelection(path); } + @Override public String getId() { return id; } + @Override public ActorRef getActor() { return actor; } + @Override public ElectionTerm getTermInformation() { return termInformation; } + @Override public long getCommitIndex() { return commitIndex; } @@ -82,15 +127,23 @@ public class RaftActorContextImpl implements RaftActorContext{ this.commitIndex = commitIndex; } + @Override public long getLastApplied() { return lastApplied; } - @Override public void setLastApplied(long lastApplied) { + @Override + public void setLastApplied(long lastApplied) { this.lastApplied = lastApplied; } - @Override public ReplicatedLog getReplicatedLog() { + @Override + public void setReplicatedLog(ReplicatedLog replicatedLog) { + this.replicatedLog = replicatedLog; + } + + @Override + public ReplicatedLog getReplicatedLog() { return replicatedLog; } @@ -98,24 +151,75 @@ public class RaftActorContextImpl implements RaftActorContext{ return context.system(); } - @Override public LoggingAdapter getLogger() { + @Override public Logger getLogger() { return this.LOG; } - @Override public Map getPeerAddresses() { - return peerAddresses; + @Override + public Collection getPeerIds() { + return peerInfoMap.keySet(); + } + + @Override + public Collection getPeers() { + return peerInfoMap.values(); + } + + @Override + public PeerInfo getPeerInfo(String peerId) { + return peerInfoMap.get(peerId); + } + + @Override + public String getPeerAddress(String peerId) { + String peerAddress = null; + PeerInfo peerInfo = peerInfoMap.get(peerId); + if(peerInfo != null) { + peerAddress = peerInfo.getAddress(); + if(peerAddress == null) { + peerAddress = configParams.getPeerAddressResolver().resolve(peerId); + peerInfo.setAddress(peerAddress); + } + } else { + peerAddress = configParams.getPeerAddressResolver().resolve(peerId); + } + + return peerAddress; + } + + @Override + public void updatePeerIds(ServerConfigurationPayload serverConfig){ + + Set currentPeers = new HashSet<>(this.getPeerIds()); + for(ServerInfo server: serverConfig.getServerConfig()) { + if(!getId().equals(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); + currentPeers.remove(server.getId()); + } + } + } + + for(String peerIdToRemove: currentPeers) { + this.removePeer(peerIdToRemove); + } + setDynamicServerConfigurationInUse(); } - @Override public String getPeerAddress(String peerId) { - return peerAddresses.get(peerId); + @Override public ConfigParams getConfigParams() { + return configParams; } - @Override public void addToPeers(String name, String address) { - peerAddresses.put(name, address); + @Override + public void addToPeers(String id, String address, VotingState votingState) { + peerInfoMap.put(id, new PeerInfo(id, address, votingState)); } @Override public void removePeer(String name) { - peerAddresses.remove(name); + peerInfoMap.remove(name); } @Override public ActorSelection getPeerActorSelection(String peerId) { @@ -126,10 +230,71 @@ public class RaftActorContextImpl implements RaftActorContext{ return null; } - @Override public void setPeerAddress(String peerId, String peerAddress) { - LOG.info("Peer address for peer {} set to {}", peerId, peerAddress); - checkState(peerAddresses.containsKey(peerId), peerId + " is unknown"); + @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); + peerInfo.setAddress(peerAddress); + } + } + + @Override + public SnapshotManager getSnapshotManager() { + if(snapshotManager == null){ + snapshotManager = new SnapshotManager(this, LOG); + } + return snapshotManager; + } + + @Override + public long getTotalMemory() { + return totalMemoryRetriever != null ? totalMemoryRetriever.get() : Runtime.getRuntime().totalMemory(); + } + + @Override + public void setTotalMemoryRetriever(Supplier retriever) { + totalMemoryRetriever = retriever; + } + + @Override + public boolean hasFollowers() { + return getPeerIds().size() > 0; + } + + @Override + public DataPersistenceProvider getPersistenceProvider() { + return persistenceProvider; + } + + + @Override + public RaftPolicy getRaftPolicy() { + return configParams.getRaftPolicy(); + } + + @Override + public boolean isDynamicServerConfigurationInUse() { + return dynamicServerConfiguration; + } + + @Override + public void setDynamicServerConfigurationInUse() { + this.dynamicServerConfiguration = true; + } + + @Override + public ServerConfigurationPayload getPeerServerInfo() { + if (!isDynamicServerConfigurationInUse()) { + return null; + } + Collection peers = getPeers(); + List newConfig = new ArrayList<>(peers.size() + 1); + for(PeerInfo peer: peers) { + newConfig.add(new ServerInfo(peer.getId(), peer.isVoting())); + } - peerAddresses.put(peerId, peerAddress); + newConfig.add(new ServerInfo(getId(), true)); + return (new ServerConfigurationPayload(newConfig)); } }