|
19 | 19 | import kafka.server.KafkaConfig;
|
20 | 20 | import kafka.server.KafkaServerStartable;
|
21 | 21 | import org.apache.commons.io.FileUtils;
|
22 |
| -import org.apache.logging.log4j.Logger; |
23 | 22 | import org.apache.logging.log4j.LogManager;
|
| 23 | +import org.apache.logging.log4j.Logger; |
24 | 24 |
|
25 |
| -import java.io.File; |
26 | 25 | import java.io.IOException;
|
27 | 26 | import java.nio.file.Files;
|
28 |
| -import java.util.Properties; |
29 | 27 |
|
30 | 28 | /**
|
31 | 29 | * Copy of https://github.com/chbatey/kafka-unit/blob/master/src/main/java/info/batey/kafka/unit/KafkaUnit.java
|
32 | 30 | * Some codes have been modified for testing from the copied code.
|
33 | 31 | */
|
34 |
| -public class KafkaUnitServer { |
35 |
| - |
36 |
| - private static final Logger logger = LogManager.getLogger(KafkaUnitServer.class); |
37 |
| - private String zookeeperString; |
38 |
| - private String brokerString; |
39 |
| - private int zkPort; |
40 |
| - private int brokerPort; |
41 |
| - private Properties kafkaBrokerConfig; |
42 |
| - private int zkMaxConnections; |
| 32 | +public class Kafka2UnitServer extends KafkaUnitServer { |
| 33 | + private static final Logger logger = LogManager.getLogger(Kafka2UnitServer.class); |
43 | 34 | private KafkaServerStartable broker;
|
44 |
| - private ZookeeperUnitServer zookeeper; |
45 |
| - private File logDir; |
46 | 35 |
|
47 |
| - public KafkaUnitServer(int zkPort, int brokerPort) { |
| 36 | + |
| 37 | + public Kafka2UnitServer(int zkPort, int brokerPort) { |
48 | 38 | this(zkPort, brokerPort, 16);
|
49 | 39 | }
|
50 | 40 |
|
51 |
| - public KafkaUnitServer(int zkPort, int brokerPort, int zkMaxConnections) { |
52 |
| - this.kafkaBrokerConfig = new Properties(); |
53 |
| - this.zkPort = zkPort; |
54 |
| - this.brokerPort = brokerPort; |
55 |
| - this.zookeeperString = "localhost:" + zkPort; |
56 |
| - this.brokerString = "localhost:" + brokerPort; |
57 |
| - this.zkMaxConnections = zkMaxConnections; |
| 41 | + public Kafka2UnitServer(int zkPort, int brokerPort, int zkMaxConnections) { |
| 42 | + super(zkPort, brokerPort, zkMaxConnections); |
58 | 43 | }
|
59 | 44 |
|
60 | 45 | public void startup() {
|
|
0 commit comments