droonga-engine 1.0.1

data/lib/droonga/plugins/groonga/column_list.rb

@@ -0,0 +1,124 @@
+# Copyright (C) 2014 Droonga Project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "groonga/command/column-list"
+
+require "droonga/plugin"
+require "droonga/plugins/groonga/generic_command"
+
+module Droonga
+  module Plugins
+    module Groonga
+      module ColumnList
+        HEADER = [
+          ["id",     "UInt32"],
+          ["name",   "ShortText"],
+          ["path",   "ShortText"],
+          ["type",   "ShortText"],
+          ["flags",  "ShortText"],
+          ["domain", "ShortText"],
+          ["range",  "ShortText"],
+          ["source", "ShortText"],
+        ].freeze
+
+        class Command < GenericCommand
+          def process_request(request)
+            command_class = ::Groonga::Command.find("column_list")
+            @command = command_class.new("column_list", request)
+
+            table_name = valid_table_name("table")
+
+            columns = list_columns(table_name)
+            [HEADER, *columns]
+          end
+
+          private
+          def list_columns(table_name)
+            table = @context[table_name]
+            table.columns.collect do |column|
+              format_column(column)
+            end
+          end
+
+          def format_column(column)
+            [
+              column.id,
+              column.local_name,
+              column.path,
+              column_type(column),
+              column_flags(column),
+              column.domain.name,
+              column.range.name,
+              column_source(column),
+            ]
+          end
+
+          def column_type(column)
+            case column
+            when ::Groonga::FixSizeColumn
+              "fix"
+            when ::Groonga::VariableSizeColumn
+              "var"
+            when ::Groonga::IndexColumn
+              "index"
+            end
+          end
+
+          def column_flags(column)
+            flags = []
+            if column.is_a?(::Groonga::IndexColumn)
+              flags << "COLUMN_INDEX"
+              flags << "WITH_SECTION" if column.with_section?
+              flags << "WITH_WEIGHT" if column.with_weight?
+              flags << "WITH_POSITION" if column.with_position?
+            else
+              if column.scalar?
+                flags << "COLUMN_SCALAR"
+              elsif column.vector?
+                flags << "COLUMN_VECTOR"
+              end
+              flags << "WITH_WEIGHT" if column.with_weight?
+            end
+            flags.join('|')
+          end
+
+          def column_source(column)
+            return [] unless column.is_a?(::Groonga::IndexColumn)
+            column.sources.collect do |source|
+              if source.is_a?(::Groonga::Table)
+                "_key"
+              else
+                source.local_name
+              end
+            end
+          end
+        end
+
+        class Handler < Droonga::Handler
+          def handle(message)
+            command = Command.new(@context)
+            command.execute(message.request)
+          end
+        end
+
+        Groonga.define_single_step do |step|
+          step.name = "column_list"
+          step.handler = Handler
+          step.collector = Collectors::Or
+        end
+      end
+    end
+  end
+end

data/lib/droonga/plugins/groonga/column_remove.rb

@@ -0,0 +1,65 @@
+# Copyright (C) 2014 Droonga Project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "groonga/command/column-remove"
+
+require "droonga/plugin"
+require "droonga/plugins/groonga/generic_command"
+
+module Droonga
+  module Plugins
+    module Groonga
+      module ColumnRemove
+        class Command < GenericCommand
+          def process_request(request)
+            command_class = ::Groonga::Command.find("column_remove")
+            @command = command_class.new("column_remove", request)
+
+            table_name = valid_table_name("table")
+            column_name = valid_column_name("name", table_name)
+
+            remove_column(table_name, column_name)
+          end
+
+          private
+          def remove_column(table_name, column_name)
+            ::Groonga::Schema.define(:context => @context) do |schema|
+              schema.change_table(table_name) do |table|
+                table.remove_column(column_name)
+              end
+            end
+            true
+          end
+        end
+
+        class Handler < Droonga::Handler
+          action.synchronous = true
+
+          def handle(message)
+            command = Command.new(@context)
+            command.execute(message.request)
+          end
+        end
+
+        Groonga.define_single_step do |step|
+          step.name = "column_remove"
+          step.write = true
+          step.handler = Handler
+          step.collector = Collectors::Or
+        end
+      end
+    end
+  end
+end

