mirror of
https://github.com/tursodatabase/limbo.git
synced 2025-08-04 10:08:20 +00:00
Unify sync I/O
This commit is contained in:
parent
890d71b17f
commit
49c19b55ab
5 changed files with 60 additions and 103 deletions
53
cli/main.rs
53
cli/main.rs
|
@ -1,12 +1,7 @@
|
|||
use anyhow::Result;
|
||||
use clap::{Parser, ValueEnum};
|
||||
use cli_table::{Cell, Table};
|
||||
use lig_core::{Database, DatabaseRef, Value};
|
||||
use lig_core::{Database, Value, SyncIO};
|
||||
use rustyline::{error::ReadlineError, DefaultEditor};
|
||||
use std::cell::RefCell;
|
||||
use std::collections::HashMap;
|
||||
use std::fs::File;
|
||||
use std::io::{Read, Seek};
|
||||
use std::path::PathBuf;
|
||||
use std::sync::Arc;
|
||||
|
||||
|
@ -35,7 +30,7 @@ struct Opts {
|
|||
|
||||
fn main() -> anyhow::Result<()> {
|
||||
let opts = Opts::parse();
|
||||
let io = IO::new();
|
||||
let io = SyncIO::new();
|
||||
let db = Database::open(Arc::new(io), opts.database.to_str().unwrap())?;
|
||||
let conn = db.connect();
|
||||
let mut rl = DefaultEditor::new()?;
|
||||
|
@ -108,47 +103,3 @@ fn main() -> anyhow::Result<()> {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
struct IO {
|
||||
inner: RefCell<IOInner>,
|
||||
}
|
||||
|
||||
struct IOInner {
|
||||
db_refs: usize,
|
||||
db_files: HashMap<DatabaseRef, File>,
|
||||
}
|
||||
|
||||
impl lig_core::IO for IO {
|
||||
fn open(&self, path: &str) -> Result<DatabaseRef> {
|
||||
let file = std::fs::File::open(path)?;
|
||||
let mut inner = self.inner.borrow_mut();
|
||||
let db_ref = inner.db_refs;
|
||||
inner.db_refs += 1;
|
||||
inner.db_files.insert(db_ref, file);
|
||||
Ok(db_ref)
|
||||
}
|
||||
|
||||
fn get(&self, database_ref: DatabaseRef, page_idx: usize, buf: &mut [u8]) -> Result<()> {
|
||||
let page_size = buf.len();
|
||||
assert!(page_idx > 0);
|
||||
assert!(page_size >= 512);
|
||||
assert!(page_size <= 65536);
|
||||
assert!((page_size & (page_size - 1)) == 0);
|
||||
let mut inner = self.inner.borrow_mut();
|
||||
let file = inner.db_files.get_mut(&database_ref).unwrap();
|
||||
let pos = (page_idx - 1) * page_size;
|
||||
file.seek(std::io::SeekFrom::Start(pos as u64))?;
|
||||
file.read_exact(buf)?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl IO {
|
||||
fn new() -> Self {
|
||||
Self {
|
||||
inner: RefCell::new(IOInner {
|
||||
db_refs: 0,
|
||||
db_files: HashMap::new(),
|
||||
}),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue