Function extractArrowValue

server/flightclient/flight_executor.go:726–810

extractArrowValue extracts a Go value from an Arrow array at the given row index.

(col arrow.Array, row int)

	})
	if err != nil {
		return err
	}

	ctx, cancel := context.WithTimeout(context.Background(), queryCloseWaitTimeout)
	defer cancel()
	if e.ctx != nil {
		go func() {
			select {
			case <-e.ctx.Done():
				cancel()
			case <-ctx.Done():
			}
		}()
	}

	stream, err := e.client.Client.DoAction(
		e.withSession(ctx),
		&flight.Action{Type: releaseQueryHandleAction, Body: payload},
	)
	if err != nil {
		if isTerminalSessionIdleWaitError(err) {
			return nil
		}
		return err
	}
	for {
		_, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			return nil
		}
		if err != nil {
			if isTerminalSessionIdleWaitError(err) {
				return nil
			}
			return err
		}
	}
}

func (e *FlightExecutor) LastProfilingOutput() string {
	v := e.lastProfiling.Load()
	if v == nil {
		return ""
	}
	return v.(string)
}

const profilingMetadataKey = "x-duckgres-profiling"

func (e *FlightExecutor) storeProfilingFromTrailer(trailer metadata.MD) {
	if vals := trailer.Get(profilingMetadataKey); len(vals) > 0 {
		e.lastProfiling.Store(vals[0])
	} else {
		e.lastProfiling.Store("")
	}
}

// FlightRowSet wraps an Arrow Flight RecordBatch reader to implement RowSet.