add key_type option to table method

Author: ogidow

lib/dynamoid/fields.rb

@@ -200,6 +200,7 @@ def range(name, type = :string, options = {})
       # @param options [Hash] options to override default table settings
       # @option options [Symbol] :name name of a table
       # @option options [Symbol] :key name of a hash key attribute
+      # @option options [Symbol] :key_type type of a hash key attribute
       # @option options [Symbol] :inheritance_field name of an attribute used for STI
       # @option options [Symbol] :capacity_mode table billing mode - either +provisioned+ or +on_demand+
       # @option options [Integer] :write_capacity table write capacity units
@@ -210,11 +211,11 @@ def range(name, type = :string, options = {})
       # @since 0.4.0
       def table(options)
         self.options = options
-
         # a default 'id' column is created when Dynamoid::Document is included
         unless attributes.key? hash_key
           remove_field :id
-          field(hash_key)
+          key_type = options[:key_type] || :string
+          field(hash_key, key_type)
         end
 
         # The created_at/updated_at fields are declared in the `included` callback first.

spec/dynamoid/fields_spec.rb

@@ -448,4 +448,16 @@ def name=(str)
       expect(obj.name_changed?).to eq false
     end
   end
+
+  describe '.table' do
+    context 'when specify key option and key_type' do
+      it 'hash_key attribute is defined by key_type type' do
+        klass = new_class do
+          table key: :hash_key, key_type: :integer
+        end
+
+        expect(klass.attributes[:hash_key][:type]).to eq(:integer)
+      end
+    end
+  end
 end