public void TestConsume()
{
const int P = 1235;
const long O = 76158134069;
_client.Consume(Topic, P, O);
VerifyConsume(Topic, P, O);
}
private static void Main(string[] args)
{
Mode mode = Mode.Profile;
bool mix = false;
var configuration = new Configuration
{
ConsumeBatchSize = 100
};
// Ugly command line parsing
string curOpt = "";
try
{
bool seeds = false;
bool topics = false;
for (int i = 0; i < args.Length; ++i)
{
curOpt = args[i];
switch (args[i])
{
case "--global":
configuration.BatchStrategy = BatchStrategy.Global;
break;
case "--mix":
mix = true;
break;
case "--stress":
mode = Mode.Stress;
break;
case "--hard":
mode = Mode.StressHard;
break;
case "--discard":
configuration.ErrorStrategy = ErrorStrategy.Discard;
break;
case "--retry":
configuration.ErrorStrategy = ErrorStrategy.Retry;
break;
case "--gzip":
configuration.CompressionCodec = CompressionCodec.Gzip;
break;
case "--snappy":
configuration.CompressionCodec = CompressionCodec.Snappy;
break;
case "--no-ack":
configuration.RequiredAcks = RequiredAcks.None;
break;
case "--all-sync-ack":
configuration.RequiredAcks = RequiredAcks.AllInSyncReplicas;
break;
case "--ttl":
configuration.MessageTtl = TimeSpan.FromSeconds(int.Parse(args[++i]));
break;
case "--batch":
configuration.ProduceBatchSize = int.Parse(args[++i]);
break;
case "--time":
configuration.ProduceBufferingTime = TimeSpan.FromMilliseconds(int.Parse(args[++i]));
break;
case "--max-messages":
configuration.MaxBufferedMessages = int.Parse(args[++i]);
break;
case "--topics":
topics = true;
_topics = args[++i].Split(',');
break;
case "--seeds":
seeds = true;
configuration.Seeds = args[++i];
break;
case "--clientid":
configuration.ClientId = args[++i];
break;
case "--concurrency":
configuration.MaximumConcurrency = int.Parse(args[++i]);
break;
case "--send-buffer":
configuration.SendBufferSize = int.Parse(args[++i]);
break;
case "--receive-buffer":
configuration.ReceiveBufferSize = int.Parse(args[++i]);
break;
case "--timeout":
configuration.RequestTimeoutMs = int.Parse(args[++i]);
break;
case "--min-bytes":
configuration.FetchMinBytes = int.Parse(args[++i]);
break;
case "--max-wait":
configuration.FetchMaxWaitTime = int.Parse(args[++i]);
break;
case "--max-bytes":
configuration.FetchMessageMaxBytes = int.Parse(args[++i]);
break;
case "--delay":
configuration.NumberOfMessagesBeforeRoundRobin = int.Parse(args[++i]);
break;
case "--consume":
{
_consumeFrom = long.Parse(args[++i]);
var p = args[++i].Split(';');
_partitions = new int[p.Length][];
for (int j = 0; j < _partitions.Length; ++j)
{
_partitions[j] = p[j].Split(',').Select(int.Parse).ToArray();
}
}
break;
}
}
// Minimal error management
if (args.Length < 1 || !seeds || !topics)
{
throw new ArgumentException();
}
}
catch
{
// Minimal error management
Console.WriteLine("Syntax error in option {0}", curOpt);
Usage();
Environment.Exit(-1);
}
var serializer = new StringSerializer();
var deserializer = new StringDeserializer();
var serializationConfig = new SerializationConfig()
{
SerializeOnProduce = true
};
foreach (var topic in _topics)
{
serializationConfig.SetSerializersForTopic(topic, serializer, serializer);
serializationConfig.SetDeserializersForTopic(topic, deserializer, deserializer);
}
configuration.SerializationConfig = serializationConfig;
var cluster =
new ClusterClient(configuration, new ConsoleLogger());
if (_partitions == null)
{
var task = Start(mode, cluster);
Console.ReadKey();
_running = false;
Console.ReadKey();
task.Wait();
}
else
{
int i = 0;
foreach (var topic in _topics)
{
var capturedTopic = topic;
cluster.Messages.Where(kr => kr.Topic == capturedTopic).Sample(TimeSpan.FromMilliseconds(15))
.Subscribe(kr => Console.WriteLine("{0}/{1} {2}: {3}", kr.Topic, kr.Partition, kr.Offset, kr.Value as string));
foreach (var p in _partitions[i])
{
cluster.Consume(topic, p, _consumeFrom);
}
++i;
}
Task task = null;
if (mix)
{
task = Start(mode, cluster);
}
Console.ReadKey();
i = 0;
foreach (var topic in _topics)
{
foreach (var p in _partitions[i])
{
if (p < 0)
{
cluster.StopConsume(topic);
}
else
{
cluster.StopConsume(topic, p);
}
}
++i;
}
if (task != null)
{
_running = false;
task.Wait();
}
}
Console.WriteLine(cluster.Statistics);
Console.ReadKey();
cluster.Dispose();
}
