Struct DataBackendSession 

pub struct DataBackendSession {
    pub chunk_size: usize,
    pub runtime: Arc<Runtime>,
    /* private fields */
}

Provides a DataFusion session and registers DataFusion queries.

The session is used to register data sources and make queries on them. A query returns a Chunk of Arrow records. It is decoded and converted into a Vec of data by types that implement DecodeDataFromRecordBatch.

Fields

chunk_size: usize

runtime: Arc<Runtime>

Implementations

impl DataBackendSession

pub fn new(chunk_size: usize) -> Self

Creates a new DataBackendSession instance.

pub fn register_object_store( &mut self, url: &Url, object_store: Arc<dyn ObjectStore>, )

Register an object store with the session context

pub fn register_object_store_from_uri( &mut self, uri: &str, storage_options: Option<HashMap<String, String>>, ) -> Result<()>

Register an object store with the session context from a URI with optional storage options

pub fn write_data<T: EncodeToRecordBatch>( data: &[T], metadata: &HashMap<String, String>, stream: &mut dyn WriteStream, ) -> Result<(), DataStreamingError>

pub fn add_file<T>( &mut self, table_name: &str, file_path: &str, sql_query: Option<&str>, ) -> Result<()>

where T: DecodeDataFromRecordBatch + Into<Data>,

Query a file for its records. the caller must specify T to indicate the kind of data expected from this query.

table_name: Logical table_name assigned to this file. Queries to this file should address the file by its table name. file_path: Path to file sql_query: A custom sql query to retrieve records from file. If no query is provided a default query “SELECT * FROM <table_name>” is run.

Safety

The file data must be ordered by the ts_init in ascending order for this to work correctly.

pub fn get_query_result(&mut self) -> QueryResult

pub fn clear_registered_tables(&mut self)

Clears all registered tables and batch streams.

This is useful when the underlying files have changed and we need to re-register tables with updated data.

Trait Implementations

impl<'py> IntoPyObject<'py> for DataBackendSession

type Target = DataBackendSession

The Python output type

type Output = Bound<'py, <DataBackendSession as IntoPyObject<'py>>::Target>

The smart pointer type to use.

type Error = PyErr

The type returned in the event of a conversion error.

fn into_pyobject( self, py: Python<'py>, ) -> Result<<Self as IntoPyObject<'_>>::Output, <Self as IntoPyObject<'_>>::Error>

Performs the conversion.

impl PyClass for DataBackendSession

type Frozen = False

Whether the pyclass is frozen.

impl PyClassImpl for DataBackendSession

const IS_BASETYPE: bool = false

#[pyclass(subclass)]

const IS_SUBCLASS: bool = false

#[pyclass(extends=…)]

const IS_MAPPING: bool = false

#[pyclass(mapping)]

const IS_SEQUENCE: bool = false

#[pyclass(sequence)]

const IS_IMMUTABLE_TYPE: bool = false

#[pyclass(immutable_type)]

const RAW_DOC: &'static CStr = /// Provides a DataFusion session and registers DataFusion queries. /// /// The session is used to register data sources and make queries on them. A /// query returns a Chunk of Arrow records. It is decoded and converted into /// a Vec of data by types that implement [`DecodeDataFromRecordBatch`].

Docstring for the class provided on the struct or enum.

const DOC: &'static CStr

Fully rendered class doc, including the text_signature if a constructor is defined.

type BaseType = PyAny

Base class

type ThreadChecker = SendablePyClass<DataBackendSession>

This handles following two situations:

type PyClassMutability = <<PyAny as PyClassBaseType>::PyClassMutability as PyClassMutability>::MutableChild

Immutable or mutable

type Dict = PyClassDummySlot

Specify this class has #[pyclass(dict)] or not.

type WeakRef = PyClassDummySlot

Specify this class has #[pyclass(weakref)] or not.

type BaseNativeType = PyAny

The closest native ancestor. This is PyAny by default, and when you declare #[pyclass(extends=PyDict)], it’s PyDict.

fn items_iter() -> PyClassItemsIter

fn lazy_type_object() -> &'static LazyTypeObject<Self>

fn dict_offset() -> Option<isize>

fn weaklist_offset() -> Option<isize>

impl PyClassNewTextSignature for DataBackendSession

const TEXT_SIGNATURE: &'static str = "(chunk_size=10000)"

impl PyMethods<DataBackendSession> for PyClassImplCollector<DataBackendSession>

fn py_methods(self) -> &'static PyClassItems

impl PyTypeInfo for DataBackendSession

const NAME: &'static str = "DataBackendSession"

Class name.

const MODULE: Option<&'static str>

Module name, if any.

fn type_object_raw(py: Python<'_>) -> *mut PyTypeObject

Returns the PyTypeObject instance for this type.

fn type_object(py: Python<'_>) -> Bound<'_, PyType>

Returns the safe abstraction over the type object.

fn is_type_of(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type or a subclass of this type.

fn is_exact_type_of(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type.

impl DerefToPyAny for DataBackendSession

impl ExtractPyClassWithClone for DataBackendSession

impl Send for DataBackendSession