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 



154
155
156
157
158
159
160
161
 
# File 'lib/rakie/channel.rb', line 154

def close
  if @io.closed?
    return nil
  end

  Event.delete(@io)
  return nil
end

#closed? ⇒ Boolean

Returns:

  • (Boolean) 


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

def closed?
  @io.closed?
end

#eof? ⇒ Boolean

Returns:

  • (Boolean) 


163
164
165
 
# File 'lib/rakie/channel.rb', line 163

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
 
# 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
    end

    len -= task
    @write_task.shift

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

#on_detach(io) ⇒ Object 



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

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 



71
72
73
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
 
# File 'lib/rakie/channel.rb', line 71

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 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 



126
127
128
129
130
131
132
133
134
135
136
137
138
139
 
# File 'lib/rakie/channel.rb', line 126

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 



141
142
143
144
145
146
147
148
149
150
151
152
 
# File 'lib/rakie/channel.rb', line 141

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

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

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

  return 0
end