commit 9815c19d7ea39b8585c2848b523e7182bb26b4c7
author zzn <zineng1991@gmail.com> Thu Jun 04 19:05:55 2015 +0800
committer Roger Meier <roger@apache.org> Sun Jun 14 20:50:03 2015 +0200
tree 7ac381d7470f6ae322fc3e7dc9407ff7401f03f7
parent 9b3b8d4987a34ed732a0c3b9edb757d913c38647

THRIFT-3180 fix framed transport

This closes #515

lib/lua/TFramedTransport.lua

diff --git a/lib/lua/TFramedTransport.lua b/lib/lua/TFramedTransport.lua
index 84ae3ec..437b701 100644
--- a/lib/lua/TFramedTransport.lua
+++ b/lib/lua/TFramedTransport.lua
@@ -38,7 +38,7 @@
     error('You must provide ' .. ttype(self) .. ' with a trans')
   end
 
-  return TTransportBase:new(obj)
+  return TTransportBase.new(self, obj)
 end
 
 function TFramedTransport:isOpen()
@@ -69,7 +69,7 @@
   end
 
   local val = string.sub(self.rBuf, 0, len)
-  self.rBuf = string.sub(self.rBuf, len)
+  self.rBuf = string.sub(self.rBuf, len+1)
   return val
 end
 
@@ -79,9 +79,6 @@
   self.rBuf = self.trans:readAll(frame_len)
 end
 
-function TFramedTransport:readAll(len)
-  return self.trans:readAll(len)
-end
 
 function TFramedTransport:write(buf, len)
   if self.doWrite == false then
@@ -91,7 +88,7 @@
   if len and len < string.len(buf) then
     buf = string.sub(buf, 0, len)
   end
-  self.wBuf = self.wBuf + buf
+  self.wBuf = self.wBuf .. buf
 end
 
 function TFramedTransport:flush()
@@ -102,6 +99,8 @@
   -- If the write fails we still want wBuf to be clear
   local tmp = self.wBuf
   self.wBuf = ''
+  local frame_len_buf = libluabpack.bpack("i", string.len(tmp))
+  self.trans:write(frame_len_buf)
   self.trans:write(tmp)
   self.trans:flush()
 end