Method QueryConn

backend/plugin/db/databricks/databricks.go:81–122

(ctx context.Context, _ *sql.Conn, statement string, queryContext db.QueryContext)

}

func (d *Driver) QueryConn(ctx context.Context, _ *sql.Conn, statement string, queryContext db.QueryContext) ([]*v1pb.QueryResult, error) {
	var results []*v1pb.QueryResult
	stmts, err := base.SplitMultiSQL(storepb.Engine_DATABRICKS, statement)
	if err != nil {
		return nil, err
	}

	for _, stmt := range stmts {
		result := &v1pb.QueryResult{}
		startTime := time.Now()
		dataArr, colInfo, err := d.execStatementSync(ctx, stmt.Text, int64(queryContext.Limit))
		if err != nil {
			return nil, err
		}
		if dataArr == nil || colInfo == nil {
			break
		}

		colNames, colTypeNames := toStrColInfo(colInfo)
		result.ColumnNames = colNames
		result.ColumnTypeNames = colTypeNames

		// process rows.
		for _, rowData := range dataArr {
			queryRow := &v1pb.QueryRow{}
			// process a single row.
			for idx, rowVal := range rowData {
				v1pbRowVal, err := toV1pbRowVal(colInfo[idx].TypeName, rowVal)
				if err != nil {
					return nil, err
				}
				queryRow.Values = append(queryRow.Values, v1pbRowVal)
			}
			result.Rows = append(result.Rows, queryRow)
		}
		result.Latency = durationpb.New(time.Since(startTime))
		result.RowsCount = int64(len(result.Rows))
		results = append(results, result)
	}

	return results, nil
}

func (d *Driver) Execute(ctx context.Context, statement string, _ db.ExecuteOptions) (int64, error) {
	// Parse transaction mode from the script