Merge pull request #218 from shogo82148/fix-execer-context

improve context interfaces for Exec and Query

xraysql/conn.go

@@ -126,7 +126,8 @@ func (conn *driverConn) Exec(query string, args []driver.Value) (driver.Result,
 
 func (conn *driverConn) ExecContext(ctx context.Context, query string, args []driver.NamedValue) (driver.Result, error) {
 	execer, ok := conn.Conn.(driver.Execer)
-	if !ok {
+	execerCtx, okCtx := conn.Conn.(driver.ExecerContext)
+	if !ok && !okCtx {
 		return nil, driver.ErrSkip
 	}
 
@@ -135,7 +136,7 @@ func (conn *driverConn) ExecContext(ctx context.Context, query string, args []dr
 
 	var err error
 	var result driver.Result
-	if execerCtx, ok := conn.Conn.(driver.ExecerContext); ok {
+	if okCtx {
 		result, err = execerCtx.ExecContext(ctx, query, args)
 	} else {
 		select {
@@ -168,7 +169,8 @@ func (conn *driverConn) Query(query string, args []driver.Value) (driver.Rows, e
 
 func (conn *driverConn) QueryContext(ctx context.Context, query string, args []driver.NamedValue) (driver.Rows, error) {
 	queryer, ok := conn.Conn.(driver.Queryer)
-	if !ok {
+	queryerCtx, okCtx := conn.Conn.(driver.QueryerContext)
+	if !ok && !okCtx {
 		return nil, driver.ErrSkip
 	}
 
@@ -177,7 +179,7 @@ func (conn *driverConn) QueryContext(ctx context.Context, query string, args []d
 
 	var err error
 	var rows driver.Rows
-	if queryerCtx, ok := conn.Conn.(driver.QueryerContext); ok {
+	if okCtx {
 		rows, err = queryerCtx.QueryContext(ctx, query, args)
 	} else {
 		select {