Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use CastToObjectVectorProcessor for cast to string (#17148)

This commit is contained in:
Clint Wylie 2024-09-24 15:21:33 -07:00 committed by GitHub
parent 83299e9882
commit 0ec7eb3ae5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 10 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
processing/src/main/java/org/apache/druid/math/expr/vector/CastToStringVectorProcessor.java
View File

@ -1,52 +0,0 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.druid.math.expr.vector;
import org.apache.druid.math.expr.Evals;
import org.apache.druid.math.expr.Expr;
import org.apache.druid.math.expr.ExpressionType;
public final class CastToStringVectorProcessor extends CastToTypeVectorProcessor<Object[]>
{
private final Object[] output;
public CastToStringVectorProcessor(ExprVectorProcessor<?> delegate, int maxVectorSize)
{
super(delegate);
this.output = new Object[maxVectorSize];
}
@Override
public ExprEvalVector<Object[]> evalVector(Expr.VectorInputBinding bindings)
{
ExprEvalVector<?> result = delegate.evalVector(bindings);
final Object[] objects = result.getObjectVector();
for (int i = 0; i < bindings.getCurrentVectorSize(); i++) {
output[i] = Evals.asString(objects[i]);
}
return new ExprEvalObjectVector(output, ExpressionType.STRING);
}
@Override
public ExpressionType getOutputType()
{
return ExpressionType.STRING;
}
}

3
processing/src/main/java/org/apache/druid/math/expr/vector/CastToTypeVectorProcessor.java
View File

@ -42,9 +42,6 @@ public abstract class CastToTypeVectorProcessor<TOutput> implements ExprVectorPr
caster = castInput;
} else {
switch (castToType.getType()) {
case STRING:
caster = new CastToStringVectorProcessor(castInput, maxVectorSize);
break;
case LONG:
caster = new CastToLongVectorProcessor(castInput);
break;

10
processing/src/test/java/org/apache/druid/math/expr/VectorExprSanityTest.java
View File

@ -271,6 +271,16 @@ public class VectorExprSanityTest extends InitializedNullHandlingTest
testExpression("array(s1, l2)", types);
}
@Test
public void testCastArraysRoundTrip()
{
testExpression("cast(cast(s1, 'ARRAY<STRING>'), 'STRING')", types);
testExpression("cast(cast(d1, 'ARRAY<DOUBLE>'), 'DOUBLE')", types);
testExpression("cast(cast(d1, 'ARRAY<STRING>'), 'DOUBLE')", types);
testExpression("cast(cast(l1, 'ARRAY<LONG>'), 'LONG')", types);
testExpression("cast(cast(l1, 'ARRAY<STRING>'), 'LONG')", types);
}
@Test
public void testJsonFns()
{