Java Code Examples for org.apache.mesos.Protos#MasterInfo

/**
 * Test Mesos registration handling.
 */
@Test
public void testRegistered() throws Exception {
	new Context() {{
		startResourceManager();

		Protos.MasterInfo masterInfo = Protos.MasterInfo.newBuilder()
			.setId("master1").setIp(0).setPort(5050).build();
		resourceManager.registered(new Registered(framework1, masterInfo));

		verify(rmServices.workerStore).setFrameworkID(Option.apply(framework1));
		resourceManager.connectionMonitor.expectMsgClass(Registered.class);
		resourceManager.reconciliationCoordinator.expectMsgClass(Registered.class);
		resourceManager.launchCoordinator.expectMsgClass(Registered.class);
		resourceManager.taskRouter.expectMsgClass(Registered.class);
	}};
}
 

/**
 * Test Mesos registration handling.
 */
@Test
public void testRegistered() throws Exception {
	new Context() {{
		startResourceManager();

		Protos.MasterInfo masterInfo = Protos.MasterInfo.newBuilder()
			.setId("master1").setIp(0).setPort(5050).build();
		resourceManager.registered(new Registered(framework1, masterInfo));

		verify(rmServices.workerStore).setFrameworkID(Option.apply(framework1));
		resourceManager.connectionMonitor.expectMsgClass(Registered.class);
		resourceManager.reconciliationCoordinator.expectMsgClass(Registered.class);
		resourceManager.launchCoordinator.expectMsgClass(Registered.class);
		resourceManager.taskRouter.expectMsgClass(Registered.class);
	}};
}
 

/**
 * Test Mesos re-registration handling.
 */
@Test
public void testReRegistered() throws Exception {
	new Context() {{
		when(rmServices.workerStore.getFrameworkID()).thenReturn(Option.apply(framework1));
		startResourceManager();

		Protos.MasterInfo masterInfo = Protos.MasterInfo.newBuilder()
			.setId("master1").setIp(0).setPort(5050).build();
		resourceManager.reregistered(new ReRegistered(masterInfo));

		resourceManager.connectionMonitor.expectMsgClass(ReRegistered.class);
		resourceManager.reconciliationCoordinator.expectMsgClass(ReRegistered.class);
		resourceManager.launchCoordinator.expectMsgClass(ReRegistered.class);
		resourceManager.taskRouter.expectMsgClass(ReRegistered.class);
	}};
}
 

/**
 * Test Mesos registration handling.
 */
@Test
public void testRegistered() throws Exception {
	new Context() {{
		startResourceManager();

		Protos.MasterInfo masterInfo = Protos.MasterInfo.newBuilder()
			.setId("master1").setIp(0).setPort(5050).build();
		resourceManager.registered(new Registered(framework1, masterInfo));

		verify(rmServices.workerStore).setFrameworkID(Option.apply(framework1));
		resourceManager.connectionMonitor.expectMsgClass(Registered.class);
		resourceManager.reconciliationCoordinator.expectMsgClass(Registered.class);
		resourceManager.launchCoordinator.expectMsgClass(Registered.class);
		resourceManager.taskRouter.expectMsgClass(Registered.class);
	}};
}
 

@Override
public void reregistered(SchedulerDriver driver, final Protos.MasterInfo masterInfo) {
	runAsync(new Runnable() {
		@Override
		public void run() {
			MesosResourceManager.this.reregistered(new ReRegistered(masterInfo));
		}
	});
}
 

@Override
public void reregistered(SchedulerDriver driver, final Protos.MasterInfo masterInfo) {
	runAsync(new Runnable() {
		@Override
		public void run() {
			MesosResourceManager.this.reregistered(new ReRegistered(masterInfo));
		}
	});
}
 

public static final Protos.MasterInfo generateMasterInfo() {
    return Protos.MasterInfo.newBuilder()
            .setId(UUID.randomUUID().toString())
            .setIp(ipToInt("127.0.0.1"))
            .setPort(5050)
            .build();
}
 

@Override
public void registered(final SchedulerDriver schedulerDriver, final Protos.FrameworkID frameworkID, final Protos.MasterInfo masterInfo) {
    log.info("call registered");
    frameworkIDService.save(frameworkID.getValue());
    taskScheduler.expireAllLeases();
    MesosStateService.register(masterInfo.getHostname(), masterInfo.getPort());
}
 

