public RequiredFieldResponse pushProjection(RequiredFieldList requiredFieldList)
throws FrontendException {
if (requiredFieldList == null) {
return null;
}
if (requiredFieldList.getFields() != null) {
int lastColumn = -1;
for (RequiredField rf : requiredFieldList.getFields()) {
if (rf.getIndex() > lastColumn) {
lastColumn = rf.getIndex();
}
}
requiredFields = new boolean[lastColumn + 1];
for (RequiredField rf : requiredFieldList.getFields()) {
if (rf.getIndex() != -1) requiredFields[rf.getIndex()] = true;
}
Properties p = UDFContext.getUDFContext().getUDFProperties(this.getClass());
try {
p.setProperty(REQUIRED_FIELDS_SIGNATURE, ObjectSerializer.serialize(requiredFields));
} catch (Exception e) {
throw new RuntimeException("Cannot serialize mRequiredColumns");
}
}
return new RequiredFieldResponse(true);
}
@Override
public RequiredFieldResponse pushProjection(RequiredFieldList requiredFieldList)
throws FrontendException {
// save the required field list to the UDFContext properties.
StringBuilder buff = new StringBuilder();
int i = 0;
for (RequiredField f : requiredFieldList.getFields()) {
if (i++ != 0) buff.append(',');
buff.append(f.getIndex());
}
Properties properties = getUDFContext();
properties.setProperty(PROJECTION_ID, buff.toString());
return new RequiredFieldResponse(true);
}
