Fix handling custom type

Dynamoid · Aug 4, 2018 · 8c04a12 · 8c04a12
1 parent 140c3e8
commit 8c04a12
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/dynamoid/primary_key_type_mapping.rb b/lib/dynamoid/primary_key_type_mapping.rb
@@ -3,6 +3,10 @@
 module Dynamoid
   class PrimaryKeyTypeMapping
     def self.dynamodb_type(type, options)
+      if Class === type
+        type = type.respond_to?(:dynamoid_field_type) ? type.dynamoid_field_type : :string
+      end
+
       case type
       when :string, :serialized
         :string
@@ -22,8 +26,6 @@ def self.dynamodb_type(type, options)
                           options[:store_as_string]
                         end
         string_format ? :string : :number
-      when Class
-        type.respond_to?(:dynamoid_field_type) ? type.dynamoid_field_type : :string
       else
         raise Errors::UnsupportedKeyType, "#{type} cannot be used as a type of table key attribute"
       end