@Override
public void reregistered(SchedulerDriver driver,
                         Protos.MasterInfo masterInfo) {
    LOGGER.info("Re-registered with master: {}", masterInfo);
    reconciler.start();
    suppressOrRevive();
}
 

public Registered(Protos.FrameworkID frameworkId, Protos.MasterInfo masterInfo) {
	requireNonNull(frameworkId);
	requireNonNull(masterInfo);

	this.frameworkId = frameworkId;
	this.masterInfo = masterInfo;
}
 

public Protos.MasterInfo getMasterInfo() {
  return masterInfo;
}
 

@Override
public void registered(SchedulerDriver driver,
                       Protos.FrameworkID frameworkId,
                       Protos.MasterInfo masterInfo) {
    // Call re-register and return early if registered() has already been called once
    // in the lifetime of this object.
    if (!isSchedulerRegistered.compareAndSet(false, true)) {
        LOGGER.info("Scheduler is already registered, calling reregistered()");
        reregistered(driver, masterInfo);
        return;
    }

    // The following is executed only once in the lifetime of this object.
    final String frameworkIdValue = frameworkId.getValue();
    LOGGER.info("Framework registered : id = {}", frameworkIdValue);
    try {
        CassandraScheduler.taskKiller = new DefaultTaskKiller(
                stateStore,
                new DefaultTaskFailureListener(stateStore),
                driver);
        this.planScheduler = new DefaultPlanScheduler(offerAccepter, new OfferEvaluator(stateStore), taskKiller);
        stateStore.storeFrameworkId(frameworkId);
        Plan plan = new CassandraPlan(
                defaultConfigurationManager,
                ReconciliationPhase.create(reconciler),
                SyncDataCenterPhase.create(seeds, executor),
                CassandraDaemonPhase.create(
                        cassandraState, offerRequirementProvider, client, defaultConfigurationManager),
                Arrays.asList(
                        backup, restore, cleanup, repair, upgrade));
        plan.subscribe(this);
        planManager = new DefaultPlanManager(plan);
        reconciler.start();
        suppressOrRevive();
        // use add() to just throw if full:
        resourcesQueue.add(Arrays.asList(
                new ServiceConfigResource(configurationManager),
                new SeedsResource(seeds),
                new ConfigurationResource(defaultConfigurationManager),
                new TasksResource(capabilities, cassandraState, client, configurationManager),
                new PlansResource(ImmutableMap.of("deploy", planManager)), // TODO(nick) include recovery
                new BackupResource(backup),
                new RestoreResource(restore),
                new CleanupResource(cleanup),
                new RepairResource(repair),
                new UpgradeSSTableResource(upgrade, enableUpgradeSSTableEndpoint),
                new DataCenterResource(seeds),
                new ConnectionResource(capabilities, cassandraState, configurationManager),
                // TODO(nick) rename upstream to StringPropertyDeserializer:
                new StateResource(stateStore, new JsonPropertyDeserializer())));
    } catch (Throwable t) {
        String error = "An error occurred when registering " +
                "the framework and initializing the execution plan.";
        LOGGER.error(error, t);
        throw new RuntimeException(error, t);
    }
}
 

public Protos.MasterInfo masterInfo() {
	return masterInfo;
}
 

@Override
public void reregistered(SchedulerDriver driver, Protos.MasterInfo mi) {
	LOGGER.info("Framework reregistered!");
}
 

public Protos.MasterInfo masterInfo() {
	return masterInfo;
}
 

public ReRegistered(Protos.MasterInfo masterInfo) {
	requireNonNull(masterInfo);

	this.masterInfo = masterInfo;
}
 

public Protos.MasterInfo masterInfo() {
	return masterInfo;
}
 

@Override
public void reregistered(SchedulerDriver driver, Protos.MasterInfo masterInfo) {
	mesosActor.tell(new ReRegistered(masterInfo), ActorRef.noSender());
}
 

@Override
public void reregistered(SchedulerDriver driver, Protos.MasterInfo masterInfo) {
	mesosActor.tell(new ReRegistered(masterInfo), ActorRef.noSender());
}
 

@Override
public void registered(SchedulerDriver schedulerDriver,
                       Protos.FrameworkID frameworkID, Protos.MasterInfo masterInfo) {
  LOG.info("Registered" + frameworkID);
}