heartbeatfix.java

jrobin,使用纯java实现的RRD数据库,使用RRD数据库来统计数据.

package org.jrobin.demo;

import org.jrobin.core.RrdToolkit;
import org.jrobin.core.RrdException;
import org.jrobin.core.RrdDb;

import java.io.FileFilter;
import java.io.File;
import java.io.IOException;

public class HeartbeatFix {
	public static void main(String[] args) throws IOException, RrdException {
		if (args.length != 2) {
			System.err.println("SYNTAX: HeartbeatFix <rrd directory> <heartbeat>");
			System.exit(-1);
		}
		File directory = new File(args[0]);
		long heartbeat = Long.parseLong(args[1]);
		File[] files = directory.listFiles(new FileFilter() {
			public boolean accept(File pathname) {
				return pathname.isFile() && pathname.getName().endsWith(".rrd.jrb");
			}
		});
		System.out.println(files.length + " files found");
		for (int i = 0; i < files.length; i++) {
			String path = files[i].getAbsolutePath();
			System.out.print((i + 1) + ": " + path + ": ");
			// fix heartbeat
			RrdToolkit.setDsHeartbeat(files[i].getAbsolutePath(), 0, heartbeat);
			RrdToolkit.setDsHeartbeat(files[i].getAbsolutePath(), 1, heartbeat);
			System.out.print("fixed");
			// check consistency of the file
			RrdDb rrd = new RrdDb(path);
			if (rrd.getRrdDef().getEstimatedSize() == files[i].length() &&
					rrd.getDatasource(0).getHeartbeat() == heartbeat &&
					rrd.getDatasource(1).getHeartbeat() == heartbeat) {
				System.out.println(", verified");
			}
			else {
				System.out.println(", ********** ERROR **********");
			}
			rrd.close();
		}
		System.out.println("FINISHED!");
	}
}