import static com.google.common.base.Preconditions.checkNotNull;
import ch.qos.logback.classic.Level;
-import com.google.common.base.Charsets;
import com.google.common.base.Preconditions;
import com.google.common.base.Stopwatch;
import com.google.common.collect.Lists;
import com.google.common.io.CharStreams;
import com.google.common.io.Files;
import java.io.File;
-import java.io.FileFilter;
+import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
+import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
-import org.opendaylight.controller.config.util.xml.XmlElement;
-import org.opendaylight.controller.config.util.xml.XmlUtil;
+import javax.xml.bind.JAXBException;
+import org.apache.karaf.features.internal.model.ConfigFile;
+import org.apache.karaf.features.internal.model.Feature;
+import org.apache.karaf.features.internal.model.Features;
+import org.apache.karaf.features.internal.model.JaxbUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-import org.w3c.dom.Document;
-import org.w3c.dom.Element;
-import org.w3c.dom.Node;
-import org.w3c.dom.NodeList;
-import org.xml.sax.SAXException;
public final class Main {
-
private static final Logger LOG = LoggerFactory.getLogger(Main.class);
+ @SuppressWarnings("checkstyle:IllegalCatch")
public static void main(final String[] args) {
final TesttoolParameters params = TesttoolParameters.parseArgs(args, TesttoolParameters.getParser());
params.validate();
- final ch.qos.logback.classic.Logger root = (ch.qos.logback.classic.Logger) LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME);
+ final ch.qos.logback.classic.Logger root = (ch.qos.logback.classic.Logger) LoggerFactory
+ .getLogger(Logger.ROOT_LOGGER_NAME);
root.setLevel(params.debug ? Level.DEBUG : Level.INFO);
- final NetconfDeviceSimulator netconfDeviceSimulator = new NetconfDeviceSimulator();
+ final NetconfDeviceSimulator netconfDeviceSimulator = new NetconfDeviceSimulator(params.threadPoolSize);
try {
+ LOG.debug("Trying to start netconf test-tool with parameters {}", params);
final List<Integer> openDevices = netconfDeviceSimulator.start(params);
if (openDevices.size() == 0) {
LOG.error("Failed to start any simulated devices, exiting...");
System.exit(1);
}
if (params.controllerDestination != null) {
- final ArrayList<ArrayList<Execution.DestToPayload>> allThreadsPayloads = params.getThreadsPayloads(openDevices);
+ final ArrayList<ArrayList<Execution.DestToPayload>> allThreadsPayloads = params
+ .getThreadsPayloads(openDevices);
final ArrayList<Execution> executions = new ArrayList<>();
for (ArrayList<Execution.DestToPayload> payloads : allThreadsPayloads) {
executions.add(new Execution(params, payloads));
final Stopwatch time = Stopwatch.createStarted();
List<Future<Void>> futures = executorService.invokeAll(executions, params.timeOut, TimeUnit.SECONDS);
int threadNum = 0;
- for(Future<Void> future : futures){
+ for (Future<Void> future : futures) {
threadNum++;
if (future.isCancelled()) {
LOG.info("{}. thread timed out.",threadNum);
private final File etcDir;
private final File loadOrderCfgFile;
- public ConfigGenerator(final File directory, final List<Integer> openDevices) {
+ ConfigGenerator(final File directory, final List<Integer> openDevices) {
this.configDir = new File(directory, ETC_OPENDAYLIGHT_KARAF_PATH);
this.etcDir = new File(directory, ETC_KARAF_PATH);
this.loadOrderCfgFile = new File(etcDir, ORG_OPS4J_PAX_URL_MVN_CFG);
public List<File> generate(final boolean useSsh, final int batchSize,
final int generateConfigsTimeout, final String address,
final int devicesPerPort) {
- if (configDir.exists() == false) {
+ if (!configDir.exists()) {
Preconditions.checkState(configDir.mkdirs(), "Unable to create directory " + configDir);
}
- for (final File file : configDir.listFiles(new FileFilter() {
- @Override
- public boolean accept(final File pathname) {
- return !pathname.isDirectory() && pathname.getName().startsWith(SIM_DEVICE_CFG_PREFIX);
- }
- })) {
+ for (final File file : configDir.listFiles(pathname ->
+ !pathname.isDirectory() && pathname.getName().startsWith(SIM_DEVICE_CFG_PREFIX))) {
Preconditions.checkState(file.delete(), "Unable to clean previous generated file %s", file);
}
try (InputStream stream = Main.class.getResourceAsStream(NETCONF_CONNECTOR_XML)) {
checkNotNull(stream, "Cannot load %s", NETCONF_CONNECTOR_XML);
- String configBlueprint = CharStreams.toString(new InputStreamReader(stream, Charsets.UTF_8));
+ String configBlueprint = CharStreams.toString(new InputStreamReader(stream, StandardCharsets.UTF_8));
final String before = configBlueprint.substring(0, configBlueprint.indexOf("<module>"));
- final String middleBlueprint = configBlueprint.substring(configBlueprint.indexOf("<module>"), configBlueprint.indexOf("</module>"));
- final String after = configBlueprint.substring(configBlueprint.indexOf("</module>") + "</module>".length());
+ final String middleBlueprint = configBlueprint.substring(
+ configBlueprint.indexOf("<module>"), configBlueprint.indexOf("</module>"));
+ final String after = configBlueprint.substring(
+ configBlueprint.indexOf("</module>") + "</module>".length());
int connectorCount = 0;
Integer batchStart = null;
- StringBuilder b = new StringBuilder();
- b.append(before);
+ StringBuilder builder = new StringBuilder();
+ builder.append(before);
final List<File> generatedConfigs = Lists.newArrayList();
}
for (int i = 0; i < devicesPerPort; i++) {
- final String name = String.valueOf(openDevice) + SIM_DEVICE_SUFFIX + (i == 0 ? "" : "-" + String.valueOf(i));
- String configContent = String.format(middleBlueprint, name, address, String.valueOf(openDevice), String.valueOf(!useSsh));
- configContent = String.format("%s%s%d%s\n%s\n", configContent, "<connection-timeout-millis>", generateConfigsTimeout, "</connection-timeout-millis>", "</module>");
-
- b.append(configContent);
+ final String name = String.valueOf(openDevice)
+ + SIM_DEVICE_SUFFIX + (i == 0 ? "" : "-" + String.valueOf(i));
+ String configContent = String.format(
+ middleBlueprint, name, address, String.valueOf(openDevice), String.valueOf(!useSsh));
+ configContent = String.format(
+ "%s%s%d%s\n%s\n", configContent, "<connection-timeout-millis>",
+ generateConfigsTimeout, "</connection-timeout-millis>", "</module>");
+
+ builder.append(configContent);
connectorCount++;
if (connectorCount == batchSize) {
- b.append(after);
- final File to = new File(configDir, String.format(SIM_DEVICE_CFG_PREFIX + "%d-%d.xml", batchStart, openDevice));
+ builder.append(after);
+ final File to = new File(
+ configDir, String.format(SIM_DEVICE_CFG_PREFIX + "%d-%d.xml", batchStart, openDevice));
generatedConfigs.add(to);
- Files.write(b.toString(), to, Charsets.UTF_8);
+ Files.write(builder.toString(), to, StandardCharsets.UTF_8);
connectorCount = 0;
- b = new StringBuilder();
- b.append(before);
+ builder = new StringBuilder();
+ builder.append(before);
batchStart = null;
}
}
// Write remaining
if (connectorCount != 0) {
- b.append(after);
- final File to = new File(configDir, String.format(SIM_DEVICE_CFG_PREFIX + "%d-%d.xml", batchStart, openDevices.get(openDevices.size() - 1)));
+ builder.append(after);
+ final File to = new File(configDir, String.format(
+ SIM_DEVICE_CFG_PREFIX + "%d-%d.xml", batchStart, openDevices.get(openDevices.size() - 1)));
generatedConfigs.add(to);
- Files.write(b.toString(), to, Charsets.UTF_8);
+ Files.write(builder.toString(), to, StandardCharsets.UTF_8);
}
LOG.info("Config files generated in {}", configDir);
}
}
-
public void updateFeatureFile(final List<File> generated) {
- // TODO karaf core contains jaxb for feature files, use that for
- // modification
- try {
- for (final File featureFile : ncFeatureFiles) {
- final Document document = XmlUtil.readXmlToDocument(Files
- .toString(featureFile, Charsets.UTF_8));
- final NodeList childNodes = document.getDocumentElement().getChildNodes();
-
- for (int i = 0; i < childNodes.getLength(); i++) {
- final Node item = childNodes.item(i);
- if (item instanceof Element == false) {
- continue;
- }
- if (item.getLocalName().equals("feature") == false) {
- continue;
- }
+ for (final File fileFeatures : ncFeatureFiles) {
+ try {
+ final Features f = JaxbUtil.unmarshal(fileFeatures.toURI().toString(), false);
- if (NETCONF_CONNECTOR_ALL_FEATURE
- .equals(((Element) item).getAttribute("name"))) {
- final Element ncAllFeatureDefinition = (Element) item;
- // Clean previous generated files
- for (final XmlElement configfile : XmlElement
- .fromDomElement(ncAllFeatureDefinition)
- .getChildElements("configfile")) {
- ncAllFeatureDefinition.removeChild(configfile.getDomElement());
- }
- for (final File file : generated) {
- final Element configfile = document.createElement("configfile");
- configfile.setTextContent("file:"
- + ETC_OPENDAYLIGHT_KARAF_PATH
- + file.getName());
- configfile.setAttribute(
- "finalname",
- ETC_OPENDAYLIGHT_KARAF_PATH
- + file.getName());
- ncAllFeatureDefinition.appendChild(configfile);
+ for (final Feature feature : f.getFeature()) {
+ if (NETCONF_CONNECTOR_ALL_FEATURE.equals(feature.getName())) {
+ //Clean all previously generated configFiles
+ feature.getConfigfile().clear();
+
+ //Create new configFiles
+ for (final File gen : generated) {
+ final ConfigFile cf = new ConfigFile();
+
+ final String generatedName = ETC_OPENDAYLIGHT_KARAF_PATH + gen.getName();
+
+ cf.setFinalname(generatedName);
+ cf.setLocation("file:" + generatedName);
+
+ feature.getConfigfile().add(cf);
}
}
}
-
- Files.write(XmlUtil.toString(document), featureFile, Charsets.UTF_8);
- LOG.info("Feature file {} updated", featureFile);
+ JaxbUtil.marshal(f, new FileWriter(fileFeatures));
+ LOG.info("Feature file {} updated", fileFeatures);
+ } catch (JAXBException | IOException e) {
+ throw new RuntimeException(e);
}
- } catch (final IOException e) {
- throw new RuntimeException("Unable to load features file as a resource");
- } catch (final SAXException e) {
- throw new RuntimeException("Unable to parse features file");
}
}
-
- private static List<File> getFeatureFile(final File distroFolder, final String featureName, final String suffix) {
+ private static List<File> getFeatureFile(final File distroFolder, final String featureName,
+ final String suffix) {
checkExistingDir(distroFolder, String.format("Folder %s does not exist", distroFolder));
- final File systemDir = checkExistingDir(new File(distroFolder, "system"), String.format("Folder %s does not contain a karaf distro, folder system is missing", distroFolder));
+ final File systemDir = checkExistingDir(new File(distroFolder, "system"),
+ String.format("Folder %s does not contain a karaf distro, folder system is missing", distroFolder));
//check if beryllium path exists, if it doesnt check for lithium and fail/succeed after
File netconfConnectorFeaturesParentDir = new File(systemDir, "org/opendaylight/netconf/" + featureName);
if (!netconfConnectorFeaturesParentDir.exists() || !netconfConnectorFeaturesParentDir.isDirectory()) {
- netconfConnectorFeaturesParentDir = checkExistingDir(new File(systemDir, "org/opendaylight/controller/" + featureName), String.format("Karaf distro in %s does not contain netconf-connector features", distroFolder));
+ netconfConnectorFeaturesParentDir = checkExistingDir(new File(systemDir,
+ "org/opendaylight/controller/" + featureName),
+ String.format("Karaf distro in %s does not contain netconf-connector features", distroFolder));
}
// Find newest version for features
final File newestVersionDir = Collections.max(
- Lists.newArrayList(netconfConnectorFeaturesParentDir.listFiles(new FileFilter() {
- @Override
- public boolean accept(final File pathname) {
- return pathname.isDirectory();
- }
- })), new Comparator<File>() {
- @Override
- public int compare(final File o1, final File o2) {
- return o1.getName().compareTo(o2.getName());
- }
- });
+ Lists.newArrayList(netconfConnectorFeaturesParentDir.listFiles(File::isDirectory)),
+ Comparator.comparing(File::getName));
- return Lists.newArrayList(newestVersionDir.listFiles(new FileFilter() {
- @Override
- public boolean accept(final File pathname) {
- return pathname.getName().contains(featureName)
- && Files.getFileExtension(pathname.getName()).equals(suffix);
- }
- }));
+ return Lists.newArrayList(newestVersionDir.listFiles(
+ pathname -> pathname.getName().contains(featureName)
+ && Files.getFileExtension(pathname.getName()).equals(suffix)));
}
private static File checkExistingDir(final File folder, final String msg) {
public void changeLoadOrder() {
try {
- Files.write(ByteStreams.toByteArray(getClass().getResourceAsStream("/" + ORG_OPS4J_PAX_URL_MVN_CFG)), loadOrderCfgFile);
+ Files.write(ByteStreams.toByteArray(getClass().getResourceAsStream(
+ "/" + ORG_OPS4J_PAX_URL_MVN_CFG)), loadOrderCfgFile);
LOG.info("Load order changed to prefer local bundles/features by rewriting file {}", loadOrderCfgFile);
} catch (IOException e) {
throw new RuntimeException("Unable to rewrite features file " + loadOrderCfgFile, e);