lib/appfuel/storage/aws_dynamodb/repository.rb in appfuel-0.5.7 vs lib/appfuel/storage/aws_dynamodb/repository.rb in appfuel-0.5.8

@@ -4,8 +4,16 @@
       class << self
         def container_class_type
           "#{super}.aws.dynamodb"
         end
       end
+
+      def storage_class(domain_name)
+        mapper.storage_class('aws.dynamodb', domain_name)
+      end
+
+      def to_entity(domain_name, storage)
+        super(domain_name, 'aws.dynamodb', storage)
+      end
     end
   end
 end