Skip to content

Commit

Permalink
[fix] [broker] Internal reader of __change_events can not started aft…
Browse files Browse the repository at this point in the history
…er metadata store session rebuilt (apache#23018)

(cherry picked from commit b955c65)
(cherry picked from commit 0e59678)
  • Loading branch information
poorbarcode authored and srinath-ctds committed Jul 30, 2024
1 parent de5ccbd commit 148077c
Show file tree
Hide file tree
Showing 6 changed files with 692 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,8 @@ public class ExtensibleLoadManagerImpl implements ExtensibleLoadManager {
private static final Set<String> INTERNAL_TOPICS =
Set.of(BROKER_LOAD_DATA_STORE_TOPIC, TOP_BUNDLES_LOAD_DATA_STORE_TOPIC, TOPIC);

private PulsarService pulsar;
@VisibleForTesting
protected PulsarService pulsar;

private ServiceConfiguration conf;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -169,8 +169,9 @@ public class ModularLoadManagerImpl implements ModularLoadManager {
// Policies used to determine which brokers are available for particular namespaces.
private SimpleResourceAllocationPolicies policies;

@VisibleForTesting
// Pulsar service used to initialize this.
private PulsarService pulsar;
protected PulsarService pulsar;

// Executor service used to update broker data.
private final ExecutorService executors;
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,197 @@
/*
* 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.broker.service;

import io.netty.bootstrap.Bootstrap;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import lombok.Getter;

public class Ipv4Proxy {
@Getter
private final int localPort;
private final String backendServerHost;
private final int backendServerPort;
private final EventLoopGroup serverGroup = new NioEventLoopGroup(1);
private final EventLoopGroup workerGroup = new NioEventLoopGroup();
private ChannelFuture localServerChannel;
private ServerBootstrap serverBootstrap = new ServerBootstrap();
private List<Channel> frontChannels = Collections.synchronizedList(new ArrayList<>());
private AtomicBoolean rejectAllConnections = new AtomicBoolean();

public Ipv4Proxy(int localPort, String backendServerHost, int backendServerPort) {
this.localPort = localPort;
this.backendServerHost = backendServerHost;
this.backendServerPort = backendServerPort;
}

public synchronized void startup() throws InterruptedException {
localServerChannel = serverBootstrap.group(serverGroup, workerGroup)
.channel(NioServerSocketChannel.class)
.handler(new LoggingHandler(LogLevel.INFO))
.childHandler(new ChannelInitializer<SocketChannel>() {
@Override
protected void initChannel(SocketChannel ch) {
ch.pipeline().addLast(new FrontendHandler());
}
}).childOption(ChannelOption.AUTO_READ, false)
.bind(localPort).sync();
}

public synchronized void stop() throws InterruptedException{
localServerChannel.channel().close().sync();
serverGroup.shutdownGracefully();
workerGroup.shutdownGracefully();
}

private static void closeOnFlush(Channel ch) {
if (ch.isActive()) {
ch.writeAndFlush(Unpooled.EMPTY_BUFFER).addListener(ChannelFutureListener.CLOSE);
}
}

public void disconnectFrontChannels() throws InterruptedException {
for (Channel channel : frontChannels) {
channel.close();
}
}

public void rejectAllConnections() throws InterruptedException {
rejectAllConnections.set(true);
}

public void unRejectAllConnections() throws InterruptedException {
rejectAllConnections.set(false);
}

private class FrontendHandler extends ChannelInboundHandlerAdapter {

private Channel backendChannel;

@Override
public void channelActive(ChannelHandlerContext ctx) {
if (rejectAllConnections.get()) {
ctx.close();
return;
}
final Channel frontendChannel = ctx.channel();
frontChannels.add(frontendChannel);
Bootstrap backendBootstrap = new Bootstrap();
backendBootstrap.group(frontendChannel.eventLoop())
.channel(ctx.channel().getClass())
.handler(new BackendHandler(frontendChannel))
.option(ChannelOption.AUTO_READ, false);
ChannelFuture backendChannelFuture =
backendBootstrap.connect(Ipv4Proxy.this.backendServerHost, Ipv4Proxy.this.backendServerPort);
backendChannel = backendChannelFuture.channel();
backendChannelFuture.addListener((ChannelFutureListener) future -> {
if (future.isSuccess()) {
frontendChannel.read();
} else {
frontChannels.remove(frontendChannel);
frontendChannel.close();
}
});
}

@Override
public void channelRead(final ChannelHandlerContext ctx, Object msg) {
if (backendChannel.isActive()) {
backendChannel.writeAndFlush(msg).addListener((ChannelFutureListener) future -> {
if (future.isSuccess()) {
ctx.channel().read();
} else {
future.channel().close();
}
});
}
}

@Override
public void channelInactive(ChannelHandlerContext ctx) {
frontChannels.remove(ctx.channel());
if (backendChannel != null) {
closeOnFlush(backendChannel);
}
}

@Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
cause.printStackTrace();
closeOnFlush(ctx.channel());
}
}

private class BackendHandler extends ChannelInboundHandlerAdapter {

private final Channel frontendChannel;

public BackendHandler(Channel inboundChannel) {
this.frontendChannel = inboundChannel;
}

@Override
public void channelActive(ChannelHandlerContext ctx) {
if (!frontendChannel.isActive()) {
closeOnFlush(ctx.channel());
} else {
ctx.read();
}
}

@Override
public void channelRead(final ChannelHandlerContext ctx, Object msg) {
frontendChannel.writeAndFlush(msg).addListener((ChannelFutureListener) future -> {
if (future.isSuccess()) {
ctx.channel().read();
} else {
future.channel().close();
}
});
}

@Override
public void channelInactive(ChannelHandlerContext ctx) {
closeOnFlush(frontendChannel);
}

@Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
cause.printStackTrace();
closeOnFlush(ctx.channel());
}
}
}
Loading

0 comments on commit 148077c

Please sign in to comment.