Python 创建TCP服务器的方法


Posted in Python onJuly 28, 2020

问题

你想实现一个服务器,通过TCP协议和客户端通信。

解决方案

创建一个TCP服务器的一个简单方法是使用 socketserver 库。例如,下面是一个简单的应答服务器:

from socketserver import BaseRequestHandler, TCPServer

class EchoHandler(BaseRequestHandler):
  def handle(self):
    print('Got connection from', self.client_address)
    while True:

      msg = self.request.recv(8192)
      if not msg:
        break
      self.request.send(msg)

if __name__ == '__main__':
  serv = TCPServer(('', 20000), EchoHandler)
  serv.serve_forever()

在这段代码中,你定义了一个特殊的处理类,实现了一个 handle() 方法,用来为客户端连接服务。 request 属性是客户端socket,client_address 有客户端地址。 为了测试这个服务器,运行它并打开另外一个Python进程连接这个服务器:

>>> from socket import socket, AF_INET, SOCK_STREAM
>>> s = socket(AF_INET, SOCK_STREAM)
>>> s.connect(('localhost', 20000))
>>> s.send(b'Hello')
5
>>> s.recv(8192)
b'Hello'
>>>

很多时候,可以很容易的定义一个不同的处理器。下面是一个使用 StreamRequestHandler 基类将一个类文件接口放置在底层socket上的例子:

from socketserver import StreamRequestHandler, TCPServer

class EchoHandler(StreamRequestHandler):
  def handle(self):
    print('Got connection from', self.client_address)
    # self.rfile is a file-like object for reading
    for line in self.rfile:
      # self.wfile is a file-like object for writing
      self.wfile.write(line)

if __name__ == '__main__':
  serv = TCPServer(('', 20000), EchoHandler)
  serv.serve_forever()

讨论

socketserver 可以让我们很容易的创建简单的TCP服务器。 但是,你需要注意的是,默认情况下这种服务器是单线程的,一次只能为一个客户端连接服务。 如果你想处理多个客户端,可以初始化一个 ForkingTCPServer 或者是 ThreadingTCPServer 对象。例如:

from socketserver import ThreadingTCPServer


if __name__ == '__main__':
  serv = ThreadingTCPServer(('', 20000), EchoHandler)
  serv.serve_forever()

使用fork或线程服务器有个潜在问题就是它们会为每个客户端连接创建一个新的进程或线程。 由于客户端连接数是没有限制的,因此一个恶意的黑客可以同时发送大量的连接让你的服务器奔溃。

如果你担心这个问题,你可以创建一个预先分配大小的工作线程池或进程池。 你先创建一个普通的非线程服务器,然后在一个线程池中使用 serve_forever() 方法来启动它们。

if __name__ == '__main__':
  from threading import Thread
  NWORKERS = 16
  serv = TCPServer(('', 20000), EchoHandler)
  for n in range(NWORKERS):
    t = Thread(target=serv.serve_forever)
    t.daemon = True
    t.start()
  serv.serve_forever()

