tcp.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. """
  4. demeter tcp
  5. name:server.py
  6. author:rabin
  7. """
  8. import socket
  9. import time
  10. from demeter.core import *
  11. from demeter.mqtt import *
  12. from tornado.tcpserver import TCPServer
  13. from tornado.ioloop import IOLoop
  14. class Connection(object):
  15. clients = set()
  16. def __init__(self, stream, address):
  17. Connection.clients.add(self)
  18. self._stream = stream
  19. self._address = address
  20. self._stream.set_close_callback(self.on_close)
  21. self.read_message()
  22. def read_message(self):
  23. self._stream.read_until('\n', self.broadcast_messages)
  24. def broadcast_messages(self, data):
  25. pub = Pub()
  26. temp = data.split(':')
  27. key = temp[0]
  28. value = temp[1]
  29. pub.push(key, value)
  30. #print "User said:", data[:-1], self._address
  31. """
  32. for conn in Connection.clients:
  33. conn.send_message(data)
  34. """
  35. self.read_message()
  36. def send_message(self, data):
  37. self._stream.write(data)
  38. def on_close(self):
  39. #print "A user has left the chat room.", self._address
  40. Connection.clients.remove(self)
  41. class Server(TCPServer):
  42. def handle_stream(self, stream, address):
  43. #print "New connection :", address, stream
  44. Connection(stream, address)
  45. #print "connection num is:", len(Connection.clients)
  46. class Client(object):
  47. def __init__(self, host='0.0.0.0', port=8000):
  48. self.connect = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  49. self.connect.connect((host, port))
  50. def send(self, msg):
  51. self.connect.sendall(msg + '\n')
  52. #data = self.connect.recv(1024)
  53. self.connect.close()