Method execStatementSync

backend/plugin/db/databricks/databricks.go:163–194

Execute SQL statement synchronously and return row data or error. rowLimit caps the result set when > 0; 0 means no limit (used by sync and dump paths). DDL/DML callers should pass 0.

(ctx context.Context, statement string, rowLimit int64)

// rowLimit caps the result set when > 0; 0 means no limit (used by sync
// and dump paths). DDL/DML callers should pass 0.
func (d *Driver) execStatementSync(ctx context.Context, statement string, rowLimit int64) ([][]string, []dbsql.ColumnInfo, error) {
	// Bind the connected Bytebase database (= Unity Catalog) to the
	// session, so unqualified `schema.table` references resolve like in
	// other RDBMSes. Equivalent to issuing `USE CATALOG <name>` first.
	req := dbsql.ExecuteStatementRequest{
		Statement:   statement,
		WarehouseId: d.WarehouseID,
		Catalog:     d.curCatalog,
	}
	// Honor the user-selected row limit (default 1000 from the SQL Editor).
	// Without it, INLINE results above ~25 MiB are aborted server-side
	// with no result available — see DispositionExternalLinks for the
	// large-result path.
	if rowLimit > 0 {
		req.RowLimit = rowLimit
	}
	resp, err := d.Client.StatementExecution.ExecuteAndWait(ctx, req)
	if err != nil {
		return nil, nil, err
	}
	if resp.Result == nil {
		return nil, nil, errors.New("no response")
	}

	if len(resp.Result.DataArray) != 0 {
		if resp.Manifest == nil || resp.Manifest.Schema == nil || len(resp.Manifest.Schema.Columns) == 0 {
			return nil, nil, errors.New("missing column info")
		}
		return resp.Result.DataArray, resp.Manifest.Schema.Columns, nil
	}
	return nil, nil, nil
}

// return a column type name array and a column name array.
func toStrColInfo(colInfo []dbsql.ColumnInfo) ([]string, []string) {