data/lib/droonga/plugins/groonga/column_rename.rb

@@ -0,0 +1,67 @@
+# Copyright (C) 2014 Droonga Project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "groonga/command/column-rename"
+
+require "droonga/plugin"
+require "droonga/plugins/groonga/generic_command"
+
+module Droonga
+  module Plugins
+    module Groonga
+      module ColumnRename
+        class Command < GenericCommand
+          def process_request(request)
+            command_class = ::Groonga::Command.find("column_rename")
+            @command = command_class.new("column_rename", request)
+
+            table_name = valid_table_name("table")
+            column_name = valid_column_name("name", table_name)
+
+            new_name = @command["new_name"]
+
+            rename_column(table_name, column_name, new_name)
+          end
+
+          private
+          def rename_column(table_name, column_name, new_name)
+            ::Groonga::Schema.define(:context => @context) do |schema|
+              schema.change_table(table_name) do |table|
+                table.rename_column(column_name, new_name)
+              end
+            end
+            true
+          end
+        end
+
+        class Handler < Droonga::Handler
+          action.synchronous = true
+
+          def handle(message)
+            command = Command.new(@context)
+            command.execute(message.request)
+          end
+        end
+
+        Groonga.define_single_step do |step|
+          step.name = "column_rename"
+          step.write = true
+          step.handler = Handler
+          step.collector = Collectors::Or
+        end
+      end
+    end
+  end
+end

data/lib/droonga/plugins/groonga/delete.rb

@@ -0,0 +1,117 @@
+# Copyright (C) 2013-2014 Droonga Project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "groonga/command/delete"
+
+require "droonga/plugin"
+require "droonga/plugins/groonga/generic_command"
+
+module Droonga
+  module Plugins
+    module Groonga
+      module Delete
+        class Command < GenericCommand
+          def process_request(request)
+            command_class = ::Groonga::Command.find("delete")
+            @command = command_class.new("delete", request)
+
+            table_name = valid_table_name("table")
+
+            key = @command["key"]
+            id = @command["id"]
+            filter = @command["filter"]
+
+            validate_parameters(key, id, filter)
+
+            table = @context[table_name]
+            if key
+              delete_record_by_key(table, key)
+            elsif id
+              delete_record_by_id(table, id)
+            else
+              delete_record_by_filter(table, filter)
+            end
+
+            true
+          end
+
+          private
+          def validate_parameters(key, id, filter)
+            if key.nil? and id.nil? and filter.nil?
+              message = "you must specify \"key\", \"id\", or \"filter\""
+              raise CommandError.new(:status => Status::INVALID_ARGUMENT,
+                                     :message => message,
+                                     :result => false)
+            end
+
+            count = 0
+            count += 1 if key
+            count += 1 if id
+            count += 1 if filter
+            if count > 1
+              message = "\"key\", \"id\", and \"filter\" are exclusive"
+              raise CommandError.new(:status => Status::INVALID_ARGUMENT,
+                                     :message => message,
+                                     :result => false)
+            end
+          end
+
+          def delete_record_by_key(table, key)
+            record = table[key]
+            record.delete unless record.nil?
+          end
+
+          def delete_record_by_id(table, id)
+            record = table[id.to_i]
+            record.delete if record and record.valid_id?
+          end
+
+          def delete_record_by_filter(table, filter)
+            condition = ::Groonga::Expression.new(:context => @context)
+            condition.define_variable(:domain => table)
+            begin
+              condition.parse(filter, :syntax => :script)
+            rescue ::Groonga::SyntaxError
+              message = "syntax error in filter: <#{filter}>"
+              raise CommandError.new(:status => Status::SYNTAX_ERROR,
+                                     :message => message,
+                                     :result => false)
+            end
+            records = table.select(condition)
+            records.each do |record|
+              record.key.delete
+            end
+          end
+        end
+
+        class Handler < Droonga::Handler
+          action.synchronous = true
+
+          def handle(message)
+            command = Command.new(@context)
+            command.execute(message.request)
+          end
+        end
+
+        Groonga.define_single_step do |step|
+          step.name = "delete"
+          step.write = true
+          step.handler = Handler
+          step.collector = Collectors::Or
+        end
+      end
+    end
+  end
+end

