source: DCWoRMS/trunk/build/classes/simulator/workload/writer/xmlJob/QcgXmlJobWriter.java @ 477

package simulator.workload.writer.xmlJob;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.exolab.castor.xml.MarshalException;
import org.exolab.castor.xml.ValidationException;

import simulator.workload.fileFilters.QcgJobFileNameFilter;

import org.qcg.broker.schemas.jobdesc.QcgJob;

/**
 * 
 * @author Marcin Krystek
 *
 */

public class QcgXmlJobWriter implements XMLJobWriter<QcgJob>{

        private Log log = LogFactory.getLog(QcgXmlJobWriter.class);
        public boolean overwrite;
        public String fileNamePrefix;
        public File directory;
        
        public QcgXmlJobWriter(String dirName, String fileNamePrefix){
                this.overwrite = true;
                this.fileNamePrefix = fileNamePrefix;
                this.directory = new File(dirName);

                if(!directory.exists())
                        directory.mkdirs();
                
        }
        
        public boolean write(QcgJob job) throws IOException{
                String fileName = QcgJobFileNameFilter.FILE_NAME_PREFIX+"_"+job.getAppId()+".xml";
                
                File file = new File(directory, fileName);
                if(file.exists()) {
                        if(this.overwrite){
                                if(!file.delete()){
                                        if(log.isErrorEnabled())
                                                log.error("File " + fileName + " can not be deleted.");
                                        return false;
                                }
                                if(!file.createNewFile()){
                                        if(log.isErrorEnabled())
                                                log.error("File " + fileName + " can not be created.");
                                        return false;
                                }
                        } else {
                                throw new IOException("File "+fileName + " exists and can not be removed. Set overwrite attribute value to true.");
                        }
                }
                
                FileWriter writer = new FileWriter(file);
                
                try {
                
                        job.marshal(writer);
        
                } catch (MarshalException e) {
                        throw new IOException(e.getMessage());
                } catch (ValidationException e) {
                        throw new IOException(e.getMessage());
                } finally {
                        writer.close();
                }
                
                
                return true;
        }

        public void setOverwrite(boolean overwrite) {
                this.overwrite = overwrite;
        }

        public void close() throws IOException{};
}