基于Tornado的TCP服务器

傷城~ 2022-10-07 06:48 219阅读 0赞

### 基于Tornado的TCP服务器 ###

先安装第三方包

pip install tornado

### 服务端 ###

# !/usr/bin/python
    # -*- coding: utf-8 -*-
    
    """ @contact: 微信 1257309054 @file: tornado_server.py @time: 2021/6/16 17:26 @author: LDC """
    import asyncio
    import re
    import threading
    import os
    
    from functools import wraps
    from tornado import gen
    from tornado.ioloop import IOLoop
    from tornado.tcpserver import TCPServer
    
    ''' 设备监控程序 开启TCP服务，与设备采集器建立连接， '''
    
    def threadIng_async(f):
        """ 使用装饰器和多线程实现异步执行 :param f: :return: """
    
        @wraps(f)
        def wrapper(*args, **kwargs):
            thr = threading.Thread(target=f, args=args, kwargs=kwargs)
            thr.start()
    
        return wrapper
    
    
    class DeviceServer(TCPServer):
        def __init__(self, account, servernum):
            super(DeviceServer, self).__init__()
            self.account = account
            self.servernum = servernum
    
        # 接收客户端连接请求
        @gen.coroutine
        def handle_stream(self, stream, address):
            conn = Connection(self.account, self.servernum, stream, address)
            yield conn.send_messages()
    
    class Connection(object):
        clients = set() # 客户端对象
    
        def __init__(self, account, servernum, stream, address):
            Connection.clients.add(self)
            self.account = account
            self.servernum = servernum
            self._read_future = None
            self._stream = stream  # TCP对象
            self._address = address  # TCP地址
            self.heart_count = 0  # 设备发送心跳包次数
            self.previous_order = 'aa0550\r\n'  # 上一条指令默认为回复心跳包
            self.machine_code = ''  # 机器编号
            self.tcp_is_activate = True  # tcp通道是否保持连接
            self._stream.set_close_callback(self.on_close)  # tcp通道关闭后相关变量处理函数
            print('第{}个客户端,地址：{}'.format(len(self.__class__.clients),self._address))
    
    
        @gen.coroutine
        def send_messages(self, data='hello\r\n'):
            # 使用coroutine实现函数分离的异步编程，即异步接收客户端发来的数据，其中数据以'\r\n'结尾
            try:
                yield self.send_message(data)
                response1 = yield self.read_message()
                self.deal_message(response1)
    
            except Exception as e:
                print('出错啦',e)
    
        def read_message(self):
            # 接收数据，遇到\r\n结束符就取出缓存区的数据
            return self._stream.read_until(b'\r\n')
    
        def deal_message(self, data):
            # 处理数据
            data = data.decode('utf-8').replace('\r\n', '')
            print(self._address, data)
            self.send_messages(data='I get it {}'.format(data))
    
        def send_message(self, data):
            # 发送数据
            return self._stream.write(data.encode('utf-8'))
    
        def on_close(self):
            # 关闭通道
            Connection.clients.remove(self)
            self.tcp_is_activate = False  # 机器tcp处于断开状态
            print("客户端已经关闭",self._address)
    
    def kill_port_process(port):
        # 根据端口号杀死进程
        ret = os.popen("netstat -nao|findstr " + str(port))
        str_list = ret.read()
    
        if not str_list:
            print('端口未使用')
            return
        if 'TCP' in str_list:
            # 只关闭处于LISTENING的端口
            ret_list = str_list.replace(' ', '')
            ret_list = re.split('\n', ret_list)
            listening_list = [rl.split('LISTENING') for rl in ret_list]
            process_pids = [ll[1] for ll in listening_list if len(ll) >= 2]
            process_pid_set = set(process_pids)
            for process_pid in process_pid_set:
                os.popen('taskkill /pid ' + str(process_pid) + ' /F')
                print(port, '端口已被释放')
                time.sleep(1)
        elif 'UDP' in str_list:
            ret_list = re.split(' ', str_list)
            process_pid = ret_list[-1].strip()
            if process_pid:
                os.popen('taskkill /pid ' + str(process_pid) + ' /F')
                print('端口已被释放')
            else:
                print("端口未被使用")
    
    def main():
        # 程序入口
        port = 5500
        ip = '0.0.0.0'
        kill_port_process(port)
    
        asyncio.set_event_loop(asyncio.new_event_loop())
        server = DeviceServer('123', 'device_server')
        print('启动服务器')
        # 启动服务器前先杀死端口
    
        server.listen(port=port, address=ip)
        IOLoop.instance().start()
    
    if __name__ == '__main__':
        main()

### 客户端 ###

# !/usr/bin/python
    # -*- coding: utf-8 -*-
    
    """ @contact: 微信 1257309054 @file: tornado客户端.py @time: 2021/6/16 17:35 @author: LDC """
    import socket
    import time
    
    def client(port):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.connect(('127.0.0.1', port))
        data = sock.recv(1024).decode("utf-8")
        print(data)
        count = 0
        while True:
            time.sleep(2)
            count += 1
            sock.send('是我 {} \r\n'.format(count).encode('utf-8'))
            data = sock.recv(1024).decode("utf-8")
            print(data)
    
    if __name__ == '__main__':
        port = 5500
        client(port)

## 后记 ##

【后记】为了让大家能够轻松学编程，我创建了一个公众号【轻松学编程】，里面有让你快速学会编程的文章，当然也有一些干货提高你的编程水平，也有一些编程项目适合做一些课程设计等课题。

也可加我微信【1257309054】，拉你进群，大家一起交流学习。  
如果文章对您有帮助，请我喝杯咖啡吧！

公众号

![公众号][20200317124808234.jpg]

![赞赏码][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xtX2lzX2Rj_size_16_color_FFFFFF_t_70]

关注我，我们一起成长~~

[20200317124808234.jpg]: /images/20221005/5a301536d7134f0399b6690a38796fe5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xtX2lzX2Rj_size_16_color_FFFFFF_t_70]: /images/20221005/16656965ef1349bea2b45559e255a10b.png