| /* |
| * 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 com.alibaba.dubbo.remoting.transport.netty4; |
| |
| import com.alibaba.dubbo.common.Constants; |
| import com.alibaba.dubbo.common.URL; |
| import com.alibaba.dubbo.common.logger.Logger; |
| import com.alibaba.dubbo.common.logger.LoggerFactory; |
| import com.alibaba.dubbo.common.utils.ExecutorUtil; |
| import com.alibaba.dubbo.common.utils.NetUtils; |
| import com.alibaba.dubbo.remoting.Channel; |
| import com.alibaba.dubbo.remoting.ChannelHandler; |
| import com.alibaba.dubbo.remoting.RemotingException; |
| import com.alibaba.dubbo.remoting.Server; |
| import com.alibaba.dubbo.remoting.transport.AbstractServer; |
| import com.alibaba.dubbo.remoting.transport.dispatcher.ChannelHandlers; |
| |
| import io.netty.bootstrap.ServerBootstrap; |
| import io.netty.buffer.PooledByteBufAllocator; |
| import io.netty.channel.ChannelFuture; |
| 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.nio.NioServerSocketChannel; |
| import io.netty.channel.socket.nio.NioSocketChannel; |
| import io.netty.util.concurrent.DefaultThreadFactory; |
| |
| import java.net.InetSocketAddress; |
| import java.util.Collection; |
| import java.util.HashSet; |
| import java.util.Map; |
| |
| /** |
| * NettyServer |
| */ |
| public class NettyServer extends AbstractServer implements Server { |
| |
| private static final Logger logger = LoggerFactory.getLogger(NettyServer.class); |
| |
| private Map<String, Channel> channels; // <ip:port, channel> |
| |
| private ServerBootstrap bootstrap; |
| |
| private io.netty.channel.Channel channel; |
| |
| private EventLoopGroup bossGroup; |
| private EventLoopGroup workerGroup; |
| |
| public NettyServer(URL url, ChannelHandler handler) throws RemotingException { |
| super(url, ChannelHandlers.wrap(handler, ExecutorUtil.setThreadName(url, SERVER_THREAD_POOL_NAME))); |
| } |
| |
| @Override |
| protected void doOpen() throws Throwable { |
| bootstrap = new ServerBootstrap(); |
| |
| bossGroup = new NioEventLoopGroup(1, new DefaultThreadFactory("NettyServerBoss", true)); |
| workerGroup = new NioEventLoopGroup(getUrl().getPositiveParameter(Constants.IO_THREADS_KEY, Constants.DEFAULT_IO_THREADS), |
| new DefaultThreadFactory("NettyServerWorker", true)); |
| |
| final NettyServerHandler nettyServerHandler = new NettyServerHandler(getUrl(), this); |
| channels = nettyServerHandler.getChannels(); |
| |
| bootstrap.group(bossGroup, workerGroup) |
| .channel(NioServerSocketChannel.class) |
| .childOption(ChannelOption.TCP_NODELAY, Boolean.TRUE) |
| .childOption(ChannelOption.SO_REUSEADDR, Boolean.TRUE) |
| .childOption(ChannelOption.ALLOCATOR, PooledByteBufAllocator.DEFAULT) |
| .childHandler(new ChannelInitializer<NioSocketChannel>() { |
| @Override |
| protected void initChannel(NioSocketChannel ch) throws Exception { |
| NettyCodecAdapter adapter = new NettyCodecAdapter(getCodec(), getUrl(), NettyServer.this); |
| ch.pipeline()//.addLast("logging",new LoggingHandler(LogLevel.INFO))//for debug |
| .addLast("decoder", adapter.getDecoder()) |
| .addLast("encoder", adapter.getEncoder()) |
| .addLast("handler", nettyServerHandler); |
| } |
| }); |
| // bind |
| ChannelFuture channelFuture = bootstrap.bind(getBindAddress()); |
| channelFuture.syncUninterruptibly(); |
| channel = channelFuture.channel(); |
| |
| } |
| |
| @Override |
| protected void doClose() throws Throwable { |
| try { |
| if (channel != null) { |
| // unbind. |
| channel.close(); |
| } |
| } catch (Throwable e) { |
| logger.warn(e.getMessage(), e); |
| } |
| try { |
| Collection<com.alibaba.dubbo.remoting.Channel> channels = getChannels(); |
| if (channels != null && channels.size() > 0) { |
| for (com.alibaba.dubbo.remoting.Channel channel : channels) { |
| try { |
| channel.close(); |
| } catch (Throwable e) { |
| logger.warn(e.getMessage(), e); |
| } |
| } |
| } |
| } catch (Throwable e) { |
| logger.warn(e.getMessage(), e); |
| } |
| try { |
| if (bootstrap != null) { |
| bossGroup.shutdownGracefully(); |
| workerGroup.shutdownGracefully(); |
| } |
| } catch (Throwable e) { |
| logger.warn(e.getMessage(), e); |
| } |
| try { |
| if (channels != null) { |
| channels.clear(); |
| } |
| } catch (Throwable e) { |
| logger.warn(e.getMessage(), e); |
| } |
| } |
| |
| @Override |
| public Collection<Channel> getChannels() { |
| Collection<Channel> chs = new HashSet<Channel>(); |
| for (Channel channel : this.channels.values()) { |
| if (channel.isConnected()) { |
| chs.add(channel); |
| } else { |
| channels.remove(NetUtils.toAddressString(channel.getRemoteAddress())); |
| } |
| } |
| return chs; |
| } |
| |
| @Override |
| public Channel getChannel(InetSocketAddress remoteAddress) { |
| return channels.get(NetUtils.toAddressString(remoteAddress)); |
| } |
| |
| @Override |
| public boolean isBound() { |
| return channel.isActive(); |
| } |
| |
| } |