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

SOLR-10303: Switching from the fieldName param to subEvaluators

This commit is contained in:
Gethin James 2017-03-17 18:11:00 +01:00 committed by Joel Bernstein
parent b13945b1ef
commit c3d205cdcc
2 changed files with 45 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
solr/solrj/src/java/org/apache/solr/client/solrj/io/eval/DatePartEvaluator.java
View File

@ -38,10 +38,9 @@ import org.apache.solr.client.solrj.io.stream.expr.StreamFactory;
*/ */
public class DatePartEvaluator extends NumberEvaluator { public class DatePartEvaluator extends NumberEvaluator {
public enum FUNCTION {year, month, day, dayofyear, dayofquarter, hour, minute, quarter, week, second, epoch}; public enum FUNCTION {year, month, day, dayofyear, dayofquarter, hour, minute, quarter, week, second, epoch}
private FUNCTION function; private FUNCTION function;
private String fieldName;
public DatePartEvaluator(StreamExpression expression, StreamFactory factory) throws IOException { public DatePartEvaluator(StreamExpression expression, StreamFactory factory) throws IOException {
super(expression, factory); super(expression, factory);
@ -51,66 +50,63 @@ public class DatePartEvaluator extends NumberEvaluator {
try { try {
this.function = FUNCTION.valueOf(functionName); this.function = FUNCTION.valueOf(functionName);
} catch (IllegalArgumentException e) { } catch (IllegalArgumentException e) {
throw new IOException(String.format(Locale.ROOT,"Invalid date expression %s - expecting one of %s",functionName, Arrays.toString(FUNCTION.values()))); throw new IOException(String.format(Locale.ROOT, "Invalid date expression %s - expecting one of %s", functionName, Arrays.toString(FUNCTION.values())));
} }
fieldName = factory.getValueOperand(expression, 0); if (1 != subEvaluators.size()) {
throw new IOException(String.format(Locale.ROOT, "Invalid expression %s - expecting one value but found %d", expression, subEvaluators.size()));
//Taken from Field evaluator
if(fieldName != null && fieldName.startsWith("'") && fieldName.endsWith("'") && fieldName.length() > 1){
fieldName = fieldName.substring(1, fieldName.length() - 1);
}
if(1 != subEvaluators.size()){
throw new IOException(String.format(Locale.ROOT,"Invalid expression %s - expecting one value but found %d",expression,subEvaluators.size()));
} }
} }
@Override @Override
public Number evaluate(Tuple tuple) throws IOException { public Number evaluate(Tuple tuple) throws IOException {
try { Instant instant = null;
Object fieldValue = tuple.get(fieldName); LocalDateTime date = null;
Instant instant = null;
LocalDateTime date = null;
if (fieldValue == null) return null; //First evaluate the parameter
StreamEvaluator streamEvaluator = subEvaluators.get(0);
Object tupleValue = streamEvaluator.evaluate(tuple);
if (fieldValue instanceof String) { if (tupleValue == null) return null;
instant = getInstant((String)fieldValue);
} else if (fieldValue instanceof Instant) {
instant = (Instant) fieldValue;
} else if (fieldValue instanceof Date) {
instant = ((Date) fieldValue).toInstant();
} else if (fieldValue instanceof LocalDateTime) {
date = ((LocalDateTime) fieldValue);
}
if (instant != null) { if (tupleValue instanceof String) {
if (function.equals(FUNCTION.epoch)) return instant.toEpochMilli(); instant = getInstant((String) tupleValue);
date = LocalDateTime.ofInstant(instant, ZoneOffset.UTC); } else if (tupleValue instanceof Instant) {
} instant = (Instant) tupleValue;
} else if (tupleValue instanceof Date) {
if (date != null) { instant = ((Date) tupleValue).toInstant();
return evaluate(date); } else if (tupleValue instanceof LocalDateTime) {
} date = ((LocalDateTime) tupleValue);
} catch (DateTimeParseException e) {
throw new IOException(String.format(Locale.ROOT,"Invalid field %s - The field must be a string formatted in the ISO_INSTANT date format.",fieldName));
} }
throw new IOException(String.format(Locale.ROOT,"Invalid field %s - The field must be a string formatted ISO_INSTANT or of type Instant,Date or LocalDateTime.",fieldName)); if (instant != null) {
if (function.equals(FUNCTION.epoch)) return instant.toEpochMilli();
date = LocalDateTime.ofInstant(instant, ZoneOffset.UTC);
}
if (date != null) {
return evaluate(date);
}
throw new IOException(String.format(Locale.ROOT, "Invalid parameter %s - The parameter must be a string formatted ISO_INSTANT or of type Instant,Date or LocalDateTime.", String.valueOf(tupleValue)));
} }
private Instant getInstant(String dateStr) { private Instant getInstant(String dateStr) throws IOException {
if (dateStr != null && !dateStr.isEmpty()) { if (dateStr != null && !dateStr.isEmpty()) {
return Instant.parse(dateStr); try {
return Instant.parse(dateStr);
} catch (DateTimeParseException e) {
throw new IOException(String.format(Locale.ROOT, "Invalid parameter %s - The String must be formatted in the ISO_INSTANT date format.", dateStr));
}
} }
return null; return null;
} }
/** /**
* Evaluate the date based on the specified function * Evaluate the date based on the specified function
*
* @param date * @param date
* @return the evaluated value * @return the evaluated value
*/ */
@ -142,7 +138,13 @@ public class DatePartEvaluator extends NumberEvaluator {
@Override @Override
public StreamExpressionParameter toExpression(StreamFactory factory) throws IOException { public StreamExpressionParameter toExpression(StreamFactory factory) throws IOException {
return new StreamExpression(function.toString()).withParameter(fieldName); StreamExpression expression = new StreamExpression(function.toString());
for (StreamEvaluator evaluator : subEvaluators) {
expression.addParameter(evaluator.toExpression(factory));
}
return expression;
} }
@Override @Override

4
solr/solrj/src/test/org/apache/solr/client/solrj/io/stream/eval/DatePartEvaluatorTest.java
View File

@ -109,7 +109,7 @@ public class DatePartEvaluatorTest {
Object result = evaluator.evaluate(new Tuple(values)); Object result = evaluator.evaluate(new Tuple(values));
assertTrue(false); assertTrue(false);
} catch (Exception e) { } catch (Exception e) {
assertEquals("Invalid field a - The field must be a string formatted ISO_INSTANT or of type Instant,Date or LocalDateTime.", e.getMessage()); assertEquals("Invalid parameter 12 - The parameter must be a string formatted ISO_INSTANT or of type Instant,Date or LocalDateTime.", e.getMessage());
} }
try { try {
@ -118,7 +118,7 @@ public class DatePartEvaluatorTest {
Object result = evaluator.evaluate(new Tuple(values)); Object result = evaluator.evaluate(new Tuple(values));
assertTrue(false); assertTrue(false);
} catch (Exception e) { } catch (Exception e) {
assertEquals("Invalid field a - The field must be a string formatted in the ISO_INSTANT date format.", e.getMessage()); assertEquals("Invalid parameter 1995-12-31 - The String must be formatted in the ISO_INSTANT date format.", e.getMessage());
} }
values.clear(); values.clear();