package org.infinispan.cli.interpreter.statement;
import java.util.List;
import org.infinispan.Cache;
import org.infinispan.cli.interpreter.codec.Codec;
import org.infinispan.cli.interpreter.logging.Log;
import org.infinispan.cli.interpreter.result.Result;
import org.infinispan.cli.interpreter.result.StatementException;
import org.infinispan.cli.interpreter.result.StringResult;
import org.infinispan.cli.interpreter.session.Session;
import org.infinispan.distribution.DistributionManager;
import org.infinispan.remoting.transport.Address;
import org.infinispan.util.logging.LogFactory;
/**
* LocateStatement locates an entry in the cluster
*
* @author Tristan Tarrant
* @since 5.2
*/
public class LocateStatement implements Statement {
private static final Log log = LogFactory.getLog(LocateStatement.class, Log.class);
private enum Options {
CODEC
};
final KeyData keyData;
final private List<Option> options;
public LocateStatement(List<Option> options, KeyData key) {
this.options = options;
this.keyData = key;
}
@Override
public Result execute(Session session) throws StatementException {
Cache<Object, Object> cache = session.getCache(keyData.getCacheName());
Codec codec = session.getCodec();
if (options.size() > 0) {
for (Option option : options) {
switch (option.toEnum(Options.class)) {
case CODEC: {
if (option.getParameter() == null) {
throw log.missingOptionParameter(option.getName());
} else {
codec = session.getCodec(option.getParameter());
}
}
}
}
}
DistributionManager distributionManager = cache.getAdvancedCache().getDistributionManager();
if(distributionManager!=null) {
Object key = keyData.getKey();
List<Address> addresses = distributionManager.locate(codec.encodeKey(key));
return new StringResult(addresses.toString());
} else {
throw log.cacheNotDistributed();
}
}
}