feat: special handle metric engine path

This commit is contained in:
evenyag
2025-03-10 13:58:46 +08:00
parent 359da62d9e
commit 30c29539a3
2 changed files with 7 additions and 1 deletions

View File

@@ -35,6 +35,7 @@ use crate::table::TableMetadataHelper;
use crate::writer::RegionWriterBuilder;
/// Input file type.
#[derive(Clone, PartialEq, Eq)]
pub enum InputFileType {
/// File type is Parquet.
Parquet,

View File

@@ -22,6 +22,7 @@ use mito2::read::BoxedBatchReader;
use mito2::region::opener::RegionMetadataLoader;
use mito2::region::options::RegionOptions;
use object_store::manager::ObjectStoreManagerRef;
use object_store::util::join_dir;
use object_store::ObjectStore;
use snafu::{OptionExt, ResultExt};
use store_api::metadata::RegionMetadataRef;
@@ -139,7 +140,11 @@ impl InputReaderBuilder {
let opts = table_info.table_info.to_region_options();
// TODO(yingwen): We ignore WAL options now. We should `prepare_wal_options()` in the future.
let region_options = RegionOptions::try_from(&opts).context(MitoSnafu)?;
let region_dir = region_dir(&table_info.region_storage_path(), region_id);
let mut region_dir = region_dir(&table_info.region_storage_path(), region_id);
if input.file_type == InputFileType::RemoteWrite {
// metric engine has two internal regions.
region_dir = join_dir(&region_dir, "data");
}
let manifest = self
.region_loader
.load_manifest(&region_dir, &region_options)