python tcp server 多线程_在TCP server中使用多线程

在TCP server中使用多线程

目前，我所实现的TCP server服务能力都很差，根本无法同时与多个客户端进行交互，只有处理完一个客户端的交互以后才能使用accept等待下一个客户端的连接。

本篇，我将使用多线程技术，在通过accept获得一个socket文件后，启动一个多线程来专门处理这个客户端的数据传输。

1. 服务端代码

import socket

import threading

def worker(client):

while True:

data=client.recv(1024) #把接收的数据实例化

if len(data) == 0 or data == None:

break

print(data.decode(encoding='utf-8'))

client.send('收到数据'.encode(encoding='utf-8'))

def start_server(port):

HOST = '0.0.0.0'

PORT = port

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) #定义socket类型，网络通信，TCP

s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

s.bind((HOST,PORT)) #套接字绑定的IP与端口

s.listen(1) #开始TCP监听

while True:

conn, addr = s.accept() #接受TCP连接，并返回新的套接字与IP地址

print('Connected by',addr) #输出客户端的IP地址

t = threading.Thread(target=worker, args=(conn, ))

t.start()

if __name__ == '__main__':

start_server(8801)

在服务端，我使用while实现了无限循环，在循环里使用accept接受TCP连接，获得连接后，这个连接何时将全部数据发送给服务端是未知的，如果一直处理这个连接，那么其他的连接就无法处理，因为在服务端，代码无法执行到下一次accept。使用了多线程以后，情况就不同了，每个TCP连接使用一个线程来处理，那么主线程就可以再次执行到accept这行代码上，不受任何操作的阻塞影响。

2. 客户端代码

import os

import time

import socket

def start_client(addr, port):

PLC_ADDR = addr

PLC_PORT = port

s = socket.socket()

s.connect((PLC_ADDR, PLC_PORT))

count = 0

while True:

msg = '进程{pid}发送数据'.format(pid=os.getpid())

msg = msg.encode(encoding='utf-8')

s.send(msg)

recv_data = s.recv(1024)

print(recv_data.decode(encoding='utf-8'))

time.sleep(3)

count += 1

if count > 20:

break

s.close()

if __name__ == '__main__':

start_client('127.0.0.1', 8801)

客户端不需要做什么改变，在代码里，每次发送完消息后，都会sleep 3秒钟，一共发送20次，这样，就有足够的时间启动多个client来测试服务端是否有能力同时处理多个客户端的请求了。