Skip to content

Commit 1b95be4

Browse files
author
Davies Liu
committed
put the generated class into sql package
1 parent 77ed72d commit 1b95be4

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/CodeGenerator.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,7 @@ abstract class CodeGenerator[InType <: AnyRef, OutType <: AnyRef] extends Loggin
366366
private[this] def doCompile(code: String): GeneratedClass = {
367367
val evaluator = new ClassBodyEvaluator()
368368
evaluator.setParentClassLoader(getClass.getClassLoader)
369+
evaluator.setClassName("org.apache.spark.sql.catalyst.expressions.codegen.GeneratedClass")
369370
evaluator.setDefaultImports(Array(
370371
classOf[PlatformDependent].getName,
371372
classOf[InternalRow].getName,

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/GenerateMutableProjection.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,17 +69,17 @@ object GenerateMutableProjection extends CodeGenerator[Seq[Expression], () => Mu
6969

7070
val code = s"""
7171
public Object generate($exprType[] expr) {
72-
return new SpecificProjection(expr);
72+
return new SpecificMutableProjection(expr);
7373
}
7474

75-
class SpecificProjection extends ${classOf[BaseMutableProjection].getName} {
75+
class SpecificMutableProjection extends ${classOf[BaseMutableProjection].getName} {
7676

7777
private $exprType[] expressions;
7878
private $mutableRowType mutableRow;
7979
${declareMutableStates(ctx)}
8080
${declareAddedFunctions(ctx)}
8181

82-
public SpecificProjection($exprType[] expr) {
82+
public SpecificMutableProjection($exprType[] expr) {
8383
expressions = expr;
8484
mutableRow = new $genericMutableRowType(${expressions.size});
8585
${initMutableStates(ctx)}

0 commit comments

Comments
 (0)