datafusion-comet
datafusion-comet copied to clipboard
[EPIC] Spark-compatible cast / try_cast operations
What is the problem the feature request solves?
Comet currently delegates to DataFusion for many cast operations, and the behavior is not guaranteed to match Spark. This epic is to track fully implementing Spark-compatible cast
and try_cast
operations in Comet, with support for ANSI mode.
For each item in this list to be considered complete, we should have scala tests demonstrating that cast
and try_cast
produce the same results as Spark, both with ANSI mode enabled and disabled, using fuzz testing to find edge cases. We can update this list with links to issues as we make progress.
For cast operations that we cannot easily support with full compatibility, we should either fall back to Spark or provide a configuration that the user can enable to allow the operation to run in Comet. We should also provide documentation explaining any differences in behavior compared to Spark.
- [ ] Cast from string to another type
- [x] Boolean - https://github.com/apache/datafusion-comet/pull/290
- [x] Integral Types (byte, short, int, long) - https://github.com/apache/datafusion-comet/pull/307
- [ ] Floating-point (float, double) - https://github.com/apache/datafusion-comet/issues/326
- [ ] Decimal - https://github.com/apache/datafusion-comet/issues/325
- [x] Date - https://github.com/apache/datafusion-comet/issues/327
- [ ] Timestamp - https://github.com/apache/datafusion-comet/issues/328 and https://github.com/apache/datafusion-comet/issues/376
- [ ] Cast to string from primitive types
- [x] Boolean - https://github.com/apache/datafusion-comet/pull/351
- [x] Integral Types (byte, short, int, long) - https://github.com/apache/datafusion-comet/pull/351
- [x] Floating-point (float, double) - https://github.com/apache/datafusion-comet/issues/312
- [ ] Decimal - seems to correct but needs tests to confirm, also should fall back to Spark if
spark.sql.legacy.allowNegativeScaleOfDecimal
is true and scale is negative - [ ] Date - seems correct but needs tests to confirm
- [ ] Timestamp - seems correct but needs tests to confirm
- [ ] Cast between numeric types
- [x] Integral to Integral - https://github.com/apache/datafusion-comet/issues/311
- [x] Integral to Boolean - https://github.com/apache/datafusion-comet/pull/351
- [ ] Integral to Decimal - needs out of range checks
- [x] Integral to Floating-point - https://github.com/apache/datafusion-comet/pull/351
- [x] Floating-point to Boolean - https://github.com/apache/datafusion-comet/pull/351
- [x] Floating-point to Decimal - https://github.com/apache/datafusion-comet/issues/371
- [x] Floating-point to Integral - https://github.com/apache/datafusion-comet/issues/350
- [ ] Decimal to Boolean - fails because arrow does not support this cast
- [x] Decimal to Integral - same issues as https://github.com/apache/datafusion-comet/issues/350
- [ ] Decimal to Floating-point - seems correct but needs tests to confirm
- [ ] https://github.com/apache/datafusion-comet/issues/375
- [ ] Cast between temporal types
- [ ] Date to boolean/int/float/decimal - results are incorrect
- [ ] Date to Timestamp / TimestampNTZ
- [ ] Timestamp to boolean/int/float/decimal - https://github.com/apache/datafusion-comet/issues/352
- [ ] Timestamp to Date
- [ ] Other
- [ ] https://github.com/apache/datafusion-comet/issues/377
- [ ] https://github.com/apache/datafusion-comet/issues/378
In addition to the above tasks, we also need to do the following:
- [x] Implement a mechanism where we can selectively fall back to Spark for specific cast operations (https://github.com/apache/datafusion-comet/pull/337 )
- [ ] Write documentation that explains any differences between Comet and Spark
- [x] https://github.com/apache/datafusion-comet/issues/374