rrdtoolkit.java

httptunnel.jar httptunnel java 源码

		if (Util.sameFilePath(sourcePath, destPath)) {
			throw new RrdException("Source and destination paths are the same");
		}
		RrdDb rrdSource = new RrdDb(sourcePath);
		RrdDef rrdDef = rrdSource.getRrdDef();
		rrdDef.setPath(destPath);
		rrdDef.removeArchive(consolFun, steps);
		RrdDb rrdDest = new RrdDb(rrdDef);
		rrdSource.copyStateTo(rrdDest);
		rrdSource.close();
		rrdDest.close();
	}

	/**
	 * <p>Removes one archive from a RRD file.</p>
	 * <p>WARNING: This method is potentialy dangerous! It will modify your RRD file.
	 * It is highly recommended to preserve the original RRD file (<i>saveBackup</i>
	 * should be set to <code>true</code>). The backup file will be created in the same
	 * directory as the original one with <code>.bak</code> extension added to the
	 * original name.</p>
	 * <p>Before applying this method, be sure that the specified RRD file is not in use
	 * (not open)</p>
	 * @param sourcePath path to a RRD file to add datasource to.
	 * @param consolFun Consolidation function of Archive which should be removed
	 * @param steps Number of steps for Archive which should be removed
	 * @param saveBackup true, if backup of the original file should be created;
	 * false, otherwise
	 * @throws IOException Thrown in case of I/O error
	 * @throws RrdException Thrown in case of JRobin specific error
	 */
	public void removeArchive(String sourcePath, String consolFun, int steps,
							  boolean saveBackup) throws IOException, RrdException {
		String destPath = Util.getTmpFilename();
		removeArchive(sourcePath, destPath, consolFun, steps);
		copyFile(destPath, sourcePath, saveBackup);
	}

	private static void copyFile(String sourcePath, String destPath, boolean saveBackup)
		throws IOException {
		File source = new File(sourcePath);
		File dest = new File(destPath);
		if (saveBackup) {
			String backupPath = getBackupPath(destPath);
			File backup = new File(backupPath);
			deleteFile(backup);
			if (!dest.renameTo(backup)) {
				throw new IOException("Could not create backup file " + backupPath);
			}
		}
		deleteFile(dest);
		if (!source.renameTo(dest)) {
			throw new IOException("Could not create file " + destPath + " from " + sourcePath);
		}
	}

	private static String getBackupPath(String destPath) {
		String backupPath = destPath;
		do {
			backupPath += ".bak";
		} while(new File(backupPath).exists());
		return backupPath;
	}

	/**
	 * Sets datasource heartbeat to a new value.
	 * @param sourcePath Path to exisiting RRD file (will be updated)
	 * @param datasourceName Name of the datasource in the specified RRD file
	 * @param newHeartbeat New datasource heartbeat
	 * @throws RrdException Thrown in case of JRobin specific error
	 * @throws IOException Thrown in case of I/O error
	 */
	public void setDsHeartbeat(String sourcePath, String datasourceName,
							   long newHeartbeat) throws RrdException, IOException {
		RrdDb rrd = new RrdDb(sourcePath);
		Datasource ds = rrd.getDatasource(datasourceName);
		ds.setHeartbeat(newHeartbeat);
		rrd.close();
	}

	/**
	 * Sets datasource min value to a new value
	 * @param sourcePath Path to exisiting RRD file (will be updated)
	 * @param datasourceName Name of the datasource in the specified RRD file
	 * @param newMinValue New min value for the datasource
	 * @param filterArchivedValues set to <code>true</code> if archived values less than
	 * <code>newMinValue</code> should be set to NaN; set to false, otherwise.
	 * @throws RrdException Thrown in case of JRobin specific error
	 * @throws IOException Thrown in case of I/O error
	 */
	public void setDsMinValue(String sourcePath, String datasourceName,
							  double newMinValue, boolean filterArchivedValues) throws RrdException, IOException {
		RrdDb rrd = new RrdDb(sourcePath);
		Datasource ds = rrd.getDatasource(datasourceName);
		ds.setMinValue(newMinValue, filterArchivedValues);
		rrd.close();
	}

	/**
	 * Sets datasource max value to a new value.
	 * @param sourcePath Path to exisiting RRD file (will be updated)
	 * @param datasourceName Name of the datasource in the specified RRD file
	 * @param newMaxValue New max value for the datasource
	 * @param filterArchivedValues set to <code>true</code> if archived values greater than
	 * <code>newMaxValue</code> should be set to NaN; set to false, otherwise.
	 * @throws RrdException Thrown in case of JRobin specific error
	 * @throws IOException Thrown in case of I/O error
	 */
	public void setDsMaxValue(String sourcePath, String datasourceName,
							  double newMaxValue, boolean filterArchivedValues) throws RrdException, IOException {
		RrdDb rrd = new RrdDb(sourcePath);
		Datasource ds = rrd.getDatasource(datasourceName);
		ds.setMaxValue(newMaxValue, filterArchivedValues);
		rrd.close();
	}

	/**
	 * Updates valid value range for the given datasource.
	 * @param sourcePath Path to exisiting RRD file (will be updated)
	 * @param datasourceName Name of the datasource in the specified RRD file
	 * @param newMinValue New min value for the datasource
	 * @param newMaxValue New max value for the datasource
	 * @param filterArchivedValues set to <code>true</code> if archived values outside
	 * of the specified min/max range should be replaced with NaNs.
	 * @throws RrdException Thrown in case of JRobin specific error
	 * @throws IOException Thrown in case of I/O error
	 */
	public void setDsMinMaxValue(String sourcePath, String datasourceName,
								 double newMinValue, double newMaxValue, boolean filterArchivedValues)
		throws RrdException, IOException {
		RrdDb rrd = new RrdDb(sourcePath);
		Datasource ds = rrd.getDatasource(datasourceName);
		ds.setMinMaxValue(newMinValue, newMaxValue, filterArchivedValues);
		rrd.close();
	}

	/**
	 * Sets single archive's X-files factor to a new value.
	 * @param sourcePath Path to existing RRD file (will be updated)
	 * @param consolFun Consolidation function of the target archive
	 * @param steps Number of sptes of the target archive
	 * @param newXff New X-files factor for the target archive
	 * @throws RrdException Thrown in case of JRobin specific error
	 * @throws IOException Thrown in case of I/O error
	 */
	public void setArcXff(String sourcePath, String consolFun, int steps,
						  double newXff) throws RrdException, IOException {
		RrdDb rrd = new RrdDb(sourcePath);
		Archive arc = rrd.getArchive(consolFun, steps);
		arc.setXff(newXff);
		rrd.close();
	}

	/**
	 * Creates new RRD file based on the existing one, but with a different
	 * size (number of rows) for a single archive. The archive to be resized
	 * is identified by its consolidation function and the number of steps.
	 * @param sourcePath Path to the source RRD file (will not be modified)
	 * @param destPath Path to the new RRD file (will be created)
	 * @param consolFun Consolidation function of the archive to be resized
	 * @param numSteps Number of steps of the archive to be resized
	 * @param newRows New archive size (number of archive rows)
	 * @throws IOException Thrown in case of I/O error
	 * @throws RrdException Thrown in case of JRobin specific error
	 */
	public void resizeArchive(String sourcePath, String destPath, String consolFun,
							  int numSteps, int newRows)
		throws IOException, RrdException {
		if (Util.sameFilePath(sourcePath, destPath)) {
			throw new RrdException("Source and destination paths are the same");
		}
		if (newRows < 2) {
			throw new RrdException("New arcihve size must be at least 2");
		}
		RrdDb rrdSource = new RrdDb(sourcePath);
		RrdDef rrdDef = rrdSource.getRrdDef();
		ArcDef arcDef = rrdDef.findArchive(consolFun, numSteps);
		if (arcDef.getRows() != newRows) {
			arcDef.setRows(newRows);
			rrdDef.setPath(destPath);
			RrdDb rrdDest = new RrdDb(rrdDef);
			rrdSource.copyStateTo(rrdDest);
			rrdDest.close();
		}
		rrdSource.close();
	}

	/**
	 * Modifies existing RRD file, by resizing its chosen archive. The archive to be resized
	 * is identified by its consolidation function and the number of steps.
	 * @param sourcePath Path to the RRD file (will be modified)
	 * @param consolFun Consolidation function of the archive to be resized
	 * @param numSteps Number of steps of the archive to be resized
	 * @param newRows New archive size (number of archive rows)
	 * @param saveBackup true, if backup of the original file should be created;
	 * false, otherwise
	 * @throws IOException Thrown in case of I/O error
	 * @throws RrdException Thrown in case of JRobin specific error
	 */
	public void resizeArchive(String sourcePath, String consolFun,
							  int numSteps, int newRows, boolean saveBackup)
		throws IOException, RrdException {
		String destPath = Util.getTmpFilename();
		resizeArchive(sourcePath, destPath, consolFun, numSteps, newRows);
		copyFile(destPath, sourcePath, saveBackup);
	}

	private static void deleteFile(File file) throws IOException {
		if (file.exists() && !file.delete()) {
			throw new IOException("Could not delete file: " + file.getCanonicalPath());
		}
	}
}