Add type argument to QueryMethods module (#108)

src/db/connection.cr

@@ -36,7 +36,7 @@ module DB
     end
 
     # :nodoc:
-    def fetch_or_build_prepared_statement(query)
+    def fetch_or_build_prepared_statement(query) : Statement
       @statements_cache.fetch(query) { build_prepared_statement(query) }
     end
 
@@ -46,7 +46,7 @@ module DB
     # :nodoc:
     abstract def build_unprepared_statement(query) : Statement
 
-    def begin_transaction
+    def begin_transaction : Transaction
       raise DB::Error.new("There is an existing transaction in this connection") if @transaction
       @transaction = true
       create_transaction

src/db/database.cr

@@ -37,7 +37,7 @@ module DB
     getter pool
 
     # Returns the uri with the connection settings to the database
-    getter uri
+    getter uri : URI
 
     getter? prepared_statements : Bool
 
@@ -86,17 +86,17 @@ module DB
     end
 
     # :nodoc:
-    def fetch_or_build_prepared_statement(query)
+    def fetch_or_build_prepared_statement(query) : PoolStatement
       @statements_cache.fetch(query) { build_prepared_statement(query) }
     end
 
     # :nodoc:
-    def build_prepared_statement(query)
+    def build_prepared_statement(query) : PoolStatement
       PoolPreparedStatement.new(self, query)
     end
 
     # :nodoc:
-    def build_unprepared_statement(query)
+    def build_unprepared_statement(query) : PoolStatement
       PoolUnpreparedStatement.new(self, query)
     end
 

src/db/pool_prepared_statement.cr

@@ -30,7 +30,7 @@ module DB
 
     # builds a statement over a real connection
     # the connection is registered in `@connections`
-    private def build_statement
+    private def build_statement : Statement
       clean_connections
       conn, existing = @db.checkout_some(@connections)
       begin

src/db/pool_unprepared_statement.cr

@@ -14,7 +14,7 @@ module DB
     end
 
     # builds a statement over a real connection
-    private def build_statement
+    private def build_statement : Statement
       conn = @db.pool.checkout
       begin
         conn.unprepared.build(@query)

src/db/query_methods.cr

@@ -17,9 +17,9 @@ module DB
   #
   # Including `QueryMethods` requires a `build(query) : Statement` method that is not expected
   # to be called directly.
-  module QueryMethods
+  module QueryMethods(Stmt)
     # :nodoc:
-    abstract def build(query) : Statement
+    abstract def build(query) : Stmt
 
     # Executes a *query* and returns a `ResultSet` with the results.
     # The `ResultSet` must be closed manually.

src/db/session_methods.cr

@@ -8,7 +8,7 @@ module DB
   #
   # This module serves for dsl reuse over session like objects.
   module SessionMethods(Session, Stmt)
-    include QueryMethods
+    include QueryMethods(Stmt)
 
     # Returns whether by default the statements should
     # be prepared or not.
@@ -49,7 +49,7 @@ module DB
     end
 
     struct PreparedQuery(Session, Stmt)
-      include QueryMethods
+      include QueryMethods(Stmt)
 
       def initialize(@session : Session)
       end
@@ -60,7 +60,7 @@ module DB
     end
 
     struct UnpreparedQuery(Session, Stmt)
-      include QueryMethods
+      include QueryMethods(Stmt)
 
       def initialize(@session : Session)
       end

src/db/statement.cr

@@ -63,12 +63,12 @@ module DB
     end
 
     # See `QueryMethods#exec`
-    def exec
+    def exec : DB::ExecResult
       perform_exec_and_release(Slice(Any).empty)
     end
 
     # See `QueryMethods#exec`
-    def exec(args : Array)
+    def exec(args : Array) : DB::ExecResult
       perform_exec_and_release(args)
     end
 
@@ -79,12 +79,12 @@ module DB
     end
 
     # See `QueryMethods#query`
-    def query
+    def query : DB::ResultSet
       perform_query_with_rescue Tuple.new
     end
 
     # See `QueryMethods#query`
-    def query(args : Array)
+    def query(args : Array) : DB::ResultSet
       perform_query_with_rescue args
     end
 

src/db/transaction.cr

@@ -36,7 +36,7 @@ module DB
   end
 
   class TopLevelTransaction < Transaction
-    getter connection
+    getter connection : Connection
     # :nodoc:
     property savepoint_name : String? = nil