93 lines
3.3 KiB
Java
93 lines
3.3 KiB
Java
|
package inr.numass.control;
|
||
|
|
|
||
|
|
import hep.dataforge.context.Context;
|
||
|
|
import hep.dataforge.context.Global;
|
||
|
|
import hep.dataforge.control.connections.Roles;
|
||
|
|
import hep.dataforge.control.connections.StorageConnection;
|
||
|
|
import hep.dataforge.control.devices.Device;
|
||
|
|
import hep.dataforge.exceptions.StorageException;
|
||
|
|
import hep.dataforge.io.MetaFileReader;
|
||
|
|
import hep.dataforge.io.XMLMetaReader;
|
||
|
|
import hep.dataforge.meta.Meta;
|
||
|
|
import hep.dataforge.storage.api.Storage;
|
||
|
|
import hep.dataforge.storage.commons.StorageFactory;
|
||
|
|
import hep.dataforge.storage.commons.StorageManager;
|
||
|
|
import inr.numass.client.ClientUtils;
|
||
|
|
import javafx.application.Application;
|
||
|
|
|
||
|
|
import java.io.File;
|
||
|
|
import java.io.IOException;
|
||
|
|
import java.text.ParseException;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.function.Predicate;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Created by darksnake on 08-May-17.
|
||
|
|
*/
|
||
|
|
public class NumassControlUtils {
|
||
|
|
public static final String DEFAULT_CONFIG_LOCATION = "numass-control.xml";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Create a single or multiple storage connections for a device
|
||
|
|
*
|
||
|
|
* @param device
|
||
|
|
* @param config
|
||
|
|
*/
|
||
|
|
public static void connectStorage(Device device, Meta config) {
|
||
|
|
//TODO add on reset listener
|
||
|
|
if (config.hasMeta("storage")&& device.acceptsRole("storge")) {
|
||
|
|
String numassRun = ClientUtils.getRunName(config);
|
||
|
|
config.getMetaList("storage").forEach(node -> {
|
||
|
|
Storage storage = StorageFactory.buildStorage(device.getContext(), node);
|
||
|
|
if (!numassRun.isEmpty()) {
|
||
|
|
try {
|
||
|
|
storage = storage.buildShelf(numassRun, Meta.empty());
|
||
|
|
} catch (StorageException e) {
|
||
|
|
device.getContext().getLogger().error("Failed to build shelf", e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
device.connect(new StorageConnection(storage), Roles.STORAGE_ROLE);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Meta readResourceMeta(String path) {
|
||
|
|
try {
|
||
|
|
return new XMLMetaReader().read(NumassControlUtils.class.getResourceAsStream(path));
|
||
|
|
} catch (IOException | ParseException e) {
|
||
|
|
throw new RuntimeException(e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Optional<Meta> getConfig(Application app) {
|
||
|
|
String configFileName = app.getParameters().getNamed().get("config");
|
||
|
|
if (configFileName == null) {
|
||
|
|
configFileName = DEFAULT_CONFIG_LOCATION;
|
||
|
|
}
|
||
|
|
File configFile = new File(configFileName);
|
||
|
|
|
||
|
|
if (configFile.exists()) {
|
||
|
|
try {
|
||
|
|
Meta config = MetaFileReader.read(configFile).build();
|
||
|
|
return Optional.of(config);
|
||
|
|
} catch (IOException | ParseException e) {
|
||
|
|
throw new RuntimeException(e);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
return Optional.empty();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
public static Optional<Meta> findDeviceMeta(Meta config, Predicate<Meta> criterion) {
|
||
|
|
return config.getMetaList("device").stream().filter(criterion).findFirst().map(it -> it);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Context setupContext(Meta meta) {
|
||
|
|
Context ctx = Global.getContext("NUMASS-CONTROL");
|
||
|
|
ctx.pluginManager().getOrLoad(StorageManager.class);
|
||
|
|
return ctx;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|