fix new context state manager

duckdb · Aug 2, 2024 · 8c9dd58 · 8c9dd58
1 parent 51b4b14
commit 8c9dd58
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/spatial/src/spatial/gdal/file_handler.cpp b/spatial/src/spatial/gdal/file_handler.cpp
@@ -385,10 +385,9 @@ string GDALClientContextState::GetPrefix(const string &value) const {
 }
 
 GDALClientContextState &GDALClientContextState::GetOrCreate(ClientContext &context) {
-	if (!context.registered_state["gdal"]) {
-		context.registered_state["gdal"] = make_uniq<GDALClientContextState>(context);
-	}
-	return *dynamic_cast<GDALClientContextState *>(context.registered_state["gdal"].get());
+	auto &manager = *context.registered_state;
+	auto state = manager.GetOrCreate<GDALClientContextState>("gdal", context);
+	return state->Cast<GDALClientContextState>();
 }
 
 } // namespace gdal