com.panet.imeta.core.playlist.FilePlayListReplay.java Source code

Introduction

Here is the source code for com.panet.imeta.core.playlist.FilePlayListReplay.java

Source

/*
 * Copyright (c) 2007 Pentaho Corporation.  All rights reserved. 
 * This software was developed by Pentaho Corporation and is provided under the terms 
 * of the GNU Lesser General Public License, Version 2.1. You may not use 
 * this file except in compliance with the license. If you need a copy of the license, 
 * please go to http://www.gnu.org/licenses/lgpl-2.1.txt. The Original Code is Pentaho 
 * Data Integration.  The Initial Developer is Pentaho Corporation.
 *
 * Software distributed under the GNU Lesser Public License is distributed on an "AS IS" 
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or  implied. Please refer to 
 * the license for the specific language governing your rights and limitations.
*/
package com.panet.imeta.core.playlist;

import java.io.IOException;
import java.util.Date;

import org.apache.commons.vfs.FileObject;

import com.panet.imeta.core.exception.KettleException;
import com.panet.imeta.trans.step.errorhandling.AbstractFileErrorHandler;

public class FilePlayListReplay implements FilePlayList {

    private final Date replayDate;

    private final String encoding;

    private final String lineNumberDirectory;

    private final String lineNumberExtension;

    private final String errorDirectory;

    private final String errorExtension;

    private FilePlayListReplayFile currentLineNumberFile;

    private FilePlayListReplayFile currentErrorFile;

    public FilePlayListReplay(Date replayDate, String lineNumberDirectory, String lineNumberExtension,
            String errorDirectory, String errorExtension, String encoding) {
        this.replayDate = replayDate;
        this.errorDirectory = errorDirectory;
        this.errorExtension = errorExtension;
        this.encoding = encoding;
        this.lineNumberDirectory = lineNumberDirectory;
        this.lineNumberExtension = lineNumberExtension;

    }

    private FileObject getCurrentProcessingFile() {
        FileObject result = null;
        if (currentLineNumberFile != null)
            result = currentLineNumberFile.getProcessingFile();
        return result;
    }

    private String getCurrentProcessingFilePart() {
        String result = null;
        if (currentLineNumberFile != null)
            result = currentLineNumberFile.getProcessingFilePart();
        return result;
    }

    public boolean isProcessingNeeded(FileObject file, long lineNr, String filePart) throws KettleException {
        initializeCurrentIfNeeded(file, filePart);
        return currentLineNumberFile.isProcessingNeeded(file, lineNr, filePart)
                || currentErrorFile.isProcessingNeeded(file, lineNr, filePart);
    }

    private void initializeCurrentIfNeeded(FileObject file, String filePart) throws KettleException {
        if (!(file.equals(getCurrentProcessingFile()) && filePart.equals(getCurrentProcessingFilePart())))
            initializeCurrent(file, filePart);
    }

    private void initializeCurrent(FileObject file, String filePart) throws KettleException {
        try {
            FileObject lineFile = AbstractFileErrorHandler.getReplayFilename(lineNumberDirectory,
                    file.getName().getBaseName(), replayDate, lineNumberExtension, filePart);
            if (lineFile.exists())
                currentLineNumberFile = new FilePlayListReplayLineNumberFile(lineFile, encoding, file, filePart);
            else
                currentLineNumberFile = new FilePlayListReplayFile(file, filePart);

            FileObject errorFile = AbstractFileErrorHandler.getReplayFilename(errorDirectory,
                    file.getName().getURI(), replayDate, errorExtension, AbstractFileErrorHandler.NO_PARTS);
            if (errorFile.exists())
                currentErrorFile = new FilePlayListReplayErrorFile(errorFile, file);
            else
                currentErrorFile = new FilePlayListReplayFile(file, AbstractFileErrorHandler.NO_PARTS);
        } catch (IOException e) {
            throw new KettleException(e);
        }
    }
}