log(Level.DEBUG, "updating %s to %s",source, dest);
Set badDs = new HashSet();
this.name = name;
}
<<<<<<< HEAD
public String getRrdName() {
String rrdName = getName().replaceAll("/","_");
return monitoredHost.getHostDir() +
Util.getFileSeparator() + rrdName + ".rrd";
}
private String parseTemplate(String template) {
Object[] arguments = {
"${host}",
"${index}",
"${url}",
"${port}",
"${index.signature}",
"${url.signature}"
};
return jrds.Util.parseOldTemplate(template, arguments, this);
}
protected DsDef[] getDsDefs() {
return getPd().getDsDefs();
}
public RrdDef getRrdDef() {
RrdDef def = new RrdDef(getRrdName());
def.setVersion(2);
def.addArchive(archives.getArchives());
def.addDatasource(getDsDefs());
def.setStep(getStep());
return def;
}
/**
* Create the probe file
* @throws IOException
*/
protected void create() throws IOException {
log(Level.INFO, "Need to create rrd");
RrdDef def = getRrdDef();
RrdDb rrdDb = new RrdDb(def);
rrdDb.close();
}
private void upgrade() {
RrdDb rrdSource = null;
try {
log(Level.WARN,"Definition is changed, the store needs to be upgraded");
File source = new File(getRrdName());
rrdSource = new RrdDb(source.getCanonicalPath());
RrdDef rrdDef = getRrdDef();
File dest = File.createTempFile("JRDS_", ".tmp", source.getParentFile());
rrdDef.setPath(dest.getCanonicalPath());
RrdDb rrdDest = new RrdDb(rrdDef);
Header header = rrdSource.getHeader();
int dsCount = header.getDsCount();
header.copyStateTo(rrdDest.getHeader());
for (int i = 0; i < dsCount; i++) {
Datasource srcDs = rrdSource.getDatasource(i);
String dsName = srcDs.getName();
Datasource dstDS = rrdDest.getDatasource(dsName);
if (dstDS != null ) {
try {
srcDs.copyStateTo(dstDS);
log(Level.TRACE, "Update %s", dsName);
} catch (RuntimeException e) {
badDs.add(dsName);
log(Level.ERROR, e, "Datasource %s can't be upgraded: %s", dsName, e.getMessage());
}
}
}
int robinMigrated = 0;
for (int i = 0; i < rrdSource.getArcCount(); i++) {
Archive srcArchive = rrdSource.getArchive(i);
ConsolFun consolFun = srcArchive.getConsolFun();
int steps = srcArchive.getSteps();
Archive dstArchive = rrdDest.getArchive(consolFun, steps);
if (dstArchive != null) {
if ( dstArchive.getConsolFun().equals(srcArchive.getConsolFun()) &&
dstArchive.getSteps() == srcArchive.getSteps() ) {
for (int k = 0; k < dsCount; k++) {
Datasource srcDs = rrdSource.getDatasource(k);
String dsName = srcDs.getName();
try {
int j = rrdDest.getDsIndex(dsName);
if (j >= 0 && ! badDs.contains(dsName)) {
log(Level.TRACE, "Upgrade of %s from %s", dsName, srcArchive);
srcArchive.getArcState(k).copyStateTo(dstArchive.getArcState(j));
srcArchive.getRobin(k).copyStateTo(dstArchive.getRobin(j));
robinMigrated++;
}
}
catch (IllegalArgumentException e) {
log(Level.TRACE, "Datastore %s removed", dsName);
}
}
log(Level.TRACE, "Update %s", srcArchive);
}
}
}
log(Level.DEBUG, "Robin migrated: %s", robinMigrated);
rrdDest.close();
rrdSource.close();
log(Level.DEBUG, "Size difference : %d", (dest.length() - source.length()));
copyFile(dest.getCanonicalPath(), source.getCanonicalPath());
} catch (IOException e) {
log(Level.ERROR, e, "Upgrade failed: %s", e);
}
finally {
if(rrdSource != null)
try {
rrdSource.close();
} catch (IOException e) {
}
}
}
private static void copyFile(String sourcePath, String destPath)
throws IOException {
File source = new File(sourcePath);
File dest = new File(destPath);
File destOld = new File(destPath + ".old");
if (!dest.renameTo(destOld)) {
throw new IOException("Could not rename file " + destPath + " from " + destOld);
}
if (!source.renameTo(dest)) {
throw new IOException("Could not rename file " + destPath + " from " + sourcePath);
}
deleteFile(destOld);
}
private static void deleteFile(File file) throws IOException {
if (file.exists() && !file.delete()) {
throw new IOException("Could not delete file: " + file.getCanonicalPath());
}
}
/**
* Check the final status of the probe. It must be called once before an probe can be used
*
* Open the rrd backend of the probe.
* it's created if it's needed
*/
public boolean checkStore() {
if(pd == null) {
log(Level.ERROR, "Missing Probe description");
return false;
}
if(monitoredHost == null) {
log(Level.ERROR, "Missing host");
return false;
}
//Name can be set by other means
if(name == null)
name = parseTemplate(getPd().getProbeName());
finished = checkStoreFile();
return finished;
}
protected boolean checkStoreFile() {
File rrdFile = new File(getRrdName());
File rrdDir = monitoredHost.getHostDir();
if (!rrdDir.isDirectory()) {
if( ! rrdDir.mkdir()) {
try {
log(Level.ERROR, "probe dir %s creation failed ", rrdDir.getCanonicalPath());
} catch (IOException e) {
}
return false;
}
}
boolean retValue = false;
RrdDb rrdDb = null;
try {
if ( rrdFile.isFile() ) {
rrdDb = new RrdDb(getRrdName());
//old definition
RrdDef tmpdef = rrdDb.getRrdDef();
Date startTime = new Date();
tmpdef.setStartTime(startTime);
String oldDef = tmpdef.dump();
long oldstep = tmpdef.getStep();
log(Level.TRACE, "Definition found: %s\n", oldDef);
//new definition
tmpdef = getRrdDef();
tmpdef.setStartTime(startTime);
String newDef = tmpdef.dump();
long newstep = tmpdef.getStep();
if(newstep != oldstep ) {
log(Level.ERROR, "step changed, you're in trouble" );
return false;
}
else if(! newDef.equals(oldDef)) {
rrdDb.close();
rrdDb = null;
upgrade();
rrdDb = new RrdDb(getRrdName());
}
log(Level.TRACE, "******");
} else
create();
retValue = true;
} catch (Exception e) {
log(Level.ERROR, e, "Store %s unusable: %s", getRrdName(), e);
}
finally {
if(rrdDb != null)
try {
rrdDb.close();
} catch (IOException e) {
}
}
return retValue;
}
=======
>>>>>>> 115986652ff4650747e059d9132fb2bd2e7cd025
/**
* The method that return a map of data collected.
* It should return return as raw as possible, they can even be opaque data tied to the probe. |