Function executeWithTimeout

backend/api/v1/sql_service.go:927–957

(
	ctx context.Context,
	driver db.Driver,
	conn *sql.Conn,
	statement string,
	queryContext db.QueryContext,
)

}

func executeWithTimeout(
	ctx context.Context,
	driver db.Driver,
	conn *sql.Conn,
	statement string,
	queryContext db.QueryContext,
) ([]*v1pb.QueryResult, time.Duration, error) {
	queryCtx := ctx
	var timeout time.Duration
	// For access control feature, we will use the timeout from request and query data policy.
	// Otherwise, no timeout will be applied.
	if queryContext.Timeout != nil {
		timeout = queryContext.Timeout.AsDuration()
		slog.Debug("create query context with timeout", slog.Duration("timeout", timeout))
		newCtx, cancelCtx := context.WithTimeout(ctx, timeout)
		defer cancelCtx()
		queryCtx = newCtx
	}

	start := time.Now()
	result, err := driver.QueryConn(queryCtx, conn, statement, queryContext)
	select {
	case <-queryCtx.Done():
		// canceled or timed out
		return nil, time.Since(start), errors.Errorf("timeout reached: %v", timeout)
	default:
		// So the select will not block
	}
	sanitizeResults(result)
	return result, time.Since(start), err
}

// Export exports the SQL query result.
func (s *SQLService) Export(ctx context.Context, req *connect.Request[v1pb.ExportRequest]) (*connect.Response[v1pb.ExportResponse], error) {