co.cask.hydrator.plugin.batch.CopybookIOUtils Maven / Gradle / Ivy
The newest version!
/*
* Copyright © 2016 Cask Data, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package co.cask.hydrator.plugin.batch;
import net.sf.JRecord.Common.CommonBits;
import net.sf.JRecord.Common.RecordException;
import net.sf.JRecord.Details.LayoutDetail;
import net.sf.JRecord.External.CobolCopybookLoader;
import net.sf.JRecord.External.CopybookLoader;
import net.sf.JRecord.External.Def.ExternalField;
import net.sf.JRecord.External.ExternalRecord;
import net.sf.JRecord.External.ToLayoutDetail;
import net.sf.JRecord.Numeric.Convert;
import net.sf.cb2xml.def.Cb2xmlConstants;
import java.io.IOException;
import java.io.InputStream;
/**
* Utility class to parse and read COBOL Copybook and binary data file contents.
*/
public class CopybookIOUtils {
public static final String FONT = "cp037";
/**
* Get the schema properties from the Copybook contents
*
* @param cblIs Input stream for COBOL Copybook contents
* @return ExternalRecord object defining the schema fields and their properties
* @throws RecordException
*/
public static ExternalRecord getExternalRecord(InputStream cblIs) throws RecordException {
CommonBits.setDefaultCobolTextFormat(Cb2xmlConstants.USE_STANDARD_COLUMNS);
CobolCopybookLoader copybookInt = new CobolCopybookLoader();
ExternalRecord record = copybookInt.loadCopyBook(cblIs, "", CopybookLoader.SPLIT_NONE, 0, FONT,
Convert.FMT_MAINFRAME, 0, null);
return record;
}
/**
* Get record length for each line
*
* @param externalRecord ExternalRecord object defining the schema fields and their properties
* @param fileStructure File structure of the data file
* @return the record length of each line
*/
public static int getRecordLength(ExternalRecord externalRecord, int fileStructure) {
int recordByteLength = 0;
for (ExternalField field : externalRecord.getRecordFields()) {
recordByteLength += field.getLen();
}
return recordByteLength;
}
/**
* Get the LayoutDetail object to read data
*
* @param externalRecord ExternalRecord object defining the schema fields and their properties
* @return
* @throws RecordException
* @throws IOException
*/
public static LayoutDetail getLayoutDetail(ExternalRecord externalRecord) throws RecordException, IOException {
LayoutDetail copybook = ToLayoutDetail.getInstance().getLayout(externalRecord);
return copybook;
}
}