Skip to main content

mj_controller/
database.rs

1pub use mj_core::storage::*;
2// Normalized controller state and composer history stored in SQLite.
3
4use std::collections::{BTreeMap, BTreeSet, HashSet};
5use std::fs;
6use std::path::{Path, PathBuf};
7use std::sync::atomic::{AtomicU64, Ordering};
8use std::sync::mpsc::{Receiver, SyncSender, sync_channel};
9use std::sync::{Arc, Mutex, OnceLock, PoisonError};
10use std::thread::{self, JoinHandle};
11use std::time::Duration;
12
13use anyhow::{Context, Result, bail, ensure};
14use chrono::Utc;
15use rusqlite::types::Type;
16use rusqlite::{Connection, OptionalExtension, Transaction, params};
17
18use mj_core::config::data_dir;
19use mj_core::state::{
20    CheckpointMetadata, HostContainerSize, ManagedWorktree, MaterializedExecutionState,
21    MaterializedQueuedPrompt, MaterializedSession, MaterializedSessionSummary, MaterializedTurn,
22    MaterializedTurnOutcome, ProjectionWindow, SessionRecord, SessionResourceAllocation,
23    SessionState, State, TargetLocator, TranscriptBody, TranscriptItem,
24    validate_relay_event_digest, validate_relay_event_frontier,
25};
26use mj_core::subagent::SubagentRecord;
27
28use crate::targets::{AdditionalMount, MountAccess};
29use mj_core::workspace::{
30    ConversationLayout, DEFAULT_WORKSPACE_ID, DetachedDraft, PaneSize, PaneSizes, WorkspaceRecord,
31    new_workspace_id, normalize_workspace_name,
32};
33
34const SCHEMA_VERSION: i64 = 55;
35
36mod session_move;
37pub use session_move::*;
38
39mod legacy_schema;
40mod schema;
41mod usage;
42pub use usage::*;
43mod events;
44pub use events::*;
45
46#[cfg(test)]
47use schema::forget_verified_schema;
48pub use schema::{check_read_compatibility, database_path};
49use schema::{open, open_reader};
50
51mod writer;
52pub use writer::*;
53mod workspaces;
54pub use workspaces::*;
55mod client_state;
56pub use client_state::*;
57mod state_io;
58pub use state_io::*;
59mod sessions;
60pub use sessions::*;
61mod native_agents;
62pub use native_agents::*;
63mod materialized;
64pub use materialized::*;
65mod mounts;
66pub use mounts::*;
67mod reviews;
68pub use reviews::*;
69mod prompts;
70pub use prompts::*;
71mod values;
72use values::*;
73mod profile_cache;
74pub use profile_cache::*;
75
76#[cfg(test)]
77mod tests;
78
79mod quota_cache;
80pub(crate) use quota_cache::*;
81
82/// Whether a failure is SQLite reporting another writer, which a later trigger
83/// simply retries.
84pub(crate) fn is_busy_error(error: &anyhow::Error) -> bool {
85    error.chain().any(|cause| {
86        matches!(
87            cause.downcast_ref::<rusqlite::Error>(),
88            Some(rusqlite::Error::SqliteFailure(failure, _))
89                if matches!(
90                    failure.code,
91                    rusqlite::ErrorCode::DatabaseBusy | rusqlite::ErrorCode::DatabaseLocked
92                )
93        )
94    })
95}