trivial change to add byte serializer to ProducerPerformance; patched by Jun Rao

This commit is contained in:
Jun Rao 2015-01-13 09:54:32 -08:00
parent e79ebdfe22
commit bfb2da3c82
1 changed files with 3 additions and 4 deletions

View File

@ -15,10 +15,7 @@ package org.apache.kafka.clients.tools;
import java.util.Arrays; import java.util.Arrays;
import java.util.Properties; import java.util.Properties;
import org.apache.kafka.clients.producer.Callback; import org.apache.kafka.clients.producer.*;
import org.apache.kafka.clients.producer.KafkaProducer;
import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.clients.producer.RecordMetadata;
public class ProducerPerformance { public class ProducerPerformance {
@ -46,6 +43,8 @@ public class ProducerPerformance {
throw new IllegalArgumentException("Invalid property: " + args[i]); throw new IllegalArgumentException("Invalid property: " + args[i]);
props.put(pieces[0], pieces[1]); props.put(pieces[0], pieces[1]);
} }
props.put(ProducerConfig.KEY_SERIALIZER_CLASS_CONFIG, "org.apache.kafka.common.serialization.ByteArraySerializer");
props.put(ProducerConfig.VALUE_SERIALIZER_CLASS_CONFIG, "org.apache.kafka.common.serialization.ByteArraySerializer");
KafkaProducer<byte[], byte[]> producer = new KafkaProducer<byte[],byte[]>(props); KafkaProducer<byte[], byte[]> producer = new KafkaProducer<byte[],byte[]>(props);
/* setup perf test */ /* setup perf test */