34 lines
916 B
C#
34 lines
916 B
C#
using RabbitMQ.Client;
|
|
using Tapeti.Cmd.RateLimiter;
|
|
using Tapeti.Cmd.Serialization;
|
|
|
|
namespace Tapeti.Cmd.Commands
|
|
{
|
|
public class ImportCommand
|
|
{
|
|
public IMessageSerializer MessageSerializer { get; set; }
|
|
|
|
public bool DirectToQueue { get; set; }
|
|
|
|
|
|
public int Execute(IModel channel, IRateLimiter rateLimiter)
|
|
{
|
|
var messageCount = 0;
|
|
|
|
foreach (var message in MessageSerializer.Deserialize())
|
|
{
|
|
rateLimiter.Execute(() =>
|
|
{
|
|
var exchange = DirectToQueue ? "" : message.Exchange;
|
|
var routingKey = DirectToQueue ? message.Queue : message.RoutingKey;
|
|
|
|
channel.BasicPublish(exchange, routingKey, message.Properties, message.Body);
|
|
messageCount++;
|
|
});
|
|
}
|
|
|
|
return messageCount;
|
|
}
|
|
}
|
|
}
|