Python3 Socket与Socket心跳机制简单实现

什么是 Socket?

Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。

---

socket()函数

Python 中，我们用 socket（）函数来创建套接字，语法格式如下：

socket.socket([family[, type[, proto]]])

参数

• family: 套接字家族可以使AF_UNIX或者AF_INET
• type: 套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM
• protocol: 一般不填默认为0.

Socket 对象(内建)方法

函数	描述
服务器端套接字
s.bind()	绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。
s.listen()	开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。
s.accept()	被动接受TCP客户端连接,(阻塞式)等待连接的到来
客户端套接字
s.connect()	主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
s.connect_ex()	connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
公共用途的套接字函数
s.recv()	接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
s.send()	发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
s.sendall()	完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
s.recvfrom()	接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
s.sendto()	发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。
s.close()	关闭套接字
s.getpeername()	返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
s.getsockname()	返回套接字自己的地址。通常是一个元组(ipaddr,port)
s.setsockopt(level,optname,value)	设置给定套接字选项的值。
s.getsockopt(level,optname[.buflen])	返回套接字选项的值。
s.settimeout(timeout)	设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
s.gettimeout()	返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。
s.fileno()	返回套接字的文件描述符。
s.setblocking(flag)	如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常。
s.makefile()	创建一个与该套接字相关连的文件

简单心跳机制实例

服务端：

    我们使用 socket 模块的 socket 函数来创建一个 socket 对象。socket 对象可以通过调用其他函数来设置一个 socket 服务。现在我们通过调用 bind(hostname, port) 函数来指定服务的 port(端口)，这里定义两个全局变量（starttime，endtime）记录客户端开始连接时间和断开连接时间。接着，我们定义Wait_connection函数，设置超时settimeout为None， 并调用 socket 对象的 accept 方法等待客户端的连接。一直等到客户端连接了，记录连接的时间starttime，打印客户端的addr信息和客户端发过来的信息，最后调用keep_alive函数。keep_alive函数与Wait_connection函数不同之处在于keep_alive可以持续监听客户端发过来的信息，并且设置了超时，如果超过5秒没有连接则判断客户端已经断开连接，报错，这里异常处理报错则记录下断开时间，并打印客户端连接了多长时间，并调用Wait_connection函数等待客户端的下次连接。

完整代码如下：

import socket
import datetime
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind(('127.0.0.1',9999))
s.listen(1)
starttime = None
endtime = None
def Wait_connection():
    global starttime
    s.settimeout(None)
    conn,addr=s.accept()
    starttime = datetime.datetime.now()
    print('client addr',addr)
    client_msg=conn.recv(1024)
    print('client msg: %s' %(str(client_msg,'utf-8')))
    keep_alive()
def keep_alive():
    global endtime
    a = 1
    while a==1:
        try:
            s.settimeout(5)
            print('---------------------------------')
            conn,addr=s.accept()
            print('client addr',addr)
            client_msg=conn.recv(1024)#客户端发送过来的消息
            print('client msg: %s' %(str(client_msg,'utf-8')))
        except:
            a = 2
            endtime = datetime.datetime.now()
    print('连接已断开，本次连接持续 %s 秒'%str((endtime - starttime).seconds))
    Wait_connection()
if __name__ == '__main__':
    Wait_connection()

客户端

接下来我们写一个简单的客户端实例每四秒连接到一次以上创建的服务。端口号为 9999。告诉服务端“我”还在线。

import socket
import time
a = 0
while True:
    time.sleep(4)
    a+=1
    s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    s.connect(('127.0.0.1',9999))
    keepclass = "我已连接"+str(a*4)+"秒"
    s.send(bytes(keepclass,'UTF-8'))#向服务端发送消息