87 lines
2.7 KiB
Rust
87 lines
2.7 KiB
Rust
use sqlx::sqlite::{SqliteConnectOptions, SqlitePool, SqlitePoolOptions};
|
|
use tauri::{AppHandle, Manager};
|
|
|
|
pub async fn init_pool(app: &AppHandle) -> Result<SqlitePool, sqlx::Error> {
|
|
// 数据库文件路径
|
|
let app_dir = app.path().app_data_dir().expect("无法获取APP数据目录");
|
|
std::fs::create_dir_all(&app_dir).unwrap();
|
|
let db_path = app_dir.join("reader.db");
|
|
// 初始化数据库连接池
|
|
let options = SqliteConnectOptions::new()
|
|
.filename(&db_path)
|
|
.create_if_missing(true);
|
|
let pool = SqlitePoolOptions::new()
|
|
.max_connections(5)
|
|
.min_connections(1)
|
|
.connect_with(options)
|
|
.await?;
|
|
sqlx::query(
|
|
"
|
|
CREATE TABLE IF NOT EXISTS books (
|
|
id INTEGER,
|
|
title TEXT DEFAULT '',
|
|
author TEXT DEFAULT '',
|
|
cover TEXT DEFAULT '',
|
|
introduction TEXT DEFAULT '',
|
|
file_path TEXT DEFAULT '',
|
|
total_chapters INTEGER DEFAULT 0,
|
|
total_chars INTEGER DEFAULT 0,
|
|
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|
last_read_chapter_id INTEGER DEFAULT 0,
|
|
last_read_position INTEGER DEFAULT 0,
|
|
last_read_time TIMESTAMP,
|
|
PRIMARY KEY (id)
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS chapters (
|
|
id INTEGER PRIMARY KEY,
|
|
book_id INTEGER NOT NULL,
|
|
number INTEGER NOT NULL,
|
|
title TEXT NOT NULL DEFAULT '',
|
|
content TEXT NOT NULL,
|
|
total_chars INTEGER DEFAULT 0,
|
|
FOREIGN KEY (book_id) REFERENCES books ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_chapters_book
|
|
ON chapters(book_id, number);
|
|
",
|
|
)
|
|
.execute(&pool)
|
|
.await?;
|
|
|
|
// 兼容旧库：确保 chapters.id 为主键（自增）
|
|
let id_is_pk: i64 = sqlx::query_scalar(
|
|
"SELECT COUNT(*) FROM pragma_table_info('chapters') WHERE name = 'id' AND pk > 0",
|
|
)
|
|
.fetch_one(&pool)
|
|
.await
|
|
.unwrap_or(0);
|
|
|
|
if id_is_pk == 0 {
|
|
sqlx::query("DROP TABLE IF EXISTS chapters")
|
|
.execute(&pool)
|
|
.await?;
|
|
sqlx::query(
|
|
"
|
|
CREATE TABLE IF NOT EXISTS chapters (
|
|
id INTEGER PRIMARY KEY,
|
|
book_id INTEGER NOT NULL,
|
|
number INTEGER NOT NULL,
|
|
title TEXT NOT NULL DEFAULT '',
|
|
content TEXT NOT NULL,
|
|
total_chars INTEGER DEFAULT 0,
|
|
FOREIGN KEY (book_id) REFERENCES books ON DELETE CASCADE
|
|
);
|
|
",
|
|
)
|
|
.execute(&pool)
|
|
.await?;
|
|
sqlx::query("CREATE INDEX IF NOT EXISTS idx_chapters_book ON chapters(book_id, number)")
|
|
.execute(&pool)
|
|
.await?;
|
|
}
|
|
|
|
Ok(pool)
|
|
}
|