2020-02-13 16:49:45 +01:00
|
|
|
package eu.dnetlib.dhp.utils.saxon;
|
|
|
|
|
2020-04-18 12:42:58 +02:00
|
|
|
import java.text.ParseException;
|
|
|
|
import java.text.SimpleDateFormat;
|
|
|
|
import java.util.Calendar;
|
|
|
|
import java.util.GregorianCalendar;
|
2020-02-13 16:49:45 +01:00
|
|
|
import net.sf.saxon.expr.XPathContext;
|
|
|
|
import net.sf.saxon.om.Item;
|
|
|
|
import net.sf.saxon.om.Sequence;
|
|
|
|
import net.sf.saxon.trans.XPathException;
|
|
|
|
import net.sf.saxon.value.SequenceType;
|
|
|
|
import net.sf.saxon.value.StringValue;
|
|
|
|
|
|
|
|
public class ExtractYear extends AbstractExtensionFunction {
|
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
private static final String[] dateFormats = {"yyyy-MM-dd", "yyyy/MM/dd"};
|
2020-02-13 16:49:45 +01:00
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
@Override
|
|
|
|
public String getName() {
|
|
|
|
return "extractYear";
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
@Override
|
|
|
|
public Sequence doCall(XPathContext context, Sequence[] arguments) throws XPathException {
|
|
|
|
if (arguments == null | arguments.length == 0) {
|
|
|
|
return new StringValue("");
|
2020-02-13 16:49:45 +01:00
|
|
|
}
|
2020-04-27 14:45:40 +02:00
|
|
|
final Item item = arguments[0].head();
|
|
|
|
if (item == null) {
|
|
|
|
return new StringValue("");
|
2020-02-13 16:49:45 +01:00
|
|
|
}
|
2020-04-27 14:45:40 +02:00
|
|
|
return new StringValue(_year(item.getStringValue()));
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
@Override
|
|
|
|
public int getMinimumNumberOfArguments() {
|
|
|
|
return 0;
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
@Override
|
|
|
|
public int getMaximumNumberOfArguments() {
|
|
|
|
return 1;
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
@Override
|
|
|
|
public SequenceType[] getArgumentTypes() {
|
|
|
|
return new SequenceType[] {SequenceType.OPTIONAL_ITEM};
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public SequenceType getResultType(SequenceType[] suppliedArgumentTypes) {
|
|
|
|
return SequenceType.SINGLE_STRING;
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
|
2020-04-27 14:45:40 +02:00
|
|
|
private String _year(String s) {
|
|
|
|
Calendar c = new GregorianCalendar();
|
|
|
|
for (String format : dateFormats) {
|
|
|
|
try {
|
|
|
|
c.setTime(new SimpleDateFormat(format).parse(s));
|
|
|
|
String year = String.valueOf(c.get(Calendar.YEAR));
|
|
|
|
return year;
|
|
|
|
} catch (ParseException e) {
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
}
|
2020-04-27 14:45:40 +02:00
|
|
|
return "";
|
|
|
|
}
|
2020-02-13 16:49:45 +01:00
|
|
|
}
|