From e6efdb5e4875dee179348ebd7a81fac78137d8f7 Mon Sep 17 00:00:00 2001
From: "Brian J. Cardiff" <bcardiff@gmail.com>
Date: Tue, 19 May 2020 12:06:27 -0300
Subject: [PATCH] Temp: Validate not_nil schemes

---
 src/db.cr | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/db.cr b/src/db.cr
index b957ccc..3c93b57 100644
--- a/src/db.cr
+++ b/src/db.cr
@@ -89,7 +89,7 @@ module DB
   record ExecResult, rows_affected : Int64, last_insert_id : Int64
 
   # :nodoc:
-  def self.driver_class(driver_name) : Driver.class
+  def self.driver_class(driver_name : String) : Driver.class
     drivers[driver_name]? ||
       raise(ArgumentError.new(%(no driver was registered for the schema "#{driver_name}", did you maybe forget to require the database driver?)))
   end
@@ -161,7 +161,7 @@ module DB
   end
 
   private def self.build_driver(uri : URI)
-    driver_class(uri.scheme).new
+    driver_class(uri.scheme.not_nil!).new
   end
 
   # :nodoc: