Examples with Mapper org.talend.sdk.component.runtime.input.Mapper used on opensource projects

Example 1 with Mapper

use of org.talend.sdk.component.runtime.input.Mapper in project component-runtime by Talend.

the class ComponentExtensionTest method manualMapper.

@Test
void manualMapper() {
    final Mapper mapper = handler.createMapper(Source.class, new Source.Config() {

        {
            values = asList("a", "b");
        }
    });
    assertFalse(mapper.isStream());
    final Input input = mapper.create();
    assertEquals("a", input.next());
    assertEquals("b", input.next());
    assertNull(input.next());
}

Example 2 with Mapper

use of org.talend.sdk.component.runtime.input.Mapper in project component-runtime by Talend.

the class ComponentExtensionTest method sourceCollectorParallel.

@Test
void sourceCollectorParallel() {
    final CountDownLatch latch = new CountDownLatch(1);
    final Mapper mapper = new PartitionMapperImpl() {

        @Override
        public long assess() {
            return 2;
        }

        @Override
        public List<Mapper> split(final long desiredSize) {
            assertEquals(1, desiredSize);
            return asList(this, this);
        }

        @Override
        public Input create() {
            return new InputImpl() {

                private final AtomicBoolean done = new AtomicBoolean();

                @Override
                public Object next() {
                    try {
                        latch.await(1, MINUTES);
                    } catch (final InterruptedException e) {
                        Thread.interrupted();
                        fail();
                    }
                    return done.compareAndSet(false, true) ? Thread.currentThread().getName() : null;
                }

                @Override
                protected Stream<Method> findMethods(final Class<? extends Annotation> marker) {
                    return Stream.empty();
                }
            };
        }

        @Override
        protected Stream<Method> findMethods(final Class<? extends Annotation> marker) {
            return Stream.empty();
        }
    };
    latch.countDown();
    final Stream<String> collect = handler.collect(String.class, mapper, 2, 2);
    final List<String> threads = collect.collect(toList());
    assertEquals(threads.toString(), 2, threads.size());
    threads.forEach(n -> assertTrue(n, n.startsWith("ComponentExtension-pool-")));
}

Example 3 with Mapper

use of org.talend.sdk.component.runtime.input.Mapper in project component-runtime by Talend.

the class SimpleComponentRuleTest method manualMapper.

@Test
public void manualMapper() {
    final Mapper mapper = COMPONENT_FACTORY.createMapper(Source.class, new Source.Config() {

        {
            values = asList("a", "b");
        }
    });
    assertFalse(mapper.isStream());
    final Input input = mapper.create();
    assertEquals("a", input.next());
    assertEquals("b", input.next());
    assertNull(input.next());
}

Example 4 with Mapper

use of org.talend.sdk.component.runtime.input.Mapper in project component-runtime by Talend.

the class SimpleComponentRuleTest method sourceCollector.

@Test
public void sourceCollector() {
    final Mapper mapper = COMPONENT_FACTORY.createMapper(Source.class, new Source.Config() {

        {
            values = asList("a", "b");
        }
    });
    assertEquals(asList("a", "b"), COMPONENT_FACTORY.collectAsList(String.class, mapper));
}

Example 5 with Mapper

use of org.talend.sdk.component.runtime.input.Mapper in project component-runtime by Talend.

the class SimpleComponentRuleTest method sourceCollectorParallel.

@Test
public void sourceCollectorParallel() {
    final CountDownLatch latch = new CountDownLatch(1);
    final Mapper mapper = new PartitionMapperImpl() {

        @Override
        public long assess() {
            return 2;
        }

        @Override
        public List<Mapper> split(final long desiredSize) {
            assertEquals(1, desiredSize);
            return asList(this, this);
        }

        @Override
        public Input create() {
            return new InputImpl() {

                private final AtomicBoolean done = new AtomicBoolean();

                @Override
                public Object next() {
                    try {
                        latch.await(1, MINUTES);
                    } catch (final InterruptedException e) {
                        Thread.interrupted();
                        fail();
                    }
                    return done.compareAndSet(false, true) ? Thread.currentThread().getName() : null;
                }

                @Override
                protected Stream<Method> findMethods(final Class<? extends Annotation> marker) {
                    return Stream.empty();
                }
            };
        }

        @Override
        protected Stream<Method> findMethods(final Class<? extends Annotation> marker) {
            return Stream.empty();
        }
    };
    latch.countDown();
    final Stream<String> collect = COMPONENT_FACTORY.collect(String.class, mapper, 2, 2);
    final List<String> threads = collect.collect(toList());
    assertEquals(threads.toString(), 2, threads.size());
    threads.forEach(n -> assertTrue(n, n.startsWith("SimpleComponentRule-pool-")));
}