Skip to content

Commit 3193302

Browse files
committed
feat: register path with datafusion
1 parent b6d4d54 commit 3193302

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/storage/gcs.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -133,9 +133,9 @@ impl ObjectStorageProvider for GcsConfig {
133133
let gcs = MetricLayer::new(gcs);
134134

135135
let object_store_registry = DefaultObjectStoreRegistry::new();
136-
// Register GCS client under the "s3://" scheme so DataFusion can route
137-
// object store calls to our GoogleCloudStorage implementatio
138-
let url = ObjectStoreUrl::parse(format!("s3://{}", &self.bucket_name)).unwrap();
136+
// Register GCS client under the "gs://" scheme so DataFusion can route
137+
// object store calls to our GoogleCloudStorage implementation
138+
let url = ObjectStoreUrl::parse(format!("gs://{}", &self.bucket_name)).unwrap();
139139
object_store_registry.register_store(url.as_ref(), Arc::new(gcs));
140140

141141
RuntimeEnvBuilder::new().with_object_store_registry(Arc::new(object_store_registry))
@@ -643,14 +643,14 @@ impl ObjectStorage for Gcs {
643643
prefixes
644644
.into_iter()
645645
.map(|prefix| {
646-
let path = format!("s3://{}/{}", &self.bucket, prefix);
646+
let path = format!("gs://{}/{}", &self.bucket, prefix);
647647
ListingTableUrl::parse(path).unwrap()
648648
})
649649
.collect()
650650
}
651651

652652
fn store_url(&self) -> url::Url {
653-
url::Url::parse(&format!("s3://{}", self.bucket)).unwrap()
653+
url::Url::parse(&format!("gs://{}", self.bucket)).unwrap()
654654
}
655655

656656
async fn list_dirs(&self) -> Result<Vec<String>, ObjectStorageError> {

0 commit comments

Comments
 (0)