Python实现多线程/多进程的TCP服务器
作者：kaichenkai
这篇文章主要为大家详细介绍了Python实现多线程/多进程的TCP服务器，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
多线程的TCP服务器，供大家参考，具体内容如下
背景：同学公司的传感器设备需要将收集的数据发到服务器上，前期想写一个简单的服务器来测试下使用效果，设备收集的数据非常的重要，所以考虑使用TCP协议来实现。
因为只是测试使用，所以采用多线程的方式，毕竟节省资源嘛（使用协程时会导致I/O阻塞）
开门见山，直接搬上来了
一、tcp_server_v1.0使用说明：
1.运行环境：python3解释器，并安装socket、threading模块；
2.该版本使用多线程实现的多任务；
3.支持多台设备同时连接，并同时提供服务。
二、代码逻辑：
指定服务器运行端口为：8125
创建一个服务器类
实例化一个服务器对象
服务器对象调用类中的方法
1 初始化服务器属性(def __init__)
1.1 创建套接字
1.2 解决程序端口占用问题
1.3 绑定本地ip地址
1.4 将套接字变为监听套接字，最大连接数量为100
2 定义设备连接方法(def run_forever)
2.1 进入循环
2.2 等待设备连接...
2.2.1 当设备连接，打印设备的 ip 和 port
2.2.2 创建线程，调用业务处理方法，为该设备提供服务
3 业务处理方法(def service_machine)
3.1 进入循环
3.2 等待接收设备发送数据...
3.2.1 当服务器接收到数据
3.2.2 判断数据是否为空
否：打印数据内容，进行业务处理，将处理后的数据结果返回给设备
是：打印设备断开连接，退出循环，关闭套接字
代码实现如下：
#!C:\Python3.6.5\python.exe
# -*- coding: gbk -*-
import socket
import threading
class WSGIServer(object):
def __init__(self, port):
"""初始化对象"""
# 创建套接字
self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 解决程序端口占用问题
self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
# 绑定本地ip地址
self.tcp_server_socket.bind(("", port))
# 将套接字变为监听套接字，最大连接数量为100
self.tcp_server_socket.listen(100)
def run_forever(self):
"""设备连接"""
while True:
# 1.等待设备连接(通过ip地址和端口建立tcp连接)
# 如果有设备连接，则会生成用于设备和服务器通讯的套接字：new_socket
# 会获取到设备的ip地址和端口
new_socket, client_addr = self.tcp_server_socket.accept()
print("设备{0}已连接".format(client_addr))
# 2.创建线程处理设备的需求
t1 = threading.Thread(target=self.service_machine, args=(new_socket, client_addr))
t1.start()
def service_machine(self, new_socket, client_addr):
"""业务处理"""
while True:
# 3.接收设备发送的数据，单次最大1024字节，按‘gbk'格式解码
receive_data = new_socket.recv(1024).decode("gbk")
# 4.如果设备发送的数据不为空
if receive_data:
# 4.1 打印接收的数据，这里可以将设备发送的数据写入到文件中
# 获取设备的ID信息
print(receive_data)
if receive_data[0:6] == "report":
response = "SET OK:" + receive_data
else:
receive_data = receive_data[6:].split(",")[0]
# 拼接响应数据
response = "alarm=" + receive_data + ",Switch:clear"
print(response)
# 4.2 返回原数据作为应答，按‘utf-8'格式编码
new_socket.send(response.encode("utf-8"))
# 5.当设备断开连接时，会收到空的字节数据，判断设备已断开连接
else:
print('设备{0}断开连接...'.format(client_addr))
break
# 关闭套接字
new_socket.close()
def main(port):
"""创建一个WEB服务器"""
wsgi_server = WSGIServer(port)
print("服务器已开启")
wsgi_server.run_forever()
if __name__ == '__main__':
port = 8125 # 指定端口
main(8125)
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。