.net异步套接字超时检查线程安全
http://msdn.microsoft.com/en-us /library/system.net.sockets.socketasynceventargs.aspx
从上面的 msdn 示例开始,我尝试编写一个超时检查,它将关闭不活动的客户端套接字并释放资源。
这就是我想出的办法。 但我不确定它是否完全线程安全以及是否有更好的方法来做到这一点。 我希望有人能给一些建议。
void IO_Completed(object sender, SocketAsyncEventArgs e)
{
if (e.SocketError != SocketError.Success)
{
CloseClientSocket(e);
return;
}
if (1 < Interlocked.CompareExchange(ref token.Status, 1, 0))
return;
switch (e.LastOperation)
{
case SocketAsyncOperation.Receive:
ProcessReceive(e);
break;
...
}
token.LastActive = Environment.TickCount;
Interlocked.CompareExchange(ref token.Status, 0, 1);
}
void ProcessReceive(SocketAsyncEventArgs e)
{
AsyncUserToken token = (AsyncUserToken)e.UserToken;
if (e.BytesTransferred > 0)
if (!token.Socket.SendAsync(e))
ProcessSend(e);
else
CloseClientSocket(e);
}
void ProcessSend(SocketAsyncEventArgs e)
{
AsyncUserToken token = (AsyncUserToken)e.UserToken;
if (!token.Socket.ReceiveAsync(e))
ProcessReceive(e);
}
TimeoutCheck 将每 20 秒执行一次。 allReadWriteArgs 是一个包含所有 SocketAsyncEventArgs 的数组。 关闭套接字后,将通过 SocketError.OperationAborted 调用 IO_Completed。
void TimeoutCheck(object state)
{
AsyncUserToken token;
int timeout = Environment.TickCount - 20000;
for (int i = 0; i < allReadWriteArgs.Length; i++)
{
token = (AsyncUserToken)allReadWriteArgs[i].UserToken;
if (token.LastActive < timeout)
if (0 == Interlocked.CompareExchange(ref token.Status, 2, 0))
Interlocked.Exchange(ref token.Socket, null).Close();
}
}
void CloseClientSocket(SocketAsyncEventArgs e)
{
AsyncUserToken token = e.UserToken as AsyncUserToken;
if (token.Socket != null)
{
try
{
token.Socket.Shutdown(SocketShutdown.Both);
}
catch (SocketException) { }
token.Socket.Close();
}
token.Status = 2;
bufferManager.FreeBuffer(e);
readWritePool.Push(e);
...
}
http://msdn.microsoft.com/en-us/library/system.net.sockets.socketasynceventargs.aspx
Starting with the above msdn example I'm trying to write a timeout check which will close inactive client sockets and free up resources.
This is what I have come up with. But I'm not sure if its completely thread-safe and if there is a better way to do this. I hope someone can give some advices.
void IO_Completed(object sender, SocketAsyncEventArgs e)
{
if (e.SocketError != SocketError.Success)
{
CloseClientSocket(e);
return;
}
if (1 < Interlocked.CompareExchange(ref token.Status, 1, 0))
return;
switch (e.LastOperation)
{
case SocketAsyncOperation.Receive:
ProcessReceive(e);
break;
...
}
token.LastActive = Environment.TickCount;
Interlocked.CompareExchange(ref token.Status, 0, 1);
}
void ProcessReceive(SocketAsyncEventArgs e)
{
AsyncUserToken token = (AsyncUserToken)e.UserToken;
if (e.BytesTransferred > 0)
if (!token.Socket.SendAsync(e))
ProcessSend(e);
else
CloseClientSocket(e);
}
void ProcessSend(SocketAsyncEventArgs e)
{
AsyncUserToken token = (AsyncUserToken)e.UserToken;
if (!token.Socket.ReceiveAsync(e))
ProcessReceive(e);
}
TimeoutCheck will execute once each 20 seconds. allReadWriteArgs is an array with all SocketAsyncEventArgs. After closing the socket IO_Completed will be invoked with SocketError.OperationAborted.
void TimeoutCheck(object state)
{
AsyncUserToken token;
int timeout = Environment.TickCount - 20000;
for (int i = 0; i < allReadWriteArgs.Length; i++)
{
token = (AsyncUserToken)allReadWriteArgs[i].UserToken;
if (token.LastActive < timeout)
if (0 == Interlocked.CompareExchange(ref token.Status, 2, 0))
Interlocked.Exchange(ref token.Socket, null).Close();
}
}
void CloseClientSocket(SocketAsyncEventArgs e)
{
AsyncUserToken token = e.UserToken as AsyncUserToken;
if (token.Socket != null)
{
try
{
token.Socket.Shutdown(SocketShutdown.Both);
}
catch (SocketException) { }
token.Socket.Close();
}
token.Status = 2;
bufferManager.FreeBuffer(e);
readWritePool.Push(e);
...
}
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论
评论(1)
你的代码看起来不错。 你也可以做这样的事情:
Your code looks good. You can do something like this, as well: