com.acgist.snail.net.torrent.utp.bootstrap.UtpService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snail Show documentation

基于Java开发的下载工具，支持下载协议：BT（BitTorrent、磁力链接、种子文件）、HLS（M3U8）、FTP、HTTP。

There is a newer version: 2.17.0

package com.acgist.snail.net.torrent.utp.bootstrap;

import java.net.InetSocketAddress;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.acgist.snail.net.UdpMessageHandler;
import com.acgist.snail.net.torrent.utp.UtpMessageHandler;
import com.acgist.snail.system.context.SystemThreadContext;

/**
 * UTP服务
 * 
 * @author acgist
 * @since 1.1.0
 */
public class UtpService {
	
	private static final Logger LOGGER = LoggerFactory.getLogger(UtpService.class);
	
	private static final UtpService INSTANCE = new UtpService();
	
	private final Map utpMessageHandlers = new ConcurrentHashMap<>();
	
	/**
	 * UTP超时定时任务
	 */
	private static final int UTP_INTERVAL = 5;
	
	private UtpService() {
		timer();
	}
	
	public static final UtpService getInstance() {
		return INSTANCE;
	}
	
	/**
	 * 连接ID，每获取一次+1。
	 */
	private int connectionId = 0;
	
	/**
	 * UTP超时定时任务：定时处理超时信息。
	 */
	private void timer() {
		SystemThreadContext.timerFixedDelay(UTP_INTERVAL, UTP_INTERVAL, TimeUnit.SECONDS, () -> {
			synchronized (this.utpMessageHandlers) {
				UtpMessageHandler handler;
				final var iterator = this.utpMessageHandlers.entrySet().iterator();
				while(iterator.hasNext()) {
					try {
						handler = iterator.next().getValue();
						if(handler.available()) {
							handler.wndTimeoutRetry();
						} else {
							handler.fin(); // 结束
							iterator.remove(); // 移除
						}
					} catch (Exception e) {
						LOGGER.error("UTP超时定时任务异常", e);
					}
				}
			}
		});
	}
	
	/**
	 * 获取连接ID
	 */
	public short connectionId() {
		synchronized (this) {
			return (short) connectionId++;
		}
	}
	
	/**
	 * 获取UTP消息处理器，如果已经存在直接返回。
	 * 
	 * @param connectionId 连接ID
	 * @param socketAddress 请求地址
	 */
	public UdpMessageHandler get(short connectionId, InetSocketAddress socketAddress) {
		final String key = buildKey(connectionId, socketAddress);
		UtpMessageHandler utpMessageHandler = this.utpMessageHandlers.get(key);
		if(utpMessageHandler != null) {
			return utpMessageHandler;
		}
		return new UtpMessageHandler(connectionId, socketAddress);
	}
	
	/**
	 * 添加消息处理器
	 */
	public void put(UtpMessageHandler utpMessageHandler) {
		synchronized (this.utpMessageHandlers) {
			this.utpMessageHandlers.put(utpMessageHandler.key(), utpMessageHandler);
		}
	}
	
	/**
	 * 删除消息处理器
	 */
	public void remove(UtpMessageHandler utpMessageHandler) {
		synchronized (this.utpMessageHandlers) {
			this.utpMessageHandlers.remove(utpMessageHandler.key());
		}
	}
	
	/**
	 * key = 地址 + 端口 + connectionId
	 */
	public String buildKey(Short connectionId, InetSocketAddress socketAddress) {
		return socketAddress.getHostString() + socketAddress.getPort() + connectionId;
	}

}