Servidor websockets encadeado Python 3

Question

May 11, 2016, 09:47 PM

Servidor websockets encadeado Python 3

Estou construindo um aplicativo de servidor Websocket em python 3. Estou usando esta implementação:https://websockets.readthedocs.io/

Basicamente, quero gerenciar vários clientes. Também quero enviar dados de 2 threads diferentes (um para GPS + um para IMU) O segmento GPS é atualizado em 1Hz, enquanto o segmento IMU é atualizado em 25Hz

Meu problema está no método MSGWorker.sendData: assim que eu descomente a linha @ asyncio.coroutine e produzo a partir de websocket.send ('{"GPS": "% s"}'% data), todo o método não faz nada (nenhuma impressão ("Enviar dados: foo") no terminal)

No entanto, com essas duas linhas comentadas, meu código funciona como eu esperava, exceto pelo fato de não enviar nada pelo websocket.

Mas, é claro, meu objetivo é enviar dados pelo soquete da web, só não entendo por que não funciona? Qualquer ideia ?

server.py

#!/usr/bin/env python3
import signal, sys
sys.path.append('.')
import time
import websockets
import asyncio
import threading

connected = set()
stopFlag = False



class GPSWorker (threading.Thread):
  def __init__(self):
    threading.Thread.__init__(self)
    self.data = 0
    self.lastData = 0
    self.inc = 0

  # Simulate GPS data
  def run(self):
    while not stopFlag:
      self.data = self.inc
      self.inc += 1
      time.sleep(1)

  def get(self):
    if self.lastData is not self.data:
      self.lastData = self.data
      return self.data



class IMUWorker (threading.Thread):
  def __init__(self):
    threading.Thread.__init__(self)
    self.data = 0
    self.lastData = 0
    self.inc = 0

  # Simulate IMU data
  def run(self):
    while not stopFlag:
      self.data = self.inc
      self.inc += 1
      time.sleep(0.04)

  def get(self):
    if self.lastData is not self.data:
      self.lastData = self.data
      return self.data



class MSGWorker (threading.Thread):
  def __init__(self):
    threading.Thread.__init__(self)

  def run(self):
    while not stopFlag:
      data = gpsWorker.get()
      if data:
        self.sendData('{"GPS": "%s"}' % data)          

      data = imuWorker.get()
      if data:
        self.sendData('{"IMU": "%s"}' % data)

      time.sleep(0.04)

  #@asyncio.coroutine
  def sendData(self, data):
    for websocket in connected.copy():
      print("Sending data: %s" % data)
      #yield from websocket.send('{"GPS": "%s"}' % data)



@asyncio.coroutine
def handler(websocket, path):
  global connected
  connected.add(websocket)
  #TODO: handle client disconnection
  # i.e connected.remove(websocket)



if __name__ == "__main__":
  print('aeroPi server')
  gpsWorker = GPSWorker()
  imuWorker = IMUWorker()
  msgWorker = MSGWorker()

  try:
    gpsWorker.start()
    imuWorker.start()
    msgWorker.start()

    start_server = websockets.serve(handler, 'localhost', 7700)
    loop = asyncio.get_event_loop()
    loop.run_until_complete(start_server)
    loop.run_forever()

  except KeyboardInterrupt:
    stopFlag = True
    loop.close()
    print("Exiting program...")

client.html

<!doctype html>
<html>
<head>
  <meta charset="UTF-8" />
</head>
<body>
</body>
</html>
<script type="text/javascript">
  var ws = new WebSocket("ws://localhost:7700", 'json');
  ws.onmessage = function (e) {
    var data = JSON.parse(e.data);
    console.log(data);
  };
</script>

Obrigado pela sua ajuda