前言

  不知道大家国庆节过的咋样，有没有学习呢？我是闲着没事就写点东西，本文篇幅较短，只是实现了心跳消息的处理，下面进入正文。

正文

  先说说为什么要增加心跳消息，Socket在连接过程中会发生很多的意外，比如网络问题、连接不上、ip地址不对、异常断连，而心跳消息说白了就是通过指定时间去收发消息，通过收发消息就能知道连接双方处于什么状态。知道是什么状态就知道该怎么处理。

一、状态判断

在开始写之前我们需要先知道状态有哪些，直接在IOException中进行处理。

			catch (e: IOException) {
    
    
                e.printStackTrace()
                when (e) {
    
    
                    is SocketTimeoutException -> {
    
    
                        Log.e(TAG, "连接超时，正在重连")
                    }
                    is NoRouteToHostException -> {
    
    
                        Log.e(TAG, "该地址不存在，请检查")
                    }
                    is ConnectException -> {
    
    
                        Log.e(TAG, "连接异常或被拒绝，请检查")
                    }
                    is SocketException -> {
    
    
                        when (e.message) {
    
    
                            "Already connected" -> Log.e(TAG, "连接异常或被拒绝，请检查")
                            "Socket closed" -> Log.e(TAG, "连接已关闭")
                        }
                    }
                }
            }

这里就可以直接做处理，我打印了日志，你可以在连接过程中，进行测试，这段代码添加的位置如下图所示：

这个代码对于服务端和客户端都一样。

二、心跳消息发送

  心跳消息需要在什么时候发送呢？在客户端连接了服务端之后，由客户端发起心跳消息，进入SocketClient，增加如下代码：

	private const val HEART_SPACETIME = 3 * 1000

    private val mHandler: Handler = Handler()

这里定义了一个心跳间隔时间，也就是说，3秒发送一次心跳，下面写一个发送心跳的方法。

	private val mHeartRunnable = Runnable {
    
     sendHeartbeat() }

    /**
     * 发送心跳消息
     */
    private fun sendHeartbeat() {
    
    
        if (clientThreadPool == null) {
    
    
            clientThreadPool = Executors.newSingleThreadExecutor()
        }
        val msg = "洞幺洞幺，呼叫洞拐，听到请回答，听到请回答，Over!"
        clientThreadPool?.execute {
    
    
            if (socket == null) {
    
    
                mCallback.otherMsg("客户端还未连接")
                return@execute
            }
            if (socket!!.isClosed) {
    
    
                mCallback.otherMsg("Socket已关闭")
                return@execute
            }
            outputStream = socket?.getOutputStream()
            try {
    
    
                outputStream?.write(msg.toByteArray())
                outputStream?.flush()
                //发送成功以后，重新建立一个心跳消息
                mHandler.postDelayed(mHeartRunnable, HEART_SPACETIME.toLong())
                Log.i(TAG, msg)
            } catch (e: IOException) {
    
    
                e.printStackTrace()
                mCallback.otherMsg("向服务端发送消息: $msg 失败")
            }
        }
    }

  心跳消息中我定义了消息的内容，服务端收到这个内容就表示为心跳消息，这个消息内容还是有待商榷的，先实现功能就行，在这里发送消息之后，进行延时发送下一次心跳消息，这里打印了发送的心跳消息了，因为是连接到服务端之后就开始发送，所以还需要在connectServer()函数中处理，增加代码如下图所示：

最后就是处理服务端回复的服务端消息，修改ClientThread中的代码如下图所示：

  这里判断一下服务端回复的内容是什么，如果是我们想要的内容就表明这是一个心跳消息，则只打印就行了，不需要回调的页面上进行显示，这里的内容因为你可能也会在输入框输入，所以会存在问题。

三、心跳消息回复

  当服务端收到客户端发送的心跳时就要回复消息给客户端，这样客户端就知道服务端一直都在，服务端的代码就相对来说简单很多了，在SocketServer中增加一个函数，代码如下：

	fun replyHeartbeat() {
    
    
        if (serverThreadPool == null) {
    
    
            serverThreadPool = Executors.newCachedThreadPool()
        }
        val msg = "洞拐收到，洞拐收到，Over!"
        serverThreadPool?.execute {
    
    
            if (socket == null) {
    
    
                mCallback.otherMsg("客户端还未连接")
                return@execute
            }
            if (socket!!.isClosed) {
    
    
                mCallback.otherMsg("Socket已关闭")
                return@execute
            }
            outputStream = socket!!.getOutputStream()
            try {
    
    
                outputStream.write(msg.toByteArray())
                outputStream.flush()
            } catch (e: IOException) {
    
    
                e.printStackTrace()
                mCallback.otherMsg("向客户端发送消息: $msg 失败")
            }
        }
    }

  在什么时候调用它呢？当然是在收到客户端发送消息的时候调用，只不过和客户端收到服务端消息一样，同样需要判断一下才行，如下图所示：

  代码是不是很简单呢？其他的地方都不需要怎么去改动了，运行一下，让你的客户端连接这个服务端，然后看控制台的日志打印，如下图所示：

四、源码

如果你觉得代码对你有帮助的话，不妨Fork或者Star一下~

源码地址：SocketDemo