smolkik_adm/deduplicator
1
0
Fork 0
mirror of https://github.com/sreedevk/deduplicator.git synced 2026-09-07 22:36:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Temporary File Path #6

Browse Source 
Better cross platform support
This commit is contained in:
Dhruva Sagar
2023-01-09 10:51:23 +05:30
parent e158a8267a
commit 8b00faf075
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/database.rs
View File

@@ -1,5 +1,6 @@
use anyhow::Result;
use crate::params::Params; use crate::params::Params;
use anyhow::Result;
use std::env::temp_dir;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct File { pub struct File {
@@ -8,9 +9,12 @@ pub struct File {
} }
pub fn get_connection(args: &Params) -> Result<sqlite::Connection, sqlite::Error> { pub fn get_connection(args: &Params) -> Result<sqlite::Connection, sqlite::Error> {
let mut tmp_file = temp_dir();
tmp_file.push("deduplicator.db");
let connection_url = match args.nocache { let connection_url = match args.nocache {
false => "/tmp/deduplicator.db", false => tmp_file.to_str().unwrap(),
true => ":memory:" true => ":memory:",
}; };
sqlite::open(connection_url).and_then(|conn| { sqlite::open(connection_url).and_then(|conn| {
@@ -36,9 +40,7 @@ pub fn put(file: &File, connection: &sqlite::Connection) -> Result<()> {
} }
pub fn indexed_paths(connection: &sqlite::Connection) -> Result<Vec<File>> { pub fn indexed_paths(connection: &sqlite::Connection) -> Result<Vec<File>> {
let query = format!( let query = format!("SELECT * FROM files");
"SELECT * FROM files"
);
let result: Vec<File> = connection let result: Vec<File> = connection
.prepare(query)? .prepare(query)?
@@ -49,7 +51,7 @@ pub fn indexed_paths(connection: &sqlite::Connection) -> Result<Vec<File>> {
let hash = row.read::<i64, _>("hash").to_string(); let hash = row.read::<i64, _>("hash").to_string();
File { path, hash } File { path, hash }
}) })
.collect(); .collect();
Ok(result) Ok(result)
} }
@@ -65,7 +67,8 @@ pub fn duplicate_hashes(connection: &sqlite::Connection, path: &String) -> Resul
ON a.hash = b.hash ON a.hash = b.hash
WHERE a.file_identifier LIKE \"{}%\" WHERE a.file_identifier LIKE \"{}%\"
ORDER BY a.file_identifier ORDER BY a.file_identifier
", path ",
path
); );
let result: Vec<File> = connection let result: Vec<File> = connection