Struct odbc_api::buffers::TextRowSet

pub struct TextRowSet { /* fields omitted */ }

This row set binds a string buffer to each column, which is large enough to hold the maximum length string representation for each element in the row set at once.

Example

//! A program executing a query and printing the result as csv to standard out. Requires
//! `anyhow` and `csv` crate.

use anyhow::Error;
use odbc_api::{buffers::TextRowSet, Cursor, Environment};
use std::{
    ffi::CStr,
    io::{stdout, Write},
    path::PathBuf,
};

/// Maximum number of rows fetched with one row set. Fetching batches of rows is usually much
/// faster than fetching individual rows.
const BATCH_SIZE: u32 = 100000;

fn main() -> Result<(), Error> {
    // Write csv to standard out
    let out = stdout();
    let mut writer = csv::Writer::from_writer(out);

    // We know this is going to be the only ODBC environment in the entire process, so this is
    // safe.
    let environment = unsafe { Environment::new() }?;

    // Connect using a DSN. Alternatively we could have used a connection string
    let mut connection = environment.connect(
        "DataSourceName",
        "Username",
        "Password",
    )?;

    // Execute a one of query without any parameters.
    match connection.execute("SELECT * FROM TableName", ())? {
        Some(cursor) => {
            // Write the column names to stdout
            let mut headline : Vec<String> = cursor.column_names()?.collect::<Result<_,_>>()?;
            writer.write_record(headline)?;

            // Use schema in cursor to initialize a text buffer large enough to hold the largest
            // possible strings for each column.
            let mut buffers = TextRowSet::for_cursor(BATCH_SIZE, &cursor)?;
            // Bind the buffer to the cursor. It is now being filled with every call to fetch.
            let mut row_set_cursor = cursor.bind_buffer(&mut buffers)?;

            // Iterate over batches
            while let Some(batch) = row_set_cursor.fetch()? {
                // Within a batch, iterate over every row
                for row_index in 0..batch.num_rows() {
                    // Within a row iterate over every column
                    let record = (0..batch.num_cols()).map(|col_index| {
                        batch
                            .at(col_index, row_index)
                            .map(CStr::to_bytes)
                            .unwrap_or(&[])
                    });
                    // Writes row as csv
                    writer.write_record(record)?;
                }
            }
        }
        None => {
            eprintln!(
                "Query came back empty. No output has been created."
            );
        }
    }

    Ok(())
}

Implementations

impl TextRowSet

pub fn for_cursor(
    batch_size: u32,
    cursor: &impl Cursor
) -> Result<TextRowSet, Error>

Use cursor to query the display size for each column of the row set and allocates the buffers accordingly.

pub fn new(
    batch_size: u32,
    max_str_lengths: impl Iterator<Item = usize>
) -> Self

Creates a text buffer large enough to hold batch_size rows with one column for each item max_str_lengths of respective size.

pub fn at(&self, col_index: usize, row_index: usize) -> Option<&CStr>

Access the element at the specified position in the row set.

pub fn at_as_str(
    &self,
    col_index: usize,
    row_index: usize
) -> Result<Option<&str>, Utf8Error>

Access the element at the specified position in the row set.

pub fn num_cols(&self) -> usize

Return the number of columns in the row set.

pub fn num_rows(&self) -> usize

Return the number of rows in the row set.

pub fn append<'a>(&mut self, row: impl Iterator<Item = Option<&'a [u8]>>)

Takes one element from the iterater for each internal column buffer and appends it to the end of the buffer. Should the buffer be not large enough to hold the element, it will be reallocated with 1.2 times its size.

This method panics if it is tried to insert elements beyond batch size. It will also panic if row does not contain at least one item for each internal column buffer.

pub fn clear(&mut self)

Sets the number of rows in the buffer to zero.

Trait Implementations

impl<'_> ParameterCollection for &'_ TextRowSet

pub fn parameter_set_size(&self) -> u32

Number of parameters in the collection. This can be different from the maximum batch size a buffer may be able to hold. Returning 0 will cause the the query not to be executed.

pub unsafe fn bind_parameters_to(
    &self,
    stmt: &mut Statement<'_>
) -> Result<(), Error>

Safety

impl RowSetBuffer for TextRowSet

pub fn row_array_size(&self) -> u32

The batch size for bulk cursors, if retrieving many rows at once.

pub fn mut_num_fetch_rows(&mut self) -> &mut usize

Mutable reference to the number of fetched rows.

pub fn bind_type(&self) -> u32

Declares the bind type of the Row set buffer. 0 Means a columnar binding is used. Any non zero number is interpreted as the size of a single row in a row wise binding style.

pub unsafe fn bind_to_cursor(
    &mut self,
    cursor: &mut impl Cursor
) -> Result<(), Error>

Binds the buffer either column or row wise to the cursor.