BlockingSocketIO.py

# -*- coding:utf-8 -*-

from io.socket import IOCallback
from java.util.concurrent.locks import ReentrantLock
from io.socket import SocketIO

# it use socket.io java client of: https://github.com/Gottox/socket.io-java-client
class BlockingSocketIO(IOCallback):
  
	respMsg = "";
	
	def __init__(self, url, logger):
		logger.info("init with url: %s" % url)
		self.transportLock = ReentrantLock()
		self.responseCondition = self.transportLock.newCondition()
		self.socketIO = SocketIO(url, self)
		self.logger = logger;
	
	def sendAndRcv(self, message):
		try:
			self.transportLock.lock()
			self.socketIO.send(message)
			self.respMsg = ""
			self.responseCondition.await()
			return self.respMsg
		except Exception, err:
			self.logger.error(err.message)
		finally:
			self.transportLock.unlock()
		return self.respMsg;

	def emitAndRcv(self, event, message):
		try:
			self.transportLock.lock();
			self.socketIO.emit(event, message)
			self.respMsg = ""
			self.responseCondition.await()
			return self.respMsg
		except Exception, err:
			self.logger.error(err.message)
		finally:
			self.transportLock.unlock()
		return self.respMsg;
	
	def onConnect(self):
		self.logger.info("connected")
	def onDisconnect(self):
		self.logger.info("disconnect")
	def onMessage(self, data, acknowledge):
		self.logger.info("message")
		setResponse(data)
	def on(self, event, acknowledge, args):
		self.logger.info("Server triggered event: %s" % event)
		for x in args:
			self.logger.info("args: %s" % x)
		respStr = ",".join(x.toString() for x in args)
		self.setResponse(respStr)
	def onError(self, exception):
		self.logger.error("error:%s" % exception.getMessage())
		
		
	def setResponse(self, data):
		self.logger.info("Server said: %s" % data)
		try:
			self.transportLock.lock()
			self.respMsg = data
			self.responseCondition.signal()
		finally:
			self.transportLock.unlock()