data/lib/droonga/plugins/groonga/generic_command.rb

@@ -0,0 +1,105 @@
+# Copyright (C) 2013-2014 Droonga Project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "groonga"
+
+module Droonga
+  module Plugins
+    module Groonga
+      module Status
+        SUCCESS          = 0
+        INVALID_ARGUMENT = -22
+        SYNTAX_ERROR     = -63
+      end
+
+      class GenericCommand
+        class CommandError < StandardError
+          attr_reader :status, :message, :result
+
+          def initialize(params={})
+            @status = params[:status]
+            @message = params[:message]
+            @result = params[:result]
+          end
+        end
+
+        def initialize(context)
+          @context = context
+        end
+
+        def execute(request)
+          @start_time = Time.now.to_f
+          result = process_request(request)
+          [header(Status::SUCCESS), result]
+        rescue CommandError => error
+          unless error.result.nil?
+            [header(error.status, error.message), error.result]
+          else
+            [header(error.status, error.message)]
+          end
+        end
+
+        private
+        def header(return_code, error_message="")
+          elapsed_time = Time.now.to_f - @start_time
+          header = [return_code, @start_time, elapsed_time]
+          header.push(error_message) unless error_message.empty?
+          header
+        end
+
+        def valid_table_name(name)
+          table_name = @command[name]
+
+          if table_name.nil?
+            message = "you must specify table via \"#{name}\""
+            raise CommandError.new(:status => Status::INVALID_ARGUMENT,
+                                   :message => message,
+                                   :result => false)
+          end
+
+          if @context[table_name].nil?
+            message = "table not found: <#{table_name.to_s}>"
+            raise CommandError.new(:status => Status::INVALID_ARGUMENT,
+                                   :message => message,
+                                   :result => false)
+          end
+
+          table_name
+        end
+
+        def valid_column_name(name, table_name)
+          column_name = @command[name]
+
+          if column_name.nil?
+            message = "you must specify column via \"#{name}\""
+            raise CommandError.new(:status => Status::INVALID_ARGUMENT,
+                                   :message => message,
+                                   :result => false)
+          end
+
+          if @context[table_name].column(column_name).nil?
+            message = "column not found: <#{column_name.to_s}> in " +
+                          "<#{table_name.to_s}>"
+            raise CommandError.new(:status => Status::INVALID_ARGUMENT,
+                                   :message => message,
+                                   :result => false)
+          end
+
+          column_name
+        end
+      end
+    end
+  end
+end

data/lib/droonga/plugins/groonga/generic_response.rb

@@ -0,0 +1,43 @@
+# Copyright (C) 2013-2014 Droonga Project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "droonga/plugin"
+
+module Droonga
+  module Plugins
+    module Groonga
+      module Generic
+        class Adapter < Droonga::Adapter
+          groonga_commands = [
+            "table_create",
+            "table_remove",
+            "table_list",
+            "column_create",
+            "column_remove",
+            "column_rename",
+            "column_list",
+            "delete",
+          ]
+          input_message.pattern  = ["type", :in, groonga_commands]
+          output_message.pattern = ["body.result", :exist]
+
+          def adapt_output(output_message)
+            output_message.body = output_message.body["result"]
+          end
+        end
+      end
+    end
+  end
+end