135 lines
3.9 KiB
Rust
135 lines
3.9 KiB
Rust
use crate::fs::*;
|
|
use rusqlite::Connection;
|
|
use std::ops::{Deref, DerefMut};
|
|
use log::error;
|
|
|
|
pub fn create_db() {
|
|
let db_file = db_path();
|
|
if !db_file.exists() {
|
|
std::fs::File::create(&db_file).expect("Database file creation failed");
|
|
let connection = Connection::open(&db_file).expect("Connection open failed");
|
|
connection
|
|
.execute("PRAGMA foreign_keys = ON;", [])
|
|
.expect("Failed to set PRAGMA");
|
|
|
|
create_tables(&connection);
|
|
}
|
|
}
|
|
|
|
fn create_tables(conn: &Connection) {
|
|
make_card_set(conn).expect("Failed to init table");
|
|
make_settings(conn).expect("Failed to init table");
|
|
make_word_group(conn).expect("Failed to init table");
|
|
make_words(conn).expect("Failed to init table");
|
|
make_card_stats(conn).expect("Failed to init table");
|
|
make_history(conn).expect("Failed to init table");
|
|
make_custom_fields(conn).expect("Failed to init table");
|
|
}
|
|
|
|
fn make_settings(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/settings.sql");
|
|
conn.execute(query, ())?;
|
|
Ok(())
|
|
}
|
|
|
|
fn make_card_set(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/decks.sql");
|
|
conn.execute(query, ())?;
|
|
Ok(())
|
|
}
|
|
|
|
fn make_word_group(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/word_groups.sql");
|
|
conn.execute(query, ())?;
|
|
conn.execute("insert into word_groups (name) values (\"Слова\");", ())?;
|
|
Ok(())
|
|
}
|
|
fn make_words(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/words.sql");
|
|
conn.execute(query, ())?;
|
|
Ok(())
|
|
}
|
|
fn make_card_stats(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/states.sql");
|
|
conn.execute(query, ())?;
|
|
Ok(())
|
|
}
|
|
|
|
fn make_history(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/history.sql");
|
|
conn.execute(query, ())?;
|
|
Ok(())
|
|
}
|
|
|
|
fn make_custom_fields(conn: &Connection) -> Result<(), rusqlite::Error> {
|
|
let query = include_str!("../../sql/custom_fields.sql");
|
|
conn.execute(query, ())?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn default_connection() -> Connection {
|
|
let db_file = db_path();
|
|
let connection = Connection::open(db_file).expect("Failed to open database");
|
|
connection
|
|
.execute("PRAGMA foreign_keys = ON;", [])
|
|
.expect("Failed to set PRAGMA");
|
|
connection
|
|
.query_one("PRAGMA journal_mode = WAL;", [], |row| {
|
|
println!("{:?}", row);
|
|
Ok(())
|
|
})
|
|
.unwrap_or_else(|e| {
|
|
println!("{}", e);
|
|
});
|
|
connection
|
|
.execute("PRAGMA synchronous = NORMAL;", [])
|
|
.expect("Failed to set PRAGMA");
|
|
connection
|
|
}
|
|
|
|
pub struct SqliteConnection(Option<Connection>);
|
|
impl Deref for SqliteConnection {
|
|
type Target = Connection;
|
|
fn deref(&self) -> &Self::Target {
|
|
self.0.as_ref().unwrap()
|
|
}
|
|
}
|
|
|
|
impl DerefMut for SqliteConnection {
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
self.0.as_mut().unwrap()
|
|
}
|
|
}
|
|
|
|
impl Default for SqliteConnection {
|
|
fn default() -> Self {
|
|
SqliteConnection(Some(default_connection()))
|
|
}
|
|
}
|
|
|
|
impl SqliteConnection {
|
|
pub fn finish(&mut self) {
|
|
let connection = self.0.take().unwrap();
|
|
connection
|
|
.query_row("PRAGMA wal_checkpoint(TRUNCATE)", [], |_| Ok(()))
|
|
.unwrap_or_else(|e| {
|
|
error!("Ошибка при checkpoint WAL: {e}");
|
|
});
|
|
connection
|
|
.execute("PRAGMA optimize", [])
|
|
.unwrap_or_else(|e| {
|
|
error!("Ошибка при оптимизации БД: {e}");
|
|
0
|
|
});
|
|
|
|
self.0 = None;
|
|
}
|
|
pub fn restart(&mut self) {
|
|
self.0 = Some(default_connection());
|
|
}
|
|
}
|
|
|
|
impl Drop for SqliteConnection {
|
|
|
|
fn drop(&mut self) { self.finish(); }
|
|
} |