@@ -219,7 +219,6 @@ impl Chunk {
219219pub 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 ,
0 commit comments