Examples with VariableField org.marc4j.marc.VariableField used on opensource projects

Example 6 with VariableField

use of org.marc4j.marc.VariableField in project mod-source-record-storage by folio-org.

the class AdditionalFieldsUtil method addFieldToMarcRecord.

/**
 * Adds field if it does not exist and a subfield with a value to that field
 *
 * @param record   record that needs to be updated
 * @param field    field that should contain new subfield
 * @param subfield new subfield to add
 * @param value    value of the subfield to add
 * @return true if succeeded, false otherwise
 */
public static boolean addFieldToMarcRecord(Record record, String field, char subfield, String value) {
    boolean result = false;
    try (ByteArrayOutputStream os = new ByteArrayOutputStream()) {
        if (record != null && record.getParsedRecord() != null && record.getParsedRecord().getContent() != null) {
            MarcReader reader = buildMarcReader(record);
            MarcWriter streamWriter = new MarcStreamWriter(new ByteArrayOutputStream());
            MarcJsonWriter jsonWriter = new MarcJsonWriter(os);
            MarcFactory factory = MarcFactory.newInstance();
            if (reader.hasNext()) {
                org.marc4j.marc.Record marcRecord = reader.next();
                VariableField variableField = getSingleFieldByIndicators(marcRecord.getVariableFields(field), INDICATOR, INDICATOR);
                DataField dataField;
                if (variableField != null && ((DataField) variableField).getIndicator1() == INDICATOR && ((DataField) variableField).getIndicator2() == INDICATOR) {
                    dataField = (DataField) variableField;
                    marcRecord.removeVariableField(variableField);
                    dataField.removeSubfield(dataField.getSubfield(subfield));
                } else {
                    dataField = factory.newDataField(field, INDICATOR, INDICATOR);
                }
                dataField.addSubfield(factory.newSubfield(subfield, value));
                marcRecord.addVariableField(dataField);
                // use stream writer to recalculate leader
                streamWriter.write(marcRecord);
                jsonWriter.write(marcRecord);
                record.setParsedRecord(record.getParsedRecord().withContent(new JsonObject(os.toString()).encode()));
                result = true;
            }
        }
    } catch (Exception e) {
        LOGGER.error("Failed to add additional subfield {} for field {} to record {}", subfield, field, record.getId(), e);
    }
    return result;
}

Example 7 with VariableField

use of org.marc4j.marc.VariableField in project mod-source-record-storage by folio-org.

the class AdditionalFieldsUtil method removeField.

/**
 * remove field from marc record
 *
 * @param record record that needs to be updated
 * @param field  tag of the field
 * @return true if succeeded, false otherwise
 */
public static boolean removeField(Record record, String field) {
    boolean result = false;
    try (ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
        if (record != null && record.getParsedRecord() != null && record.getParsedRecord().getContent() != null) {
            MarcReader reader = buildMarcReader(record);
            MarcWriter marcStreamWriter = new MarcStreamWriter(new ByteArrayOutputStream());
            MarcJsonWriter marcJsonWriter = new MarcJsonWriter(baos);
            if (reader.hasNext()) {
                org.marc4j.marc.Record marcRecord = reader.next();
                VariableField variableField = marcRecord.getVariableField(field);
                if (variableField != null) {
                    marcRecord.removeVariableField(variableField);
                }
                // use stream writer to recalculate leader
                marcStreamWriter.write(marcRecord);
                marcJsonWriter.write(marcRecord);
                record.setParsedRecord(record.getParsedRecord().withContent(new JsonObject(baos.toString()).encode()));
                result = true;
            }
        }
    } catch (Exception e) {
        LOGGER.error("Failed to remove controlled field {) from record {}", field, record.getId(), e);
    }
    return result;
}