Skip to content

Commit 8d8b669

Browse files
committed
address review comments
1 parent 53661eb commit 8d8b669

File tree

1 file changed

+30
-30
lines changed

1 file changed

+30
-30
lines changed

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/regexpExpressions.scala

Lines changed: 30 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ case class Like(left: Expression, right: Expression) extends StringRegexExpressi
119119
val regexStr =
120120
StringEscapeUtils.escapeJava(escape(rVal.asInstanceOf[UTF8String].toString()))
121121
val pattern = ctx.addMutableState(patternClass, "patternLike",
122-
v => s"""$v = ${patternClass}.compile("$regexStr");""")
122+
v => s"""$v = $patternClass.compile("$regexStr");""")
123123

124124
// We don't use nullSafeCodeGen here because we don't want to re-evaluate right again.
125125
val eval = left.genCode(ctx)
@@ -142,9 +142,9 @@ case class Like(left: Expression, right: Expression) extends StringRegexExpressi
142142
val rightStr = ctx.freshName("rightStr")
143143
nullSafeCodeGen(ctx, ev, (eval1, eval2) => {
144144
s"""
145-
String $rightStr = ${eval2}.toString();
146-
${patternClass} $pattern = ${patternClass}.compile($escapeFunc($rightStr));
147-
${ev.value} = $pattern.matcher(${eval1}.toString()).matches();
145+
String $rightStr = $eval2.toString();
146+
$patternClass $pattern = $patternClass.compile($escapeFunc($rightStr));
147+
${ev.value} = $pattern.matcher($eval1.toString()).matches();
148148
"""
149149
})
150150
}
@@ -194,7 +194,7 @@ case class RLike(left: Expression, right: Expression) extends StringRegexExpress
194194
val regexStr =
195195
StringEscapeUtils.escapeJava(rVal.asInstanceOf[UTF8String].toString())
196196
val pattern = ctx.addMutableState(patternClass, "patternRLike",
197-
v => s"""$v = ${patternClass}.compile("$regexStr");""")
197+
v => s"""$v = $patternClass.compile("$regexStr");""")
198198

199199
// We don't use nullSafeCodeGen here because we don't want to re-evaluate right again.
200200
val eval = left.genCode(ctx)
@@ -217,9 +217,9 @@ case class RLike(left: Expression, right: Expression) extends StringRegexExpress
217217
val pattern = ctx.freshName("pattern")
218218
nullSafeCodeGen(ctx, ev, (eval1, eval2) => {
219219
s"""
220-
String $rightStr = ${eval2}.toString();
221-
${patternClass} $pattern = ${patternClass}.compile($rightStr);
222-
${ev.value} = $pattern.matcher(${eval1}.toString()).find(0);
220+
String $rightStr = $eval2.toString();
221+
$patternClass $pattern = $patternClass.compile($rightStr);
222+
${ev.value} = $pattern.matcher($eval1.toString()).find(0);
223223
"""
224224
})
225225
}
@@ -336,25 +336,25 @@ case class RegExpReplace(subject: Expression, regexp: Expression, rep: Expressio
336336

337337
nullSafeCodeGen(ctx, ev, (subject, regexp, rep) => {
338338
s"""
339-
if (!$regexp.equals(${termLastRegex})) {
339+
if (!$regexp.equals($termLastRegex)) {
340340
// regex value changed
341-
${termLastRegex} = $regexp.clone();
342-
${termPattern} = ${classNamePattern}.compile(${termLastRegex}.toString());
341+
$termLastRegex = $regexp.clone();
342+
$termPattern = $classNamePattern.compile($termLastRegex.toString());
343343
}
344-
if (!$rep.equals(${termLastReplacementInUTF8})) {
344+
if (!$rep.equals($termLastReplacementInUTF8)) {
345345
// replacement string changed
346-
${termLastReplacementInUTF8} = $rep.clone();
347-
${termLastReplacement} = ${termLastReplacementInUTF8}.toString();
346+
$termLastReplacementInUTF8 = $rep.clone();
347+
$termLastReplacement = $termLastReplacementInUTF8.toString();
348348
}
349-
$classNameStringBuffer ${termResult} = new $classNameStringBuffer();
350-
java.util.regex.Matcher ${matcher} = ${termPattern}.matcher($subject.toString());
349+
$classNameStringBuffer $termResult = new $classNameStringBuffer();
350+
java.util.regex.Matcher $matcher = $termPattern.matcher($subject.toString());
351351

352-
while (${matcher}.find()) {
353-
${matcher}.appendReplacement(${termResult}, ${termLastReplacement});
352+
while ($matcher.find()) {
353+
$matcher.appendReplacement($termResult, $termLastReplacement);
354354
}
355-
${matcher}.appendTail(${termResult});
356-
${ev.value} = UTF8String.fromString(${termResult}.toString());
357-
${termResult} = null;
355+
$matcher.appendTail($termResult);
356+
${ev.value} = UTF8String.fromString($termResult.toString());
357+
$termResult = null;
358358
$setEvNotNull
359359
"""
360360
})
@@ -423,19 +423,19 @@ case class RegExpExtract(subject: Expression, regexp: Expression, idx: Expressio
423423

424424
nullSafeCodeGen(ctx, ev, (subject, regexp, idx) => {
425425
s"""
426-
if (!$regexp.equals(${termLastRegex})) {
426+
if (!$regexp.equals($termLastRegex)) {
427427
// regex value changed
428-
${termLastRegex} = $regexp.clone();
429-
${termPattern} = ${classNamePattern}.compile(${termLastRegex}.toString());
428+
$termLastRegex = $regexp.clone();
429+
$termPattern = $classNamePattern.compile($termLastRegex.toString());
430430
}
431-
java.util.regex.Matcher ${matcher} =
432-
${termPattern}.matcher($subject.toString());
433-
if (${matcher}.find()) {
434-
java.util.regex.MatchResult ${matchResult} = ${matcher}.toMatchResult();
435-
if (${matchResult}.group($idx) == null) {
431+
java.util.regex.Matcher $matcher =
432+
$termPattern.matcher($subject.toString());
433+
if ($matcher.find()) {
434+
java.util.regex.MatchResult $matchResult = $matcher.toMatchResult();
435+
if ($matchResult.group($idx) == null) {
436436
${ev.value} = UTF8String.EMPTY_UTF8;
437437
} else {
438-
${ev.value} = UTF8String.fromString(${matchResult}.group($idx));
438+
${ev.value} = UTF8String.fromString($matchResult.group($idx));
439439
}
440440
$setEvNotNull
441441
} else {

0 commit comments

Comments
 (0)