private static void addModuleJarToCurrentClassloader(Map<String, ComponentDependencies> componentsToStartWithDependencies) { componentsToStartWithDependencies.values().forEach(c -> { try { ClassLoader currentThreadClassLoader = Thread.currentThread().getContextClassLoader(); URLClassLoader urlClassLoader = new URLClassLoader(new URL[]{c.getDependencies().get(0).toURL()}, currentThreadClassLoader); Thread.currentThread().setContextClassLoader(urlClassLoader); } catch (MalformedURLException e) { LOGGER.error("unable to locate {}", c.getDependencies().get(0)); } }); }
private static List<String> computeStartingOrder(Map<String, ComponentDependencies> componentsToStartWithDependencies) { Map<String, ComponentMetadata> commands = new HashMap<>(); componentsToStartWithDependencies.entrySet().stream().forEach(entry -> { try { ClassLoader currentThreadClassLoader = Thread.currentThread().getContextClassLoader(); Class<?> componentClass = currentThreadClassLoader.loadClass(hadoopUnitConfiguration.getString(entry.getKey() + ".metadataClass")); ComponentMetadata componentInstance = (ComponentMetadata) componentClass.getConstructor().newInstance(); commands.put(componentInstance.getName(), componentInstance); } catch (Exception e) { LOGGER.error("unable to instantiate {}", entry.getValue().getName(), e); } }); Graph<String, DefaultEdge> dependenciesGraph = generateGraph(commands); Map<String, List<String>> dependenciesMapByComponent = commands.values().stream().collect(Collectors.toMap(ComponentMetadata::getName, c -> DependenciesCalculator.calculateParents(dependenciesGraph, c.getName()))); Map<String, List<String>> transitiveDependenciesMapByComponent = DependenciesCalculator.findTransitiveDependenciesByComponent(dependenciesMapByComponent); return DependenciesCalculator.dryRunToDefineCorrectOrder(transitiveDependenciesMapByComponent); }
ComponentProperties componentProperties = loadAndRun(c, hadoopUnitConfiguration.getString(c.toLowerCase() + ".mainClass"), componentsToStartWithDependencies.get(c.toLowerCase()).getDependencies());
artifacts.add(a.getArtifact().getFile()) ); ComponentDependencies component = new ComponentDependencies(c, artifacts); componentsToStartWithDependencies.put(c, component);