summaryrefslogtreecommitdiffstats
path: root/module/remote/thriftbackend/Protocol.py
diff options
context:
space:
mode:
Diffstat (limited to 'module/remote/thriftbackend/Protocol.py')
-rw-r--r--module/remote/thriftbackend/Protocol.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/module/remote/thriftbackend/Protocol.py b/module/remote/thriftbackend/Protocol.py
new file mode 100644
index 000000000..c42d01459
--- /dev/null
+++ b/module/remote/thriftbackend/Protocol.py
@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+
+from thrift.protocol import TBinaryProtocol
+
+class Protocol(TBinaryProtocol.TBinaryProtocol):
+ def writeString(self, str):
+ try:
+ str = str.encode("utf8", "ignore")
+ except Exception, e:
+ pass
+
+ self.writeI32(len(str))
+ self.trans.write(str)
+
+ def readString(self):
+ len = self.readI32()
+ str = self.trans.readAll(len)
+ try:
+ str = str.decode("utf8", "ignore")
+ except:
+ pass
+
+ return str
+
+
+class ProtocolFactory(TBinaryProtocol.TBinaryProtocolFactory):
+
+ def getProtocol(self, trans):
+ prot = Protocol(trans, self.strictRead, self.strictWrite)
+ return prot \ No newline at end of file