MCPcopy Create free account
hub / github.com/apache/datafusion / build_ctx

Function build_ctx

datafusion/core/src/datasource/file_format/parquet.rs:1244–1269  ·  view source on GitHub ↗
(store_url: &url::Url)

Source from the content-addressed store, hash-verified

1242 }
1243
1244 fn build_ctx(store_url: &url::Url) -> Arc<TaskContext> {
1245 let tmp_dir = tempfile::TempDir::new().unwrap();
1246 let local = Arc::new(
1247 LocalFileSystem::new_with_prefix(&tmp_dir)
1248 .expect("should create object store"),
1249 );
1250
1251 let mut session = SessionConfig::default();
1252 let mut parquet_opts = ParquetOptions {
1253 allow_single_file_parallelism: true,
1254 ..Default::default()
1255 };
1256 parquet_opts.allow_single_file_parallelism = true;
1257 session.options_mut().execution.parquet = parquet_opts;
1258
1259 let runtime = RuntimeEnv::default();
1260 runtime
1261 .object_store_registry
1262 .register_store(store_url, local);
1263
1264 Arc::new(
1265 TaskContext::default()
1266 .with_session_config(session)
1267 .with_runtime(Arc::new(runtime)),
1268 )
1269 }
1270
1271 #[tokio::test]
1272 async fn parquet_sink_write() -> Result<()> {

Calls 5

newFunction · 0.85
options_mutMethod · 0.80
with_runtimeMethod · 0.80
with_session_configMethod · 0.80
register_storeMethod · 0.45

Tested by 1

test_memory_reservationFunction · 0.68

Used in the wild real call sites across dependent graphs

searching dependent graphs…