Skip to main content

DatabaseConnection

cognee_lib::prelude

Enum DatabaseConnection 

Source 
pub enum DatabaseConnection {
    SqlxPostgresPoolConnection(SqlxPostgresPoolConnection),
    SqlxSqlitePoolConnection(SqlxSqlitePoolConnection),
    Disconnected,
}
Expand description

Handle a database connection depending on the backend enabled by the feature flags. This creates a database pool. This will be Clone unless the feature flag mock is enabled.

Variants§

§

SqlxPostgresPoolConnection(SqlxPostgresPoolConnection)

Create a PostgreSQL database connection and pool

§

SqlxSqlitePoolConnection(SqlxSqlitePoolConnection)

Create a SQLite database connection and pool

§

Disconnected

The connection to the database has been severed

Implementations§

Source§

impl DatabaseConnection

Source

pub fn set_metric_callback<F>(&mut self, _callback: F)
where F: Fn(&Info<'_>) + Send + Sync + 'static,

Sets a callback to metric this connection

Source

pub async fn ping(&self) -> Result<(), DbErr>

Checks if a connection to the database is still valid.

Source

pub async fn close(self) -> Result<(), DbErr>

Explicitly close the database connection. See Self::close_by_ref for usage with references.

Source

pub async fn close_by_ref(&self) -> Result<(), DbErr>

Explicitly close the database connection

Source§

impl DatabaseConnection

Source

pub fn get_postgres_connection_pool(&self) -> &Pool<Postgres>

Get sqlx::PgPool

§Panics

Panics if DbConn is not a Postgres connection.

Source

pub fn get_sqlite_connection_pool(&self) -> &Pool<Sqlite>

Get sqlx::SqlitePool

§Panics

Panics if DbConn is not a SQLite connection.

Trait Implementations§

Source§

impl Clone for DatabaseConnection

Source§

fn clone(&self) -> DatabaseConnection

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl ConnectionTrait for DatabaseConnection

Source§

fn get_database_backend(&self) -> DatabaseBackend

Fetch the database backend as specified in DbBackend. This depends on feature flags enabled.
Source§

fn execute<'life0, 'async_trait>( &'life0 self, stmt: Statement, ) -> Pin<Box<dyn Future<Output = Result<ExecResult, DbErr>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Execute a Statement
Source§

fn execute_unprepared<'life0, 'life1, 'async_trait>( &'life0 self, sql: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<ExecResult, DbErr>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Execute a unprepared Statement
Source§

fn query_one<'life0, 'async_trait>( &'life0 self, stmt: Statement, ) -> Pin<Box<dyn Future<Output = Result<Option<QueryResult>, DbErr>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Execute a Statement and return a query
Source§

fn query_all<'life0, 'async_trait>( &'life0 self, stmt: Statement, ) -> Pin<Box<dyn Future<Output = Result<Vec<QueryResult>, DbErr>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Execute a Statement and return a collection Vec<QueryResult> on success
Source§

fn support_returning(&self) -> bool

Check if the connection supports RETURNING syntax on insert and update
Source§

fn is_mock_connection(&self) -> bool

Check if the connection is a test connection for the Mock database
Source§

impl DatasetConfigDb for DatabaseConnection

Source§

fn get_by_dataset_id<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Option<DatasetConfiguration>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn upsert<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, patch: DatasetConfigurationPatch, ) -> Pin<Box<dyn Future<Output = Result<DatasetConfiguration, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

impl Debug for DatabaseConnection

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter. Read more
Source§

impl Default for DatabaseConnection

Source§

fn default() -> DatabaseConnection

Returns the “default value” for a type. Read more
Source§

impl DeleteDb for DatabaseConnection

Source§

fn get_data<'life0, 'async_trait>( &'life0 self, id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Option<Data>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn delete_data<'life0, 'async_trait>( &'life0 self, id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn list_datasets_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn get_dataset_by_name<'life0, 'life1, 'async_trait>( &'life0 self, name: &'life1 str, owner_id: Uuid, tenant_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<Option<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn get_dataset_data<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<Data>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn count_dataset_data<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<usize, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Count the number of data items linked to a dataset without loading them.
Source§

fn list_datasets_by_owner<'life0, 'async_trait>( &'life0 self, owner_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn list_datasets<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<Vec<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn delete_dataset<'life0, 'async_trait>( &'life0 self, id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn detach_data_from_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, data_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn delete_pipeline_runs_by_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<u64, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete all pipeline_runs rows for a given dataset. Read more
Source§

fn clear_pipeline_status_for_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<usize, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Clear pipeline_status JSON entries keyed by dataset_id from all Data records linked to that dataset. Read more
Source§

fn clear_cognify_pipeline_status_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Clear only the cognify_pipeline key from the pipeline_status JSON of a single Data record, removing only the entry keyed by dataset_id.
Source§

fn get_nodes_by_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<GraphNode>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Get all provenance node rows for a dataset.
Source§

fn get_edges_by_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<GraphEdge>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Get all provenance edge rows for a dataset.
Source§

fn get_unique_nodes_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<GraphNode>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Get nodes belonging to (data_id, dataset_id) whose slug is NOT shared with other data items in the same dataset. Safe for targeted deletion.
Source§

fn get_unique_edges_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<GraphEdge>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Get edges belonging to (data_id, dataset_id) whose slug is NOT shared with other data items in the same dataset.
Source§

fn delete_provenance_nodes_for_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete all provenance node rows for a dataset.
Source§

fn delete_provenance_edges_for_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete all provenance edge rows for a dataset.
Source§

fn delete_provenance_nodes_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete provenance node rows for a specific (data_id, dataset_id) pair.
Source§

fn delete_provenance_edges_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete provenance edge rows for a specific (data_id, dataset_id) pair.
Source§

fn get_provenance_node_count_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<usize, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Count provenance node rows for a specific (data_id, dataset_id) pair.
Source§

fn get_provenance_edge_count_for_data<'life0, 'async_trait>( &'life0 self, data_id: Uuid, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<usize, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Count provenance edge rows for a specific (data_id, dataset_id) pair.
Source§

fn delete_search_history_for_user<'life0, 'async_trait>( &'life0 self, user_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<u64, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete all search history (queries + cascaded results) for a user. Read more
Source§

fn delete_all_search_history<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<u64, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete all search history (queries + cascaded results). Read more
Source§

fn count_search_history_for_user<'life0, 'async_trait>( &'life0 self, user_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<u64, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Count search history query rows for a specific user.
Source§

fn count_all_search_history<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<u64, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Count all search history query rows.
Source§

impl From<Pool<Postgres>> for DatabaseConnection

Source§

fn from(pool: Pool<Postgres>) -> DatabaseConnection

Converts to this type from the input type.
Source§

impl From<Pool<Sqlite>> for DatabaseConnection

Source§

fn from(pool: Pool<Sqlite>) -> DatabaseConnection

Converts to this type from the input type.
Source§

impl IngestDb for DatabaseConnection

Source§

fn get_dataset_by_name<'life0, 'life1, 'async_trait>( &'life0 self, name: &'life1 str, owner_id: Uuid, tenant_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<Option<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn get_dataset<'life0, 'async_trait>( &'life0 self, id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Option<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Look up a dataset by its UUID.
Source§

fn create_dataset<'life0, 'async_trait>( &'life0 self, dataset: Dataset, ) -> Pin<Box<dyn Future<Output = Result<Dataset, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn list_datasets_by_owner<'life0, 'async_trait>( &'life0 self, owner_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<Dataset>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

List all datasets owned by the given user.
Source§

fn get_data<'life0, 'async_trait>( &'life0 self, id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Option<Data>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn create_data<'life0, 'async_trait>( &'life0 self, d: Data, ) -> Pin<Box<dyn Future<Output = Result<Data, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn attach_data_to_dataset<'life0, 'async_trait>( &'life0 self, dataset_id: Uuid, data_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn update_last_accessed<'life0, 'life1, 'async_trait>( &'life0 self, data_ids: &'life1 [Uuid], timestamp: DateTime<Utc>, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Update the last_accessed timestamp on the given Data records. Read more
Source§

fn get_latest_pipeline_status<'life0, 'life1, 'async_trait>( &'life0 self, pipeline_name: &'life1 str, dataset_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Option<PipelineRunStatus>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Get the latest pipeline run status for a (pipeline_name, dataset_id) pair. Read more
Source§

impl<'c> IntoSchemaManagerConnection<'c> for &'c DatabaseConnection

Source§

fn into_schema_manager_connection(self) -> SchemaManagerConnection<'c>

Source§

impl NotebookDb for DatabaseConnection

Source§

fn list_by_owner<'life0, 'async_trait>( &'life0 self, owner_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Vec<Notebook>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Return all notebooks owned by owner_id, ordered by created_at asc.
Source§

fn create<'life0, 'async_trait>( &'life0 self, owner_id: Uuid, name: String, cells: Value, deletable: bool, ) -> Pin<Box<dyn Future<Output = Result<Notebook, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Insert a new notebook row and return it. Read more
Source§

fn create_seeded<'life0, 'async_trait>( &'life0 self, id: Uuid, owner_id: Uuid, name: String, cells: Value, deletable: bool, ) -> Pin<Box<dyn Future<Output = Result<Notebook, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Insert a notebook with a caller-supplied id (used by the tutorial seeder to guarantee deterministic UUID5 ids across SDK restarts).
Source§

fn get_by_id_and_owner<'life0, 'async_trait>( &'life0 self, id: Uuid, owner_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Option<Notebook>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Fetch a notebook by id, scoped to the owner.
Source§

fn update<'life0, 'async_trait>( &'life0 self, id: Uuid, owner_id: Uuid, patch: NotebookUpdatePatch, ) -> Pin<Box<dyn Future<Output = Result<Option<Notebook>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Apply a partial update. Returns the updated row or None when not found (ownership check included).
Source§

fn delete<'life0, 'async_trait>( &'life0 self, id: Uuid, owner_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<bool, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Delete a notebook. Returns true if a row was actually removed.
Source§

impl SearchHistoryDb for DatabaseConnection

Source§

fn log_query<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, query_text: &'life1 str, query_type: &'life2 str, user_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<Uuid, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn log_result<'life0, 'life1, 'async_trait>( &'life0 self, query_id: Uuid, serialized_result: &'life1 str, user_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<Uuid, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Source§

fn get_history<'life0, 'async_trait>( &'life0 self, user_id: Option<Uuid>, limit: Option<usize>, ) -> Pin<Box<dyn Future<Output = Result<Vec<SearchHistoryEntry>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Source§

impl SessionLifecycleDb for DatabaseConnection

Source§

fn ensure_and_touch_session<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 str, user_id: Uuid, dataset_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Upsert a session row, bumping last_activity_at if the row is already running. Mirrors metrics.py:62-130.
Source§

fn accumulate_usage<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, session_id: &'life1 str, user_id: Uuid, model: Option<&'life2 str>, tokens_in: i64, tokens_out: i64, cost_usd: f64, errored: bool, ) -> Pin<Box<dyn Future<Output = Result<(), DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, DatabaseConnection: 'async_trait,

Atomically add usage counters to the session row + per-model row. Mirrors metrics.py:133-241.
Source§

fn get_session_row<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, session_id: &'life1 str, user_id: Uuid, permitted_dataset_ids: &'life2 [Uuid], prefer_other_owner: bool, ) -> Pin<Box<dyn Future<Output = Result<Option<SessionRowWithStatus>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, DatabaseConnection: 'async_trait,

Visibility-checked single-row read. Mirrors metrics.py:295-333.
Source§

fn list_session_rows<'life0, 'async_trait>( &'life0 self, filters: SessionListFilters, ) -> Pin<Box<dyn Future<Output = Result<SessionListPage, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Paginated list with effective_status filter support. Mirrors metrics.py:365-438.
Source§

fn aggregate_stats<'life0, 'life1, 'async_trait>( &'life0 self, user_id: Uuid, permitted_dataset_ids: &'life1 [Uuid], since: Option<DateTime<Utc>>, ) -> Pin<Box<dyn Future<Output = Result<SessionStats, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Dashboard counters for GET /sessions/stats. Mirrors get_sessions_router.py:112-196.
Source§

fn cost_by_model<'life0, 'life1, 'async_trait>( &'life0 self, user_id: Uuid, permitted_dataset_ids: &'life1 [Uuid], since: Option<DateTime<Utc>>, ) -> Pin<Box<dyn Future<Output = Result<Vec<CostByModelRow>, DatabaseError>> + Send + 'async_trait>>
where 'life0: 'async_trait, 'life1: 'async_trait, DatabaseConnection: 'async_trait,

Per-model attribution for GET /sessions/cost-by-model. Mirrors get_sessions_router.py:198-252.
Source§

impl StreamTrait for DatabaseConnection

Source§

type Stream<'a> = QueryStream

Create a stream for the QueryResult
Source§

fn stream<'a>( &'a self, stmt: Statement, ) -> Pin<Box<dyn Future<Output = Result<<DatabaseConnection as StreamTrait>::Stream<'a>, DbErr>> + Send + 'a>>

Execute a Statement and return a stream of results
Source§

impl TransactionTrait for DatabaseConnection

Source§

fn transaction<'life0, 'async_trait, F, T, E>( &'life0 self, _callback: F, ) -> Pin<Box<dyn Future<Output = Result<T, TransactionError<E>>> + Send + 'async_trait>>
where 'life0: 'async_trait, F: for<'c> FnOnce(&'c DatabaseTransaction) -> Pin<Box<dyn Future<Output = Result<T, E>> + Send + 'c>> + Send + 'async_trait, T: Send + 'async_trait, E: Display + Debug + Send + 'async_trait, DatabaseConnection: 'async_trait,

Execute the function inside a transaction. If the function returns an error, the transaction will be rolled back. If it does not return an error, the transaction will be committed.

Source§

fn transaction_with_config<'life0, 'async_trait, F, T, E>( &'life0 self, _callback: F, _isolation_level: Option<IsolationLevel>, _access_mode: Option<AccessMode>, ) -> Pin<Box<dyn Future<Output = Result<T, TransactionError<E>>> + Send + 'async_trait>>
where 'life0: 'async_trait, F: for<'c> FnOnce(&'c DatabaseTransaction) -> Pin<Box<dyn Future<Output = Result<T, E>> + Send + 'c>> + Send + 'async_trait, T: Send + 'async_trait, E: Display + Debug + Send + 'async_trait, DatabaseConnection: 'async_trait,

Execute the function inside a transaction. If the function returns an error, the transaction will be rolled back. If it does not return an error, the transaction will be committed.

Source§

fn begin<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<DatabaseTransaction, DbErr>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Execute SQL BEGIN transaction. Returns a Transaction that can be committed or rolled back
Source§

fn begin_with_config<'life0, 'async_trait>( &'life0 self, _isolation_level: Option<IsolationLevel>, _access_mode: Option<AccessMode>, ) -> Pin<Box<dyn Future<Output = Result<DatabaseTransaction, DbErr>> + Send + 'async_trait>>
where 'life0: 'async_trait, DatabaseConnection: 'async_trait,

Execute SQL BEGIN transaction with isolation level and/or access mode. Returns a Transaction that can be committed or rolled back

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> At for T

Source§

fn at<M>(self, metadata: M) -> Meta<T, M>

Wraps self inside a Meta<Self, M> using the given metadata. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> BorrowStripped for T

Source§

fn stripped(&self) -> &Stripped<T>

Source§

impl<T> BorrowUnordered for T

Source§

fn unordered(&self) -> &Unordered<T>

Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> Conv for T

Source§

fn conv<T>(self) -> T
where Self: Into<T>,

Converts self into T using Into<T>. Read more
Source§

impl<T> DatasetDb for T
where T: IngestDb + DeleteDb + Send + Sync,

Source§

impl<T> DynClone for T
where T: Clone,

Source§

fn __clone_box(&self, _: Private) -> *mut ()

Source§

impl<T> FmtForward for T

Source§

fn fmt_binary(self) -> FmtBinary<Self>
where Self: Binary,

Causes self to use its Binary implementation when Debug-formatted.
Source§

fn fmt_display(self) -> FmtDisplay<Self>
where Self: Display,

Causes self to use its Display implementation when Debug-formatted.
Source§

fn fmt_lower_exp(self) -> FmtLowerExp<Self>
where Self: LowerExp,

Causes self to use its LowerExp implementation when Debug-formatted.
Source§

fn fmt_lower_hex(self) -> FmtLowerHex<Self>
where Self: LowerHex,

Causes self to use its LowerHex implementation when Debug-formatted.
Source§

fn fmt_octal(self) -> FmtOctal<Self>
where Self: Octal,

Causes self to use its Octal implementation when Debug-formatted.
Source§

fn fmt_pointer(self) -> FmtPointer<Self>
where Self: Pointer,

Causes self to use its Pointer implementation when Debug-formatted.
Source§

fn fmt_upper_exp(self) -> FmtUpperExp<Self>
where Self: UpperExp,

Causes self to use its UpperExp implementation when Debug-formatted.
Source§

fn fmt_upper_hex(self) -> FmtUpperHex<Self>
where Self: UpperHex,

Causes self to use its UpperHex implementation when Debug-formatted.
Source§

fn fmt_list(self) -> FmtList<Self>
where &'a Self: for<'a> IntoIterator,

Formats each item in a sequence. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, C> FromWithContext<T, C> for T

Source§

fn from_with(value: T, _context: &C) -> T

Source§

impl<T> FutureExt for T

Source§

fn with_context(self, otel_cx: Context) -> WithContext<Self>

Attaches the provided Context to this type, returning a WithContext wrapper. Read more
Source§

fn with_current_context(self) -> WithContext<Self>

Attaches the current Context to this type, returning a WithContext wrapper. Read more
Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> IntoRequest<T> for T

Source§

fn into_request(self) -> Request<T>

Wrap the input message T in a tonic::Request
Source§

impl<T, U, C> IntoWithContext<U, C> for T
where U: FromWithContext<T, C>,

Source§

fn into_with(self, context: &C) -> U

Source§

impl<L> LayerExt<L> for L

Source§

fn named_layer<S>(&self, service: S) -> Layered<<L as Layer<S>>::Service, S>
where L: Layer<S>,

Applies the layer to a service and wraps it in Layered.
Source§

impl<T> Pipe for T
where T: ?Sized,

Source§

fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> R
where Self: Sized,

Pipes by value. This is generally the method you want to use. Read more
Source§

fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> R
where R: 'a,

Borrows self and passes that borrow into the pipe function. Read more
Source§

fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> R
where R: 'a,

Mutably borrows self and passes that borrow into the pipe function. Read more
Source§

fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
where Self: Borrow<B>, B: 'a + ?Sized, R: 'a,

Borrows self, then passes self.borrow() into the pipe function. Read more
Source§

fn pipe_borrow_mut<'a, B, R>( &'a mut self, func: impl FnOnce(&'a mut B) -> R, ) -> R
where Self: BorrowMut<B>, B: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.borrow_mut() into the pipe function. Read more
Source§

fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
where Self: AsRef<U>, U: 'a + ?Sized, R: 'a,

Borrows self, then passes self.as_ref() into the pipe function.
Source§

fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
where Self: AsMut<U>, U: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.as_mut() into the pipe function.
Source§

fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
where Self: Deref<Target = T>, T: 'a + ?Sized, R: 'a,

Borrows self, then passes self.deref() into the pipe function.
Source§

fn pipe_deref_mut<'a, T, R>( &'a mut self, func: impl FnOnce(&'a mut T) -> R, ) -> R
where Self: DerefMut<Target = T> + Deref, T: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.deref_mut() into the pipe function.
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T> PolicyExt for T
where T: ?Sized,

Source§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
Source§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T> Tap for T

Source§

fn tap(self, func: impl FnOnce(&Self)) -> Self

Immutable access to a value. Read more
Source§

fn tap_mut(self, func: impl FnOnce(&mut Self)) -> Self

Mutable access to a value. Read more
Source§

fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
where Self: Borrow<B>, B: ?Sized,

Immutable access to the Borrow<B> of a value. Read more
Source§

fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
where Self: BorrowMut<B>, B: ?Sized,

Mutable access to the BorrowMut<B> of a value. Read more
Source§

fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
where Self: AsRef<R>, R: ?Sized,

Immutable access to the AsRef<R> view of a value. Read more
Source§

fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
where Self: AsMut<R>, R: ?Sized,

Mutable access to the AsMut<R> view of a value. Read more
Source§

fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
where Self: Deref<Target = T>, T: ?Sized,

Immutable access to the Deref::Target of a value. Read more
Source§

fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
where Self: DerefMut<Target = T> + Deref, T: ?Sized,

Mutable access to the Deref::Target of a value. Read more
Source§

fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self

Calls .tap() only in debug builds, and is erased in release builds.
Source§

fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self

Calls .tap_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
where Self: Borrow<B>, B: ?Sized,

Calls .tap_borrow() only in debug builds, and is erased in release builds.
Source§

fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
where Self: BorrowMut<B>, B: ?Sized,

Calls .tap_borrow_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
where Self: AsRef<R>, R: ?Sized,

Calls .tap_ref() only in debug builds, and is erased in release builds.
Source§

fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
where Self: AsMut<R>, R: ?Sized,

Calls .tap_ref_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
where Self: Deref<Target = T>, T: ?Sized,

Calls .tap_deref() only in debug builds, and is erased in release builds.
Source§

fn tap_deref_mut_dbg<T>(self, func: impl FnOnce(&mut T)) -> Self
where Self: DerefMut<Target = T> + Deref, T: ?Sized,

Calls .tap_deref_mut() only in debug builds, and is erased in release builds.
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T> ToOwned for T

Source§

type Owned = T

Source§

impl<T> TryConv for T

Source§

fn try_conv<T>(self) -> Result<T, Self::Error>
where Self: TryInto<T>,

Attempts to convert self into T using TryInto<T>. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U, C> TryFromWithContext<U, C> for T
where U: IntoWithContext<T, C>,

Source§

type Error = Infallible

Source§

fn try_from_with( value: U, context: &C, ) -> Result<T, <T as TryFromWithContext<U, C>>::Error>

Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<T, U, C> TryIntoWithContext<U, C> for T
where U: TryFromWithContext<T, C>,

Source§

type Error = <U as TryFromWithContext<T, C>>::Error

Source§

fn try_into_with( self, context: &C, ) -> Result<U, <T as TryIntoWithContext<U, C>>::Error>

Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> Value for T
where T: Any + Send + Sync + 'static,

Source§

fn as_any(&self) -> &(dyn Any + 'static)

Source§

fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)

Source§

fn into_any(self: Box<T>) -> Box<dyn Any>

Source§

impl<T> WithContext for T
where T: ?Sized,

Source§

fn with<C>(&self, context: C) -> Contextual<&T, C>

Source§

fn into_with<C>(self, context: C) -> Contextual<T, C>
where T: Sized,

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more