一般来讲,一个 TCPServer 在实例化的时候会绑定并激活相应的 socket 。 不过,有时候你想通过设置某些选项去调整底下的 socket` ,可以设置参数 bind_and_activate=False 。如下:

if __name__ == '__main__':
  serv = TCPServer(('', 20000), EchoHandler, bind_and_activate=False)
  # Set up various socket options
  serv.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
  # Bind and activate
  serv.server_bind()
  serv.server_activate()
  serv.serve_forever()

上面的 socket 选项是一个非常普遍的配置项,它允许服务器重新绑定一个之前使用过的端口号。 由于要被经常使用到,它被放置到类变量中,可以直接在 TCPServer 上面设置。 在实例化服务器的时候去设置它的值,如下所示:

if __name__ == '__main__':
  TCPServer.allow_reuse_address = True
  serv = TCPServer(('', 20000), EchoHandler)
  serv.serve_forever()

在上面示例中,我们演示了两种不同的处理器基类( BaseRequestHandler 和 StreamRequestHandler )。 StreamRequestHandler 更加灵活点,能通过设置其他的类变量来支持一些新的特性。比如:

import socket

class EchoHandler(StreamRequestHandler):
  # Optional settings (defaults shown)
  timeout = 5           # Timeout on all socket operations
  rbufsize = -1          # Read buffer size
  wbufsize = 0           # Write buffer size
  disable_nagle_algorithm = False # Sets TCP_NODELAY socket option
  def handle(self):
    print('Got connection from', self.client_address)
    try:
      for line in self.rfile:
        # self.wfile is a file-like object for writing
        self.wfile.write(line)
    except socket.timeout:
      print('Timed out!')

最后,还需要注意的是绝大部分Python的高层网络模块(比如HTTP、XML-RPC等)都是建立在 socketserver 功能之上。 也就是说,直接使用 socket 库来实现服务器也并不是很难。 下面是一个使用 socket 直接编程实现的一个服务器简单例子:

from socket import socket, AF_INET, SOCK_STREAM

def echo_handler(address, client_sock):
  print('Got connection from {}'.format(address))
  while True:
    msg = client_sock.recv(8192)
    if not msg:
      break
    client_sock.sendall(msg)
  client_sock.close()

def echo_server(address, backlog=5):
  sock = socket(AF_INET, SOCK_STREAM)
  sock.bind(address)
  sock.listen(backlog)
  while True:
    client_sock, client_addr = sock.accept()
    echo_handler(client_addr, client_sock)

if __name__ == '__main__':
  echo_server(('', 20000))

以上就是Python 创建TCP服务器的方法的详细内容,更多关于Python 创建TCP服务器的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
python中文乱码的解决方法
Nov 04 Python
Linux下Python获取IP地址的代码
Nov 30 Python
Python中的异常处理学习笔记
Jan 28 Python
python绘制圆柱体的方法
Jul 02 Python
python实现三维拟合的方法
Dec 29 Python
Python实现点阵字体读取与转换的方法
Jan 29 Python
基于python实现把图片转换成素描
Nov 13 Python
Python图片的横坐标汉字实例
Dec 04 Python
浅谈对pytroch中torch.autograd.backward的思考
Dec 27 Python
python3中关于excel追加写入格式被覆盖问题(实例代码)
Jan 10 Python
Python趣味入门教程之循环语句while
Aug 26 Python
python按照list中字典的某key去重的示例代码
Oct 13 Python
Python实现画图软件功能方法详解
Jul 28 #Python
Python绘图之柱形图绘制详解
Jul 28 #Python
Python如何定义接口和抽象类
Jul 28 #Python
Python爬虫之爬取淘女郎照片示例详解
Jul 28 #Python
Python selenium键盘鼠标事件实现过程详解
Jul 28 #Python
用python写爬虫简单吗
Jul 28 #Python
公认8个效率最高的爬虫框架
Jul 28 #Python
You might like
收音机史话 - 1960年代前后的DIY
2021/03/02 无线电
用PHP4访问Oracle815
2006/10/09 PHP
php支持中文字符串分割的函数
2015/05/28 PHP
php socket通信(tcp/udp)实例分析
2016/02/14 PHP
php实现通过stomp协议连接ActiveMQ操作示例
2020/02/23 PHP
基于JQuery模仿苹果桌面的Dock效果(初级版)
2012/10/15 Javascript
Bootstrap多级导航栏(级联导航)的实现代码
2016/03/08 Javascript
快速解决jquery.touchSwipe左右滑动和垂直滚动条冲突
2016/04/15 Javascript
简单谈谈ES6的六个小特性
2016/11/18 Javascript
Angular中响应式表单的三种更新值方法详析
2017/08/22 Javascript
Chart.js 轻量级HTML5图表绘制工具库(知识整理)
2018/05/22 Javascript
vue打包相关细节整理(小结)
2018/09/28 Javascript
Vue多组件仓库开发与发布详解
2019/02/28 Javascript
Python常见数据结构详解
2014/07/24 Python
python处理csv数据动态显示曲线实例代码
2018/01/23 Python
Python基于更相减损术实现求解最大公约数的方法
2018/04/04 Python
用TensorFlow实现多类支持向量机的示例代码
2018/04/28 Python
实践Vim配置python开发环境
2018/07/02 Python
pandas 把数据写入txt文件每行固定写入一定数量的值方法
2018/12/28 Python
python 随机生成10位数密码的实现代码
2019/06/27 Python
django框架用户权限中的session缓存到redis中的方法
2019/08/06 Python
Python threading的使用方法解析
2019/08/28 Python
浅谈pytorch torch.backends.cudnn设置作用
2020/02/20 Python
canvas画布实现手写签名效果的示例代码
2019/04/23 HTML / CSS
HTML5 新标签全部总汇(推荐)
2016/06/13 HTML / CSS
联想墨西哥官方网站:Lenovo墨西哥
2016/08/17 全球购物
受外贸欢迎的美国主机:BlueHost
2017/05/16 全球购物
俄罗斯最大的隐形眼镜销售网站:Ochkov.Net
2021/02/07 全球购物
技术总监的工作职责
2013/11/13 职场文书
歌唱比赛获奖感言
2014/01/21 职场文书
管理学院毕业生自荐信范文
2014/03/10 职场文书
公司门卫岗位职责范本
2014/07/08 职场文书
2014年社区教育工作总结
2014/12/02 职场文书
离婚被告代理词
2015/05/23 职场文书
践行三严三实心得体会(2016推荐篇)
2016/01/06 职场文书
如何避免mysql启动时错误及sock文件作用分析
2022/01/22 MySQL