schema: support vectorOptions

droonga · Mar 14, 2014 · 2735180 · 2735180
1 parent fd63909
commit 2735180
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/lib/droonga/catalog/schema.rb b/lib/droonga/catalog/schema.rb
@@ -65,11 +65,12 @@ def flags
       end
 
       class Column
-        attr_reader :table, :name, :data, :index_options
+        attr_reader :table, :name, :data, :vector_options, :index_options
         def initialize(table, name, data)
           @table = table
           @name = name
           @data = data
+          @vector_options = ColumnVectorOptions.new(vector_options_data)
           @index_options = ColumnIndexOptions.new(index_options_data)
         end
 
@@ -97,7 +98,7 @@ def type_flag
         end
 
         def flags
-          [type_flag] + index_options.flags
+          [type_flag] + vector_options.flags + index_options.flags
         end
 
         def value_type
@@ -128,6 +129,10 @@ def to_column_create_body
         end
 
         private
+        def vector_options_data
+          @data["vectorOptions"] || {}
+        end
+
         def index_options_data
           @data["indexOptions"] || {}
         end

diff --git a/test/unit/catalog/test_schema.rb b/test/unit/catalog/test_schema.rb
@@ -241,6 +241,17 @@ def test_type
                        flags(data))
         end
 
+        def test_weight_options
+          data = {
+            "type" => "Vector",
+            "vectorOptions" => {
+              "weight" => true
+            }
+          }
+          assert_equal(["COLUMN_VECTOR", "WITH_WEIGHT"],
+                       flags(data))
+        end
+
         def test_index_options
           data = {
             "type" => "Index",