DBZ-8166 Provide explicit engine type to avoid raw types
This commit is contained in:
parent
be9c70d631
commit
17a3908403
@ -109,7 +109,7 @@ public abstract class AbstractConnectorTest implements Testing {
|
||||
private static final String TEST_PROPERTY_PREFIX = "debezium.test.";
|
||||
|
||||
private ExecutorService executor;
|
||||
protected TestingDebeziumEngine engine;
|
||||
protected TestingDebeziumEngine<SourceRecord> engine;
|
||||
protected BlockingQueue<SourceRecord> consumedLines;
|
||||
protected long pollTimeoutInMs = TimeUnit.SECONDS.toMillis(10);
|
||||
protected final Logger logger = LoggerFactory.getLogger(getClass());
|
||||
@ -331,7 +331,7 @@ protected void start(Class<? extends SourceConnector> connectorClass, Configurat
|
||||
* @param changeConsumer {@link io.debezium.engine.DebeziumEngine.ChangeConsumer} invoked when a record arrives and is stored in the queue
|
||||
*/
|
||||
protected void start(Class<? extends SourceConnector> connectorClass, Configuration connectorConfig,
|
||||
DebeziumEngine.ChangeConsumer changeConsumer) {
|
||||
DebeziumEngine.ChangeConsumer<SourceRecord> changeConsumer) {
|
||||
start(connectorClass, connectorConfig, loggingCompletion(), null, x -> {
|
||||
}, true, changeConsumer);
|
||||
}
|
||||
@ -369,7 +369,7 @@ protected void start(Class<? extends SourceConnector> connectorClass, Configurat
|
||||
*/
|
||||
protected void start(Class<? extends SourceConnector> connectorClass, Configuration connectorConfig,
|
||||
DebeziumEngine.CompletionCallback callback, Predicate<SourceRecord> isStopRecord,
|
||||
Consumer<SourceRecord> recordArrivedListener, boolean ignoreRecordsAfterStop, DebeziumEngine.ChangeConsumer changeConsumer) {
|
||||
Consumer<SourceRecord> recordArrivedListener, boolean ignoreRecordsAfterStop, DebeziumEngine.ChangeConsumer<SourceRecord> changeConsumer) {
|
||||
Configuration config = Configuration.copy(connectorConfig)
|
||||
.with(EmbeddedEngineConfig.ENGINE_NAME, "testing-connector")
|
||||
.with(EmbeddedEngineConfig.CONNECTOR_CLASS, connectorClass.getName())
|
||||
@ -413,7 +413,7 @@ public void connectorStopped() {
|
||||
};
|
||||
|
||||
// Create the connector ...
|
||||
DebeziumEngine.Builder builder = createEngineBuilder();
|
||||
DebeziumEngine.Builder<SourceRecord> builder = createEngineBuilder();
|
||||
builder.using(config.asProperties())
|
||||
.notifying(getConsumer(isStopRecord, recordArrivedListener, ignoreRecordsAfterStop))
|
||||
.using(this.getClass().getClassLoader())
|
||||
@ -444,11 +444,11 @@ public void connectorStopped() {
|
||||
}
|
||||
}
|
||||
|
||||
protected DebeziumEngine.Builder createEngineBuilder() {
|
||||
protected DebeziumEngine.Builder<SourceRecord> createEngineBuilder() {
|
||||
return new EmbeddedEngine.EngineBuilder();
|
||||
}
|
||||
|
||||
protected TestingDebeziumEngine createEngine(DebeziumEngine.Builder builder) {
|
||||
protected TestingDebeziumEngine<SourceRecord> createEngine(DebeziumEngine.Builder<SourceRecord> builder) {
|
||||
return new TestingEmbeddedEngine((EmbeddedEngine) builder.build());
|
||||
}
|
||||
|
||||
|
@ -8,10 +8,13 @@
|
||||
import java.io.IOException;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
import org.apache.kafka.connect.source.SourceRecord;
|
||||
import org.apache.kafka.connect.source.SourceTask;
|
||||
|
||||
/**
|
||||
* Implementation of {@link TestingDebeziumEngine} for {@link EmbeddedEngine}.
|
||||
*/
|
||||
public class TestingEmbeddedEngine implements TestingDebeziumEngine {
|
||||
public class TestingEmbeddedEngine implements TestingDebeziumEngine<SourceRecord> {
|
||||
|
||||
private final EmbeddedEngine engine;
|
||||
|
||||
@ -30,7 +33,7 @@ public void close() throws IOException {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void runWithTask(Consumer consumer) {
|
||||
public void runWithTask(Consumer<SourceTask> consumer) {
|
||||
engine.runWithTask(consumer);
|
||||
}
|
||||
|
||||
|
@ -5,6 +5,8 @@
|
||||
*/
|
||||
package io.debezium.embedded.async;
|
||||
|
||||
import org.apache.kafka.connect.source.SourceRecord;
|
||||
|
||||
import io.debezium.embedded.AbstractConnectorTest;
|
||||
import io.debezium.embedded.TestingDebeziumEngine;
|
||||
import io.debezium.engine.DebeziumEngine;
|
||||
@ -17,13 +19,13 @@
|
||||
public class AbstractAsyncEngineConnectorTest extends AbstractConnectorTest {
|
||||
|
||||
@Override
|
||||
protected DebeziumEngine.Builder createEngineBuilder() {
|
||||
protected DebeziumEngine.Builder<SourceRecord> createEngineBuilder() {
|
||||
return new AsyncEmbeddedEngine.AsyncEngineBuilder();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected TestingDebeziumEngine createEngine(DebeziumEngine.Builder builder) {
|
||||
return new TestingAsyncEmbeddedEngine((AsyncEmbeddedEngine) builder.build());
|
||||
protected TestingDebeziumEngine<SourceRecord> createEngine(DebeziumEngine.Builder<SourceRecord> builder) {
|
||||
return new TestingAsyncEmbeddedEngine<SourceRecord>((AsyncEmbeddedEngine<SourceRecord>) builder.build());
|
||||
}
|
||||
|
||||
protected DebeziumEngine.Signaler getSignaler() {
|
||||
|
@ -8,6 +8,8 @@
|
||||
import java.io.IOException;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
import org.apache.kafka.connect.source.SourceTask;
|
||||
|
||||
import io.debezium.embedded.TestingDebeziumEngine;
|
||||
|
||||
/**
|
||||
@ -15,10 +17,10 @@
|
||||
*
|
||||
* @author vjuranek
|
||||
*/
|
||||
public class TestingAsyncEmbeddedEngine implements TestingDebeziumEngine {
|
||||
private final AsyncEmbeddedEngine engine;
|
||||
public class TestingAsyncEmbeddedEngine<T> implements TestingDebeziumEngine<T> {
|
||||
private final AsyncEmbeddedEngine<T> engine;
|
||||
|
||||
public TestingAsyncEmbeddedEngine(AsyncEmbeddedEngine engine) {
|
||||
public TestingAsyncEmbeddedEngine(AsyncEmbeddedEngine<T> engine) {
|
||||
this.engine = engine;
|
||||
}
|
||||
|
||||
@ -33,7 +35,7 @@ public void close() throws IOException {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void runWithTask(Consumer consumer) {
|
||||
public void runWithTask(Consumer<SourceTask> consumer) {
|
||||
engine.runWithTask(consumer);
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user