de.gwdg.metadataqa.marc.utils.pica.PicaPlainReader Maven / Gradle / Ivy

Go to download
package de.gwdg.metadataqa.marc.utils.pica;

import de.gwdg.metadataqa.marc.MarcFactory;
import org.apache.commons.lang3.StringUtils;
import org.marc4j.marc.Record;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

public class PicaPlainReader extends PicaReader {

  private static final Logger logger = Logger.getLogger(PicaPlainReader.class.getCanonicalName());

  private BufferedReader bufferedReader = null;
  private String line = null;
  private int lineNumber = 0;
  private List lines = new ArrayList<>();
  private String defauultSubfieldSeparator = "$";

  public PicaPlainReader(String fileName) {
    try {
      bufferedReader = new BufferedReader(new FileReader(fileName));
    } catch (IOException e) {
      logger.log(Level.WARNING, "error in PicaReader()", e);
    }
  }

  public PicaPlainReader(InputStream stream, String encoding) {
    try {
      bufferedReader = new BufferedReader(new InputStreamReader(stream, encoding));
    } catch (IOException e) {
      logger.log(Level.WARNING, "error in PicaReader()", e);
    }
  }

  @Override
  public boolean hasNext() {
    if (lineNumber == 0)
      checkSubfieldSeparator();

    if (lineNumber == 0) {
      try {
        line = bufferedReader.readLine();
      } catch (IOException e) {
        logger.log(Level.WARNING, "error in hasNext()", e);
      }
      lineNumber++;
    }
    return (line != null);
  }

  @Override
  public Record next() {
    Record marc4jRecord = null;
    boolean finished = false;
    while (line != null && !finished) {
      PicaLine picaLine = new PicaLine(line, subfieldSeparator);
      if (picaLine.isSkippable() && !lines.isEmpty()) {
        if (!parsed && StringUtils.isNotEmpty(idField) && StringUtils.isNotBlank(idField))
          parseIdField();
        marc4jRecord = MarcFactory.createRecordFromPica(lines, idTag, idCode, schema);
        finished = true;
        lines = new ArrayList<>();
      }

      if (picaLine.isValidTag()) {
        lines.add(picaLine);
      }

      try {
        line = bufferedReader.readLine();
        lineNumber++;
      } catch (IOException e) {
        logger.log(Level.SEVERE, "next", e);
      }
    } // while

    if (line == null && !lines.isEmpty()) {
      marc4jRecord = MarcFactory.createRecordFromPica(lines, idTag, idCode, schema);
    }
    return marc4jRecord;
  }

  private void checkSubfieldSeparator() {
    if (subfieldSeparator == null)
      subfieldSeparator = defauultSubfieldSeparator;
  }
}