看起来NodeLocator.updateLocator(List<MemcachedNode> newNodes)
应该做这项工作。
但是要连接 MemcachedNode 有点困难。您必须覆盖MemcachedClient
,MemcachedConnection
和DefaultConnectionFactory
.
您想添加或删除客户端是合理的,MemcachedClient
因此您可以添加remove(MemcachedNode node)
和add(MemcachedNode node)
方法。
在删除的情况下,您应该断开节点(请参阅MemcachedConnection.shutdown()
)并将其从中删除NodeLocator.getAll()
并调用NodeLocator.updateLocator(List<MemcachedNode> newNodes)
。
在添加的情况下,您应该通过 MemcachedConnection.createConnections(final Collection a) 连接节点,将其合并NodeLocator.getAll()
并调用NodeLocator.updateLocator(List<MemcachedNode> newNodes)
。
好吧,我从来没有尝试过,所以它可能不起作用。祝你好运!
ExtMemCachedConnection.java
公共类 ExtMemCachedConnection 扩展 MemcachedConnection {
protected final OperationFactory opFact;
/**
* Construct a memcached connection.
*
* @param bufSize the size of the buffer used for reading from the server
* @param f the factory that will provide an operation queue
* @param a the addresses of the servers to connect to
* @throws java.io.IOException if a connection attempt fails early
*/
public ExtendableMemcachedConnection(int bufSize, ConnectionFactory f,
List<InetSocketAddress> a,
Collection<ConnectionObserver> obs,
FailureMode fm, OperationFactory opfactory)
throws IOException {
super(bufSize, f, a, obs, fm, opfactory);
this.opFact = opfactory;
}
public void add(InetSocketAddress nodeAddress) throws IOException {
final List<InetSocketAddress> nodeToAdd = new ArrayList<InetSocketAddress>(1);
nodeToAdd.add(nodeAddress);
List<MemcachedNode> newNodesList = createConnections(nodeToAdd);
newNodesList.addAll(getLocator().getAll());
getLocator().updateLocator(newNodesList);
}
//The node should be obtain from locator to ensure currentNode.equals(node) will return true
public void remove(MemcachedNode node) throws IOException {
for(MemcachedNode currentNode : getLocator().getAll()) {
if(currentNode.equals(node)) {
Collection<Operation> notCompletedOperations = currentNode.destroyInputQueue();
if (currentNode.getChannel() != null) {
currentNode.getChannel().close();
currentNode.setSk(null);
if (currentNode.getBytesRemainingToWrite() > 0) {
getLogger().warn("Shut down with %d bytes remaining to write",
currentNode.getBytesRemainingToWrite());
}
getLogger().debug("Shut down channel %s", currentNode.getChannel());
}
//Unfortunatelly, redistributeOperations is private so it cannot be used or override. I put copy/paste the implementation
redistributeOperations(notCompletedOperations);
}
}
}
protected void redistributeOperations(Collection<Operation> ops) {
for (Operation op : ops) {
if (op.isCancelled() || op.isTimedOut()) {
continue;
}
if (op instanceof KeyedOperation) {
KeyedOperation ko = (KeyedOperation) op;
int added = 0;
for (String k : ko.getKeys()) {
for (Operation newop : opFact.clone(ko)) {
addOperation(k, newop);
added++;
}
}
assert added > 0 : "Didn't add any new operations when redistributing";
} else {
// Cancel things that don't have definite targets.
op.cancel();
}
}
}
}
ExtMemcachedClient.java
public void add(InetSocketAddress nodeAddress) {
if(mconn instanceof ExtMemcachedConnection) {
((ExtMemcachedConnection)mconn).add(nodeAddress);
}
}
public boolean remove(MemcachedNode node) {
if(mconn instanceof ExtMemcachedConnection) {
((ExtMemcachedConnection)mconn).remove(nodeAddress);
}
}
ExtMemcachedConnectionfactory.java
@Override
public MemcachedConnection createConnection(List<InetSocketAddress> addrs) throws IOException {
return new ExtendableMemcachedConnection(getReadBufSize(), this, addrs,
getInitialObservers(), getFailureMode(), getOperationFactory());
}