Function arrowTypeToDuckDB

server/flightclient/flight_executor.go:658–723

arrowTypeToDuckDB maps an Arrow DataType back to a DuckDB type name string.

(dt arrow.DataType)

		}
	}
}

// SetS3CacheMode selects the cache transport mode on the session worker.
func (e *FlightExecutor) SetS3CacheMode(ctx context.Context, mode string) (err error) {
	if e.dead.Load() {
		return ErrWorkerDead
	}
	if e.client == nil || e.client.Client == nil {
		return ErrWorkerDead
	}
	defer recoverClientPanic(&err)

	payload, err := json.Marshal(wire.WorkerSetS3CachePayload{
		WorkerControlMetadata: wire.WorkerControlMetadata{
			WorkerID:     e.workerID,
			OwnerEpoch:   e.ownerEpoch,
			CPInstanceID: e.cpInstanceID,
		},
		Mode: mode,
	})
	if err != nil {
		return err
	}

	merged, cancel := e.mergedContext(ctx)
	defer cancel()

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

func (e *FlightExecutor) releaseQueryHandle(ticket *flight.Ticket) (err error) {
	if e.dead.Load() {
		return nil
	}
	if e.client == nil || e.client.Client == nil || ticket == nil || len(ticket.Ticket) == 0 {
		return nil
	}
	defer func() {
		recoverClientPanic(&err)
		if err != nil && isTerminalSessionIdleWaitError(err) {
			err = nil
		}
	}()