ahgittin commented on a change in pull request #1115:
URL: https://github.com/apache/brooklyn-server/pull/1115#discussion_r504314052
##########
File path:
core/src/main/java/org/apache/brooklyn/core/typereg/BrooklynCatalogBundleResolvers.java
##########
@@ -18,26 +18,149 @@
*/
package org.apache.brooklyn.core.typereg;
+import com.google.common.annotations.Beta;
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.collect.ArrayListMultimap;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Iterables;
+import com.google.common.collect.Multimap;
import java.io.InputStream;
+import java.util.*;
+import org.apache.brooklyn.api.framework.FrameworkLookup;
import org.apache.brooklyn.api.mgmt.ManagementContext;
+import org.apache.brooklyn.api.typereg.RegisteredType;
+import org.apache.brooklyn.api.typereg.RegisteredTypeLoadingContext;
import org.apache.brooklyn.core.mgmt.ha.OsgiBundleInstallationResult;
+import
org.apache.brooklyn.core.typereg.BrooklynCatalogBundleResolver.BundleInstallationOptions;
+import org.apache.brooklyn.util.collections.MutableList;
+import org.apache.brooklyn.util.exceptions.Exceptions;
+import org.apache.brooklyn.util.exceptions.PropagatedRuntimeException;
import org.apache.brooklyn.util.exceptions.ReferenceWithError;
+import org.apache.brooklyn.util.guava.Maybe;
+import org.apache.brooklyn.util.text.Strings;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
public class BrooklynCatalogBundleResolvers {
- public static ReferenceWithError<OsgiBundleInstallationResult>
install(ManagementContext managementContext, InputStream input,
-
BrooklynCatalogBundleResolver.BundleInstallationOptions options) {
+ private static Logger LOG =
LoggerFactory.getLogger(BrooklynCatalogBundleResolvers.class);
- if (options==null) options = new
BrooklynCatalogBundleResolver.BundleInstallationOptions();
+ private static Collection<BrooklynCatalogBundleResolver> getAll() {
+ return
ImmutableList.copyOf(FrameworkLookup.lookupAll(BrooklynCatalogBundleResolver.class));
+ }
+
+ private static Collection<Class<? extends BrooklynCatalogBundleResolver>>
OVERRIDE;
+ @SafeVarargs
+ @VisibleForTesting
+ public synchronized static void forceAvailable(Class<? extends
BrooklynCatalogBundleResolver> ...classes) {
+ OVERRIDE = Arrays.asList(classes);
+ }
+ public synchronized static void clearForced() {
+ OVERRIDE = null;
+ }
+
+ public static Collection<BrooklynCatalogBundleResolver>
all(ManagementContext mgmt) {
+ // TODO cache these in the TypeRegistry, looking for new ones
periodically or supplying a way to register them
+ Collection<Class<? extends BrooklynCatalogBundleResolver>> override =
OVERRIDE;
+ Collection<BrooklynCatalogBundleResolver> result = new ArrayList<>();
+ if (override!=null) {
+ for (Class<? extends BrooklynCatalogBundleResolver> o1: override) {
+ try {
+ result.add(o1.newInstance());
+ } catch (Exception e) {
+ Exceptions.propagate(e);
+ }
+ }
+ } else {
+ result.addAll(getAll());
+ }
+ for(BrooklynCatalogBundleResolver t : result) {
+ t.setManagementContext(mgmt);
+ }
+ return result;
+ }
+
+ /** returns a list of {@link BrooklynCatalogBundleResolver} instances for
this {@link ManagementContext}
+ * which may be able to handle the given bundle; the list is sorted with
highest-score transformer first */
+ @Beta
+ public static List<BrooklynCatalogBundleResolver>
forBundle(ManagementContext mgmt, InputStream input,
+
BrooklynCatalogBundleResolver.BundleInstallationOptions options) {
+ Multimap<Double,BrooklynCatalogBundleResolver> byScoreMulti =
ArrayListMultimap.create();
+ Collection<BrooklynCatalogBundleResolver> transformers = all(mgmt);
+ for (BrooklynCatalogBundleResolver transformer : transformers) {
+ double score = transformer.scoreForBundle(options.format, input);
+ if (LOG.isTraceEnabled()) {
+ LOG.trace("SCORE for '" + input + "' at " + transformer + ": "
+ score);
+ }
+ if (score>0) byScoreMulti.put(score, transformer);
+ }
+ Map<Double, Collection<BrooklynCatalogBundleResolver>> tree = new
TreeMap<Double,
Collection<BrooklynCatalogBundleResolver>>(byScoreMulti.asMap());
+ List<Collection<BrooklynCatalogBundleResolver>> highestFirst = new
ArrayList<Collection<BrooklynCatalogBundleResolver>>(tree.values());
Review comment:
nice one. i just copied `TypePlanTransformers` so I've applied this
tidy-up there also
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
