Class: Droonga::Client::Connection::DroongaProtocol::Coolio

Inherits:
Object
  • Object
show all
Defined in:
lib/droonga/client/connection/droonga-protocol/coolio.rb

Defined Under Namespace

Classes: InfiniteRequest, Receiver, Request, Sender

Instance Method Summary collapse

Constructor Details

#initialize(host, port, tag, options = {}) ⇒ Coolio

Returns a new instance of Coolio.



136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
# File 'lib/droonga/client/connection/droonga-protocol/coolio.rb', line 136

def initialize(host, port, tag, options={})
  @host = host
  @port = port
  @tag = tag
  default_options = {
  }
  @options = default_options.merge(options)
  @loop = options[:loop] || ::Coolio::Loop.default

  @sender = Sender.connect(@host, @port)
  @sender.attach(@loop)
  @receiver_host = @options[:receiver_host] || Socket.gethostname
  @receiver_port = @options[:receiver_port] || 0
  @receiver = Receiver.new(@receiver_host, @receiver_port)
  @receiver.attach(@loop)
end

Instance Method Details

#closeObject



219
220
221
222
# File 'lib/droonga/client/connection/droonga-protocol/coolio.rb', line 219

def close
  @sender.close
  @receiver.close
end

#request(message, options = {}, &block) ⇒ Object



153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
# File 'lib/droonga/client/connection/droonga-protocol/coolio.rb', line 153

def request(message, options={}, &block)
  id = message["id"] || generate_id
  message = message.merge("id" => id,
                          "replyTo" => @receiver.droonga_name)
  send(message, options)

  sync = block.nil?
  if sync
    response = nil
    block = lambda do |_response|
      response = _response
    end
  end
  @receiver.register(id) do |response|
    @receiver.unregister(id)
    block.call(response)
  end
  request = Request.new(@receiver, id, @loop)
  if sync
    request.wait
    response
  else
    request
  end
end

#send(message, options = {}, &block) ⇒ Object



210
211
212
213
214
215
216
217
# File 'lib/droonga/client/connection/droonga-protocol/coolio.rb', line 210

def send(message, options={}, &block)
  if message["id"].nil? or message["date"].nil?
    id = message["id"] || generate_id
    date = message["date"] || Time.now
    message = message.merge("id" => id, "date" => date)
  end
  @sender.send("#{@tag}.message", message)
end

#subscribe(message, options = {}, &block) ⇒ Object



179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
# File 'lib/droonga/client/connection/droonga-protocol/coolio.rb', line 179

def subscribe(message, options={}, &block)
  id = message["id"] || generate_id
  message = message.merge("id" => id,
                          "from" => @receiver.droonga_name)
  send(message, options)

  request = InfiniteRequest.new(@loop)
  sync = block.nil?
  if sync
    yielder = nil
    buffer = []
    @receiver.register(id) do |response|
      if yielder
        while (old_response = buffer.shift)
          yielder << old_response
        end
        yielder << response
      else
        buffer << response
      end
    end
    Enumerator.new do |_yielder|
      yielder = _yielder
      request.wait
    end
  else
    @receiver.register(id, &block)
    request
  end
end