com.netflix.fenzo.functions.Action1 Java Examples

/**
 * Creates the Fenzo optimizer (builder).
 * The builder is an indirection to facilitate unit testing of the Launch Coordinator.
 */
private static TaskSchedulerBuilder createOptimizer() {
	return new TaskSchedulerBuilder() {
		TaskScheduler.Builder builder = new TaskScheduler.Builder();

		@Override
		public TaskSchedulerBuilder withLeaseRejectAction(Action1<VirtualMachineLease> action) {
			builder.withLeaseRejectAction(action);
			return this;
		}

		@Override
		public TaskScheduler build() {
			return builder.build();
		}
	};
}
 

/**
 * Creates the Fenzo optimizer (builder).
 * The builder is an indirection to facilitate unit testing of the Launch Coordinator.
 */
private static TaskSchedulerBuilder createOptimizer() {
	return new TaskSchedulerBuilder() {
		TaskScheduler.Builder builder = new TaskScheduler.Builder();

		@Override
		public TaskSchedulerBuilder withLeaseRejectAction(Action1<VirtualMachineLease> action) {
			builder.withLeaseRejectAction(action);
			return this;
		}

		@Override
		public TaskScheduler build() {
			return builder.build();
		}
	};
}
 

private TaskScheduler getTaskScheduler() {
    return new TaskScheduler.Builder()
            .withLeaseOfferExpirySecs(1000000000L)
            .withLeaseRejectAction(new Action1<VirtualMachineLease>() {
                
                @Override
                public void call(final VirtualMachineLease lease) {
                    log.warn("Declining offer on '{}'", lease.hostname());
                    schedulerDriver.declineOffer(lease.getOffer().getId());
                }
            }).build();
}
 

@Before
public void setUp() throws Exception {
    taskScheduler = new TaskScheduler.Builder().withLeaseOfferExpirySecs(1000000000L).withLeaseRejectAction(new Action1<VirtualMachineLease>() {
        
        @Override
        public void call(final VirtualMachineLease virtualMachineLease) {
        }
    }).build();
}
 

/**
 * Creates the Fenzo optimizer (builder).
 * The builder is an indirection to facilitate unit testing of the Launch Coordinator.
 */
private static TaskSchedulerBuilder createOptimizer() {
	return new TaskSchedulerBuilder() {
		TaskScheduler.Builder builder = new TaskScheduler.Builder();

		@Override
		public TaskSchedulerBuilder withLeaseRejectAction(Action1<VirtualMachineLease> action) {
			builder.withLeaseRejectAction(action);
			return this;
		}

		@Override
		public TaskSchedulerBuilder withRejectAllExpiredOffers() {
			builder.withRejectAllExpiredOffers();
			return this;
		}

		@Override
		public TaskSchedulerBuilder withLeaseOfferExpirySecs(long leaseOfferExpirySecs) {
			builder.withLeaseOfferExpirySecs(leaseOfferExpirySecs);
			return this;
		}

		@Override
		public TaskScheduler build() {
			return builder.build();
		}
	};
}
 

@Override
public void setVMLeaseHandler(Action1<List<? extends VirtualMachineLease>> leaseHandler) {
    this.leaseHandler = virtualMachineLeases -> {
        drainKillTaskQueue();
        leaseHandler.call(virtualMachineLeases);
    };
}
 

@Override
public void setVMLeaseHandler(Action1<List<? extends VirtualMachineLease>> leaseHandler) {
    this.leaseHandler = leaseHandler;
}
 

@Override
public void setRescindLeaseHandler(Action1<List<LeaseRescindedEvent>> rescindLeaseHandler) {
    this.rescindLeaseHandler = rescindLeaseHandler;
}
 

@Override
public void setRescindLeaseHandler(Action1<List<LeaseRescindedEvent>> rescindLeaseHandler) {
    throw new IllegalStateException("method not supported");
}
 

