Skip to content

Commit 5935222

Browse files
committed
Merge #123: Drop: Unused chunks_metadata table
Approved-by: ReinierMaas Priority: Normal Auto-deploy: false
2 parents ff4003f + 3fecec7 commit 5935222

3 files changed

Lines changed: 2 additions & 113 deletions

File tree

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
DROP TABLE chunks_metadata;

opsqueue/src/common/chunk.rs

Lines changed: 0 additions & 89 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,6 @@ impl Chunk {
219219
pub mod db {
220220
use super::*;
221221
use crate::common::errors::{ChunkNotFound, DatabaseError, SubmissionNotFound, E};
222-
use crate::common::StrategicMetadataMap;
223222
use crate::db::{Connection, True, WriterConnection};
224223
use axum_prometheus::metrics::{counter, gauge};
225224
use sqlx::{query, query_as};
@@ -277,33 +276,6 @@ pub mod db {
277276
Ok(())
278277
}
279278

280-
#[tracing::instrument(skip(conn))]
281-
pub async fn insert_chunk_metadata(
282-
chunk: Chunk,
283-
metadata_key: &[u8],
284-
metadata_value: &[u8],
285-
mut conn: impl WriterConnection,
286-
) -> sqlx::Result<()> {
287-
query!(
288-
"
289-
INSERT INTO chunks_metadata
290-
( submission_id
291-
, chunk_index
292-
, metadata_key
293-
, metadata_value
294-
)
295-
VALUES ($1, $2, $3, $4)
296-
",
297-
chunk.submission_id,
298-
chunk.chunk_index,
299-
metadata_key,
300-
metadata_value,
301-
)
302-
.execute(conn.get_inner())
303-
.await?;
304-
Ok(())
305-
}
306-
307279
#[tracing::instrument(skip(conn))]
308280
pub async fn complete_chunk(
309281
chunk_id: ChunkId,
@@ -530,33 +502,6 @@ pub mod db {
530502
.await
531503
}
532504

533-
/// Retrieves the earlier stored strategic metadata.
534-
///
535-
/// Primarily for testing and introspection.
536-
///
537-
/// Be aware that the strategic metadata for individual chunks
538-
/// is cleaned up once the chunk is marked as completed or failed.
539-
/// (At that time it is still available on the submission level).
540-
pub async fn get_chunk_strategic_metadata(
541-
full_chunk_id: ChunkId,
542-
mut conn: impl Connection,
543-
) -> Result<StrategicMetadataMap, DatabaseError> {
544-
use futures::{future, TryStreamExt};
545-
let metadata = query!(
546-
r#"
547-
SELECT metadata_key, metadata_value FROM chunks_metadata
548-
WHERE submission_id = $1 AND chunk_index = $2
549-
"#,
550-
full_chunk_id.submission_id,
551-
full_chunk_id.chunk_index,
552-
)
553-
.fetch(conn.get_inner())
554-
.and_then(|row| future::ok((row.metadata_key, row.metadata_value)))
555-
.try_collect()
556-
.await?;
557-
Ok(metadata)
558-
}
559-
560505
#[tracing::instrument(skip(chunks, conn))]
561506
pub async fn insert_many_chunks(
562507
chunks: &[Chunk],
@@ -584,40 +529,6 @@ pub mod db {
584529
Ok(())
585530
}
586531

587-
pub async fn insert_many_chunks_metadata(
588-
chunks: &[Chunk],
589-
metadata: &StrategicMetadataMap,
590-
mut conn: impl WriterConnection,
591-
) -> sqlx::Result<()> {
592-
use itertools::Itertools;
593-
const ROWS_PER_QUERY: usize = 1000;
594-
595-
let mut iter = chunks.iter().cartesian_product(metadata).peekable();
596-
while iter.peek().is_some() {
597-
let query_rows = iter.by_ref().take(ROWS_PER_QUERY);
598-
599-
let mut query_builder: QueryBuilder<Sqlite> = QueryBuilder::new(
600-
"
601-
INSERT INTO chunks_metadata
602-
( submission_id
603-
, chunk_index
604-
, metadata_key
605-
, metadata_value
606-
)
607-
",
608-
);
609-
query_builder.push_values(query_rows, |mut b, (chunk, metadata)| {
610-
b.push_bind(chunk.submission_id)
611-
.push_bind(chunk.chunk_index)
612-
.push_bind(metadata.0)
613-
.push_bind(metadata.1);
614-
});
615-
let query = query_builder.build();
616-
query.execute(conn.get_inner()).await?;
617-
}
618-
Ok(())
619-
}
620-
621532
#[tracing::instrument(skip(conn))]
622533
pub async fn skip_remaining_chunks(
623534
submission_id: SubmissionId,

opsqueue/src/common/submission.rs

Lines changed: 1 addition & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -392,12 +392,6 @@ pub mod db {
392392
)
393393
.await?;
394394
super::chunk::db::insert_many_chunks(&chunks, &mut tx).await?;
395-
super::chunk::db::insert_many_chunks_metadata(
396-
&chunks,
397-
&submission.strategic_metadata,
398-
&mut tx,
399-
)
400-
.await?;
401395
Ok(())
402396
}
403397
.boxed()
@@ -1255,24 +1249,7 @@ pub mod test {
12551249
let fetched_metadata = get_submission_strategic_metadata(submission_id, &mut conn)
12561250
.await
12571251
.unwrap();
1258-
assert!(fetched_metadata == strategic_metadata);
1259-
1260-
let res = sqlx::query!("SELECT * FROM chunks_metadata;")
1261-
.fetch_all(conn.get_inner())
1262-
.await
1263-
.unwrap();
1264-
dbg!(res);
1265-
1266-
let chunk_id = (submission_id, ChunkIndex::zero()).into();
1267-
let chunk_fetched_metadata = chunk::db::get_chunk_strategic_metadata(chunk_id, &mut conn)
1268-
.await
1269-
.unwrap();
1270-
1271-
dbg!(&strategic_metadata);
1272-
dbg!(&fetched_metadata);
1273-
dbg!(&chunk_fetched_metadata);
1274-
1275-
assert_eq!(chunk_fetched_metadata, strategic_metadata);
1252+
assert_eq!(fetched_metadata, strategic_metadata);
12761253
}
12771254

12781255
#[sqlx::test]

0 commit comments

Comments
 (0)