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  连接字符串(join %)
Sep 06 Python
Python入门篇之数字
Oct 20 Python
Python Tkinter GUI编程入门介绍
Mar 10 Python
python web框架学习笔记
May 03 Python
Python3.6正式版新特性预览
Dec 15 Python
Python PyQt4实现QQ抽屉效果
Apr 20 Python
Python装饰器简单用法实例小结
Dec 03 Python
使用Python 统计高频字数的方法
Jan 31 Python
Python常用特殊方法实例总结
Mar 22 Python
python循环嵌套的多种使用方法解析
Nov 29 Python
pyx文件 生成pyd 文件用于 cython调用的实现
Mar 04 Python
总结Pyinstaller打包的高级用法
Jun 28 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
PHP开发过程中常用函数收藏
2009/12/14 PHP
JavaScript 学习 - 提高篇
2007/02/02 Javascript
Extjs在exlipse中设置自动提示的方法
2010/04/07 Javascript
Node.js:Windows7下搭建的Node.js服务(来玩玩服务器端的javascript吧,这可不是前端js插件)
2011/06/27 Javascript
图片动画横条广告带上下滚动可自定义图片、链接等等
2013/10/20 Javascript
javascript替换已有元素replaceChild()使用介绍
2014/04/03 Javascript
js中定义一个变量并判断其是否为空的方法
2014/05/13 Javascript
javascript实现的淘宝旅行通用日历组件用法实例
2015/08/03 Javascript
JavaScript实现的浮动层框架用法实例分析
2015/10/10 Javascript
JQuery ztree带筛选、异步加载实例讲解
2016/02/25 Javascript
javascript中数组和字符串的方法对比
2016/07/20 Javascript
微信小程序图片横向左右滑动案例
2017/05/19 Javascript
vue-cli启动本地服务局域网不能访问的原因分析
2018/01/22 Javascript
解决vue-router进行build无法正常显示路由页面的问题
2018/03/06 Javascript
JavaScript之解构赋值的理解
2019/01/30 Javascript
用Python编写简单的定时器的方法
2015/05/02 Python
今天 平安夜 Python 送你一顶圣诞帽 @微信官方
2017/12/25 Python
Ubuntu下使用Python实现游戏制作中的切分图片功能
2018/03/30 Python
PyTorch的Optimizer训练工具的实现
2019/08/18 Python
python制作朋友圈九宫格图片
2019/11/03 Python
python pygame实现球球大作战
2019/11/25 Python
Python中logging日志库实例详解
2020/02/19 Python
jupyter notebook 多环境conda kernel配置方式
2020/04/10 Python
浅析CSS3 用text-overflow解决文字排版问题
2020/10/28 HTML / CSS
html5 canvas手势解锁源码分享
2020/01/07 HTML / CSS
英国著名国际平价时尚男装品牌:Topman
2016/08/27 全球购物
节能环保口号
2014/06/12 职场文书
水污染治理工程专业自荐信
2014/06/21 职场文书
新郎答谢词
2015/01/04 职场文书
收银员岗位职责
2015/02/03 职场文书
2015年财务部年度工作总结
2015/05/19 职场文书
小学生读书笔记
2015/07/01 职场文书
小学生作文写作技巧100例,非常实用!
2019/07/08 职场文书
关于拾金不昧的感谢信(五篇)
2019/10/18 职场文书
详解nginx.conf 中 root 目录设置问题
2021/04/01 Servers
Python中的 No Module named ***问题及解决
2022/07/23 Python