kafka.utils.ToolsUtils.scala Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package kafka.utils
import joptsimple.OptionParser
import org.apache.kafka.common.{Metric, MetricName}
import scala.collection.mutable
object ToolsUtils {
def validatePortOrDie(parser: OptionParser, hostPort: String) = {
val hostPorts: Array[String] = if(hostPort.contains(','))
hostPort.split(",")
else
Array(hostPort)
val validHostPort = hostPorts.filter { hostPortData =>
org.apache.kafka.common.utils.Utils.getPort(hostPortData) != null
}
val isValid = !validHostPort.isEmpty && validHostPort.size == hostPorts.length
if(!isValid)
CommandLineUtils.printUsageAndDie(parser, "Please provide valid host:port like host1:9091,host2:9092\n ")
}
/**
* print out the metrics in alphabetical order
* @param metrics the metrics to be printed out
*/
def printMetrics(metrics: mutable.Map[MetricName, _ <: Metric]): Unit = {
var maxLengthOfDisplayName = 0
val sortedMap = metrics.toSeq.sortWith( (s,t) =>
Array(s._1.group(), s._1.name(), s._1.tags()).mkString(":")
.compareTo(Array(t._1.group(), t._1.name(), t._1.tags()).mkString(":")) < 0
).map {
case (key, value) =>
val mergedKeyName = Array(key.group(), key.name(), key.tags()).mkString(":")
if (maxLengthOfDisplayName < mergedKeyName.length) {
maxLengthOfDisplayName = mergedKeyName.length
}
(mergedKeyName, value.metricValue)
}
println(s"\n%-${maxLengthOfDisplayName}s %s".format("Metric Name", "Value"))
sortedMap.foreach {
case (metricName, value) =>
println(s"%-${maxLengthOfDisplayName}s : %.3f".format(metricName, value))
}
}
}