blob: 16136c0e1645d61473a1a3c0ddec84ef40562589 [file] [log] [blame]
/**
* 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 org.apache.pulsar.proxy.server;
import static com.google.common.base.Preconditions.checkNotNull;
import static org.apache.commons.lang3.StringUtils.isBlank;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.AdaptiveRecvByteBufAllocator;
import io.netty.channel.Channel;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.util.concurrent.DefaultThreadFactory;
import io.prometheus.client.Counter;
import io.prometheus.client.Gauge;
import lombok.Getter;
import lombok.Setter;
import java.io.Closeable;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
import org.apache.pulsar.broker.ServiceConfigurationUtils;
import org.apache.pulsar.broker.authentication.AuthenticationService;
import org.apache.pulsar.broker.authorization.AuthorizationService;
import org.apache.pulsar.broker.cache.ConfigurationCacheService;
import org.apache.pulsar.broker.cache.ConfigurationMetadataCacheService;
import org.apache.pulsar.broker.resources.PulsarResources;
import org.apache.pulsar.common.allocator.PulsarByteBufAllocator;
import org.apache.pulsar.common.configuration.PulsarConfigurationLoader;
import org.apache.pulsar.common.util.netty.EventLoopUtil;
import org.apache.pulsar.metadata.api.MetadataStoreException;
import org.apache.pulsar.metadata.api.extended.MetadataStoreExtended;
import org.apache.pulsar.broker.web.plugin.servlet.AdditionalServlets;
import org.apache.pulsar.proxy.stats.TopicStats;
import org.apache.pulsar.zookeeper.ZooKeeperClientFactory;
import org.apache.pulsar.zookeeper.ZookeeperClientFactoryImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
/**
* Pulsar proxy service
*/
public class ProxyService implements Closeable {
private final ProxyConfiguration proxyConfig;
private String serviceUrl;
private String serviceUrlTls;
private ConfigurationMetadataCacheService configurationCacheService;
private final AuthenticationService authenticationService;
private AuthorizationService authorizationService;
private ZooKeeperClientFactory zkClientFactory = null;
private MetadataStoreExtended localMetadataStore;
private MetadataStoreExtended configMetadataStore;
private PulsarResources pulsarResources;
private final EventLoopGroup acceptorGroup;
private final EventLoopGroup workerGroup;
private Channel listenChannel;
private Channel listenChannelTls;
private final DefaultThreadFactory acceptorThreadFactory = new DefaultThreadFactory("pulsar-proxy-acceptor");
private final DefaultThreadFactory workersThreadFactory = new DefaultThreadFactory("pulsar-proxy-io");
private BrokerDiscoveryProvider discoveryProvider;
protected final AtomicReference<Semaphore> lookupRequestSemaphore;
@Getter
@Setter
protected int proxyLogLevel;
private final ScheduledExecutorService statsExecutor;
private static final int numThreads = Runtime.getRuntime().availableProcessors();
static final Gauge activeConnections = Gauge
.build("pulsar_proxy_active_connections", "Number of connections currently active in the proxy").create()
.register();
static final Counter newConnections = Counter
.build("pulsar_proxy_new_connections", "Counter of connections being opened in the proxy").create()
.register();
static final Counter rejectedConnections = Counter
.build("pulsar_proxy_rejected_connections", "Counter for connections rejected due to throttling").create()
.register();
static final Counter opsCounter = Counter
.build("pulsar_proxy_binary_ops", "Counter of proxy operations").create().register();
static final Counter bytesCounter = Counter
.build("pulsar_proxy_binary_bytes", "Counter of proxy bytes").create().register();
@Getter
private final Set<ProxyConnection> clientCnxs;
@Getter
private final Map<String, TopicStats> topicStats;
@Getter
private AdditionalServlets proxyAdditionalServlets;
public ProxyService(ProxyConfiguration proxyConfig,
AuthenticationService authenticationService) throws IOException {
checkNotNull(proxyConfig);
this.proxyConfig = proxyConfig;
this.clientCnxs = Sets.newConcurrentHashSet();
this.topicStats = Maps.newConcurrentMap();
this.lookupRequestSemaphore = new AtomicReference<Semaphore>(
new Semaphore(proxyConfig.getMaxConcurrentLookupRequests(), false));
if (proxyConfig.getProxyLogLevel().isPresent()) {
proxyLogLevel = proxyConfig.getProxyLogLevel().get();
} else {
proxyLogLevel = 0;
}
this.acceptorGroup = EventLoopUtil.newEventLoopGroup(1, acceptorThreadFactory);
this.workerGroup = EventLoopUtil.newEventLoopGroup(numThreads, workersThreadFactory);
this.authenticationService = authenticationService;
statsExecutor = Executors
.newSingleThreadScheduledExecutor(new DefaultThreadFactory("proxy-stats-executor"));
statsExecutor.schedule(()->{
this.clientCnxs.forEach(cnx -> {
if (cnx.getDirectProxyHandler() != null
&& cnx.getDirectProxyHandler().getInboundChannelRequestsRate() != null) {
cnx.getDirectProxyHandler().getInboundChannelRequestsRate().calculateRate();
}
});
this.topicStats.forEach((topic, stats) -> {
stats.calculate();
});
}, 60, TimeUnit.SECONDS);
this.proxyAdditionalServlets = AdditionalServlets.load(proxyConfig);
}
public void start() throws Exception {
if (!isBlank(proxyConfig.getZookeeperServers()) && !isBlank(proxyConfig.getConfigurationStoreServers())) {
localMetadataStore = createLocalMetadataStore();
configMetadataStore = createConfigurationMetadataStore();
pulsarResources = new PulsarResources(localMetadataStore, configMetadataStore);
discoveryProvider = new BrokerDiscoveryProvider(this.proxyConfig, getZooKeeperClientFactory(),
pulsarResources);
this.configurationCacheService = new ConfigurationMetadataCacheService(pulsarResources, null);
authorizationService = new AuthorizationService(PulsarConfigurationLoader.convertFrom(proxyConfig),
configurationCacheService);
}
ServerBootstrap bootstrap = new ServerBootstrap();
bootstrap.childOption(ChannelOption.ALLOCATOR, PulsarByteBufAllocator.DEFAULT);
bootstrap.group(acceptorGroup, workerGroup);
bootstrap.childOption(ChannelOption.TCP_NODELAY, true);
bootstrap.childOption(ChannelOption.RCVBUF_ALLOCATOR,
new AdaptiveRecvByteBufAllocator(1024, 16 * 1024, 1 * 1024 * 1024));
bootstrap.channel(EventLoopUtil.getServerSocketChannelClass(workerGroup));
EventLoopUtil.enableTriggeredMode(bootstrap);
bootstrap.childHandler(new ServiceChannelInitializer(this, proxyConfig, false));
// Bind and start to accept incoming connections.
if (proxyConfig.getServicePort().isPresent()) {
try {
listenChannel = bootstrap.bind(proxyConfig.getBindAddress(), proxyConfig.getServicePort().get()).sync().channel();
LOG.info("Started Pulsar Proxy at {}", listenChannel.localAddress());
} catch (Exception e) {
throw new IOException("Failed to bind Pulsar Proxy on port " + proxyConfig.getServicePort().get(), e);
}
}
if (proxyConfig.getServicePortTls().isPresent()) {
ServerBootstrap tlsBootstrap = bootstrap.clone();
tlsBootstrap.childHandler(new ServiceChannelInitializer(this, proxyConfig, true));
listenChannelTls = tlsBootstrap.bind(proxyConfig.getBindAddress(), proxyConfig.getServicePortTls().get()).sync().channel();
LOG.info("Started Pulsar TLS Proxy on {}", listenChannelTls.localAddress());
}
final String hostname =
ServiceConfigurationUtils.getDefaultOrConfiguredAddress(proxyConfig.getAdvertisedAddress());
if (proxyConfig.getServicePort().isPresent()) {
this.serviceUrl = String.format("pulsar://%s:%d/", hostname, getListenPort().get());
} else {
this.serviceUrl = null;
}
if (proxyConfig.getServicePortTls().isPresent()) {
this.serviceUrlTls = String.format("pulsar+ssl://%s:%d/", hostname, getListenPortTls().get());
} else {
this.serviceUrlTls = null;
}
}
public ZooKeeperClientFactory getZooKeeperClientFactory() {
if (zkClientFactory == null) {
zkClientFactory = new ZookeeperClientFactoryImpl();
}
// Return default factory
return zkClientFactory;
}
public BrokerDiscoveryProvider getDiscoveryProvider() {
return discoveryProvider;
}
public void close() throws IOException {
if (discoveryProvider != null) {
discoveryProvider.close();
}
if (listenChannel != null) {
listenChannel.close();
}
if (listenChannelTls != null) {
listenChannelTls.close();
}
if (statsExecutor != null) {
statsExecutor.shutdown();
}
if (proxyAdditionalServlets != null) {
proxyAdditionalServlets.close();
proxyAdditionalServlets = null;
}
if (localMetadataStore != null) {
try {
localMetadataStore.close();
} catch (Exception e) {
throw new IOException(e);
}
}
if (configMetadataStore != null) {
try {
configMetadataStore.close();
} catch (Exception e) {
throw new IOException(e);
}
}
acceptorGroup.shutdownGracefully();
workerGroup.shutdownGracefully();
}
public String getServiceUrl() {
return serviceUrl;
}
public String getServiceUrlTls() {
return serviceUrlTls;
}
public ProxyConfiguration getConfiguration() {
return proxyConfig;
}
public AuthenticationService getAuthenticationService() {
return authenticationService;
}
public AuthorizationService getAuthorizationService() {
return authorizationService;
}
public ConfigurationCacheService getConfigurationCacheService() {
return configurationCacheService;
}
public void setConfigurationCacheService(ConfigurationMetadataCacheService configurationCacheService) {
this.configurationCacheService = configurationCacheService;
}
public Semaphore getLookupRequestSemaphore() {
return lookupRequestSemaphore.get();
}
public EventLoopGroup getWorkerGroup() {
return workerGroup;
}
public Optional<Integer> getListenPort() {
if (listenChannel != null) {
return Optional.of(((InetSocketAddress) listenChannel.localAddress()).getPort());
} else {
return Optional.empty();
}
}
public Optional<Integer> getListenPortTls() {
if (listenChannelTls != null) {
return Optional.of(((InetSocketAddress) listenChannelTls.localAddress()).getPort());
} else {
return Optional.empty();
}
}
public MetadataStoreExtended createLocalMetadataStore() throws MetadataStoreException {
return PulsarResources.createMetadataStore(proxyConfig.getZookeeperServers(),
proxyConfig.getZookeeperSessionTimeoutMs());
}
public MetadataStoreExtended createConfigurationMetadataStore() throws MetadataStoreException {
return PulsarResources.createMetadataStore(proxyConfig.getConfigurationStoreServers(),
proxyConfig.getZookeeperSessionTimeoutMs());
}
private static final Logger LOG = LoggerFactory.getLogger(ProxyService.class);
}