wip

Author: ZENOTME

crates/iceberg/src/writer/file_writer/location_generator.rs

@@ -26,8 +26,13 @@ use crate::{Error, ErrorKind, Result};
 
 /// LocationGenerator will generate a file location for the writer.
 pub trait LocationGenerator: Send + Sync + 'static {
-    /// Generate a related file location for the writer.
-    fn generate_name(&self) -> String;
+    /// Generate a full file location for the writer.
+    fn generate_name(&self, file_name: &str) -> String;
+
+    /// 
+    fn generate_name_with_extra_dir(&self, extra_dir: &str, file_name: &str) -> String {
+        self.generate_name(&format!("{}/{}", extra_dir, file_name))
+    }
 }
 
 const DEFAULT_FILE_FORMAT: &str = "write.format.default";