cn.novelweb.tool.upload.fastdfs.pool.TrackerLocator Maven / Gradle / Ivy
package cn.novelweb.tool.upload.fastdfs.pool;
import cn.novelweb.tool.upload.fastdfs.exception.FastDfsUnavailableException;
import cn.novelweb.tool.upload.fastdfs.utils.Log;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import java.net.InetSocketAddress;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
/**
* 表示Tracker服务器位置
* 支持负载均衡对IP轮询
* 2020-02-03 17:27
*
* @author LiZW
**/
@Slf4j
public class TrackerLocator {
/**
* 连接不可经过10分钟以后重试连接
*/
private static final int DEFAULT_RETRY_AFTER_SECOND = 10 * 60;
/**
* 连接中断以后经过N秒后可以重试
*/
private int retryAfterSecond = DEFAULT_RETRY_AFTER_SECOND;
/**
* 方便随意快速读取
*/
private Map trackerAddressMap = new HashMap();
/**
* 轮询圈(用户轮询获取连接地址)
*/
private CircularList trackerAddressCircular = new CircularList();
/**
* 初始化Tracker服务器地址
* 配置方式为 ip:port 如 192.168.1.2:21000
*/
public TrackerLocator(Set trackerSet) {
Log.debug("开始初始化Tracker Server地址:{}", trackerSet);
for (String addressStr : trackerSet) {
if (StringUtils.isBlank(addressStr)) {
continue;
}
String[] parts = StringUtils.split(addressStr, ":", 2);
if (parts.length != 2) {
log.warn("Tracker Server地址格式无效[{}], 跳过此配置(正确格式 host:port)", addressStr);
continue;
}
InetSocketAddress address;
try {
address = new InetSocketAddress(parts[0].trim(), Integer.parseInt(parts[1].trim()));
} catch (Throwable e) {
log.warn("创建InetSocketAddress失败, Tracker Server地址[{}], 跳过此配置", addressStr);
continue;
}
if (trackerAddressMap.get(addressStr) == null) {
TrackerAddressState holder = new TrackerAddressState(address);
trackerAddressCircular.add(holder);
trackerAddressMap.put(address.toString(), holder);
}
}
if (log.isDebugEnabled()) {
String tmp = "\r\n" +
"#=======================================================================================================================#\r\n" +
"# 初始化Tracker Server地址完毕\r\n" +
"#\t trackerAddressMap#keySet: " + trackerAddressMap.keySet() + "\r\n" +
"#\t trackerAddressCircular: " + trackerAddressCircular + "\r\n" +
"#=======================================================================================================================#\r\n";
Log.debug(tmp);
}
}
/**
* 获取Tracker服务器地址(使用轮询)
*/
public InetSocketAddress getTrackerAddress() {
TrackerAddressState holder;
// 遍历连接地址,抓取当前有效的地址
for (int i = 0; i < trackerAddressCircular.size(); i++) {
holder = trackerAddressCircular.next();
if (holder.canTryToConnect(retryAfterSecond)) {
return holder.getAddress();
}
}
throw new FastDfsUnavailableException("找不到可用的 Tracker Server - {}" + trackerAddressMap.keySet());
}
/**
* 设置连接是否有效
*
* @param address 连接地址
* @param available true有效，false无效
*/
public void setActive(InetSocketAddress address, boolean available) {
TrackerAddressState holder = trackerAddressMap.get(address.toString());
if (holder == null) {
Log.debug("TrackerAddressMap获取TrackerAddressState为null, key={}, 设置连接是否有效失败[{}]", address, available);
return;
}
holder.setAvailable(available);
}
public void setRetryAfterSecond(int retryAfterSecond) {
this.retryAfterSecond = retryAfterSecond;
}
}