1. Socket 介绍

Socket套接字，是一种通用的网络编程接口，python中提供了socket.py标准库，非常底层的接口库。

接口相关参数：

  • 协议族
  • Socket类型 |
    | 名称 | 含义 | | —- | —- | —- | | 协议族 | AF_INET | IPv4 | | | AF_INET6 | IPv6 | | Socket类型 | SOCK_STREAM | 面向连接的流套接字。默认值是TCP协议 | | | SOCK_DGRAM | 无连接的数据报文套接字。默认为UDP协议 |

Socket编程，一般需要两端，一个服务端Server，一个客户端Client。

2. TCP 编程

服务端编程步骤：

  1. 创建Socket对象；
  2. 绑定IP和端口；
  3. 开始监听
  4. 建链通信，获取用于传输数据的Socket对象

实现一个简单的TCP server：

  1. import time
  2. import socket
  4. # TCP Server
  5. sock = socket.socket()   # step1:创建Socket对象
  6. addr = ('127.0.0.1', 9999)
  7. sock.bind(addr)   # step2: 绑定地址&端口
  8. sock.listen()   # step3：开始监听
  10. time.sleep(10)
  12. sock.close()   # step4：释放资源

image.png

使用工具模拟client，实现发送/接收消息：

  1. import socket
  2. import logging
  4. logging.basicConfig(format='%(thread)s %(threadName)s %(message)s', level=logging.INFO)
  6. # TCP Server
  7. sock = socket.socket()   # step1:创建Socket对象
  8. addr = ('127.0.0.1', 9999)
  9. sock.bind(addr)   # step2: 绑定地址&端口
  10. sock.listen()   # step3：开始监听
  12. conn, client_addr = sock.accept()   # step4：完成三次握手，建立连接，下面就可以发送/接收数据了。默认是阻塞的，返回一个新对象
  13. logging.info(client_addr)
  15. # 接收消息
  16. data = conn.recv(1024)
  17. logging.info(data.decode())
  18. # 发送消息
  19. msg = 'ack: {}'.format(data.decode())
  20. conn.send(msg.encode())
  22. conn.close()
  23. sock.close()

image.png
image.png

client 模拟工具：image.png

优化tcp server，实现一个连接发送/接收多次消息：

  1. import socket
  2. import logging
  4. logging.basicConfig(format='[%(thread)s %(threadName)s] %(message)s', level=logging.INFO)
  6. # TCP Server
  7. sock = socket.socket()
  8. ip = '127.0.0.1'
  9. port = 9999
  10. addr = (ip, port)
  11. sock.bind(addr)
  12. sock.listen()
  14. conn, client_addr = sock.accept()
  15. while True:
  16.     data = conn.recv(1024).decode()
  17.     msg = "[ACK] {}".format(data)
  18.     conn.send(msg.encode())
  19.     if data == 'Disconnect':
  20.         break
  22. conn.close()
  23. sock.close()

注意：上面代码里的 accept() 、recv() 是阻塞性函数（此阻塞是一个I/O请求，或叫I/O密集性阻塞，等网络I/O的缓存区填满到你要求的时候，才可以继续执行后面的程序），主线程经常被阻塞住而不能工作，怎么办呢？

查看某个TCP进程端口的网络状态：netstat -ant | find “9999” image.png