1919
2020package org .apache .comet .serde
2121
22- import org .apache .spark .sql .catalyst .expressions .{Attribute , BloomFilterMightContain , KnownFloatingPointNormalized }
22+ import org .apache .spark .sql .catalyst .expressions .{Attribute , KnownFloatingPointNormalized }
2323import org .apache .spark .sql .catalyst .optimizer .NormalizeNaNAndZero
2424
2525import org .apache .comet .CometSparkSessionExtensions .withInfo
26- import org .apache .comet .serde .QueryPlanSerde .{exprToProtoInternal , serializeDataType }
26+ import org .apache .comet .serde .QueryPlanSerde .{exprToProtoInternal , optExprWithInfo , serializeDataType }
2727
2828object CometKnownFloatingPointNormalized
2929 extends CometExpressionSerde [KnownFloatingPointNormalized ] {
3030
3131 override def getSupportLevel (expr : KnownFloatingPointNormalized ): SupportLevel = {
3232 expr.child match {
3333 case _ : NormalizeNaNAndZero => Compatible ()
34- case _ => Unsupported (Some (" KnownFloatingPointNormalized only supports NormalizeNaNAndZero child expressions" ))
34+ case _ =>
35+ Unsupported (
36+ Some (
37+ " KnownFloatingPointNormalized only supports NormalizeNaNAndZero child expressions" ))
3538 }
3639 }
3740
@@ -58,30 +61,3 @@ object CometKnownFloatingPointNormalized
5861 optExprWithInfo(optExpr, expr, wrapped)
5962 }
6063}
61-
62- object CometBloomFilterMightContain extends CometExpressionSerde [BloomFilterMightContain ] {
63-
64- override def convert (
65- expr : BloomFilterMightContain ,
66- inputs : Seq [Attribute ],
67- binding : Boolean ): Option [ExprOuterClass .Expr ] = {
68-
69- val bloomFilter = expr.left
70- val value = expr.right
71- val bloomFilterExpr = exprToProtoInternal(bloomFilter, inputs, binding)
72- val valueExpr = exprToProtoInternal(value, inputs, binding)
73- if (bloomFilterExpr.isDefined && valueExpr.isDefined) {
74- val builder = ExprOuterClass .BloomFilterMightContain .newBuilder()
75- builder.setBloomFilter(bloomFilterExpr.get)
76- builder.setValue(valueExpr.get)
77- Some (
78- ExprOuterClass .Expr
79- .newBuilder()
80- .setBloomFilterMightContain(builder)
81- .build())
82- } else {
83- withInfo(expr, bloomFilter, value)
84- None
85- }
86- }
87- }
0 commit comments