Skip to content

Commit c1fe3bd

Browse files
committed
add sasl when there is kafka config file
1 parent 8a0fc0f commit c1fe3bd

File tree

2 files changed

+20
-1
lines changed

2 files changed

+20
-1
lines changed

plugins/kafkaplugin/build.gradle

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,4 +21,7 @@ dependencies {
2121
compileOnly ("org.projectlombok:lombok:1.16.18") {
2222
exclude group: "org.slf4j"
2323
}
24+
25+
compile group: 'com.typesafe', name: 'config', version: '1.3.2'
26+
2427
}

plugins/kafkaplugin/src/main/java/org/tron/eventplugin/MessageSenderImpl.java

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
11
package org.tron.eventplugin;
2+
import com.typesafe.config.Config;
3+
import com.typesafe.config.ConfigFactory;
4+
import java.io.File;
25
import org.apache.kafka.clients.producer.*;
36
import org.slf4j.Logger;
47
import org.slf4j.LoggerFactory;
@@ -94,7 +97,20 @@ private KafkaProducer createProducer(int eventType){
9497
props.put("bootstrap.servers", this.serverAddress);
9598
props.put("key.serializer", "org.apache.kafka.common.serialization.StringSerializer");
9699
props.put("value.serializer", "org.apache.kafka.common.serialization.StringSerializer");
97-
100+
String defaultConfig = "kafka.config";
101+
File configFile = new File(defaultConfig);
102+
if(configFile.exists()) {
103+
Config config = ConfigFactory.load(defaultConfig);
104+
if (config.hasPath("authorization.user") && config.hasPath("authorization.passwd")) {
105+
String user = config.getString("authorization.user");
106+
String passwd = config.getString("authorization.passwd");
107+
props.put("security.protocol", "SASL_PLAINTEXT");
108+
props.put("sasl.mechanism", "SCRAM-SHA-512");
109+
props.put("sasl.jaas.config",
110+
"org.apache.kafka.common.security.scram.ScramLoginModule required username=\"" +
111+
user + "\" password=\"" + passwd + "\"");
112+
}
113+
}
98114
producer = new KafkaProducer<String, String>(props);
99115

100116
producerMap.put(eventType, producer);

0 commit comments

Comments
 (0)