Question

我正在开发一款Netty服务器，可用作Android应用的后端。在我当前的实现中，在逻辑处理程序中实现对DB的访问，由特殊的Netty线程池（而不是I / O线程）执行，通过每个Netty通道使用一个DB连接，如下所示：

初始化：

EventExecutorGroup logicExecutor = new DefaultEventExecutorGroup(4);
EventLoopGroup acceptGroup = new NioEventLoopGroup();
EventLoopGroup workerGroup = new NioEventLoopGroup();

try
{
    ServerBootstrap b = new ServerBootstrap();
    b.group(acceptGroup, workerGroup)
        .channel(NioServerSocketChannel.class)
        .option(ChannelOption.SO_BACKLOG, 50)
        .childOption(ChannelOption.SO_KEEPALIVE, false)
        .childHandler(new ChannelInitializer<SocketChannel>()
        {
        @Override
        public void initChannel(SocketChannel ch) throws Exception
        {
            ChannelPipeline pipeline = ch.pipeline();
            pipeline.addLast(new IdleStateHandler(5*60, 0, 0));
            pipeline.addLast(new ProtobufDelimitedFrameDecoder(65536));
            pipeline.addLast(new ProtobufVarint32LengthFieldPrepender());
            pipeline.addLast(new ProtobufDecoder(NetMsg.ClientMsg.getDefaultInstance()));
            pipeline.addLast(new ProtobufEncoder());
            pipeline.addLast(logicExecutor, "logic", new ChannelLogicHandler());
        }
        });

在通道激活时打开与DB的连接：

@Override
public void channelActive(ChannelHandlerContext ctx) throws Exception
{
    dbConnection = DriverManager.getConnection(dbConnectParams[0], dbConnectParams[1], dbConnectParams[2]);
    if (dbConnection == null)
    throw new SQLException("Connection to database failed");

    super.channelActive(ctx);
}

...并且关闭通道上的非活动连接。

但据我所知，Netty将其自己的线程池中的每个线程与其所有生命周期的通道相关联，因此在我的情况下，使用DefaultEventExecutorGroup（4）作为逻辑处理程序意味着所有通道将以四个线程提供并且对于任何给定的通道，将仅使用来自线程池的一个线程。因此，每个执行程序线程维护一个数据库连接就足以提供数据完整性而无需任何锁定（具有适当的事务隔离级别）。所以我的问题是可以在线程池中为每个线程关联一个数据库连接，以便在启动线程时建立每个连接（或者在第一个通道与它关联时），以及如何实现它？

Answer 1

猜猜我自己找到了解决方案 - 每个ChannelHandlerContext（ctx）都有自己的EventExecutor，它本质上就是线程。所以我使用hashmap来分配与执行程序的数据库连接。在代码中：

<div id="mydiv2" style="display:none;">

// ........................

//Declare hashmap in main server class
private final HashMap<EventExecutor,java.sql.Connection> execConsMap = new HashMap<>(4);

数据库连接在服务器停止时关闭：

    public class ChannelLogicHandler extends ChannelInboundHandlerAdapter
{
private java.sql.Connection dbConnection = null; //DB connection saved as private member of logic handler
//........................

@Override
public void channelActive(ChannelHandlerContext ctx) throws Exception
{
    EventExecutor ex = ctx.executor(); //Get channel executor
    synchronized(execConsMap)
    {
    if (execConsMap.containsKey(ex)) //If already processed get DB connection from hashmap
    {
        dbConnection = execConsMap.get(ex);
    }
    else //Else create new connection and save in hashmap
    {
        java.sql.Connection dbc = DriverManager.getConnection(dbConnectParams[0], dbConnectParams[1], dbConnectParams[2]);
        if (dbc != null)
        {
        execConsMap.put(ex, dbc);
        dbConnection = dbc;
        }
        else
        {
        throw new SQLException("Connection to database failed");
        }
    }
    }

    System.out.println("New client connected");
    super.channelActive(ctx);
}

@Override
public void channelInactive(ChannelHandlerContext ctx) throws Exception
{
    try
    {
    if (!dbConnection.getAutoCommit())
    {
        dbConnection.rollback();
        dbConnection.setAutoCommit(true);
    }
    }
    catch (SQLException e) 
    { 
    e.printStackTrace(); 
    }

    System.out.println("Client disconnected");
    super.channelInactive(ctx);
}

Answer 2

如果您想要每个线程的变量，可以使用ThreadLocal字段。

这样的事情：

private static final ThreadLocal<DatabaseConnection> databaseConnection =
     new ThreadLocal<DatabaseConnection>() {
         @Override protected DatabaseConnection initialValue() {
             return DriverManager.getConnection(dbConnectParams[0], dbConnectParams[1], dbConnectParams[2]);
     }
 };

这样做，每个线程将具有不同的数据库连接。首次调用get方法时，将通过调用创建数据库连接的initialValue()方法初始化线程的连接。后续调用将返回相同的先前值，但您也可以手动为此字段设置新值。

将数据库连接与Netty 4事件执行程序

2 个答案: