71 lines
2.2 KiB
C#
71 lines
2.2 KiB
C#
|
|
/***************************************************************************
|
|
* MessagePump.cs
|
|
* -------------------
|
|
* begin : May 1, 2002
|
|
* copyright : (C) The RunUO Software Team
|
|
* email : info@runuo.com
|
|
*
|
|
* $Id$
|
|
*
|
|
***************************************************************************/
|
|
|
|
/***************************************************************************
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
***************************************************************************/
|
|
|
|
using System.Buffers;
|
|
using System.Collections.Concurrent;
|
|
|
|
namespace Server.Network
|
|
{
|
|
public interface IMessagePumpService
|
|
{
|
|
void QueueWork(NetState ns, IMemoryOwner<byte> memOwner, OnPacketReceive onReceive);
|
|
void DoWork();
|
|
}
|
|
|
|
public class MessagePumpService : IMessagePumpService
|
|
{
|
|
private readonly ConcurrentQueue<Work> m_WorkQueue = new ConcurrentQueue<Work>();
|
|
|
|
public void QueueWork(NetState ns, IMemoryOwner<byte> memOwner, OnPacketReceive onReceive)
|
|
{
|
|
m_WorkQueue.Enqueue(new Work(ns, memOwner, onReceive));
|
|
Core.Set();
|
|
}
|
|
|
|
public void DoWork()
|
|
{
|
|
int count = 0;
|
|
while (!m_WorkQueue.IsEmpty && count++ < 250)
|
|
{
|
|
if (!m_WorkQueue.TryDequeue(out Work work))
|
|
break;
|
|
|
|
work.OnReceive(work.State, new PacketReader(new ReadOnlySequence<byte>(work.MemoryOwner.Memory)));
|
|
work.MemoryOwner.Dispose();
|
|
}
|
|
}
|
|
|
|
private class Work
|
|
{
|
|
public readonly NetState State;
|
|
// TODO: Force dispose?
|
|
public readonly IMemoryOwner<byte> MemoryOwner;
|
|
public readonly OnPacketReceive OnReceive;
|
|
|
|
public Work(NetState ns, IMemoryOwner<byte> memOwner, OnPacketReceive onReceive)
|
|
{
|
|
State = ns;
|
|
MemoryOwner = memOwner;
|
|
OnReceive = onReceive;
|
|
}
|
|
}
|
|
}
|
|
}
|