Python Tornado如何實現(xiàn)WEB服務器Socket服務器共存并實現(xiàn)交互

from tornado.iostream import IOStream  # 這句可以沒有，只是作為參數(shù)的代碼提示
from tornado.tcpserver import TCPServer

class Connecter:

  clients = set()   # 存放連接的客戶端

  async def init(self, stream: IOStream, address: tuple):
    """
    注意這個不是構(gòu)造方法，這里不用構(gòu)造方法是為了方便后續(xù)的與web端相互通信
    """
    self.stream, self.address = stream, address
    self.clients.add(self)
    print("{address} 上線!".format(address=address))
    self.stream.set_close_callback(self.onClose) # 客戶端離線的時候會被調(diào)用
    await self.receive()    # 接受消息

  async def receive(self):
    """
    接受消息
    """
    while True:
      try:  # 因為異步的原因?？赡茉O(shè)備離線后還在接收消息，所以try一下，不讓錯誤打印出來，其實打印了錯誤也不影響程序運行
        data = await self.stream.read_bytes(num_bytes=1024, partial=True)  # num_bytes:每次最多接收字節(jié)，partial:數(shù)據(jù)中斷后視為接收完成
        print(data)
        # TODO:接收到數(shù)據(jù)的處理
      except: 
        pass

  def send(self, data):
    """
    發(fā)送消息
    :param data: 消息內(nèi)容
    """
    self.stream.write(bytes(data.encode('utf8')))

  def onClose(self):
    """
    客戶端離線
    """
    print("{address} 離線!".format(address=self.address))
    self.clients.remove(self) # 在clients內(nèi)刪掉該客戶端

class SocketServer(TCPServer):   # 需要繼承TCPServer這個類
  async def handle_stream(self, stream: IOStream, address: tuple):  # 實現(xiàn)類里面的handle_stream方法
    await Connecter().init(stream, address)   # 每次有客戶端連入都實例化一個Connecter類

from tornado.web import RequestHandler   # 導入RequestHandler類


class TestApiHandler(RequestHandler):    # 繼承RequestHandler類

  def get(self):   # 實現(xiàn)GET方法，GET請求會執(zhí)行這個方法
    pass

  def post(self):   # 實現(xiàn)POST方法，POST請求會執(zhí)行這個方法
    pass

from tornado.web import Application     # 導入Tornado的Application類
from .src.web_test import TestApiHandler  # 導入我們自己寫的TestApiHandler類


def webServerApp():   # 構(gòu)造出webApp
  return Application([
    (r'/api_test/', TestApiHandler),  # 把/api_test/路由到TestApiHandler
  ])

from web_server.web_server import webServerApp
from socket_server.socket_server import SocketServer
from tornado import ioloop
from tornado.options import define, options

#這里用define定義端口，可以方便使用命令行參數(shù)的形式修改端口
define("socketPort", 8888, type=int)  # socket默認使用8888端口
define("webPort", 8080, type=int)    # web默認使用8080端口

def main():
  socket_server = SocketServer()
  socket_server.listen(options.socketPort, '0.0.0.0')
  print("socket服務器啟動，端口：{port}".format(port=options.socketPort))
  app = webServerApp()
  app.listen(options.webPort, '0.0.0.0')
  print("web服務器啟動，端口：{port}".format(port=options.webPort))
  ioloop.IOLoop.current().start()


if __name__ == '__main__':
  main()

import socket
import datetime


class Client:

  def __init__(self):
    with socket.create_connection(("127.0.0.1", 8888)) as sock:
      while True:
        msg = sock.recv(1024)
        if len(msg) > 0:
          print(msg)
          sock.send(bytes(str(datetime.datetime.now).encode('utf8')))
        msg = []
            
        

if __name__ == "__main__":
  Client()

from socket_server.socket_server import Connecter

 def post(self):   # 實現(xiàn)POST方法，POST請求會執(zhí)行這個方法
    msg = self.get_argument("msg") # 得到post請求中的msg的值
    ip = self.get_argument('ip')  # 得到要發(fā)送的ip
    c = Connecter()   # 實例化Connecter類
    counter = 0   # 記錄發(fā)送到客戶端的個數(shù)
    for client in c.clients:  # type:Connecter
      if client.address[0] == ip:   # 根據(jù)ip發(fā)送
        client.send(msg)  # 發(fā)送消息
        counter += 1    # 計數(shù)加1
    self.write("{'send_counter':" + str(counter) + "}")

class SocketData:
  msg = ""

 async def receive(self):
    """
    接受消息
    """
    while True:
      try:  # 因為異步的原因?？赡茉O(shè)備離線后還在接收消息，所以try一下，不讓錯誤打印出來，其實打印了錯誤也不影響程序運行
        data = await self.stream.read_bytes(num_bytes=1024, partial=True)  # num_bytes:每次最多接收字節(jié)，partial:數(shù)據(jù)中斷后視為接收完成
        print(data)
        from .src.socket_data_processing import SocketData
        SocketData.msg = data.decode('utf8')
      except:
        pass

 def get(self):   # 實現(xiàn)GET方法，GET請求會執(zhí)行這個方法
    from socket_server.src.socket_data_processing import SocketData
    self.write(SocketData.msg)