in client/sql/dataframe.go [59:106]
func (df *dataFrameImpl) Show(numRows int, truncate bool) error {
truncateValue := 0
if truncate {
truncateValue = 20
}
vertical := false
plan := &proto.Plan{
OpType: &proto.Plan_Root{
Root: &proto.Relation{
Common: &proto.RelationCommon{
PlanId: newPlanId(),
},
RelType: &proto.Relation_ShowString{
ShowString: &proto.ShowString{
Input: df.relation,
NumRows: int32(numRows),
Truncate: int32(truncateValue),
Vertical: vertical,
},
},
},
},
}
responseClient, err := df.sparkSession.executePlan(plan)
if err != nil {
return fmt.Errorf("failed to show dataframe: %w", err)
}
for {
response, err := responseClient.Recv()
if err != nil {
return fmt.Errorf("failed to receive show response: %w", err)
}
arrowBatch := response.GetArrowBatch()
if arrowBatch == nil {
continue
}
err = showArrowBatch(arrowBatch)
if err != nil {
return err
}
return nil
}
return fmt.Errorf("did not get arrow batch in response")
}