lib/rb/lib/thrift/transport/ttransport.rb

   1 #!/usr/bin/env ruby
   2 #
   3 # Copyright (c) 2006- Facebook
   4 # Distributed under the Thrift Software License
   5 #
   6 # See accompanying file LICENSE or visit the Thrift site at:
   7 # http://developers.facebook.com/thrift/
   8 #
   9 # Author: Mark Slee <mcslee@facebook.com>
  10 #
  11
  12 require 'thrift/thrift'
  13
  14 class TTransportException < TException
  15
  16   UNKNOWN = 0
  17   NOT_OPEN = 1
  18   ALREADY_OPEN = 2
  19   TIMED_OUT = 3
  20   END_OF_FILE = 4
  21
  22   attr_reader :type
  23
  24   def initialize(type=UNKNOWN, message=nil)
  25     super(message)
  26     @type = type
  27   end
  28
  29 end
  30
  31 class TTransport
  32   def isOpen(); nil; end
  33
  34   def open(); nil; end
  35
  36   def close(); nil; end
  37
  38   def read(sz); nil; end
  39
  40   def readAll(sz)
  41     buff = ''
  42     have = 0
  43     while (have < sz)
  44       chunk = read(sz - have)
  45       have += chunk.length
  46       buff += chunk
  47     end
  48     return buff
  49   end
  50
  51   def write(buf); nil; end
  52
  53   def flush(); nil; end
  54
  55 end
  56
  57 class TServerTransport
  58   def listen(); nil; end
  59
  60   def accept(); nil; end
  61
  62   def close(); nil; end
  63
  64 end
  65
  66 class TTransportFactory
  67   def getTransport(trans)
  68     return trans
  69   end
  70 end
  71
  72 class TBufferedTransport < TTransport
  73   def initialize(transport)
  74     @transport = transport
  75     @wbuf = ''
  76   end
  77
  78   def isOpen()
  79     return @transport.isOpen()
  80   end
  81
  82   def open()
  83     @transport.open()
  84   end
  85
  86   def close()
  87     @transport.close()
  88   end
  89
  90   def read(sz)
  91     return @transport.read(sz)
  92   end
  93
  94   def write(buf)
  95     @wbuf += buf
  96   end
  97
  98   def flush()
  99     @transport.write(@wbuf)
 100     @transport.flush()
 101     @wbuf = ''
 102   end
 103 end
 104
 105 class TBufferedTransportFactory < TTransportFactory
 106   def getTransport(transport)
 107     return TBufferedTransport.new(transport)
 108   end
 109 end
 110
 111
 112 class TFramedTransport < TTransport
 113   def initialize(transport, read=true, write=true)
 114     @transport = transport
 115     @rbuf      = ''
 116     @wbuf      = ''
 117     @read      = read
 118     @write     = write
 119   end
 120
 121   def isOpen()
 122     return @transport.isOpen
 123   end
 124
 125   def open()
 126     @transport.open
 127   end
 128
 129   def close()
 130     @transport.close
 131   end
 132
 133   def read(sz)
 134     if !@read
 135       return @transport.read(sz)
 136     end
 137
 138     if (sz <= 0)
 139       return ''
 140     end
 141
 142     if (@rbuf.length == 0)
 143       self.readFrame
 144     end
 145
 146     # return full buf
 147     if (sz > @rbuf.length)
 148       out = @rbuf
 149       @rbuf = ''
 150       return out
 151     end
 152
 153     # Return substr
 154     out = @rbuf.slice(0, sz)
 155     @rbuf = @rbuf.slice(sz, @rbuf.length)
 156     return out
 157
 158   end
 159
 160   def write(buf,sz=nil)
 161
 162     if !@write
 163       return @transport.write(buf);
 164     end
 165
 166     if (sz != nil && sz < buf.length)
 167       buf = buf.slice(0,sz)
 168     end
 169
 170     @wbuf += buf
 171
 172   end
 173
 174   #
 175   # Writes the output buffer to the stream in the format of a 4-byte length
 176   # followed by the actual data.
 177   #
 178   def flush
 179     if !@write
 180       return @transport.flush
 181     end
 182
 183     out = [@wbuf.length].pack('N')
 184     out += @wbuf
 185     @transport.write(out)
 186     @transport.flush
 187     @wbuf = ''
 188   end
 189
 190   private
 191
 192   def readFrame
 193     buf  = @transport.readAll(4)
 194     val  = buf.unpack('N')
 195     sz   = val[0]
 196
 197     @rbuf = @transport.readAll(sz)
 198   end
 199
 200 end
 201
 202
 203 class TFramedTransportFactory < TTransportFactory
 204   def getTransport(transport)
 205     return TFramedTransport.new(transport)
 206   end
 207 end
 208
 209 class TMemoryBuffer < TTransport
 210   def initialize(sz=1024)
 211     @buf = ''
 212     @sz  = sz
 213     wpos = 0
 214     rpos = 0
 215   end
 216
 217   def isOpen
 218     return 1
 219   end
 220
 221   def open
 222   end
 223
 224   def close
 225   end
 226
 227   def peek
 228     return rpos < wpos
 229   end
 230
 231   def getBuffer
 232     return @buf
 233   end
 234
 235   def resetBuffer(new_buf = '')
 236      @buf  = new_buf
 237      @sz   = new_buf.length
 238      @wpos = new_buf.length
 239      @rpos = 0
 240   end
 241
 242   def available
 243     return @wpos - @rpos
 244   end
 245
 246   def read(len)
 247     avail = available()
 248
 249     return '' if avail == 0
 250
 251     #how much to give
 252     give = len
 253     give = avail if avail < len
 254
 255     ret = @buf.slice(@rpos,give)
 256
 257     @rpos += give;
 258
 259     return ret;
 260   end
 261
 262   def write(wbuf)
 263     @buf  += wbuf
 264     @wpos += wbuf.length()
 265   end
 266
 267   def flush
 268   end
 269 end