Examples with UPID com.groupon.mesos.util.UPID used on opensource projects

Example 1 with UPID

use of com.groupon.mesos.util.UPID in project jesos by groupon.

the class LocalSchedulerMessageProcessor method frameworkRescindOffer.

@Subscribe
public void frameworkRescindOffer(final RescindResourceOfferMessageEnvelope envelope) {
    checkState(envelope.getRecipient().equals(context.getDriverUPID()), "Received a remote message for local delivery");
    final UPID sender = envelope.getSender();
    if (!driverIsConnected(sender)) {
        return;
    }
    final RescindResourceOfferMessage rescindResourceOfferMessage = envelope.getMessage();
    context.removeAllOffers(rescindResourceOfferMessage.getOfferId());
    eventBus.post(new SchedulerCallback() {

        @Override
        public Runnable getCallback(final Scheduler scheduler, final SchedulerDriver schedulerDriver) {
            return new Runnable() {

                @Override
                public void run() {
                    scheduler.offerRescinded(schedulerDriver, rescindResourceOfferMessage.getOfferId());
                }
            };
        }
    });
}

Example 2 with UPID

use of com.groupon.mesos.util.UPID in project jesos by groupon.

the class LocalSchedulerMessageProcessor method frameworkLostSlave.

@Subscribe
public void frameworkLostSlave(final LostSlaveMessageEnvelope envelope) {
    checkState(envelope.getRecipient().equals(context.getDriverUPID()), "Received a remote message for local delivery");
    final UPID sender = envelope.getSender();
    if (!driverIsConnected(sender)) {
        return;
    }
    final LostSlaveMessage lostSlaveMessage = envelope.getMessage();
    final SlaveID slaveId = lostSlaveMessage.getSlaveId();
    context.removeSlave(slaveId);
    eventBus.post(new SchedulerCallback() {

        @Override
        public Runnable getCallback(final Scheduler scheduler, final SchedulerDriver schedulerDriver) {
            return new Runnable() {

                @Override
                public void run() {
                    scheduler.slaveLost(schedulerDriver, slaveId);
                }
            };
        }
    });
}

Example 3 with UPID

use of com.groupon.mesos.util.UPID in project jesos by groupon.

the class LocalSchedulerMessageProcessor method frameworkFrameworkMessage.

@Subscribe
public void frameworkFrameworkMessage(final ExecutorToFrameworkMessageEnvelope envelope) {
    checkState(envelope.getRecipient().equals(context.getDriverUPID()), "Received a remote message for local delivery");
    final UPID sender = envelope.getSender();
    if (!driverIsConnected(sender)) {
        return;
    }
    final ExecutorToFrameworkMessage executorToFrameworkMessage = envelope.getMessage();
    eventBus.post(new SchedulerCallback() {

        @Override
        public Runnable getCallback(final Scheduler scheduler, final SchedulerDriver schedulerDriver) {
            return new Runnable() {

                @Override
                public void run() {
                    scheduler.frameworkMessage(schedulerDriver, executorToFrameworkMessage.getExecutorId(), executorToFrameworkMessage.getSlaveId(), executorToFrameworkMessage.getData().toByteArray());
                }
            };
        }
    });
}

Example 4 with UPID

use of com.groupon.mesos.util.UPID in project jesos by groupon.

the class LocalSchedulerMessageProcessor method frameworkResourceOffers.

@Subscribe
public void frameworkResourceOffers(final ResourceOffersMessageEnvelope envelope) {
    checkState(envelope.getRecipient().equals(context.getDriverUPID()), "Received a remote message for local delivery");
    final UPID sender = envelope.getSender();
    if (!driverIsConnected(sender)) {
        return;
    }
    final ResourceOffersMessage resourceOffersMessage = envelope.getMessage();
    final List<Offer> offers = resourceOffersMessage.getOffersList();
    final List<UPID> pids = ImmutableList.copyOf(Lists.transform(resourceOffersMessage.getPidsList(), UPID.getCreateFunction()));
    checkState(offers.size() == pids.size(), "Received %s offers but only %s pids!", offers.size(), pids.size());
    int pidIndex = 0;
    for (final Offer offer : offers) {
        context.addOffer(offer.getId(), offer.getSlaveId(), pids.get(pidIndex++));
    }
    eventBus.post(new SchedulerCallback() {

        @Override
        public Runnable getCallback(final Scheduler scheduler, final SchedulerDriver schedulerDriver) {
            return new Runnable() {

                @Override
                public void run() {
                    scheduler.resourceOffers(schedulerDriver, resourceOffersMessage.getOffersList());
                }
            };
        }
    });
}

Example 5 with UPID

use of com.groupon.mesos.util.UPID in project jesos by groupon.

the class InternalExecutorDriver method sendMessage.

/**
     * Remote message delivery. Called by the event bus on an event bus thread to transfer a message
     * to another mesos host. This must <b>not</b> called directly to ensure the thread separation between
     * caller threads, event bus threads and executor threads.
     */
@Subscribe
public void sendMessage(final RemoteMessageEnvelope envelope) throws Exception {
    final Message message = envelope.getMessage();
    final UPID recipient = envelope.getRecipient();
    checkState(!recipient.equals(context.getDriverUPID()), "Received a message with local recipient! (%s)", message);
    sender.sendHttpMessage(recipient, message);
}