Class: Thrift::BufferedTransport

Inherits:
BaseTransport show all
Defined in:
lib/thrift/transport/buffered_transport.rb

Constant Summary collapse

DEFAULT_BUFFER = 
4096

Instance Method Summary collapse

Methods inherited from BaseTransport

#read_all

Constructor Details

#initialize(transport) ⇒ BufferedTransport

Returns a new instance of BufferedTransport.



25
26
27
28
29
30
 
# File 'lib/thrift/transport/buffered_transport.rb', line 25

def initialize(transport)
  @transport = transport
  @wbuf = ''
  @rbuf = ''
  @index = 0
end

Instance Method Details

#close ⇒ Object 



40
41
42
43
 
# File 'lib/thrift/transport/buffered_transport.rb', line 40

def close
  flush
  @transport.close
end

#flush ⇒ Object 



93
94
95
96
97
98
99
100
 
# File 'lib/thrift/transport/buffered_transport.rb', line 93

def flush
  if @wbuf != ''
    @transport.write(@wbuf)
    @wbuf = ''
  end
  
  @transport.flush
end

#open ⇒ Object 



36
37
38
 
# File 'lib/thrift/transport/buffered_transport.rb', line 36

def open
  @transport.open
end

#open? ⇒ Boolean

Returns:

  • (Boolean) 


32
33
34
 
# File 'lib/thrift/transport/buffered_transport.rb', line 32

def open?
  return @transport.open?
end

#read(sz) ⇒ Object 



45
46
47
48
49
50
51
52
53
54
55
56
 
# File 'lib/thrift/transport/buffered_transport.rb', line 45

def read(sz)
  @index += sz
  ret = @rbuf.slice(@index - sz, sz) || ''

  if ret.length == 0
    @rbuf = @transport.read([sz, DEFAULT_BUFFER].max)
    @index = sz
    ret = @rbuf.slice(0, sz) || ''
  end

  ret
end

#read_byte ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/thrift/transport/buffered_transport.rb', line 58

def read_byte
  # If the read buffer is exhausted, try to read up to DEFAULT_BUFFER more bytes into it.
  if @index >= @rbuf.size
    @rbuf = @transport.read(DEFAULT_BUFFER)
    @index = 0
  end

  # The read buffer has some data now, read a single byte. Using get_string_byte() avoids
  # allocating a temp string of size 1 unnecessarily.
  @index += 1
  return ::Thrift::TransportUtils.get_string_byte(@rbuf, @index - 1)
end

#read_into_buffer(buffer, size) ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/thrift/transport/buffered_transport.rb', line 71

def read_into_buffer(buffer, size)
  i = 0
  while i < size
    # If the read buffer is exhausted, try to read up to DEFAULT_BUFFER more bytes into it.
    if @index >= @rbuf.size
      @rbuf = @transport.read(DEFAULT_BUFFER)
      @index = 0
    end

    # The read buffer has some data now, so copy bytes over to the output buffer.
    byte = ::Thrift::TransportUtils.get_string_byte(@rbuf, @index)
    ::Thrift::TransportUtils.set_string_byte(buffer, i, byte)
    @index += 1
    i += 1
  end
  i
end

#write(buf) ⇒ Object 



89
90
91
 
# File 'lib/thrift/transport/buffered_transport.rb', line 89

def write(buf)
  @wbuf << buf
end