netty-客户端连接.connect()方法 源码分析.md

应用程序

官方demo

/*

 * Copyright 2012 The Netty Project

 *

 * The Netty Project 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 io.netty.example.echo;



import io.netty.bootstrap.Bootstrap;

import io.netty.channel.ChannelFuture;

import io.netty.channel.ChannelInitializer;

import io.netty.channel.ChannelOption;

import io.netty.channel.ChannelPipeline;

import io.netty.channel.EventLoopGroup;

import io.netty.channel.nio.NioEventLoopGroup;

import io.netty.channel.socket.SocketChannel;

import io.netty.channel.socket.nio.NioSocketChannel;

import io.netty.handler.ssl.SslContext;

import io.netty.handler.ssl.SslContextBuilder;

import io.netty.handler.ssl.util.InsecureTrustManagerFactory;



/**

 * Sends one message when a connection is open and echoes back any received

 * data to the server.  Simply put, the echo client initiates the ping-pong

 * traffic between the echo client and server by sending the first message to

 * the server.

 */

public final class EchoClient {



    static final boolean SSL = System.getProperty("ssl") != null;

    static final String HOST = System.getProperty("host", "127.0.0.1");

    static final int PORT = Integer.parseInt(System.getProperty("port", "8007"));

    static final int SIZE = Integer.parseInt(System.getProperty("size", "256"));



    public static void main(String[] args) throws Exception {

        // Configure SSL.git

        final SslContext sslCtx;

        if (SSL) {

            sslCtx = SslContextBuilder.forClient()

                .trustManager(InsecureTrustManagerFactory.INSTANCE).build();

        } else {

            sslCtx = null;

        }



        // Configure the client.

        EventLoopGroup group = new NioEventLoopGroup();

        try {

            Bootstrap b = new Bootstrap();

            b.group(group)

             .channel(NioSocketChannel.class)

             .option(ChannelOption.TCP_NODELAY, true)

             .handler(new ChannelInitializer<SocketChannel>() {

                 @Override

                 public void initChannel(SocketChannel ch) throws Exception {

                     ChannelPipeline p = ch.pipeline();

                     if (sslCtx != null) {

                         p.addLast(sslCtx.newHandler(ch.alloc(), HOST, PORT));

                     }

                     //p.addLast(new LoggingHandler(LogLevel.INFO));

                     p.addLast(new EchoClientHandler());

                 }

             });



            // Start the client.

            ChannelFuture f = b.connect(HOST, PORT).sync(); //连接方法



            // Wait until the connection is closed.

            f.channel().closeFuture().sync();

        } finally {

            // Shut down the event loop to terminate all threads.

            group.shutdownGracefully();

        }

    }

}


复制代码

netty源码

1.启动类
public class Bootstrap extends AbstractBootstrap<Bootstrap, Channel> {


/**

     * @see #connect()

     */

    private ChannelFuture doResolveAndConnect(final SocketAddress remoteAddress, final SocketAddress localAddress) {

        final ChannelFuture regFuture = initAndRegister();

        final Channel channel = regFuture.channel();



        if (regFuture.isDone()) {

            if (!regFuture.isSuccess()) {

                return regFuture;

            }

            return doResolveAndConnect0(channel, remoteAddress, localAddress, channel.newPromise());

        } else {

            // Registration future is almost always fulfilled already, but just in case it's not.

            final PendingRegistrationPromise promise = new PendingRegistrationPromise(channel);

            regFuture.addListener(new ChannelFutureListener() {

                @Override

                public void operationComplete(ChannelFuture future) throws Exception {

                    // Directly obtain the cause and do a null check so we only need one volatile read in case of a

                    // failure.

                    Throwable cause = future.cause();

                    if (cause != null) {

                        // Registration on the EventLoop failed so fail the ChannelPromise directly to not cause an

                        // IllegalStateException once we try to access the EventLoop of the Channel.

                        promise.setFailure(cause);

                    } else {

                        // Registration was successful, so set the correct executor to use.

                        // See https://github.com/netty/netty/issues/2586

                        promise.registered();

                        doResolveAndConnect0(channel, remoteAddress, localAddress, promise);

                    }

                }

            });

            return promise;

        }

    }





private ChannelFuture doResolveAndConnect0(final Channel channel, SocketAddress remoteAddress,

                                               final SocketAddress localAddress, final ChannelPromise promise) {

        try {

            final EventLoop eventLoop = channel.eventLoop();

            final AddressResolver<SocketAddress> resolver = this.resolver.getResolver(eventLoop);



            if (!resolver.isSupported(remoteAddress) || resolver.isResolved(remoteAddress)) {

                // Resolver has no idea about what to do with the specified remote address or it's resolved already.

                doConnect(remoteAddress, localAddress, promise);

                return promise;

            }



            final Future<SocketAddress> resolveFuture = resolver.resolve(remoteAddress);



            if (resolveFuture.isDone()) {

                final Throwable resolveFailureCause = resolveFuture.cause();



                if (resolveFailureCause != null) {

                    // Failed to resolve immediately

                    channel.close();

                    promise.setFailure(resolveFailureCause);

                } else {

                    // Succeeded to resolve immediately; cached? (or did a blocking lookup)

                    doConnect(resolveFuture.getNow(), localAddress, promise);

                }

                return promise;

            }



            // Wait until the name resolution is finished.

            resolveFuture.addListener(new FutureListener<SocketAddress>() {

                @Override

                public void operationComplete(Future<SocketAddress> future) throws Exception {

                    if (future.cause() != null) {

                        channel.close();

                        promise.setFailure(future.cause());

                    } else {

                        doConnect(future.getNow(), localAddress, promise);

                    }

                }

            });

        } catch (Throwable cause) {

            promise.tryFailure(cause);

        }

        return promise;

    }



private static void doConnect(

            final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise connectPromise) {



        // This method is invoked before channelRegistered() is triggered.  Give user handlers a chance to set up

        // the pipeline in its channelRegistered() implementation.

        final Channel channel = connectPromise.channel();

        channel.eventLoop().execute(new Runnable() {

            @Override

            public void run() {

                if (localAddress == null) {

                    channel.connect(remoteAddress, connectPromise);

                } else {

                    channel.connect(remoteAddress, localAddress, connectPromise);

                }

                connectPromise.addListener(ChannelFutureListener.CLOSE_ON_FAILURE);

            }

        });

    }





复制代码
2.通道.连接()

abstract class AbstractChannelHandlerContext extends DefaultAttributeMap

