Use Int instead of Integer to configure #partitions and replication f…

…actor
tomzhang · Sep 29, 2014 · 3f476ae · 3f476ae
1 parent cb2efad
commit 3f476ae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/main/scala/com/miguno/kafkastorm/kafka/KafkaEmbedded.scala b/src/main/scala/com/miguno/kafkastorm/kafka/KafkaEmbedded.scala
@@ -81,7 +81,7 @@ class KafkaEmbedded(config: Properties = new Properties) extends LazyLogging {
     logger.debug(s"Shutdown of embedded Kafka broker at $brokerList completed (with ZK server at $zookeeperConnect)")
   }
 
-  def createTopic(topic: String, partitions: Int =  1, replicationFactor: Int = 1, config: Properties = new Properties): Unit = {
+  def createTopic(topic: String, partitions: Int = 1, replicationFactor: Int = 1, config: Properties = new Properties): Unit = {
     logger.debug(s"Creating topic { name: $topic, partitions: $partitions, replicationFactor: $replicationFactor, config: $config }")
     val sessionTimeout = 10.seconds
     val connectionTimeout = 8.seconds

diff --git a/src/test/scala/com/miguno/kafkastorm/spark/KafkaSparkStreamingSpec.scala b/src/test/scala/com/miguno/kafkastorm/spark/KafkaSparkStreamingSpec.scala
@@ -171,7 +171,7 @@ class KafkaSparkStreamingSpec extends FeatureSpec with Matchers with BeforeAndAf
         // And yes, the way we do this looks quite strange -- we combine a hardcoded `1` in the topic map with a
         // subsequent `(1..N)` construct.  But this approach is the recommended way.
         val topics = Map(inputTopic.name -> 1)
-        val streams = (1 to inputTopic.partitions.toInt) map { _ =>
+        val streams = (1 to inputTopic.partitions) map { _ =>
           KafkaUtils.createStream[Array[Byte], Array[Byte], DefaultDecoder, DefaultDecoder](
             ssc,
             kafkaParams,

diff --git a/src/test/scala/com/miguno/kafkastorm/testing/EmbeddedKafkaZooKeeperCluster.scala b/src/test/scala/com/miguno/kafkastorm/testing/EmbeddedKafkaZooKeeperCluster.scala
@@ -120,7 +120,7 @@ class EmbeddedKafkaZooKeeperCluster(zookeeperPort: Integer = InstanceSpec.getRan
 
 }
 
-case class KafkaTopic(name: String, partitions: Integer = 1, replicationFactor: Integer = 1,
+case class KafkaTopic(name: String, partitions: Int = 1, replicationFactor: Int = 1,
                       config: Properties = new Properties) {
 
   require(partitions > 0)