Return SnowflakeResult instead of driver.ResultNoRows

connection.go

@@ -335,7 +335,11 @@ func (sc *snowflakeConn) ExecContext(
 		return sc.handleMultiExec(ctx, data.Data)
 	}
 	logger.Debug("DDL")
-	return driver.ResultNoRows, nil
+	return &snowflakeResult{
+		affectedRows: 0,
+		insertID:     -1,
+		queryID:      data.Data.QueryID,
+	}, nil
 }
 
 func (sc *snowflakeConn) QueryContext(

connection_util.go

@@ -194,10 +194,13 @@ func getResultType(ctx context.Context) resultType {
 
 // isDml returns true if the statement type code is in the range of DML.
 func isDml(v int64) bool {
-	return statementTypeIDDml <= v && v <= statementTypeIDMultiTableInsert
+	return (statementTypeIDDml <= v && v <= statementTypeIDMultiTableInsert) || v == statementTypeIDSelect
 }
 
 func updateRows(data execResponseData) (int64, error) {
+	if data.RowSet == nil {
+		return 0, nil
+	}
 	var count int64
 	for i, n := 0, len(data.RowType); i < n; i++ {
 		v, err := strconv.ParseInt(*data.RowSet[0][i], 10, 64)

statement_test.go

@@ -40,6 +40,68 @@ func openConn(t *testing.T) *sql.Conn {
 	return conn
 }
 
+func TestDMLExec(t *testing.T) {
+	query := "SELECT 1"
+	runDBTest(t, func(dbt *DBTest) {
+		testcases := []struct {
+			name string
+			f    func(dbt *DBTest) (any, error)
+		}{
+			{
+				name: "Exec",
+				f: func(dbt *DBTest) (any, error) {
+					stmt, _ := dbt.prepare(query)
+					return stmt.Exec()
+				},
+			},
+			{
+				name: "ExecContext",
+				f: func(dbt *DBTest) (any, error) {
+					stmt, _ := dbt.prepare(query)
+					return stmt.ExecContext(context.Background())
+				},
+			},
+		}
+		for _, tc := range testcases {
+			_, err := tc.f(dbt)
+			if err != nil {
+				t.Error(err)
+			}
+		}
+	})
+}
+
+func TestDDLExec(t *testing.T) {
+	query := "CREATE OR REPLACE TABLE TestDDLExec (num NUMBER)"
+	runDBTest(t, func(dbt *DBTest) {
+		testcases := []struct {
+			name string
+			f    func(dbt *DBTest) (any, error)
+		}{
+			{
+				name: "Exec",
+				f: func(dbt *DBTest) (any, error) {
+					stmt, _ := dbt.prepare(query)
+					return stmt.Exec()
+				},
+			},
+			{
+				name: "ExecContext",
+				f: func(dbt *DBTest) (any, error) {
+					stmt, _ := dbt.prepare(query)
+					return stmt.ExecContext(context.Background())
+				},
+			},
+		}
+		for _, tc := range testcases {
+			_, err := tc.f(dbt)
+			if err != nil {
+				t.Error(err)
+			}
+		}
+	})
+}
+
 func TestFailedQueryIdInSnowflakeError(t *testing.T) {
 	failingQuery := "SELECTT 1"
 	failingExec := "INSERT 1 INTO NON_EXISTENT_TABLE"