Examples with TaskClient com.netflix.conductor.client.http.TaskClient used on opensource projects

Example 11 with TaskClient

use of com.netflix.conductor.client.http.TaskClient in project conductor by Netflix.

the class WorkflowTaskCoordinatorTests method testTaskException.

@Test
public void testTaskException() {
    Worker worker = Worker.create("test", task -> {
        throw new NoSuchMethodError();
    });
    TaskClient client = Mockito.mock(TaskClient.class);
    WorkflowTaskCoordinator coordinator = new WorkflowTaskCoordinator.Builder().withWorkers(worker).withThreadCount(1).withWorkerQueueSize(1).withSleepWhenRetry(100000).withUpdateRetryCount(1).withTaskClient(client).withWorkerNamePrefix("test-worker-").build();
    when(client.batchPollTasksInDomain(anyString(), isNull(), anyString(), anyInt(), anyInt())).thenReturn(ImmutableList.of(new Task()));
    when(client.ack(any(), any())).thenReturn(true);
    CountDownLatch latch = new CountDownLatch(1);
    doAnswer(invocation -> {
        assertEquals("test-worker-0", Thread.currentThread().getName());
        Object[] args = invocation.getArguments();
        TaskResult result = (TaskResult) args[0];
        assertEquals(TaskResult.Status.FAILED, result.getStatus());
        latch.countDown();
        return null;
    }).when(client).updateTask(any());
    coordinator.init();
    Uninterruptibles.awaitUninterruptibly(latch);
    Mockito.verify(client).updateTask(any());
}

Example 12 with TaskClient

use of com.netflix.conductor.client.http.TaskClient in project conductor by Netflix.

the class WorkflowTaskCoordinator method init.

/**
 * Starts the polling.
 * Must be called after the constructor {@link #WorkflowTaskCoordinator(EurekaClient, TaskClient, int, int, int, int, Iterable, String)}
 * or the builder {@link Builder#build()} method
 */
public synchronized void init() {
    if (threadCount == -1) {
        threadCount = workers.size();
    }
    logger.info("Initialized the worker with {} threads", threadCount);
    MetricsContainer.incrementInitializationCount(this.getClass().getCanonicalName());
    this.workerQueue = new LinkedBlockingQueue<Runnable>(workerQueueSize);
    AtomicInteger count = new AtomicInteger(0);
    this.executorService = new ThreadPoolExecutor(threadCount, threadCount, 0L, TimeUnit.MILLISECONDS, workerQueue, (runnable) -> {
        Thread thread = new Thread(runnable);
        thread.setName(workerNamePrefix + count.getAndIncrement());
        return thread;
    });
    this.scheduledExecutorService = Executors.newScheduledThreadPool(workers.size());
    workers.forEach(worker -> {
        scheduledExecutorService.scheduleWithFixedDelay(() -> pollForTask(worker), worker.getPollingInterval(), worker.getPollingInterval(), TimeUnit.MILLISECONDS);
    });
}

Example 13 with TaskClient

use of com.netflix.conductor.client.http.TaskClient in project conductor by Netflix.

the class ExclusiveJoinEndToEndTest method setUp.

@BeforeClass
public static void setUp() throws Exception {
    TestEnvironment.setup();
    System.setProperty(ElasticSearchConfiguration.EMBEDDED_PORT_PROPERTY_NAME, "9205");
    System.setProperty(ElasticSearchConfiguration.ELASTIC_SEARCH_URL_PROPERTY_NAME, "localhost:9305");
    System.setProperty(Configuration.EXECUTION_LOCK_ENABLED_PROPERTY_NAME, "false");
    Injector bootInjector = Guice.createInjector(new BootstrapModule());
    Injector serverInjector = Guice.createInjector(bootInjector.getInstance(ModulesProvider.class).get());
    search = serverInjector.getInstance(EmbeddedElasticSearchProvider.class).get().get();
    search.start();
    JettyServer server = new JettyServer(SERVER_PORT, false);
    server.start();
    String apiRoot = String.format("http://localhost:%d/api/", SERVER_PORT);
    taskClient = new TaskClient();
    taskClient.setRootURI(apiRoot);
    workflowClient = new WorkflowClient();
    workflowClient.setRootURI(apiRoot);
    metadataClient = new MetadataClient();
    metadataClient.setRootURI(apiRoot);
}

Example 14 with TaskClient

use of com.netflix.conductor.client.http.TaskClient in project conductor by Netflix.

the class ESRestClientHttpEndToEndTest method setup.

@BeforeClass
public static void setup() throws Exception {
    TestEnvironment.setup();
    System.setProperty(ElasticSearchConfiguration.EMBEDDED_PORT_PROPERTY_NAME, "9203");
    System.setProperty(ElasticSearchConfiguration.ELASTIC_SEARCH_URL_PROPERTY_NAME, "http://localhost:9203");
    System.setProperty(ElasticSearchConfiguration.ELASTIC_SEARCH_INDEX_BATCH_SIZE_PROPERTY_NAME, "1");
    Injector bootInjector = Guice.createInjector(new BootstrapModule());
    Injector serverInjector = Guice.createInjector(bootInjector.getInstance(ModulesProvider.class).get());
    search = serverInjector.getInstance(EmbeddedElasticSearchProvider.class).get().get();
    search.start();
    search.waitForGreenCluster();
    SystemPropertiesElasticSearchConfiguration configuration = new SystemPropertiesElasticSearchConfiguration();
    ElasticSearchRestClientProvider restClientProvider = new ElasticSearchRestClientProvider(configuration);
    elasticSearchAdminClient = restClientProvider.get();
    JettyServer server = new JettyServer(SERVER_PORT, false);
    server.start();
    apiRoot = String.format("http://localhost:%d/api/", SERVER_PORT);
    taskClient = new TaskClient();
    taskClient.setRootURI(apiRoot);
    workflowClient = new WorkflowClient();
    workflowClient.setRootURI(apiRoot);
    metadataClient = new MetadataClient();
    metadataClient.setRootURI(apiRoot);
}

Example 15 with TaskClient

use of com.netflix.conductor.client.http.TaskClient in project conductor by Netflix.

the class HttpEndToEndTest method setup.

@BeforeClass
public static void setup() throws Exception {
    TestEnvironment.setup();
    System.setProperty(ElasticSearchConfiguration.EMBEDDED_PORT_PROPERTY_NAME, "9201");
    System.setProperty(ElasticSearchConfiguration.ELASTIC_SEARCH_URL_PROPERTY_NAME, "localhost:9301");
    System.setProperty(Configuration.EXECUTION_LOCK_ENABLED_PROPERTY_NAME, "false");
    Injector bootInjector = Guice.createInjector(new BootstrapModule());
    Injector serverInjector = Guice.createInjector(bootInjector.getInstance(ModulesProvider.class).get());
    search = serverInjector.getInstance(EmbeddedElasticSearchProvider.class).get().get();
    search.start();
    JettyServer server = new JettyServer(SERVER_PORT, false);
    server.start();
    apiRoot = String.format("http://localhost:%d/api/", SERVER_PORT);
    taskClient = new TaskClient();
    taskClient.setRootURI(apiRoot);
    workflowClient = new WorkflowClient();
    workflowClient.setRootURI(apiRoot);
    metadataClient = new MetadataClient();
    metadataClient.setRootURI(apiRoot);
}