[Golang] 从零开始写Socket Server(3): 对长、短连接的处理策略(模拟心跳)

原创 2015年08月17日 18:36:32

    通过前两章,我们成功是写出了一套凑合能用的Server和Client,并在二者之间实现了通过协议交流。这么一来,一个简易的socket通讯框架已经初具雏形了,那么我们接下来做的,就是想办法让这个框架更加稳定,茁壮~

    作为一个可能会和很多Client进行通讯交互的Server,首先要保证的就是整个Server运行状态的稳定性,因此在和Client建立连接通讯的时候,确保连接的及时断开非常重要,否则一旦和多个客户端建立不关闭的长连接,对于服务器资源的占用是很可怕的。因此,我们需要针对可能出现的短连接和长连接,设定不同的限制策略。

    针对短连接,我们可以使用golang中的net包自带的timeout函数,一共有三个,分别是:


func (*IPConn) SetDeadline
func (c *IPConn) SetDeadline(t time.Time) error

func (*IPConn) SetReadDeadline
func (c *IPConn) SetReadDeadline(t time.Time) error

func (*IPConn) SetWriteDeadline
func (c *IPConn) SetWriteDeadline(t time.Time) error


    如果想要给服务器设置短连接的timeout,我们就可以这么写:

netListen, err := net.Listen("tcp", Port)
	Log("Waiting for clients")
	for {
		conn, err := netListen.Accept()
		if err != nil {
			continue
		}

		conn.SetReadDeadline(time.Now().Add(time.Duration(10) * time.Second))
    这里的三个函数都是用于设置每次socket连接能够维持的最长时间,一旦超过设置的timeout后,便会在Server端自动断开连接。其中SetReadline, SetWriteline设置的是读取和写入的最长持续时间,而SetDeadline则同时包含了 SetReadline, SetWriteline两个函数。

    通过这样设定,每个和Server通讯的Client连接时长最长也不会超过10s了~~


    搞定短连接后,接下来就是针对长连接的处理策略了~~

    作为长连接,由于我们往往很难确定什么时候会中断连接,因此并不能像处理短连接那样简单粗暴的设定一个timeout就可以搞定,而在Golang的net包中,并没有针对长连接的函数,因此需要我们自己设计并实现针对长连接的处理策略啦~

    针对socke长连接,常见的做法是在Server和Socket之间设计通讯机制,当两者之间没有信息交互时,双方便会定时发送数据包(心跳),以维持连接状态。


    这种方法是目前使用相对比较多的做法,但是开销相对也较大,特别是当Server和多个client保持长连接的时候,并发会比较高,考虑到公司的业务需求,我最后选择了逻辑相对简单,开销相对较小的策略:

    当Server每次收到Client发到的信息之后,便会开始心跳计时,如果在心跳计时结束之前没有再次收到Client发来的信息,那么便会断开跟Client的连接。而一旦在设定时间内再次收到Client发来的信息,那么Server便会重置计时器,再次重新进行心跳计时,直到超时断开连接为止。

下面就是实现该计时的代码:

//长连接入口
func handleConnection(conn net.Conn,timeout int) {

	buffer := make([]byte, 2048)
	for {
		n, err := conn.Read(buffer)

		if err != nil {
			LogErr(conn.RemoteAddr().String(), " connection error: ", err)
			return
		}
		Data :=(buffer[:n])
		messnager := make(chan byte)
		//心跳计时
		go HeartBeating(conn,messnager,timeout)
		//检测每次Client是否有数据传来
		go GravelChannel(Data,messnager)
		Log( "receive data length:",n)
		Log(conn.RemoteAddr().String(), "receive data string:", string(Data))

	}
}

//心跳计时,根据GravelChannel判断Client是否在设定时间内发来信息
func HeartBeating(conn net.Conn, readerChannel chan byte,timeout int) {
		select {
		case fk := <-readerChannel:
			Log(conn.RemoteAddr().String(), "receive data string:", string(fk))
			conn.SetDeadline(time.Now().Add(time.Duration(timeout) * time.Second))
			//conn.SetReadDeadline(time.Now().Add(time.Duration(5) * time.Second))
			break
		case <-time.After(time.Second*5):
			Log("It's really weird to get Nothing!!!")
			conn.Close()
		}

}

func GravelChannel(n []byte,mess chan byte){
	for _ , v := range n{
		mess <- v
	}
	close(mess)
}


func Log(v ...interface{}) {
	log.Println(v...)
}


    这样,就可以成功实现对于长连接的处理了~~,我们可以这么进行测试:

func sender(conn net.Conn) {
	for i := 0; i <5; i++ {
		words:= strconv.Itoa(i)+"This is a test for long conn" 
		conn.Write([]byte(words))
		time.Sleep(2*time.Second)

	}
	fmt.Println("send over")

}

    可以发现,Sender函数中time.Sleep阻塞的时间设定的比Server中的timeout短的时候,Client端的信息可以自由的发送到循环结束,而当我们设定Sender函数的阻塞时间较长时,就只能发出第一次循环的信息。



我已经把SocketServer系列的代码整合到了一起,发布到了我个人的github上:点击链接, 希望大家有兴趣的可以学习star一下~


版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ahlxt123/article/details/47726783

如何实现支持数亿用户的长连消息系统 | Golang高并发案例

此文是根据周洋在【高可用架构群】中的分享内容整理而成,转发请注明出处。 周洋,360手机助手技术经理及架构师,负责360长连接消息系统,360手机助手架构的开发与维护。 不知道咱们群名什么时候改为...
  • RandyLeonard
  • RandyLeonard
  • 2016-01-18 16:52:45
  • 9585

golang websocket长连接

设计思路:每个websocket允许的连接都是有时间限制的,超时后服务端会自动断开连接,那么长连接就在服务端发出断开连接信息后客户端检测断开信息再次发起连接请求,中间再通过握手信息确保客户端同服务器处...
  • strangerLove
  • strangerLove
  • 2016-06-07 12:18:33
  • 1751

golang 长短连接处理

golang 长连接 短连接
  • chenwenhao0304
  • chenwenhao0304
  • 2016-03-07 21:53:30
  • 2331

go-代码收集-net/http客户端长连接

net/http客户端长连接   新建一个http client,后面复用这个client,借助go的高度集成包,来复用连接。 client.go package mainimport ( ...
  • u013870094
  • u013870094
  • 2017-12-10 20:50:54
  • 251

如何保证socket长连接

主要分为两部分: 一,如何更好的检测TCP连接是否正常 二,如何提取本机TCP连接状态 一,如何更好的检测TCP连接是否正常 这方面问题,我上网查了很久,一般来说比较成熟的有两种方法: ...
  • chengyingzhilian
  • chengyingzhilian
  • 2012-06-05 11:07:22
  • 14334

go 短连接和长连接 超时处理

作为一个可能会和很多Client进行通讯交互的Server,首先要保证的就是整个Server运行状态的稳定性,因此在和Client建立连接通讯的时候,确保连接的及时断开非常重要,否则一旦和多个客户端建...
  • whatday
  • whatday
  • 2017-12-03 11:26:49
  • 592

MarioTCP:一个单机可日30亿的百万并发长连接服务器

注:如果用此服务器做变长data的传输,请在业务处理函数中为input buffer增加清空功能(一行memset搞定;也可以在mariotcp核心代码mario_network.c的read功能中增...
  • educast
  • educast
  • 2014-06-27 09:56:24
  • 12121

go net/http Client使用——长连接客户端的使用

go net/http Client使用总结 Client数据结构 // A Client is an HTTP client. Its zero value (DefaultClient) i...
  • u013870094
  • u013870094
  • 2017-12-06 15:46:40
  • 1243

消息推送之Socket----长连接和心跳

最近学习郭霖讲的消息推送其中讲到使用原生的Socket保持长链接来实现消息的推送,这是我的短板,所以跟着仔细学习了一遍,郭神还是依旧的那么仔细那么认真,赞赞,不要骄傲要保持,而对于我就是不要急躁,要踏...
  • qq_33923079
  • qq_33923079
  • 2016-08-23 16:04:23
  • 506

Go语言TCP网络编程(详细)

一、序言 Golang的主要 设计目标之一就是面向大规模后端服务程序,网络通信这块是服务端 程序必不可少也是至关重要的一部分。在日常应用中,我们也可以看到Go中的net以及其subdirectori...
  • hacker00011000
  • hacker00011000
  • 2016-12-28 17:04:24
  • 7440
收藏助手
不良信息举报
您举报文章:[Golang] 从零开始写Socket Server(3): 对长、短连接的处理策略(模拟心跳)
举报原因:
原因补充:

(最多只允许输入30个字)