coveragedatafilehandler.java

java覆盖率测试工具

/*
 * Cobertura - http://cobertura.sourceforge.net/
 *
 * Copyright (C) 2003 jcoverage ltd.
 * Copyright (C) 2005 Mark Doliner
 * Copyright (C) 2007 Joakim Erdfelt
 *
 * Cobertura is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published
 * by the Free Software Foundation; either version 2 of the License,
 * or (at your option) any later version.
 *
 * Cobertura is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Cobertura; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

package net.sourceforge.cobertura.coveragedata;

import net.sourceforge.cobertura.util.ConfigurationUtil;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;

/**
 * This contains methods used for reading and writing the
 * "cobertura.ser" file.
 */
public abstract class CoverageDataFileHandler implements HasBeenInstrumented
{
	private static File defaultFile = null;

	public static File getDefaultDataFile()
	{
		// return cached defaultFile
		if (defaultFile != null) 
		{
			return defaultFile;
		}

		// load and cache datafile configuration
		ConfigurationUtil config = new ConfigurationUtil();
		defaultFile = new File(config.getDatafile());
        
		return defaultFile;
	}

	public static ProjectData loadCoverageData(File dataFile)
	{
		InputStream is = null;

		//System.out.println("Cobertura: Loading coverage data from " + dataFile.getAbsolutePath());
		try
		{
			is = new FileInputStream(dataFile);
			return loadCoverageData(is);
		}
		catch (IOException e)
		{
			System.err.println("Cobertura: Error reading file "
					+ dataFile.getAbsolutePath() + ": "
					+ e.getLocalizedMessage());
			return null;
		}
		finally
		{
			if (is != null)
				try
				{
					is.close();
				}
				catch (IOException e)
				{
					System.err.println("Cobertura: Error closing file "
							+ dataFile.getAbsolutePath() + ": "
							+ e.getLocalizedMessage());
				}
		}
	}

	private static ProjectData loadCoverageData(InputStream dataFile) throws IOException
	{
		ObjectInputStream objects = null;

		try
		{
			objects = new ObjectInputStream(dataFile);
			ProjectData projectData = (ProjectData)objects.readObject();
			System.out.println("Cobertura: Loaded information on "
					+ projectData.getNumberOfClasses() + " classes.");
			return projectData;
		}
		catch (IOException e) {
			throw e;
		}
		catch (Exception e)
		{
			System.err.println("Cobertura: Error reading from object stream.");
			e.printStackTrace();
			return null;
		}
		finally
		{
			if (objects != null)
			{
				try
				{
					objects.close();
				}
				catch (IOException e)
				{
					System.err
							.println("Cobertura: Error closing object stream.");
					e.printStackTrace();
				}
			}
		}
	}

	public static void saveCoverageData(ProjectData projectData,
			File dataFile)
	{
		FileOutputStream os = null;

		//System.out.println("Cobertura: Saving coverage data to " + dataFile.getAbsolutePath());
		try
		{
			File dataDir = dataFile.getParentFile();
			if( (dataDir != null) && !dataDir.exists() )
			{
				dataDir.mkdirs();
			}
			os = new FileOutputStream(dataFile);
			saveCoverageData(projectData, os);
		}
		catch (IOException e)
		{
			System.err.println("Cobertura: Error writing file "
					+ dataFile.getAbsolutePath());
			e.printStackTrace();
		}
		finally
		{
			if (os != null)
			{
				try
				{
					os.close();
				}
				catch (IOException e)
				{
					System.err.println("Cobertura: Error closing file "
							+ dataFile.getAbsolutePath());
					e.printStackTrace();
				}
			}
		}
	}

	private static void saveCoverageData(ProjectData projectData,
			OutputStream dataFile)
	{
		ObjectOutputStream objects = null;
        
		try
		{
			objects = new ObjectOutputStream(dataFile);
			objects.writeObject(projectData);
			System.out.println("Cobertura: Saved information on " + projectData.getNumberOfClasses() + " classes.");
		}
		catch (IOException e)
		{
			System.err.println("Cobertura: Error writing to object stream.");
			e.printStackTrace();
		}
		finally
		{
			if (objects != null)
			{
				try
				{
					objects.close();
				}
				catch (IOException e)
				{
					System.err
							.println("Cobertura: Error closing object stream.");
					e.printStackTrace();
				}
			}
		}
	}

}