Dynamically control the resources allocated to task

package stkparallelcomputingserversdk.howto;

import agi.parallel.client.ClusterJobScheduler;
import agi.parallel.client.IJobScheduler;
import agi.parallel.client.Job;
import agi.parallel.infrastructure.ConsumableResources;
import agi.parallel.infrastructure.IJobSchedulerContext;
import agi.parallel.infrastructure.Task;
import agi.parallel.infrastructure.TaskProperties;

public class DynamicallyControlConsumedResources {
    public static void main(String[] args) {
        try (IJobScheduler scheduler = new ClusterJobScheduler("localhost")) {
            scheduler.connect();

            Task task = new YieldResourceDemo();
            Job job = scheduler.createJob();
            job.addTask(task);

            job.submit();
            job.waitUntilDone();

            System.out.print(task.getStandardOutput());
        }

        /*
         * The output of the application should resemble:
         * Initial: Consuming 1 Cores
         * After YieldResource: Consuming 0 Cores
         * After ReserveResource: Consuming 1 Cores
         */
    }

    public static class YieldResourceDemo extends Task {
        @Override
        public void execute() {
            IJobSchedulerContext jobSchedulerContext = (IJobSchedulerContext) this.getProperty(TaskProperties.JOB_SCHEDULER_CONTEXT);

            // You can pass 0 to query that amount the task is currently consuming.
            long currentConsumption = jobSchedulerContext.yieldResource(ConsumableResources.CORES, 0);
            System.out.format("Initial: Consuming %s Cores%n", currentConsumption);

            // Let's yield the resource to the global resource poll
            currentConsumption = jobSchedulerContext.yieldResource(ConsumableResources.CORES, 1);
            System.out.format("After YieldResource: Consuming %s Cores%n", currentConsumption);

            // We can also dynamically increase our resource consumption.
            currentConsumption = jobSchedulerContext.reserveResource(ConsumableResources.CORES, 1);
            System.out.format("After ReserveResource: Consuming %s Cores%n", currentConsumption);
        }
    }
}