using System;
|
using System.Collections.Concurrent;
|
using System.Linq;
|
|
namespace Sodao.FastSocket.SocketBase
|
{
|
/// <summary>
|
/// socket connection collection
|
/// </summary>
|
public sealed class ConnectionCollection
|
{
|
#region Private Members
|
/// <summary>
|
/// key:ConnectionID
|
/// </summary>
|
private readonly ConcurrentDictionary<long, IConnection> _dic =
|
new ConcurrentDictionary<long, IConnection>();
|
#endregion
|
|
#region Public Methods
|
/// <summary>
|
/// add
|
/// </summary>
|
/// <param name="connection"></param>
|
/// <returns></returns>
|
/// <exception cref="ArgumentNullException">connection is null</exception>
|
public bool Add(IConnection connection)
|
{
|
if (connection == null) throw new ArgumentNullException("connection");
|
return this._dic.TryAdd(connection.ConnectionID, connection);
|
}
|
/// <summary>
|
/// remove connection by id.
|
/// </summary>
|
/// <param name="connectionID"></param>
|
/// <returns></returns>
|
public bool Remove(long connectionID)
|
{
|
IConnection connection;
|
return this._dic.TryRemove(connectionID, out connection);
|
}
|
/// <summary>
|
/// get by connection id
|
/// </summary>
|
/// <param name="connectionID"></param>
|
/// <returns></returns>
|
public IConnection Get(long connectionID)
|
{
|
IConnection connection;
|
this._dic.TryGetValue(connectionID, out connection);
|
return connection;
|
}
|
/// <summary>
|
/// to array
|
/// </summary>
|
/// <returns></returns>
|
public IConnection[] ToArray()
|
{
|
return this._dic.ToArray().Select(c => c.Value).ToArray();
|
}
|
/// <summary>
|
/// count.
|
/// </summary>
|
/// <returns></returns>
|
public int Count()
|
{
|
return this._dic.Count;
|
}
|
/// <summary>
|
/// 断开所有连接
|
/// </summary>
|
public void DisconnectAll()
|
{
|
var connections = this.ToArray();
|
foreach (var conn in connections) conn.BeginDisconnect();
|
}
|
#endregion
|
}
|
}
|
