Class: Rakie::Channel

Inherits:
Object
  • Object
show all
Defined in:
lib/rakie/channel.rb

Direct Known Subclasses

TCPChannel

Constant Summary collapse

DEFAULT_BUFFER_SIZE = 
512 * 1024

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(io, delegate = nil) ⇒ Channel

Returns a new instance of Channel.



7
8
9
10
11
12
13
14
 
# File 'lib/rakie/channel.rb', line 7

def initialize(io, delegate=nil)
  @io = io
  @read_buffer = String.new
  @write_buffer = String.new
  @write_task = []
  @delegate = delegate
  Event.push(io, self, Event::READ_EVENT)
end

Instance Attribute Details

#delegate ⇒ Object

Returns the value of attribute delegate.



3
4
5
 
# File 'lib/rakie/channel.rb', line 3

def delegate
  @delegate
end

Instance Method Details

#close ⇒ Object 



159
160
161
162
163
164
165
166
 
# File 'lib/rakie/channel.rb', line 159

def close
  if @io.closed?
    return nil
  end

  Event.delete(@io)
  return nil
end

#closed? ⇒ Boolean

Returns:

  • (Boolean) 


172
173
174
 
# File 'lib/rakie/channel.rb', line 172

def closed?
  @io.closed?
end

#eof? ⇒ Boolean

Returns:

  • (Boolean) 


168
169
170
 
# File 'lib/rakie/channel.rb', line 168

def eof?
  @read_buffer.empty?
end

#handle_write(len) ⇒ Object 



53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/rakie/channel.rb', line 53

def handle_write(len)
  task = @write_task[0]

  while len > 0
    if len < task
      @write_task[0] = task - len
      return
    end

    len -= task
    @write_task.shift

    if @delegate != nil
      @delegate.on_send(self)
       Log.debug("Channel handle on_send")
    end

    task = @write_task[0]
  end
end

#on_detach(io) ⇒ Object 



109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
 
# File 'lib/rakie/channel.rb', line 109

def on_detach(io)
  if io.closed?
    return
  end

  begin
    io.close

  rescue
    Log.debug("Channel is already closed")
    return
  end

  if @delegate
    @delegate.on_close(self)
  end

  Log.debug("Channel close ok")
end

#on_read(io) ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/rakie/channel.rb', line 16

def on_read(io)
  begin
    loop do
      @read_buffer << io.read_nonblock(DEFAULT_BUFFER_SIZE)
    end

  rescue IO::EAGAINWaitReadable
    Log.debug("Channel read pending")

  rescue IO::EWOULDBLOCKWaitReadable
    Log.debug("Channel read pending")

  rescue Exception => e
    # Process the last message on exception
    if @delegate != nil
      @delegate.on_recv(self, @read_buffer)
      @read_buffer = String.new # Reset buffer
    end

    Log.debug("Channel error #{io}: #{e}")
    return Event::HANDLE_FAILED
  end

  if @delegate != nil
    Log.debug("Channel handle on_recv")
    len = @delegate.on_recv(self, @read_buffer)

    if len > @read_buffer.length
      len = @read_buffer.length
    end

    @read_buffer = @read_buffer[len .. -1]
  end

  return Event::HANDLE_CONTINUED
end

#on_write(io) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/rakie/channel.rb', line 74

def on_write(io)
  len = 0
  offset = 0

  begin
    while @write_buffer.length > 0
      len = io.write_nonblock(@write_buffer)
      offset += len
      @write_buffer = @write_buffer[len .. -1]
    end

    Log.debug("Channel write #{len} bytes finished")

  rescue IO::EAGAINWaitWritable
    self.handle_write(offset)

    Log.debug("Channel write pending")
    return Event::HANDLE_CONTINUED

  rescue IO::EWOULDBLOCKWaitWritable
    self.handle_write(offset)

    Log.debug("Channel write pending")
    return Event::HANDLE_CONTINUED

  rescue
    Log.debug("Channel close #{io}")
    return Event::HANDLE_FAILED
  end

  self.handle_write(offset)

  return Event::HANDLE_FINISHED
end

#read(size) ⇒ Object 



129
130
131
132
133
134
135
136
137
138
139
140
141
142
 
# File 'lib/rakie/channel.rb', line 129

def read(size)
  if self.eof?
    return ""
  end

  if size > data.length
    size = data.length
  end

  data = @read_buffer[0 .. (size - 1)]
  @read_buffer = @read_buffer[size .. -1]

  return data
end

#write(data) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
156
157
 
# File 'lib/rakie/channel.rb', line 144

def write(data)
  if @io.closed?
    return -1
  end

  @write_buffer << data
  @write_task << data.length

  Log.debug("write buffer append size: #{data.length}")

  Event.modify(@io, self, (Event::READ_EVENT | Event::WRITE_EVENT))

  return 0
end