NIO源码阅读(4)-SocketChannel

概述

ServerSocketChannel主要用于服务端,而在客户端,经常打交道的是SocketChannel,这篇文章将介绍SocketChannel是如何实现的。

实例化

在之前介绍SelectorProvider的时候曾经介绍过,NIO channel的创建都是通过SelecorProvider实现的:

public SocketChannel openSocketChannel() throws IOException {
        return new SocketChannelImpl(this);
    }

当然SocketChannel也提供了快捷方法open:

public static SocketChannel open() throws IOException {
    return SelectorProvider.provider().openSocketChannel();
}

public static SocketChannel open(SocketAddress remote)
    throws IOException
{
    SocketChannel sc = open();//调用SelectorProvider实现,默认是堵塞的
    try {
        sc.connect(remote);
    } catch (Throwable x) {
        try {
            sc.close();
        } catch (Throwable suppressed) {
            x.addSuppressed(suppressed);
        }
        throw x;
    }
    assert sc.isConnected();
    return sc;
}

SocketChannel是个抽象类,SelectorProvider返回的是SocketChannelImpl,继承自SocketChannel;一般情况下,当采用异步方式时,使用不带参数的open方法比较常见,而且会调用configureBlocking设置非堵塞;

SocketChannelImpl构造函数定义如下:

SocketChannelImpl(SelectorProvider sp) throws IOException {
    super(sp);
    //调用inux的socket函数,true表示TCP
    this.fd = Net.socket(true);
    //由于FileDescriptor未提供访问fdVal的方法,通过JNI获取
    this.fdVal = IOUtil.fdVal(fd);
    this.state = ST_UNCONNECTED;//设置状态为未连接
}

connect

调用connect方法连接到远程服务器,其源码如下:

public boolean connect(SocketAddress sa) throws IOException {
    int localPort = 0;
    //注意加速顺序,整个类保存一致
    synchronized (readLock) {
        synchronized (writeLock) {
            ensureOpenAndUnconnected();//检查连接状态
            InetSocketAddress isa = Net.checkAddress(sa);
            synchronized (blockingLock()) {
                int n = 0;
                try {
                    try {
                       //支持线程中断,通过设置当前线程的Interruptible blocker属性实现,由于前面已经介绍过多次,此处不再介绍
                        begin();
                        synchronized (stateLock) {
                           //默认为open, 除非调用了close方法
                            if (!isOpen()) {
                                return false;
                            }
                            //只有未绑定本地地址也就是说未调用bind方法才执行,该方法在ServerSocketChannel中介绍过
                            if (localAddress == null) {
                                NetHooks.beforeTcpConnect(fd,
                                                       isa.getAddress(),
                                                       isa.getPort());
                            }
                            //记录当前线程
                            readerThread = NativeThread.current();
                        }
                        for (;;) {
                            InetAddress ia = isa.getAddress();
                            if (ia.isAnyLocalAddress())
                                ia = InetAddress.getLocalHost();
                           //调用Linux的connect函数实现,如果采用堵塞模式,会一直等待,直到成功或出现异常,后面会介绍
                            n = Net.connect(fd,
                                            ia,
                                            isa.getPort());
                            if (  (n == IOStatus.INTERRUPTED)
                                  && isOpen())
                                continue;
                            break;
                        }

                    } finally {
                       //清空readerThread
                        readerCleanup();
                       //和begin成对出现,当线程中断时,抛出ClosedByInterruptException
                        end((n > 0) || (n == IOStatus.UNAVAILABLE));
                        assert IOStatus.check(n);
                    }
                } catch (IOException x) {
                    close(); //出现异常,关闭channel
                    throw x;
                }
                synchronized (stateLock) {
                    remoteAddress = isa;
                    if (n > 0) {//如果连接成功,更新状态为ST_CONNECTED
                        state = ST_CONNECTED;
                       //如果未调用bind方法,操作系统内核会自动分配地址和端口;否则返回bind的地址和端口
                        if (isOpen())
                            localAddress = Net.localAddress(fd);
                        return true;
                    }
                   //如果是非堵塞模式,而且未立即返回成功,更新状态为ST_PENDING;
                   //由此可见,该状态只有非堵塞时才会存在
                    if (!isBlocking())
                        state = ST_PENDING;
                    else
                        assert false;
                }
            }
            return false;
        }
    }
}

上面的代码中会调用Net.connect方法,该方法最终会调用native方法:

JNIEXPORT jint JNICALL
Java_sun_nio_ch_Net_connect0(JNIEnv *env, jclass clazz, jboolean preferIPv6,
                             jobject fdo, jobject iao, jint port)
{
    SOCKADDR sa;
    int sa_len = SOCKADDR_LEN;
    int rv;
    //地址转换为struct sockaddr格式
    if (NET_InetAddressToSockaddr(env, iao, port, (struct sockaddr *) &sa,
                                  &sa_len, preferIPv6) != 0)
    {
      return IOS_THROWN;
    }
   //传入fd和sockaddr,与远程服务器建立连接,一般就是TCP三次握手
   //如果设置了configureBlocking(false),不会堵塞,否则会堵塞一直到超时或出现异常
    rv = connect(fdval(env, fdo), (struct sockaddr *)&sa, sa_len);
    if (rv != 0) {//0表示连接成功,失败时通过errno获取具体原因
        if (errno == EINPROGRESS) {//非堵塞,连接还未建立(-2)
            return IOS_UNAVAILABLE;
        } else if (errno == EINTR) {//中断(-3)
            return IOS_INTERRUPTED;
        }
        return handleSocketError(env, errno); //出错
    }
    return 1;//连接建立,一般TCP连接连接都需要时间,因此除非是本地网络,一般情况下非堵塞模式返回IOS_UNAVAILABLE比较多;
}

