feat: add media type and indexed status filters to FilesView
Frontend: - Add media type filter (全部/影片/照片) - Add indexed status filter (未入庫/已入庫) - Show media type column with icons - Fix status filter to handle indexed/unindexed correctly - Determine media type from file extension Backend: - Add total_chunks field to FileItem API response - Query chunk counts efficiently in batch with IN clause - Frontend uses total_chunks to determine is_indexed status
This commit is contained in:
+42
-3
@@ -96,11 +96,21 @@ async fn list_files(
|
||||
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))?;
|
||||
|
||||
let data = if let Some(v) = video {
|
||||
let chunk_count: i64 = sqlx::query_scalar(&format!(
|
||||
"SELECT COUNT(*) FROM {} WHERE file_uuid = $1",
|
||||
crate::core::db::schema::table_name("chunk")
|
||||
))
|
||||
.bind(&v.file_uuid)
|
||||
.fetch_one(state.db.pool())
|
||||
.await
|
||||
.unwrap_or(0);
|
||||
|
||||
vec![FileItem {
|
||||
file_uuid: v.file_uuid,
|
||||
file_name: v.file_name,
|
||||
file_path: v.file_path,
|
||||
status: v.status.as_str().to_string(),
|
||||
total_chunks: chunk_count,
|
||||
}]
|
||||
} else {
|
||||
vec![]
|
||||
@@ -124,18 +134,45 @@ async fn list_files(
|
||||
.await
|
||||
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))?;
|
||||
|
||||
let data = records
|
||||
let total = records.1;
|
||||
|
||||
let mut data: Vec<FileItem> = records
|
||||
.0
|
||||
.into_iter()
|
||||
.map(|r| FileItem {
|
||||
file_uuid: r.file_uuid,
|
||||
file_uuid: r.file_uuid.clone(),
|
||||
file_name: r.file_name,
|
||||
file_path: r.file_path,
|
||||
status: r.status.as_str().to_string(),
|
||||
total_chunks: 0,
|
||||
})
|
||||
.collect();
|
||||
|
||||
let total = records.1;
|
||||
// Fetch chunk counts for all files in one query
|
||||
let uuids: Vec<String> = data.iter().map(|f| f.file_uuid.clone()).collect();
|
||||
if !uuids.is_empty() {
|
||||
let chunk_table = crate::core::db::schema::table_name("chunk");
|
||||
let placeholders: Vec<String> = (1..=uuids.len()).map(|i| format!("${}", i)).collect();
|
||||
let query_str = format!(
|
||||
"SELECT file_uuid, COUNT(*) as cnt FROM {} WHERE file_uuid IN ({}) GROUP BY file_uuid",
|
||||
chunk_table,
|
||||
placeholders.join(", ")
|
||||
);
|
||||
|
||||
let chunk_counts: Vec<(String, i64)> = sqlx::query_as(&query_str)
|
||||
.fetch_all(state.db.pool())
|
||||
.await
|
||||
.unwrap_or_default();
|
||||
|
||||
let count_map: std::collections::HashMap<String, i64> =
|
||||
chunk_counts.into_iter().collect();
|
||||
|
||||
for item in &mut data {
|
||||
if let Some(cnt) = count_map.get(&item.file_uuid) {
|
||||
item.total_chunks = *cnt;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(Json(FilesResponse {
|
||||
success: true,
|
||||
@@ -161,6 +198,8 @@ pub struct FileItem {
|
||||
pub file_name: String,
|
||||
pub file_path: String,
|
||||
pub status: String,
|
||||
#[serde(default)]
|
||||
pub total_chunks: i64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize)]
|
||||
|
||||
Reference in New Issue
Block a user