예제 #1
0
 @Override
 public MesosSlave getMesosSlave(String hostname) {
   Collection<Entity> slaves = sensors().get(MESOS_SLAVES).getMembers();
   Optional<Entity> found =
       Iterables.tryFind(
           slaves,
           Predicates.or(
               EntityPredicates.attributeEqualTo(MesosSlave.HOSTNAME, hostname),
               EntityPredicates.attributeEqualTo(MesosSlave.ADDRESS, hostname)));
   if (found.isPresent()) {
     return (MesosSlave) found.get();
   } else {
     throw new IllegalStateException("Cannot find slave for host: " + hostname);
   }
 }
 @Override
 public List<ApplicationSummary> list(String typeRegex) {
   if (Strings.isBlank(typeRegex)) {
     typeRegex = ".*";
   }
   return FluentIterable.from(mgmt().getApplications())
       .filter(
           EntitlementPredicates.isEntitled(
               mgmt().getEntitlementManager(), Entitlements.SEE_ENTITY))
       .filter(EntityPredicates.hasInterfaceMatching(typeRegex))
       .transform(ApplicationTransformer.FROM_APPLICATION)
       .toList();
 }
예제 #3
0
  /** De-register our {@link MesosLocation} and its children. */
  @Override
  public void stop() {
    disconnectSensors();

    sensors().set(SERVICE_UP, Boolean.FALSE);

    deleteLocation();

    Duration timeout = config().get(SHUTDOWN_TIMEOUT);

    // Find all applications and stop, blocking for up to five minutes until ended
    try {
      Iterable<Entity> entities =
          Iterables.filter(
              getManagementContext().getEntityManager().getEntities(),
              Predicates.and(
                  MesosUtils.sameCluster(this),
                  Predicates.not(EntityPredicates.applicationIdEqualTo(getApplicationId()))));
      Set<Application> applications =
          ImmutableSet.copyOf(
              Iterables.transform(
                  entities,
                  new Function<Entity, Application>() {
                    @Override
                    public Application apply(Entity input) {
                      return input.getApplication();
                    }
                  }));
      LOG.debug("Stopping applications: {}", Iterables.toString(applications));
      Entities.invokeEffectorList(this, applications, Startable.STOP).get(timeout);
    } catch (Exception e) {
      LOG.warn("Error stopping applications", e);
    }

    // Stop all framework tasks in parallel
    try {
      Group frameworks = sensors().get(MESOS_FRAMEWORKS);
      LOG.debug("Stopping framework tasks in: {}", Iterables.toString(frameworks.getMembers()));
      Entities.invokeEffectorList(this, frameworks.getMembers(), Startable.STOP).get(timeout);
    } catch (Exception e) {
      LOG.warn("Error stopping frameworks", e);
    }

    // Stop anything else left over
    // TODO Stop slave entities
    try {
      super.stop();
    } catch (Exception e) {
      LOG.warn("Error stopping children", e);
    }
  }
  @Test
  public void testEntityCreatingItsEnricherDoesNotReCreateItUnlessUniqueTagDifferent()
      throws Exception {
    TestEntity e1 =
        origApp.createAndManageChild(
            EntitySpec.create(TestEntity.class, MyTestEntityWithEnricher.class));
    Collection<Enricher> e1e = e1.getEnrichers();
    log.info("enrichers1: " + e1e);
    Entities.dumpInfo(e1);
    assertEquals(e1e.size(), 5);

    newApp = (TestApplication) rebind();
    Entity e2 =
        Iterables.getOnlyElement(
            Entities.descendants(newApp, EntityPredicates.idEqualTo(e1.getId())));
    Collection<Enricher> e2e = e2.getEnrichers();
    log.info("enrichers2: " + e2e);
    Entities.dumpInfo(e2);

    assertEquals(e2e.size(), e1e.size() + 1);
  }
