summaryrefslogtreecommitdiff
path: root/opendc-trace/opendc-trace-parquet/src
diff options
context:
space:
mode:
Diffstat (limited to 'opendc-trace/opendc-trace-parquet/src')
-rw-r--r--opendc-trace/opendc-trace-parquet/src/main/kotlin/org/opendc/trace/util/parquet/LocalParquetReader.kt8
1 files changed, 6 insertions, 2 deletions
diff --git a/opendc-trace/opendc-trace-parquet/src/main/kotlin/org/opendc/trace/util/parquet/LocalParquetReader.kt b/opendc-trace/opendc-trace-parquet/src/main/kotlin/org/opendc/trace/util/parquet/LocalParquetReader.kt
index 031bad60..de8a56d0 100644
--- a/opendc-trace/opendc-trace-parquet/src/main/kotlin/org/opendc/trace/util/parquet/LocalParquetReader.kt
+++ b/opendc-trace/opendc-trace-parquet/src/main/kotlin/org/opendc/trace/util/parquet/LocalParquetReader.kt
@@ -38,10 +38,12 @@ import kotlin.io.path.isDirectory
*
* @param path The path to the Parquet file or directory to read.
* @param readSupport Helper class to perform conversion from Parquet to [T].
+ * @param strictTyping A flag to disable strict typing of primitive types.
*/
public class LocalParquetReader<out T>(
path: Path,
- private val readSupport: ReadSupport<T>
+ private val readSupport: ReadSupport<T>,
+ private val strictTyping: Boolean = true
) : AutoCloseable {
/**
* The input files to process.
@@ -119,6 +121,8 @@ public class LocalParquetReader<out T>(
private fun createReader(input: InputFile): ParquetReader<T> {
return object : ParquetReader.Builder<T>(input) {
override fun getReadSupport(): ReadSupport<@UnsafeVariance T> = this@LocalParquetReader.readSupport
- }.build()
+ }
+ .set("parquet.strict.typing", strictTyping.toString())
+ .build()
}
}