Add driver.Connector implementation for SQLite (#1001)

Author: ljluestc

sqlite3.go

@@ -2282,3 +2282,28 @@ func (rc *SQLiteRows) nextSyncLocked(dest []driver.Value) error {
 	}
 	return nil
 }
+
+// SQLiteConnector implements driver.Connector for custom connection handling.
+type SQLiteConnector struct {
+	DSN            string
+	DriverInstance *SQLiteDriver
+}
+
+// Connect implements driver.Connector.
+func (c *SQLiteConnector) Connect(ctx context.Context) (driver.Conn, error) {
+	// Context is ignored for now, as SQLiteDriver.Open does not use it.
+	return c.DriverInstance.Open(c.DSN)
+}
+
+// Driver returns the underlying driver.
+func (c *SQLiteConnector) Driver() driver.Driver {
+	return c.DriverInstance
+}
+
+// NewConnector returns a new SQLiteConnector.
+func NewConnector(dsn string) *SQLiteConnector {
+	return &SQLiteConnector{
+		DSN:            dsn,
+		DriverInstance: &SQLiteDriver{},
+	}
+}

sqlite3_test.go

@@ -10,6 +10,7 @@ package sqlite3
 
 import (
 	"bytes"
+	"context"
 	"database/sql"
 	"database/sql/driver"
 	"errors"
@@ -2586,3 +2587,14 @@ func benchmarkQueryParallel(b *testing.B) {
 		}
 	})
 }
+
+func TestSQLiteConnector(t *testing.T) {
+	connector := NewConnector(":memory:")
+	db := sql.OpenDB(connector)
+	defer db.Close()
+
+	ctx := context.Background()
+	if err := db.PingContext(ctx); err != nil {
+		t.Fatalf("PingContext failed: %v", err)
+	}
+}