Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* 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.cluster
import java.util
import kafka.common.BrokerEndPointNotAvailableException
import kafka.server.KafkaConfig
import org.apache.kafka.common.{ClusterResource, Endpoint, Node}
import org.apache.kafka.common.network.ListenerName
import org.apache.kafka.common.security.auth.SecurityProtocol
import org.apache.kafka.server.authorizer.AuthorizerServerInfo
import scala.collection.Seq
import scala.collection.JavaConverters._
object Broker {
private[cluster] case class ServerInfo(clusterResource: ClusterResource,
brokerId: Int,
endpoints: util.List[Endpoint],
interBrokerEndpoint: Endpoint) extends AuthorizerServerInfo
}
/**
* A Kafka broker.
* A broker has an id, a collection of end-points, an optional rack and a listener to security protocol map.
* Each end-point is (host, port, listenerName).
*/
case class Broker(id: Int, endPoints: Seq[EndPoint], rack: Option[String]) {
private val endPointsMap = endPoints.map { endPoint =>
endPoint.listenerName -> endPoint
}.toMap
if (endPointsMap.size != endPoints.size)
throw new IllegalArgumentException(s"There is more than one end point with the same listener name: ${endPoints.mkString(",")}")
override def toString: String =
s"$id : ${endPointsMap.values.mkString("(",",",")")} : ${rack.orNull}"
def this(id: Int, host: String, port: Int, listenerName: ListenerName, protocol: SecurityProtocol) = {
this(id, Seq(EndPoint(host, port, listenerName, protocol)), None)
}
def this(bep: BrokerEndPoint, listenerName: ListenerName, protocol: SecurityProtocol) = {
this(bep.id, bep.host, bep.port, listenerName, protocol)
}
def node(listenerName: ListenerName): Node =
getNode(listenerName).getOrElse {
throw new BrokerEndPointNotAvailableException(s"End point with listener name ${listenerName.value} not found " +
s"for broker $id")
}
def getNode(listenerName: ListenerName): Option[Node] =
endPointsMap.get(listenerName).map(endpoint => new Node(id, endpoint.host, endpoint.port, rack.orNull))
def brokerEndPoint(listenerName: ListenerName): BrokerEndPoint = {
val endpoint = endPoint(listenerName)
new BrokerEndPoint(id, endpoint.host, endpoint.port)
}
def endPoint(listenerName: ListenerName): EndPoint = {
endPointsMap.getOrElse(listenerName,
throw new BrokerEndPointNotAvailableException(s"End point with listener name ${listenerName.value} not found for broker $id"))
}
def toServerInfo(clusterId: String, config: KafkaConfig): AuthorizerServerInfo = {
val clusterResource: ClusterResource = new ClusterResource(clusterId)
val interBrokerEndpoint: Endpoint = endPoint(config.interBrokerListenerName).toJava
val brokerEndpoints: util.List[Endpoint] = endPoints.toList.map(_.toJava).asJava
Broker.ServerInfo(clusterResource, id, brokerEndpoints, interBrokerEndpoint)
}
}