From f96351f84115a26b4dd992719421ea91f4f757e3 Mon Sep 17 00:00:00 2001
From: Ivan Trubach <mr.trubach@icloud.com>
Date: Mon, 28 Sep 2020 15:46:11 +0300
Subject: [PATCH 1/2] Remove unused code

---
 sql/driver/driver.go | 38 ++------------------------------------
 1 file changed, 2 insertions(+), 36 deletions(-)

diff --git a/sql/driver/driver.go b/sql/driver/driver.go
index 26af9fb26..026d11d7c 100644
--- a/sql/driver/driver.go
+++ b/sql/driver/driver.go
@@ -33,45 +33,11 @@ func (d sqlDriver) Open(name string) (driver.Conn, error) {
 	return &conn{db: db}, nil
 }
 
-// proxyDriver is used to turn an existing DB into a driver.Driver.
-type proxyDriver struct {
-	db *genji.DB
-}
-
-func newDriver(db *genji.DB) driver.Driver {
-	return proxyDriver{
-		db: db,
-	}
-}
-
-func (d proxyDriver) Open(name string) (driver.Conn, error) {
-	return &conn{db: d.db}, nil
-}
-
-type proxyConnector struct {
-	driver driver.Driver
-}
-
-func newProxyConnector(db *genji.DB) driver.Connector {
-	return proxyConnector{
-		driver: newDriver(db),
-	}
-}
-
-func (c proxyConnector) Connect(ctx context.Context) (driver.Conn, error) {
-	return c.driver.Open("")
-}
-
-func (c proxyConnector) Driver() driver.Driver {
-	return c.driver
-}
-
 // conn represents a connection to the Genji database.
 // It implements the database/sql/driver.Conn interface.
 type conn struct {
-	db            *genji.DB
-	tx            *genji.Tx
-	nonPromotable bool
+	db *genji.DB
+	tx *genji.Tx
 }
 
 // Prepare returns a prepared statement, bound to this connection.

From 26956b2b0cbb11468928acc2c0ca04876ca6bac5 Mon Sep 17 00:00:00 2001
From: Ivan Trubach <mr.trubach@icloud.com>
Date: Thu, 1 Oct 2020 20:46:38 +0300
Subject: [PATCH 2/2] Use driver.DriverContext in sql/driver

---
 sql/driver/driver.go | 47 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/sql/driver/driver.go b/sql/driver/driver.go
index 026d11d7c..d7d9cb6c3 100644
--- a/sql/driver/driver.go
+++ b/sql/driver/driver.go
@@ -6,6 +6,7 @@ import (
 	"database/sql/driver"
 	"errors"
 	"io"
+	"runtime"
 	"sync"
 
 	"github.com/genjidb/genji"
@@ -20,17 +21,61 @@ func init() {
 	sql.Register("genji", sqlDriver{})
 }
 
+var (
+	_ driver.Driver        = (*sqlDriver)(nil)
+	_ driver.DriverContext = (*sqlDriver)(nil)
+)
+
 // sqlDriver is a driver.Driver that can open a new connection to a Genji database.
 // It is the driver used to register Genji against the database/sql package.
 type sqlDriver struct{}
 
 func (d sqlDriver) Open(name string) (driver.Conn, error) {
+	return nil, errors.New("requires go1.10 or greater")
+}
+
+func (d sqlDriver) OpenConnector(name string) (driver.Connector, error) {
 	db, err := genji.Open(name)
 	if err != nil {
 		return nil, err
 	}
 
-	return &conn{db: db}, nil
+	c := &connector{
+		db:     db,
+		driver: d,
+	}
+	runtime.SetFinalizer(c, (*connector).Close)
+
+	return c, nil
+}
+
+var (
+	_ driver.Connector = (*connector)(nil)
+	_ io.Closer        = (*connector)(nil)
+)
+
+type connector struct {
+	driver driver.Driver
+
+	db *genji.DB
+
+	closeOnce sync.Once
+}
+
+func (c *connector) Connect(ctx context.Context) (driver.Conn, error) {
+	return &conn{db: c.db}, nil
+}
+
+func (c *connector) Driver() driver.Driver {
+	return c.driver
+}
+
+func (c *connector) Close() error {
+	var err error
+	c.closeOnce.Do(func() {
+		err = c.db.Close()
+	})
+	return err
 }
 
 // conn represents a connection to the Genji database.