Support mappable classes in query_one and query_all methods

spec/mapping_spec.cr

@@ -131,4 +131,23 @@ describe "DB.mapping" do
     end
   end
 
+  it "should initialize from a query_one" do
+    with_dummy do |db|
+      obj = db.query_one "1,a", as: SimpleMapping
+      obj.c0.should eq(1)
+      obj.c1.should eq("a")
+    end
+  end
+
+  it "should initialize from a query_all" do
+    with_dummy do |db|
+      objs = db.query_all "1,a 2,b", as: SimpleMapping
+      objs.size.should eq(2)
+      objs[0].c0.should eq(1)
+      objs[0].c1.should eq("a")
+      objs[1].c0.should eq(2)
+      objs[1].c1.should eq("b")
+    end
+  end
+
 end

src/db/mapping.cr

@@ -1,5 +1,8 @@
 module DB
 
+  # Empty module used for marking a class as supporting DB:Mapping
+  module Mappable; end
+
   # The `DB.mapping` macro defines how an object is built from a DB::ResultSet.
   #
   # It takes hash literal as argument, in which attributes and types are defined.
@@ -56,6 +59,8 @@ module DB
   #
   # This macro also declares instance variables of the types given in the mapping.
   macro mapping(properties, strict = true)
+    include DB::Mappable
+
     {% for key, value in properties %}
       {% properties[key] = {type: value} unless value.is_a?(HashLiteral) || value.is_a?(NamedTupleLiteral) %}
     {% end %}

src/db/result_set.cr

@@ -64,6 +64,11 @@ module DB
     # Reads the next column value
     abstract def read
 
+    # Reads the next columns and maps them to a class
+    def read(type : DB::Mappable.class)
+      type.new(self)
+    end
+
     # Reads the next column value as a **type**
     def read(type : T.class) : T
       read.as(T)