Add tracing to EXIF DAO methods

src/main.rs

@@ -158,6 +158,7 @@ async fn get_file_metadata(
     let tracer = global_tracer();
     let context = extract_context_from_request(&request);
     let mut span = tracer.start_with_context("get_file_metadata", &context);
+    let span_context = opentelemetry::Context::current();
 
     let full_path = is_valid_full_path(&app_state.base_path, &path.path, false);
 
@@ -171,7 +172,7 @@ async fn get_file_metadata(
 
             // Query EXIF data if available
             if let Ok(mut dao) = exif_dao.lock()
-                && let Ok(Some(exif)) = dao.get_exif(&path.path)
+                && let Ok(Some(exif)) = dao.get_exif(&span_context, &path.path)
             {
                 response.exif = Some(exif.into());
             }
@@ -205,6 +206,7 @@ async fn upload_image(
     let tracer = global_tracer();
     let context = extract_context_from_request(&request);
     let mut span = tracer.start_with_context("upload_image", &context);
+    let span_context = opentelemetry::Context::current();
 
     let mut file_content: BytesMut = BytesMut::new();
     let mut file_name: Option<String> = None;
@@ -305,7 +307,7 @@ async fn upload_image(
                         };
 
                         if let Ok(mut dao) = exif_dao.lock() {
-                            if let Err(e) = dao.store_exif(insert_exif) {
+                            if let Err(e) = dao.store_exif(&span_context, insert_exif) {
                                 error!("Failed to store EXIF data for {}: {:?}", relative_path, e);
                             } else {
                                 debug!("EXIF data stored for {}", relative_path);
@@ -877,6 +879,7 @@ fn process_new_files(
     exif_dao: Arc<Mutex<Box<dyn ExifDao>>>,
     modified_since: Option<SystemTime>,
 ) {
+    let context = opentelemetry::Context::new();
     let thumbs = dotenv::var("THUMBNAILS").expect("THUMBNAILS not defined");
     let thumbnail_directory = Path::new(&thumbs);
 
@@ -922,7 +925,7 @@ fn process_new_files(
 
     let existing_exif_paths: HashMap<String, bool> = {
         let mut dao = exif_dao.lock().expect("Unable to lock ExifDao");
-        match dao.get_exif_batch(&file_paths) {
+        match dao.get_exif_batch(&context, &file_paths) {
             Ok(exif_records) => exif_records
                 .into_iter()
                 .map(|record| (record.file_path, true))
@@ -995,7 +998,7 @@ fn process_new_files(
                     };
 
                     let mut dao = exif_dao.lock().expect("Unable to lock ExifDao");
-                    if let Err(e) = dao.store_exif(insert_exif) {
+                    if let Err(e) = dao.store_exif(&context, insert_exif) {
                         error!("Failed to store EXIF data for {}: {:?}", relative_path, e);
                     } else {
                         debug!("EXIF data stored for {}", relative_path);