YUKI Hiroshi
null+****@clear*****
Fri Apr 10 13:08:34 JST 2015
YUKI Hiroshi 2015-04-10 13:08:34 +0900 (Fri, 10 Apr 2015) New Revision: 67ce4de554987959aca21ea639e3cda9cb982df6 https://github.com/droonga/droonga-engine/commit/67ce4de554987959aca21ea639e3cda9cb982df6 Message: Implement system.absorb-data command Modified files: lib/droonga/plugins/system/absorb_data.rb Modified: lib/droonga/plugins/system/absorb_data.rb (+49 -0) =================================================================== --- lib/droonga/plugins/system/absorb_data.rb 2015-04-10 12:38:19 +0900 (ba2c6c0) +++ lib/droonga/plugins/system/absorb_data.rb 2015-04-10 13:08:34 +0900 (26c8150) @@ -14,6 +14,10 @@ # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA require "droonga/plugin" +require "droonga/catalog/dataset" +require "droonga/node_name" + +require "drndump/dumper" module Droonga module Plugins @@ -21,9 +25,54 @@ module Droonga class AbsorbDataHandler < Droonga::Handler action.synchronous = true + class MissingHostParameter < BadRequest + def initialize + super("\"host\" must be specified.") + end + end + + class DumpFailed < InternalServerError + def initialize(error) + super("source node returns an error.", + error) + end + end + def handle(message) + raise MissingHostParameter.new unless message.include?("host") + + dumper = Drndump::Dumper.new(dumper_params(message)) + error_message = dumper.run do |message| + @messenger.forward(message, + "to" => my_node_name, + "type" => message["type"]) + end + + raise DumpFailed.new(error_message) if error_message + true end + + private + def dumper_params(message) + { + :host => message["host"], + :port => message["port"] ||= NodeName::DEFAULT_PORT, + :tag => message["tag"] ||= NodeName::DEFAULT_TAG, + :dataset => message["dataset"] ||= Catalog::Dataset::DEFAULT_NAME, + + :receiver_host => myself.host, + :receiver_port => 0, + } + end + + def myself + @myself ||= NodeName.parse(my_node_name) + end + + def my_node_name + @messenger.engine_state.name + end end define_single_step do |step| -------------- next part -------------- HTML����������������������������...다운로드
