importjava.util.HashSet;importjava.util.Properties;importjava.util.Set;importjava.util.regex.Pattern;importorg.apache.commons.pool2.impl.GenericObjectPoolConfig;importorg.springframework.beans.factory.FactoryBean;importorg.springframework.beans.factory.InitializingBean;importorg.springframework.core.io.Resource;importredis.clients.jedis.HostAndPort;importredis.clients.jedis.JedisCluster;public class JedisClusterFactory implements FactoryBean, InitializingBean {privateResource addressConfig;privateString addressKeyPrefix ;privateJedisCluster jedisCluster;privateInteger timeout;privateInteger maxRedirections;privateGenericObjectPoolConfig genericObjectPoolConfig;private Pattern p = Pattern.compile("^.+[:]\\d{1,5}\\s*$");
@Overridepublic JedisCluster getObject() throwsException {returnjedisCluster;
}
@Overridepublic Class extends JedisCluster>getObjectType() {return (this.jedisCluster != null ? this.jedisCluster.getClass() : JedisCluster.class);
}
@Overridepublic booleanisSingleton() {return true;
}private Set parseHostAndPort() throwsException {try{
Properties prop= newProperties();
prop.load(this.addressConfig.getInputStream());
Set haps = new HashSet();for(Object key : prop.keySet()) {if (!((String) key).startsWith(addressKeyPrefix)) {continue;
}
String val=(String) prop.get(key);boolean isIpPort =p.matcher(val).matches();if (!isIpPort) {throw new IllegalArgumentException("ip 或 port 不合法");
}
String[] ipAndPort= val.split(":");
HostAndPort hap= new HostAndPort(ipAndPort[0], Integer.parseInt(ipAndPort[1]));
haps.add(hap);
}returnhaps;
}catch(IllegalArgumentException ex) {throwex;
}catch(Exception ex) {throw new Exception("解析 jedis 配置文件失败", ex);
}
}
@Overridepublic void afterPropertiesSet() throwsException {
Set haps = this.parseHostAndPort();
jedisCluster= newJedisCluster(haps, timeout, maxRedirections,genericObjectPoolConfig);
}public voidsetAddressConfig(Resource addressConfig) {this.addressConfig =addressConfig;
}public void setTimeout(inttimeout) {this.timeout =timeout;
}public void setMaxRedirections(intmaxRedirections) {this.maxRedirections =maxRedirections;
}public voidsetAddressKeyPrefix(String addressKeyPrefix) {this.addressKeyPrefix =addressKeyPrefix;
}public voidsetGenericObjectPoolConfig(GenericObjectPoolConfig genericObjectPoolConfig) {this.genericObjectPoolConfig =genericObjectPoolConfig;
}
}