public MesosSchedulerCallbackHandler(
        Action1<List<? extends VirtualMachineLease>> leaseHandler,
        Observer<LeaseRescindedEvent> vmLeaseRescindedObserver,
        Observer<ContainerEvent> vmTaskStatusObserver,
        V3JobOperations v3JobOperations,
        Optional<FitInjection> taskStatusUpdateFitInjection,
        MasterConfiguration config,
        MesosConfiguration mesosConfiguration,
        TitusRuntime titusRuntime) {
    this.leaseHandler = leaseHandler;
    this.vmLeaseRescindedObserver = vmLeaseRescindedObserver;
    this.vmTaskStatusObserver = vmTaskStatusObserver;
    this.v3JobOperations = v3JobOperations;
    this.taskStatusUpdateFitInjection = taskStatusUpdateFitInjection;
    this.config = config;
    this.mesosConfiguration = mesosConfiguration;
    this.registry = titusRuntime.getRegistry();
    this.mesosStateTracker = new MesosStateTracker(config, titusRuntime, Schedulers.computation());

    numMesosRegistered = registry.counter(MetricConstants.METRIC_MESOS + "numMesosRegistered");
    numMesosDisconnects = registry.counter(MetricConstants.METRIC_MESOS + "numMesosDisconnects");
    numOfferRescinded = registry.counter(MetricConstants.METRIC_MESOS + "numOfferRescinded");
    numReconcileTasks = registry.counter(MetricConstants.METRIC_MESOS + "numReconcileTasks");
    lastOfferReceivedMillis = registry.gauge(MetricConstants.METRIC_MESOS + "lastOfferReceivedMillis", new AtomicLong());
    lastValidOfferReceiveMillis = registry.gauge(MetricConstants.METRIC_MESOS + "lastValidOfferReceiveMillis", new AtomicLong());
    numInvalidOffers = registry.counter(MetricConstants.METRIC_MESOS + "numInvalidOffers");
    numOfferTooSmall = registry.counter(MetricConstants.METRIC_MESOS + "numOfferTooSmall");

    this.invalidRequestMessageMatcherFactory = RegExpExt.dynamicMatcher(mesosConfiguration::getInvalidRequestMessagePattern, "invalidRequestMessagePattern", Pattern.DOTALL, logger);
    this.crashedMessageMatcherFactory = RegExpExt.dynamicMatcher(mesosConfiguration::getCrashedMessagePattern, "crashedMessagePattern", Pattern.DOTALL, logger);
    this.transientSystemErrorMessageMatcherFactory = RegExpExt.dynamicMatcher(mesosConfiguration::getTransientSystemErrorMessagePattern, "transientSystemErrorMessagePattern", Pattern.DOTALL, logger);
    this.localSystemErrorMessageMatcherFactory = RegExpExt.dynamicMatcher(mesosConfiguration::getLocalSystemErrorMessagePattern, "localSystemErrorMessagePattern", Pattern.DOTALL, logger);
    this.unknownSystemErrorMessageMatcherFactory = RegExpExt.dynamicMatcher(mesosConfiguration::getUnknownSystemErrorMessagePattern, "unknownSystemErrorMessagePattern", Pattern.DOTALL, logger);

    this.subscription = Observable
            .interval(10, 10, TimeUnit.SECONDS)
            .doOnNext(tick -> {
                lastOfferReceivedMillis.set(System.currentTimeMillis() - lastOfferReceivedAt.get());
                lastValidOfferReceiveMillis.set(System.currentTimeMillis() - lastValidOfferReceivedAt.get());
            })
            .subscribe();
}
 

@Override
public void setRescindLeaseHandler(Action1<List<LeaseRescindedEvent>> rescindLeaseHandler) {
    // do nothing
}
 

@Override
public void setVMLeaseHandler(Action1<List<? extends VirtualMachineLease>> leaseHandler) {
    throw new IllegalStateException("method not supported");
}
 

/**
 * Set the callback action for rejecting a lease.
 */
TaskSchedulerBuilder withLeaseRejectAction(Action1<VirtualMachineLease> action);
 

/**
 * Set the callback action for rejecting a lease.
 */
TaskSchedulerBuilder withLeaseRejectAction(Action1<VirtualMachineLease> action);
 

/**
 * Set the callback action for rejecting a lease.
 */
TaskSchedulerBuilder withLeaseRejectAction(Action1<VirtualMachineLease> action);
 

void setRescindLeaseHandler(Action1<List<LeaseRescindedEvent>> rescindLeaseHandler); 

void setVMLeaseHandler(Action1<List<? extends VirtualMachineLease>> leaseHandler);