예제 #5
0
  @Override
  public void init() {
    LOG.info("Starting Mesos cluster id {}", getId());
    registerLocationResolver();
    super.init();

    Group slaves = addChild(EntitySpec.create(BasicGroup.class).displayName("Mesos Slaves"));

    Group frameworks =
        addChild(EntitySpec.create(BasicGroup.class).displayName("Mesos Frameworks"));

    DynamicGroup tasks =
        addChild(
            EntitySpec.create(DynamicGroup.class)
                .configure(
                    DynamicGroup.ENTITY_FILTER,
                    Predicates.and(
                        Predicates.instanceOf(MesosTask.class),
                        EntityPredicates.attributeEqualTo(MesosAttributes.MESOS_CLUSTER, this)))
                .displayName("Mesos Tasks"));

    DynamicMultiGroup applications =
        addChild(
            EntitySpec.create(DynamicMultiGroup.class)
                .configure(
                    DynamicMultiGroup.ENTITY_FILTER,
                    Predicates.and(
                        MesosUtils.sameCluster(this),
                        Predicates.not(EntityPredicates.applicationIdEqualTo(getApplicationId()))))
                .configure(DynamicMultiGroup.RESCAN_INTERVAL, 15L)
                .configure(
                    DynamicMultiGroup.BUCKET_FUNCTION,
                    new Function<Entity, String>() {
                      @Override
                      public String apply(@Nullable Entity input) {
                        return input.getApplication().getDisplayName()
                            + ":"
                            + input.getApplicationId();
                      }
                    })
                .configure(DynamicMultiGroup.BUCKET_SPEC, EntitySpec.create(BasicGroup.class))
                .displayName("Mesos Applications"));

    if (config().get(SDN_ENABLE) && config().get(SDN_PROVIDER_SPEC) != null) {
      EntitySpec entitySpec = EntitySpec.create(config().get(SDN_PROVIDER_SPEC));
      entitySpec.configure(MesosAttributes.MESOS_CLUSTER, this);
      Entity sdn = addChild(entitySpec);
      sensors().set(SDN_PROVIDER, sdn);
    }

    sensors().set(MESOS_SLAVES, slaves);
    sensors().set(MESOS_FRAMEWORKS, frameworks);
    sensors().set(MESOS_TASKS, tasks);
    sensors().set(MESOS_APPLICATIONS, applications);

    // Override the health-check: just interested in the slaves, frameworks and sdn (rather than
    // the groups that show the tasks or apps).
    Entity sdn = sensors().get(SDN_PROVIDER);
    enrichers()
        .add(
            EnricherSpec.create(ComputeServiceIndicatorsFromChildrenAndMembers.class)
                .uniqueTag(ComputeServiceIndicatorsFromChildrenAndMembers.DEFAULT_UNIQUE_TAG)
                .configure(ComputeServiceIndicatorsFromChildrenAndMembers.FROM_CHILDREN, true)
                .configure(
                    ComputeServiceIndicatorsFromChildrenAndMembers.ENTITY_FILTER,
                    Predicates.or(
                        ImmutableList.of(
                            Predicates.<Entity>equalTo(slaves),
                            Predicates.<Entity>equalTo(frameworks),
                            (sdn == null
                                ? Predicates.<Entity>alwaysFalse()
                                : Predicates.equalTo(sdn))))));
  }
  @Test
  public void testParameterDefaultsUsedInConfig() throws Exception {
    addCatalogItems(
        "brooklyn.catalog:",
        "  version: " + TEST_VERSION,
        "  items:",
        "    - id: " + ConfigEntityForTest.class.getSimpleName() + "WithParams",
        "      itemType: entity",
        "      item:",
        "        type: " + ConfigEntityForTest.class.getName(),
        "        brooklyn.parameters:",
        "          - name: num",
        "            type: integer",
        "            default: 1234",
        "        brooklyn.children:",
        "          - type: " + BasicStartable.class.getName(),
        "            name: s",
        "            brooklyn.config:",
        "              test: $brooklyn:parent().config(\"num\")",
        "    - id: " + SYMBOLIC_NAME,
        "      itemType: entity",
        "      item:",
        "        type: " + BasicApplication.class.getName(),
        "        brooklyn.children:",
        "          - type: " + ConfigEntityForTest.class.getSimpleName() + "WithParams",
        "            name: a",
        "          - type: " + ConfigEntityForTest.class.getSimpleName() + "WithParams",
        "            name: b",
        "            brooklyn.config:",
        "              num: 5678",
        "          - type: " + ConfigEntityForTest.class.getSimpleName() + "WithParams",
        "            name: c",
        "            brooklyn.config:",
        "              test: $brooklyn:config(\"num\")");

    CatalogItem<?, ?> item =
        catalog.getCatalogItem(
            ConfigEntityForTest.class.getSimpleName() + "WithParams", TEST_VERSION);
    AbstractBrooklynObjectSpec<?, ?> spec = createSpec(item);
    List<SpecParameter<?>> params = spec.getParameters();
    assertEquals(params.size(), 3);
    assertTrue(Iterables.tryFind(params, nameEqualTo("num")).isPresent());

    Application app =
        (Application) createAndStartApplication("services:", "  - type: " + ver(SYMBOLIC_NAME));

    Iterable<Entity> children = app.getChildren();
    Optional<Entity> a = Iterables.tryFind(children, EntityPredicates.displayNameEqualTo("a"));
    assertTrue(a.isPresent());
    assertEquals(a.get().config().get(NUM).intValue(), 1234);
    Optional<Entity> as =
        Iterables.tryFind(a.get().getChildren(), EntityPredicates.displayNameEqualTo("s"));
    assertTrue(as.isPresent());
    assertEquals(as.get().config().get(ConfigKeys.newIntegerConfigKey("test")).intValue(), 1234);
    Optional<Entity> b = Iterables.tryFind(children, EntityPredicates.displayNameEqualTo("b"));
    assertTrue(b.isPresent());
    assertEquals(b.get().config().get(NUM).intValue(), 5678);
    Optional<Entity> bs =
        Iterables.tryFind(b.get().getChildren(), EntityPredicates.displayNameEqualTo("s"));
    assertTrue(bs.isPresent());
    assertEquals(bs.get().config().get(ConfigKeys.newIntegerConfigKey("test")).intValue(), 5678);
    Optional<Entity> c = Iterables.tryFind(children, EntityPredicates.displayNameEqualTo("c"));
    assertTrue(c.isPresent());
    assertEquals(c.get().config().get(ConfigKeys.newIntegerConfigKey("test")).intValue(), 1234);
    Optional<Entity> cs =
        Iterables.tryFind(c.get().getChildren(), EntityPredicates.displayNameEqualTo("s"));
    assertTrue(cs.isPresent());
    assertEquals(cs.get().config().get(ConfigKeys.newIntegerConfigKey("test")).intValue(), 1234);
  }