protected override bool RequireGetPeersRegisteredInfo(TransactionId msgId, DhtNode node, out byte[] infoHash)
{
var nodeId = node.CompactEndPoint().ToInt64();
var srcript = _unRegisterScript[(int)(nodeId % _registerScript.Count)];
var result = srcript.Evaluate(_database, new { point = nodeId, msgId = ((byte[])msgId) });
if (result.IsNull)
{
infoHash = null;
return(false);
}
var hash = (byte[])result;
infoHash = hash;
return(true);
}
protected override bool RequireGetPeersRegisteredInfo(TransactionId msgId, DhtNode node, out byte[] infoHash)
{
if (_mappingInfo.TryGetValue(msgId, out var mapInfo))
{
if (_idMappingInfo.TryGetValue(mapInfo.InfoHash, out var idMap))
{
lock (idMap)
{
var nodeId = node.CompactEndPoint().ToInt64();
var isok = idMap.Remove(nodeId);
if (idMap.Count <= 0)
{
_idMappingInfo.TryRemove(mapInfo.InfoHash, out var rm);
_mappingInfo.TryRemove(msgId, out var obj);
_bucket.Add(msgId);
}
if (!isok)
{
infoHash = null;
return(false);
}
}
}
else
{
_mappingInfo.TryRemove(msgId, out var obj);
_bucket.Add(msgId);
}
}
if (mapInfo?.InfoHash == null)
{
infoHash = null;
return(false);
}
infoHash = mapInfo.InfoHash;
return(true);
}
protected override bool RequireGetPeersRegisteredInfo(TransactionId msgId, DhtNode node, out byte[] infoHash)
{
infoHash = null;
var nodeKey = node.CompactEndPoint().ToInt64();
if (!NodeMap.TryGetValue(nodeKey, out var nodeMap))
{
return(false);
}
var result = false;
nodeMap.LastTime = DateTime.Now.Ticks;
if (nodeMap.InfoHashMap.TryRemove(msgId, out var infohash))
{
if (infohash?.InfoHash == null)
{
return(false);
}
infoHash = infohash.InfoHash;
result = true;
}
if (nodeMap.InfoHashMap.Count <= 0)
{
if (NodeMap.TryRemove(nodeKey, out var rm) && rm.InfoHashMap.Count > 0)
{
NodeMap.AddOrUpdate(nodeKey, rm, (key, old) =>
{
foreach (var item in rm.InfoHashMap)
{
old.InfoHashMap.TryAdd(item.Key, item.Value);
}
return(old);
});
}
}
return(result);
}
protected override bool RegisterGetPeersMessage(byte[] infoHash, DhtNode node, out TransactionId msgId)
{
var nodeId = node.CompactEndPoint().ToInt64();
if (_filter.Contain(nodeId))
{
msgId = null;
return(false);
}
TransactionId messageId;
if (_idMappingInfo.TryGetValue(infoHash, out var idMap))
{
if (!idMap.Add(nodeId))
{
msgId = null;
return(false);
}
messageId = idMap.TransactionId;
if (_mappingInfo.TryGetValue(messageId, out var info))
{
info.LastTime = DateTime.Now;
}
}
else
{
msgId = null;
var cleared = false;
while (!_bucket.TryTake(out messageId, 1000))
{
if (cleared) //清理以后没有过期命令则丢弃该消息
{
return(false);
}
ClearExpireMessage();
cleared = true;
}
if (!_mappingInfo.TryAdd(messageId, new MapInfo()
{
LastTime = DateTime.Now,
InfoHash = infoHash
}))
{
msgId = null;
return(false);
}
idMap = new IdMapInfo()
{
TransactionId = messageId
};
idMap.Add(nodeId);
if (_idMappingInfo.TryAdd(infoHash, idMap))
{
msgId = messageId;
return(true);
}
return(false);
}
msgId = messageId;
return(true);
}
protected override bool RegisterGetPeersMessage(byte[] infoHash, DhtNode node, out TransactionId msgId)
{
var nodeId = node.CompactEndPoint().ToInt64();
var srcript = _registerScript[(int)(nodeId % _registerScript.Count)];
lock (this)
{
_index++;
if (_index >= _bucketArray.Length)
{
_index = 0;
}
msgId = _bucketArray[_index];
}
var result = srcript.Evaluate(_database, new { point = nodeId, hash = infoHash, msgId = ((byte[])msgId) });
if (result.IsNull)
{
msgId = null;
return(false);
}
var resultInt = (int)result;
if (resultInt == 1)
{
return(true);
}
msgId = null;
return(false);
}
protected override Task <(bool IsOk, byte[] InfoHash)> RequireGetPeersRegisteredInfoAsync(TransactionId msgId, DhtNode node)
{
var nodeId = node.CompactEndPoint().ToInt64();
var srcript = _unRegisterScript[(int)(nodeId % _registerScript.Count)];
return(srcript.EvaluateAsync(_database, new { point = nodeId, msgId = ((byte[])msgId) }).ContinueWith(
t =>
{
if (t.IsCanceled || t.IsFaulted)
{
return (false, null);
}
var result = t.Result;
if (result.IsNull)
{
return (false, null);
}
var hash = (byte[])result;
return (true, hash);
}));
}
protected override bool RegisterGetPeersMessage(byte[] infoHash, DhtNode node, out TransactionId msgId)
{
var nodeKey = node.CompactEndPoint();
var key = new byte[nodeKey.Length + 2];
Array.Copy(nodeKey, 0, key, 2, nodeKey.Length);
var tryTimes = 0;
while (true)
{
if (tryTimes > 10)
{
break;
}
lock (this)
{
_bucketIndex++;
if (_bucketIndex >= _bucketArray.Length)
{
_bucketIndex = 0;
}
}
msgId = _bucketArray[_bucketIndex];
key[0] = msgId[0];
key[1] = msgId[1];
lock (_syncRoot)
{
var old = _table.Find(msgId);
if (old == null || old.AddTime.AddMinutes(30) < DateTime.Now)
{
_operateSize++;
_table[key] = new Record()
{
InfoHash = infoHash, AddTime = DateTime.Now
};
if (_operateSize > 5120)
{
_storageEngine.Commit();
_operateSize = 0;
}
return(true);
}
}
tryTimes++;
}
msgId = null;
return(false);
}
protected virtual Task <(bool IsOk, byte[] InfoHash)> RequireGetPeersRegisteredInfoAsync(TransactionId msgId, DhtNode node)
{
var result = RequireGetPeersRegisteredInfo(msgId, node, out var infoHash);
return(Task.FromResult((result, infoHash)));
}
protected abstract bool RequireGetPeersRegisteredInfo(TransactionId msgId, DhtNode node, out byte[] infoHash);
protected abstract bool RegisterGetPeersMessage(byte[] infoHash, DhtNode node, out TransactionId msgId);
protected override bool RegisterGetPeersMessage(byte[] infoHash, DhtNode node, out TransactionId msgId)
{
var sendPeers = InfoHashPeers.GetOrAdd(infoHash, new HashSet <long>());
var nodeKey = node.CompactEndPoint().ToInt64();
lock (sendPeers)
{
if (sendPeers.Contains(nodeKey))
{
msgId = null;
return(false);
}
sendPeers.Add(nodeKey);
}
var nodeMapInfo = NodeMap.GetOrAdd(nodeKey, new NodeMapInfo());
if (nodeMapInfo.IsExpire(_expireTimeSpan))
{
NodeMap.TryRemove(nodeKey, out nodeMapInfo);
msgId = null;
lock (sendPeers)
{
sendPeers.Remove(nodeKey);
}
return(false);
}
if (DateTime.Now >= _lastClearTime)
{
lock (_syncRoot)
{
if (DateTime.Now >= _lastClearTime)
{
ClearExpireMessage();
_lastClearTime = DateTime.Now.AddTicks(_expireTimeSpan);
}
}
}
var mapItem = new NodeMapInfoItem()
{
InfoHash = infoHash, LastTime = DateTime.Now.Ticks
};
lock (_bucketArray)
{
_bucketIndex++;
if (_bucketIndex >= _bucketArray.Length)
{
_bucketIndex = 0;
}
msgId = _bucketArray[_bucketIndex];
}
if (nodeMapInfo.InfoHashMap.TryAdd(msgId, mapItem))
{
return(true);
}
lock (sendPeers)
{
sendPeers.Remove(nodeKey);
}
msgId = null;
return(false);
}
protected override bool RegisterGetPeersMessage(byte[] infoHash, DhtNode node, out TransactionId msgId)
{
var nodeId = (ulong)node.CompactEndPoint().ToInt64();
lock (this)
{
_index++;
if (_index >= _bucketArray.Length)
{
_index = 0;
}
}
var nowTick = DateTime.Now.Ticks;
if (nowTick - _lastClearTime >= _clearDuration)
{
lock (this)
{
if (nowTick - _lastClearTime >= _clearDuration)
{
_treeStore.Clear();
_lastClearTime = nowTick;
}
}
}
msgId = _bucketArray[_index];
var path = nodeId << 16 | (uint)(msgId[0] << 8) | msgId[1];
return(_treeStore.TryAdd(path, infoHash));
}
