public boolean init(StepMetaInterface smi, StepDataInterface sdi) { meta = (JsonInputMeta) smi; data = (JsonInputData) sdi; if (super.init(smi, sdi)) { data.rownr = 1L; data.nrInputFields = meta.getInputFields().length; // Take care of variable substitution for (int i = 0; i < data.nrInputFields; i++) { JsonInputField field = meta.getInputFields()[i]; field.setPath(environmentSubstitute(field.getPath())); } try { // Init a new JSON reader data.jsonReader = new JsonReader(); data.jsonReader.SetIgnoreMissingPath(meta.isIgnoreMissingPath()); } catch (KettleException e) { logError(e.getMessage()); return false; } return true; } return false; }
private void parseJson() throws Exception { // Read JSON source if (data.file != null) { data.jsonReader.readFile(data.filename); } else { if (meta.isReadUrl()) { data.jsonReader.readUrl(data.stringToParse); } else { // read string data.jsonReader.readString(data.stringToParse); } } List<NJSONArray> resultList = new ArrayList<NJSONArray>(); data.nrrecords = -1; data.recordnr = 0; String prevPath = ""; for (int i = 0; i < data.nrInputFields; i++) { String path = meta.getInputFields()[i].getPath(); NJSONArray ja = data.jsonReader.getPath(path); if (data.nrrecords != -1 && data.nrrecords != ja.size() && !ja.isNull()) { throw new KettleException( BaseMessages.getString( PKG, "JsonInput.Error.BadStructure", ja.size(), path, prevPath, data.nrrecords)); } resultList.add(ja); if (data.nrrecords == -1 && !ja.isNull()) { data.nrrecords = ja.size(); } prevPath = path; } data.resultList = new ArrayList<NJSONArray>(); Iterator<NJSONArray> it = resultList.iterator(); while (it.hasNext()) { NJSONArray j = it.next(); if (j.isNull()) { if (data.nrrecords == -1) { data.nrrecords = 1; } // The object is empty means that we do not // find Json path // We need here to create a dummy structure j = new NJSONArray(); for (int i = 0; i < data.nrrecords; i++) { j.add(null); } } data.resultList.add(j); } resultList = null; if (log.isDetailed()) { logDetailed(BaseMessages.getString(PKG, "JsonInput.Log.NrRecords", data.nrrecords)); } }
private boolean openNextFile() { try { if (data.filenr >= data.files.nrOfFiles()) { if (log.isDetailed()) logDetailed(BaseMessages.getString(PKG, "JsonInput.Log.FinishedProcessing")); return false; } // Close previous file if needed if (data.file != null) data.file.close(); // get file data.file = (FileObject) data.files.getFile(data.filenr); if (meta.isIgnoreEmptyFile() && data.file.getContent().getSize() == 0) { // log only basic as a warning (was before logError) logBasic( BaseMessages.getString(PKG, "JsonInput.Error.FileSizeZero", "" + data.file.getName())); openNextFile(); } readFileOrString(); } catch (Exception e) { logError( BaseMessages.getString( PKG, "JsonInput.Log.UnableToOpenFile", "" + data.filenr, data.file.toString(), e.toString())); stopAll(); setErrors(1); return false; } return true; }
public void dispose(StepMetaInterface smi, StepDataInterface sdi) { meta = (JsonInputMeta) smi; data = (JsonInputData) sdi; if (data.file != null) { try { data.file.close(); } catch (Exception e) { } } data.resultList = null; super.dispose(smi, sdi); }
private void readFileOrString() throws Exception { if (data.file != null) { data.filename = KettleVFS.getFilename(data.file); // Add additional fields? if (meta.getShortFileNameField() != null && meta.getShortFileNameField().length() > 0) { data.shortFilename = data.file.getName().getBaseName(); } if (meta.getPathField() != null && meta.getPathField().length() > 0) { data.path = KettleVFS.getFilename(data.file.getParent()); } if (meta.isHiddenField() != null && meta.isHiddenField().length() > 0) { data.hidden = data.file.isHidden(); } if (meta.getExtensionField() != null && meta.getExtensionField().length() > 0) { data.extension = data.file.getName().getExtension(); } if (meta.getLastModificationDateField() != null && meta.getLastModificationDateField().length() > 0) { data.lastModificationDateTime = new Date(data.file.getContent().getLastModifiedTime()); } if (meta.getUriField() != null && meta.getUriField().length() > 0) { data.uriName = data.file.getName().getURI(); } if (meta.getRootUriField() != null && meta.getRootUriField().length() > 0) { data.rootUriName = data.file.getName().getRootURI(); } // Check if file is empty long fileSize = data.file.getContent().getSize(); if (meta.getSizeField() != null && meta.getSizeField().length() > 0) { data.size = fileSize; } // Move file pointer ahead! data.filenr++; if (log.isDetailed()) logDetailed(BaseMessages.getString(PKG, "JsonInput.Log.OpeningFile", data.file.toString())); addFileToResultFilesname(data.file); } parseJson(); }
private Object[] buildRow() throws KettleException { // Create new row... Object[] outputRowData = null; if (data.readrow != null) outputRowData = data.readrow.clone(); else outputRowData = buildEmptyRow(); // Read fields... for (int i = 0; i < data.nrInputFields; i++) { // Get field JsonInputField field = meta.getInputFields()[i]; // get json array for field JSONArray jsona = data.resultList.get(i).getJSONArray(); String nodevalue = null; if (jsona != null) { Object jo = (Object) jsona.get(data.recordnr); if (jo != null) { nodevalue = jo.toString(); } } // Do trimming switch (field.getTrimType()) { case JsonInputField.TYPE_TRIM_LEFT: nodevalue = Const.ltrim(nodevalue); break; case JsonInputField.TYPE_TRIM_RIGHT: nodevalue = Const.rtrim(nodevalue); break; case JsonInputField.TYPE_TRIM_BOTH: nodevalue = Const.trim(nodevalue); break; default: break; } if (meta.isInFields()) { // Add result field to input stream outputRowData = RowDataUtil.addValueData(outputRowData, data.totalpreviousfields + i, nodevalue); } // Do conversions // ValueMetaInterface targetValueMeta = data.outputRowMeta.getValueMeta(data.totalpreviousfields + i); ValueMetaInterface sourceValueMeta = data.convertRowMeta.getValueMeta(data.totalpreviousfields + i); outputRowData[data.totalpreviousfields + i] = targetValueMeta.convertData(sourceValueMeta, nodevalue); // Do we need to repeat this field if it is null? if (meta.getInputFields()[i].isRepeated()) { if (data.previousRow != null && Const.isEmpty(nodevalue)) { outputRowData[data.totalpreviousfields + i] = data.previousRow[data.totalpreviousfields + i]; } } } // End of loop over fields... // When we have an input stream // the row index take care of previous fields int rowIndex = data.totalpreviousfields + data.nrInputFields; // See if we need to add the filename to the row... if (meta.includeFilename() && !Const.isEmpty(meta.getFilenameField())) { outputRowData[rowIndex++] = data.filename; } // See if we need to add the row number to the row... if (meta.includeRowNumber() && !Const.isEmpty(meta.getRowNumberField())) { outputRowData[rowIndex++] = new Long(data.rownr); } // Possibly add short filename... if (meta.getShortFileNameField() != null && meta.getShortFileNameField().length() > 0) { outputRowData[rowIndex++] = data.shortFilename; } // Add Extension if (meta.getExtensionField() != null && meta.getExtensionField().length() > 0) { outputRowData[rowIndex++] = data.extension; } // add path if (meta.getPathField() != null && meta.getPathField().length() > 0) { outputRowData[rowIndex++] = data.path; } // Add Size if (meta.getSizeField() != null && meta.getSizeField().length() > 0) { outputRowData[rowIndex++] = new Long(data.size); } // add Hidden if (meta.isHiddenField() != null && meta.isHiddenField().length() > 0) { outputRowData[rowIndex++] = new Boolean(data.path); } // Add modification date if (meta.getLastModificationDateField() != null && meta.getLastModificationDateField().length() > 0) { outputRowData[rowIndex++] = data.lastModificationDateTime; } // Add Uri if (meta.getUriField() != null && meta.getUriField().length() > 0) { outputRowData[rowIndex++] = data.uriName; } // Add RootUri if (meta.getRootUriField() != null && meta.getRootUriField().length() > 0) { outputRowData[rowIndex++] = data.rootUriName; } data.recordnr++; RowMetaInterface irow = getInputRowMeta(); data.previousRow = irow == null ? outputRowData : (Object[]) irow.cloneRow(outputRowData); // copy it to make // surely the next step doesn't change it in between... return outputRowData; }
public boolean processRow(StepMetaInterface smi, StepDataInterface sdi) throws KettleException { if (first && !meta.isInFields()) { first = false; data.files = meta.getFiles(this); if (!meta.isdoNotFailIfNoFile() && data.files.nrOfFiles() == 0) { throw new KettleException(BaseMessages.getString(PKG, "JsonInput.Log.NoFiles")); } handleMissingFiles(); // Create the output row meta-data data.outputRowMeta = new RowMeta(); meta.getFields(data.outputRowMeta, getStepname(), null, null, this); // Create convert meta-data objects that will contain Date & Number formatters data.convertRowMeta = data.outputRowMeta.clone(); for (int i = 0; i < data.convertRowMeta.size(); i++) data.convertRowMeta.getValueMeta(i).setType(ValueMetaInterface.TYPE_STRING); // For String to <type> conversions, we allocate a conversion meta data row as well... // data.convertRowMeta = data.outputRowMeta.clone(); for (int i = 0; i < data.convertRowMeta.size(); i++) { data.convertRowMeta.getValueMeta(i).setType(ValueMetaInterface.TYPE_STRING); } } Object[] r = null; try { // Grab a row r = getOneRow(); if (r == null) { setOutputDone(); // signal end to receiver(s) return false; // end of data or error. } if (log.isRowLevel()) logRowlevel( BaseMessages.getString(PKG, "JsonInput.Log.ReadRow", data.outputRowMeta.getString(r))); incrementLinesInput(); data.rownr++; putRow(data.outputRowMeta, r); // copy row to output rowset(s); if (meta.getRowLimit() > 0 && data.rownr > meta.getRowLimit()) { // limit has been reached: stop now. setOutputDone(); return false; } } catch (Exception e) { boolean sendToErrorRow = false; String errorMessage = null; if (getStepMeta().isDoingErrorHandling()) { sendToErrorRow = true; errorMessage = e.toString(); } else { logError( BaseMessages.getString( PKG, "JsonInput.ErrorInStepRunning", e.getMessage())); // $NON-NLS-1$ setErrors(1); stopAll(); setOutputDone(); // signal end to receiver(s) return false; } if (sendToErrorRow) { // Simply add this row to the error row putError(getInputRowMeta(), r, 1, errorMessage, null, "JsonInput001"); } } return true; }
private boolean ReadNextString() { try { data.readrow = getRow(); // Grab another row ... if (data.readrow == null) { // finished processing! if (log.isDetailed()) logDetailed(BaseMessages.getString(PKG, "JsonInput.Log.FinishedProcessing")); return false; } if (first) { first = false; data.inputRowMeta = getInputRowMeta(); data.outputRowMeta = data.inputRowMeta.clone(); meta.getFields(data.outputRowMeta, getStepname(), null, null, this); // Get total previous fields data.totalpreviousfields = data.inputRowMeta.size(); // Create convert meta-data objects that will contain Date & Number formatters data.convertRowMeta = data.outputRowMeta.clone(); for (int i = 0; i < data.convertRowMeta.size(); i++) data.convertRowMeta.getValueMeta(i).setType(ValueMetaInterface.TYPE_STRING); // For String to <type> conversions, we allocate a conversion meta data row as well... // data.convertRowMeta = data.outputRowMeta.clone(); for (int i = 0; i < data.convertRowMeta.size(); i++) { data.convertRowMeta.getValueMeta(i).setType(ValueMetaInterface.TYPE_STRING); } // Check if source field is provided if (Const.isEmpty(meta.getFieldValue())) { logError(BaseMessages.getString(PKG, "JsonInput.Log.NoField")); throw new KettleException(BaseMessages.getString(PKG, "JsonInput.Log.NoField")); } // cache the position of the field if (data.indexSourceField < 0) { data.indexSourceField = getInputRowMeta().indexOfValue(meta.getFieldValue()); if (data.indexSourceField < 0) { // The field is unreachable ! logError( BaseMessages.getString( PKG, "JsonInput.Log.ErrorFindingField", meta.getFieldValue())); // $NON-NLS-1$ //$NON-NLS-2$ throw new KettleException( BaseMessages.getString( PKG, "JsonInput.Exception.CouldnotFindField", meta.getFieldValue())); // $NON-NLS-1$ //$NON-NLS-2$ } } } // get source field value String fieldValue = getInputRowMeta().getString(data.readrow, data.indexSourceField); if (log.isDetailed()) logDetailed( BaseMessages.getString( PKG, "JsonInput.Log.SourceValue", meta.getFieldValue(), fieldValue)); if (meta.getIsAFile()) { // source is a file. data.file = KettleVFS.getFileObject(fieldValue, getTransMeta()); if (meta.isIgnoreEmptyFile() && data.file.getContent().getSize() == 0) { // log only basic as a warning (was before logError) logBasic( BaseMessages.getString(PKG, "JsonInput.Error.FileSizeZero", data.file.getName())); ReadNextString(); } } else { // read string data.stringToParse = fieldValue; } readFileOrString(); } catch (Exception e) { logError(BaseMessages.getString(PKG, "JsonInput.Log.UnexpectedError", e.toString())); stopAll(); logError(Const.getStackTracker(e)); setErrors(1); return false; } return true; }