从上面可以看到,如果是非堵塞,而且连接未马上建立成功,此时状态为ST_PENDING,那么什么时候会变为ST_CONNECTED呢?是否有什么方法可以查询状态或者等待连接完成呢?

finishConnect

带着上面的问题,我们一起看看finishConnect的实现,代码比较长,我只保留比较重要的部分:

public boolean finishConnect() throws IOException {
    synchronized (readLock) {
        synchronized (writeLock) {
            int n = 0;
            try {
                try {
                    synchronized (blockingLock()) {
                        if (!isBlocking()) {//非堵塞模式
                            for (;;) {
                                n = checkConnect(fd, false,
                                                 readyToConnect);
                                if (  (n == IOStatus.INTERRUPTED)
                                      && isOpen())
                                    continue;
                                break;//除非被中断,否则退出
                            }
                        } else {//堵塞模式
                            for (;;) {
                                n = checkConnect(fd, true,
                                                 readyToConnect);
                                if (n == 0) {//除非>0,否则自旋,继续等待
                                    continue;
                                }
                                if (  (n == IOStatus.INTERRUPTED)
                                      && isOpen())
                                    continue;
                                break;
                            }
                        }
                    }
                } finally {
                    synchronized (stateLock) {
                        if (state == ST_KILLPENDING) {//调用了close方法
                            kill();
                            n = 0;
                        }
                    }
                }
            } catch (IOException x) { //异常发生,关闭channel
                close();
                throw x;
            }
            if (n > 0) {//连接成功
                synchronized (stateLock) {
                    state = ST_CONNECTED;//更新状态
                    if (isOpen())
                        localAddress = Net.localAddress(fd);
                }
                return true;
            }
            return false;
        }
    }
}

从上面看到,如果是堵塞模式,会一直循环检查状态,直到成功或发生异常;而非堵塞模式下,检查完,马上结束循环;
上面的代码是通过checkConnect检查连接状态,下面看看它是如何实现的:

JNIEXPORT jint JNICALL
Java_sun_nio_ch_SocketChannelImpl_checkConnect(JNIEnv *env, jobject this,
                                               jobject fdo, jboolean block,
                                               jboolean ready)
{
    int error = 0;
    socklen_t n = sizeof(int);
    jint fd = fdval(env, fdo);//获取FileDescriptor中的fd
    int result = 0;
    struct pollfd poller;

    poller.revents = 1;//返回的事件
    if (!ready) {
        poller.fd = fd;//文件描述符
        poller.events = POLLOUT;//请求的事件:写事件
        poller.revents = 0;//返回的事件
       //第3个参数表示超时时间(毫秒)
       //-1表示永远不会超时,0表示立即返回,不阻塞进程
        result = poll(&poller, 1, block ? -1 : 0);
        if (result < 0) {//小于0表示调用失败
            JNU_ThrowIOExceptionWithLastError(env, "Poll failed");
            return IOS_THROWN;
        }
        //非堵塞时,0表示没有准备好的连接
        if (!block && (result == 0))
            return IOS_UNAVAILABLE;
    }

    if (poller.revents) {//准备好写或出现错误的socket数量>0
        errno = 0;
        result = getsockopt(fd, SOL_SOCKET, SO_ERROR, &error, &n);
        if (result < 0) {//出错
            handleSocketError(env, errno);
            return JNI_FALSE;
        } else if (error) {//发生错误
            handleSocketError(env, error);
            return JNI_FALSE;
        }
        return 1;//socket已经准备好,可写,即连接已经建立好
    }
    return 0;
}

从上面的源码看到,底层是通过poll查询socket的状态,从而判断连接是否建立成功;
由于在非堵塞模式下,finishConnect方法会立即返回,因此不大建议用循环的方式判断连接是否建立,而是建议注册到Selector,通过ops=OP_CONNECT获取连接完成的SelectionKey,然后调用finishConnect完成连接的建立;
那么finishConnect是否可以不调用呢?答案是否,因为只有finishConnect中会将状态更新为ST_CONNECTED,而在调用read和write时都会对状态进行判断;

另外还有特别说一下的是translateReadyOps方法,在EpollSelectorImpl的doSelect方法中会调用channel的translateAndSetReadyOps方法,在该方法中设置SocketChannel的readyToConnect变量;从上面代码知道,finishConnect的时候,如果发现readyToConnect=true,将不会调用poll来查询状态;

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,590评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 86,808评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,151评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,779评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,773评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,656评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,022评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,678评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,038评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,659评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,756评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,411评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,005评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,973评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,053评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,495评论 2 343

推荐阅读更多精彩内容