Struct Database 

pub struct Database { /* private fields */ }

Database connection wrapper with thread-safe access

Implementations

impl Database

pub fn new<P: AsRef<Path>>(path: P) -> Result<Self>

Create a new database connection and run migrations

Arguments

• path - Path to the SQLite database file

Returns

A new Database instance

pub fn insert_artifact(&self, artifact: &Artifact) -> Result<()>

Insert an artifact into the database

Arguments

• artifact - The artifact to insert

Returns

Ok(()) if successful

pub fn insert_spans(&self, spans: &[Span]) -> Result<()>

Insert multiple spans in a transaction

Arguments

• spans - Vector of spans to insert

Returns

Ok(()) if successful

pub fn get_vector_index(&self) -> Result<Arc<VectorIndex>>

Get or build the cached vector index

The index is cached and only rebuilt when data changes (on ingest). Phase 2.1: Tries to load from disk first, then builds if needed.

Returns

A reference-counted vector index

pub fn get_vector_index_with_kind( &self, ) -> Result<(Arc<VectorIndex>, IndexLoadKind)>

Get or build the cached vector index and return how it was obtained

Returns the index and an indicator of whether it was loaded from cache or freshly built from spans.

pub fn get_all_spans(&self) -> Result<Vec<Span>>

Get all spans from the database

Returns

Vector of all spans

pub fn get_artifact(&self, artifact_id: &str) -> Result<Option<Artifact>>

Get artifact by ID

Arguments

• artifact_id - The artifact ID to look up

Returns

The artifact if found

pub fn get_artifact_by_path(&self, path: &str) -> Result<Option<Artifact>>

Get artifact by path

Returns the artifact row matching the unique path, if present.

pub fn determine_ingest_action( &self, path: &str, content_hash: &str, ) -> Result<IngestAction>

Determine what action to take when ingesting a document

Compares content hash to detect if document needs update or can be skipped.

Arguments

• path - The document path
• content_hash - SHA256 hash of the new content

Returns

• IngestAction::Skip if document exists with same content hash
• IngestAction::Update if document exists but content changed
• IngestAction::Create if document doesn’t exist

pub fn delete_artifact(&self, artifact_id: &str) -> Result<usize>

Delete an artifact and its spans

Arguments

• artifact_id - The artifact ID to delete

Returns

Number of spans deleted

pub fn search_spans(&self, query: &str, limit: usize) -> Result<Vec<Span>>

Search spans by text content (simple keyword matching)

Arguments

• query - The search query
• limit - Maximum number of results

Returns

Vector of matching spans

pub fn get_stats(&self) -> Result<(usize, usize, usize)>

Get database statistics

Returns

(artifacts_count, spans_count, total_tokens)

pub fn clear(&self) -> Result<()>

Clear all data from the database

pub fn create_session( &self, user_id: Option<&str>, title: Option<&str>, ) -> Result<Session>

Create a new session

Arguments

• user_id - Optional user identifier
• title - Optional session title

Returns

The newly created session

pub fn get_session(&self, session_id: &str) -> Result<Option<Session>>

Get a session by ID

Arguments

• session_id - The session ID to look up

Returns

The session if found

pub fn list_sessions( &self, user_id: Option<&str>, limit: Option<usize>, ) -> Result<Vec<Session>>

List sessions for a user (or all sessions if user_id is None)

Arguments

• user_id - Optional user ID to filter by
• limit - Maximum number of sessions to return

Returns

Vector of sessions, sorted by updated_at descending

pub fn update_session( &self, session_id: &str, title: Option<&str>, metadata: Option<&Value>, ) -> Result<()>

Update session metadata

Arguments

• session_id - The session ID to update
• title - Optional new title
• metadata - Optional new metadata

Returns

Ok(()) if successful

pub fn delete_session(&self, session_id: &str) -> Result<()>

Delete a session (cascades to messages and working sets)

Arguments

• session_id - The session ID to delete

Returns

Ok(()) if successful

pub fn add_message( &self, session_id: &str, role: MessageRole, content: &str, metadata: Option<&Value>, ) -> Result<Message>

Add a message to a session

Arguments

• session_id - The session to add the message to
• role - Message role (user, assistant, system, tool)
• content - Message content
• metadata - Optional metadata

Returns

The newly created message

pub fn get_messages( &self, session_id: &str, limit: Option<usize>, ) -> Result<Vec<Message>>

Get messages for a session

Arguments

• session_id - The session ID
• limit - Optional limit on number of messages (most recent first)

Returns

Vector of messages in chronological order

pub fn associate_working_set( &self, session_id: &str, message_id: Option<&str>, working_set: &WorkingSet, query: &str, config: &CompilerConfig, ) -> Result<SessionWorkingSet>

Associate a working set with a session

Arguments

• session_id - The session ID
• message_id - Optional message ID that triggered this compilation
• working_set - The working set to associate
• query - Query that generated this working set
• config - Configuration used for compilation

Returns

The newly created SessionWorkingSet

pub fn get_session_full( &self, session_id: &str, ) -> Result<Option<SessionWithMessages>>

Get session with all messages and working sets

Arguments

• session_id - The session ID

Returns

SessionWithMessages if found

Trait Implementations

impl Clone for Database

fn clone(&self) -> Database

Returns a duplicate of the value.

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

Performs copy-assignment from source.