        implements ChannelHandlerContext, ResourceLeakHint {



@Override

    public ChannelFuture connect(

            final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise promise) {



        if (remoteAddress == null) {

            throw new NullPointerException("remoteAddress");

        }

        if (isNotValidPromise(promise, false)) {

            // cancelled

            return promise;

        }



        final AbstractChannelHandlerContext next = findContextOutbound();

        EventExecutor executor = next.executor();

        if (executor.inEventLoop()) {

            next.invokeConnect(remoteAddress, localAddress, promise);

        } else {

            safeExecute(executor, new Runnable() {

                @Override

                public void run() {

                    next.invokeConnect(remoteAddress, localAddress, promise);

                }

            }, promise, null);

        }

        return promise;

    }







private void invokeConnect(SocketAddress remoteAddress, SocketAddress localAddress, ChannelPromise promise) {

        if (invokeHandler()) {

            try {

                ((ChannelOutboundHandler) handler()).connect(this, remoteAddress, localAddress, promise);

            } catch (Throwable t) {

                notifyOutboundHandlerException(t, promise);

            }

        } else {

            connect(remoteAddress, localAddress, promise);

        }

    }





复制代码
3.通道实现类
class SocketChannelImpl

    extends SocketChannel

    implements SelChImpl

{


public boolean connect(SocketAddress sa) throws IOException {

        int localPort = 0;



        synchronized (readLock) {

            synchronized (writeLock) {

                ensureOpenAndUnconnected();

                InetSocketAddress isa = Net.checkAddress(sa);

                SecurityManager sm = System.getSecurityManager();

                if (sm != null)

                    sm.checkConnect(isa.getAddress().getHostAddress(),

                                    isa.getPort());

                synchronized (blockingLock()) {

                    int n = 0;

                    try {

                        try {

                            begin();

                            synchronized (stateLock) {

                                if (!isOpen()) {

                                    return false;

                                }

                                // notify hook only if unbound

                                if (localAddress == null) {

                                    NetHooks.beforeTcpConnect(fd,

                                                           isa.getAddress(),

                                                           isa.getPort());

                                }

                                readerThread = NativeThread.current();

                            }

                            for (;;) {

                                InetAddress ia = isa.getAddress();

                                if (ia.isAnyLocalAddress())

                                    ia = InetAddress.getLocalHost();

                                n = Net.connect(fd,

                                                ia,

                                                isa.getPort()); //最终还是调用nio.channel包里的Net类——》native方法 所以netty和nio一样 只是封装了nio nio也是调用Net.native()方法

                                if (  (n == IOStatus.INTERRUPTED)

                                      && isOpen())

                                    continue;

                                break;

                            }



                        } finally {

                            readerCleanup();

                            end((n > 0) || (n == IOStatus.UNAVAILABLE));

                            assert IOStatus.check(n);

                        }

                    } catch (IOException x) {

                        // If an exception was thrown, close the channel after

                        // invoking end() so as to avoid bogus

                        // AsynchronousCloseExceptions

                        close();

                        throw x;

                    }

                    synchronized (stateLock) {

                        remoteAddress = isa;

                        if (n > 0) {



                            // Connection succeeded; disallow further

                            // invocation

                            state = ST_CONNECTED;

                            if (isOpen())

                                localAddress = Net.localAddress(fd);

                            return true;

                        }

                        // If nonblocking and no exception then connection

                        // pending; disallow another invocation

                        if (!isBlocking())

                            state = ST_PENDING;

                        else

                            assert false;

                    }

                }

                return false;

            }

        }

    }

复制代码

转载于:https://juejin.im/post/5c54126ae51d457f9f2c0f45

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值