internal async ValueTask ProcessChannel(RabbitMQProtocolReader protocol, FrameHeader header, MethodHeader method, CancellationToken token = default)
{
switch (method.MethodId)
{
case 11: //open-ok
{
var openOk = await protocol.ReadChannelOpenOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnChannelOpenOkAsync(header.Channel).ConfigureAwait(false);
break;
}
case 40: //close
{
var closeInfo = await protocol.ReadCloseAsync(token).ConfigureAwait(false);;
await _channelHandler.OnChannelCloseAsync(header.Channel, closeInfo).ConfigureAwait(false);
break;
}
case 41: //close-ok
{
var closeOk = await protocol.ReadCloseOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnChannelCloseOkAsync(header.Channel).ConfigureAwait(false);
break;
}
default:
throw new RabbitMQMethodException(nameof(ProcessChannel), method.ClassId, method.MethodId);
}
}
//internal ValueTask ProcessMethod(RabbitMQProtocolReader protocol, ref Frame frame)
internal ValueTask ProcessMethod(RabbitMQProtocolReader protocol, ref FrameHeader header, ref MethodHeader method, CancellationToken token = default)
{
switch (method.ClassId)
{
case 10:
{
return(ProcessConnection(protocol, header, method, token));
}
case 20:
{
return(ProcessChannel(protocol, header, method, token));
}
case 40:
{
return(ProcessExchange(protocol, header, method, token));
}
case 50:
{
return(ProcessQueue(protocol, header, method, token));
}
case 60:
{
return(ProcessBasic(protocol, header, method, token));
}
default:
{
throw new RabbitMQMethodException(nameof(ProcessMethod), method.ClassId, method.MethodId);
}
}
}
internal async ValueTask ProcessConnection(RabbitMQProtocolReader protocol, FrameHeader header, MethodHeader method, CancellationToken token = default)
{
if (header.Channel != 0)
{
ReaderThrowHelper.ThrowIfFrameTypeMissmatch();
}
switch (method.MethodId)
{
case 10:
{
var serverConf = await protocol.ReadStartAsync(token).ConfigureAwait(false);
await _connectionHandler.OnStartAsync(serverConf).ConfigureAwait(false);
break;
}
case 30:
{
var tuneConf = await protocol.ReadTuneMethodAsync(token).ConfigureAwait(false);
await _connectionHandler.OnTuneAsync(tuneConf).ConfigureAwait(false);
break;
}
case 41:
{
var result = await protocol.ReadConnectionOpenOkAsync(token).ConfigureAwait(false);
await _connectionHandler.OnOpenOkAsync();
break;
}
case 50: //close
{
var closeInfo = await protocol.ReadCloseAsync(token).ConfigureAwait(false);
await _connectionHandler.OnCloseAsync(closeInfo).ConfigureAwait(false);
break;
}
case 51: //close-ok
{
var result = await protocol.ReadCloseOkAsync(token).ConfigureAwait(false);
await _connectionHandler.OnCloseOkAsync().ConfigureAwait(false);
break;
}
default:
throw new RabbitMQMethodException(nameof(ProcessConnection), method.ClassId, method.MethodId);
}
}
internal async ValueTask ProcessBasic(RabbitMQProtocolReader protocol, FrameHeader header, MethodHeader method, CancellationToken token = default)
{
switch (method.MethodId)
{
case 60: //deliver method
{
var deliver = await protocol.ReadBasicDeliverAsync(token).ConfigureAwait(false);
await _channelHandler.OnBeginDeliveryAsync(header.Channel, deliver, protocol).ConfigureAwait(false);
break;
}
case 21: // consume-ok
{
var tag = await protocol.ReadBasicConsumeOkAsync(token).ConfigureAwait(false);;
await _channelHandler.OnConsumeOkAsync(header.Channel, tag).ConfigureAwait(false);
break;
}
case 11: // qos-ok
{
var qosOk = await protocol.ReadBasicQoSOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnQosOkAsync(header.Channel).ConfigureAwait(false);
break;
}
case 31: //consumer cancel-ok
{
var tag = await protocol.ReadBasicConsumeCancelOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnConsumeCancelOkAsync(header.Channel, tag).ConfigureAwait(false);
break;
}
case 30:
{
var cancelInfo = await protocol.ReadBasicConsumeCancelAsync(token).ConfigureAwait(false);
await _channelHandler.OnConsumeCancelAsync(header.Channel, cancelInfo);
break;
}
default:
throw new RabbitMQMethodException(nameof(ProcessBasic), method.ClassId, method.MethodId);
}
}
internal async ValueTask ProcessQueue(RabbitMQProtocolReader protocol, FrameHeader header, MethodHeader method, CancellationToken token = default)
{
switch (method.MethodId)
{
case 11: //declare-ok
{
var declareOk = await protocol.ReadQueueDeclareOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnQueueDeclareOkAsync(header.Channel, declareOk).ConfigureAwait(false);
break;
}
case 21: //bind-ok
{
var bindOk = await protocol.ReadQueueBindOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnQueueBindOkAsync(header.Channel).ConfigureAwait(false);
break;
}
case 51: //unbind-ok
{
var unbindOk = await protocol.ReadQueueUnbindOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnQueueUnbindOkAsync(header.Channel).ConfigureAwait(false);
break;
}
case 31: //purge-ok
{
var purged = await protocol.ReadQueuePurgeOkAsync(token).ConfigureAwait(false);
await _channelHandler.OnQueuePurgeOkAsync(header.Channel, purged).ConfigureAwait(false);
break;
}
case 41: //delete-ok
{
var deleted = await protocol.ReadQueueDeleteOkAsync(token).ConfigureAwait(false);;
await _channelHandler.OnQueueDeleteOkAsync(header.Channel, deleted).ConfigureAwait(false);;
break;
}
default:
throw new RabbitMQMethodException(nameof(ProcessQueue), method.ClassId, method.MethodId);
}
}
internal async ValueTask ProcessExchange(RabbitMQProtocolReader protocol, FrameHeader header, MethodHeader method, CancellationToken token = default)
{
switch (method.MethodId)
{
case 11: //declare-ok
{
var declareOk = await protocol.ReadExchangeDeclareOkAsync(token).ConfigureAwait(false);;
await _channelHandler.OnExchangeDeclareOkAsync(header.Channel).ConfigureAwait(false);;
break;
}
case 21: //delete-ok
{
var declareOk = protocol.ReadExchangeDeleteOkAsync(token).ConfigureAwait(false);;
await _channelHandler.OnExchangeDeleteOkAsync(header.Channel).ConfigureAwait(false);;
break;
}
default:
throw new RabbitMQMethodException(nameof(ProcessExchange), method.ClassId, method.MethodId);
}
}
public async Task StartAsync(RabbitMQProtocolReader reader, IConnectionHandler connection, IChannelHandler channel, CancellationToken token = default)
{
_connectionHandler = connection;
_channelHandler = channel;
var headerReader = new FrameHeaderReader();
_isClosed = false;
//while (true)
while (!_isClosed)
{
var result = await reader.ReadAsync(headerReader, token).ConfigureAwait(false);
switch (result.FrameType)
{
case RabbitMQConstants.FrameMethod:
{
var method = await reader.ReadAsync(_methodHeaderReader, token).ConfigureAwait(false);
await ProcessMethod(reader, ref result, ref method, token).ConfigureAwait(false);
break;
}
case RabbitMQConstants.FrameHeartbeat:
{
await reader.ReadNoPayloadAsync(token).ConfigureAwait(false);
await _connectionHandler.OnHeartbeatAsync().ConfigureAwait(false);
break;
}
default:
{
throw new RabbitMQFrameException(result.FrameType);
}
}
}
}
