18
18
package org .apache .rocketmq .connect .kafka .connector ;
19
19
20
20
import io .openmessaging .KeyValue ;
21
- import io .openmessaging .connect .runtime .common .ConnectKeyValue ;
22
- import io .openmessaging .connect .runtime .config .RuntimeConfigDefine ;
23
21
import io .openmessaging .connector .api .Task ;
24
22
import io .openmessaging .connector .api .source .SourceConnector ;
25
- import org .apache .rocketmq .connect .kafka .Config ;
23
+ import io .openmessaging .internal .DefaultKeyValue ;
24
+ import org .apache .rocketmq .connect .kafka .config .ConfigDefine ;
26
25
import org .slf4j .Logger ;
27
26
import org .slf4j .LoggerFactory ;
28
27
29
28
import java .util .ArrayList ;
30
29
import java .util .List ;
31
30
32
- public class KafkaSourceConnector extends SourceConnector {
31
+ public class KafkaSourceConnector extends SourceConnector {
33
32
private static final Logger log = LoggerFactory .getLogger (KafkaSourceConnector .class );
34
33
35
34
private KeyValue connectConfig ;
@@ -42,12 +41,12 @@ public KafkaSourceConnector() {
42
41
public String verifyAndSetConfig (KeyValue config ) {
43
42
44
43
log .info ("KafkaSourceConnector verifyAndSetConfig enter" );
45
- for ( String key : config .keySet ()) {
44
+ for (String key : config .keySet ()) {
46
45
log .info ("connector verifyAndSetConfig: key:{}, value:{}" , key , config .getString (key ));
47
46
}
48
47
49
- for (String requestKey : Config .REQUEST_CONFIG ){
50
- if (!config .containsKey (requestKey )){
48
+ for (String requestKey : ConfigDefine .REQUEST_CONFIG ) {
49
+ if (!config .containsKey (requestKey )) {
51
50
return "Request Config key: " + requestKey ;
52
51
}
53
52
}
@@ -82,20 +81,22 @@ public Class<? extends Task> taskClass() {
82
81
83
82
@ Override
84
83
public List <KeyValue > taskConfigs () {
84
+ if (connectConfig == null ) {
85
+ return new ArrayList <KeyValue >();
86
+ }
85
87
86
88
log .info ("Source Connector taskConfigs enter" );
87
89
List <KeyValue > configs = new ArrayList <>();
88
- int task_num = connectConfig .getInt (Config .TASK_NUM );
90
+ int task_num = connectConfig .getInt (ConfigDefine .TASK_NUM );
89
91
log .info ("Source Connector taskConfigs: task_num:" + task_num );
90
- for (int i =0 ; i < task_num ; ++i ) {
91
- KeyValue config = new ConnectKeyValue ();
92
- config .put (Config .BOOTSTRAP_SERVER , connectConfig .getString (Config .BOOTSTRAP_SERVER ));
93
- config .put (Config .TOPICS , connectConfig .getString (Config .TOPICS ));
94
- config .put (Config .GROUP_ID , connectConfig .getString (Config .GROUP_ID ));
95
-
96
- config .put (RuntimeConfigDefine .CONNECTOR_CLASS , connectConfig .getString (RuntimeConfigDefine .CONNECTOR_CLASS ));
97
- config .put (RuntimeConfigDefine .SOURCE_RECORD_CONVERTER , connectConfig .getString (RuntimeConfigDefine .SOURCE_RECORD_CONVERTER ));
98
- config .put (RuntimeConfigDefine .OMS_DRIVER_URL , connectConfig .getString (RuntimeConfigDefine .OMS_DRIVER_URL ));
92
+ for (int i = 0 ; i < task_num ; ++i ) {
93
+ KeyValue config = new DefaultKeyValue ();
94
+ config .put (ConfigDefine .BOOTSTRAP_SERVER , connectConfig .getString (ConfigDefine .BOOTSTRAP_SERVER ));
95
+ config .put (ConfigDefine .TOPICS , connectConfig .getString (ConfigDefine .TOPICS ));
96
+ config .put (ConfigDefine .GROUP_ID , connectConfig .getString (ConfigDefine .GROUP_ID ));
97
+
98
+ config .put (ConfigDefine .CONNECTOR_CLASS , connectConfig .getString (ConfigDefine .CONNECTOR_CLASS ));
99
+ config .put (ConfigDefine .SOURCE_RECORD_CONVERTER , connectConfig .getString (ConfigDefine .SOURCE_RECORD_CONVERTER ));
99
100
configs .add (config );
100
101
}
101
102
return configs ;
0 commit comments