use crate::{csv_source::CsvSource, parquet_source::ParquetSource};
pub enum DataSource {
Csv(CsvSource),
Parquet(ParquetSource),
}
impl DataSource {
pub fn open(path: &str) -> Self {
if path.ends_with(".csv") {
DataSource::Csv(CsvSource::new(path))
} else if path.ends_with(".parquet") {
DataSource::Parquet(ParquetSource::new(path))
} else {
panic!("Unsupported file extension");
}
}
}