jffwang
/
hetu-core
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add kafka test conf

This commit is contained in:
Zhang Jianming 2022-06-23 22:46:13 +08:00
parent e5817f83d7
commit d20985a5cb
3 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
presto-kafka/src/main/java/io/prestosql/plugin/kafka/KafkaConnectorConfig.java
View File

@ -108,7 +108,7 @@ public class KafkaConnectorConfig
/**
* whether to use kerberos
*/
private boolean kerberosOn;
private String kerberosOn;
public String getKrb5Conf()
{
@ -206,7 +206,7 @@ public class KafkaConnectorConfig
return this;
}
public boolean isKerberosOn()
public String isKerberosOn()
{
return kerberosOn;
}
@ -216,7 +216,7 @@ public class KafkaConnectorConfig
defaultValue = "false",
required = true)
@Config("kerberos.on")
public KafkaConnectorConfig setKerberosOn(boolean kerberosOn)
public KafkaConnectorConfig setKerberosOn(String kerberosOn)
{
this.kerberosOn = kerberosOn;
return this;

4
presto-kafka/src/main/java/io/prestosql/plugin/kafka/KafkaSimpleConsumerManager.java
View File

@ -47,7 +47,7 @@ public class KafkaSimpleConsumerManager
private final int connectTimeoutMillis;
private final int bufferSizeBytes;
private final boolean kerberosOn;
private final String kerberosOn;
private final String loginConfig;
private final String krb5Conf;
private String groupId;
@ -116,7 +116,7 @@ public class KafkaSimpleConsumerManager
{
log.info("Creating new SaslConsumer for %s", host);
Properties props = new Properties();
if (kerberosOn) {
if ("true".equalsIgnoreCase(kerberosOn)) {
System.setProperty("java.security.auth.login.config", loginConfig);
System.setProperty("java.security.krb5.conf", krb5Conf);
props.put("security.protocol", securityProtocol);

16
presto-kafka/src/test/java/io/prestosql/plugin/kafka/TestKafkaConnectorConfig.java
View File

@ -32,13 +32,13 @@ public class TestKafkaConnectorConfig
.setDefaultSchema("default")
.setTableNames("")
.setTableDescriptionDir(new File("etc/kafka/"))
.setGroupId("test")
.setKerberosOn(false)
.setSecurityProtocol("SASL_PLAINTEXT")
.setKrb5Conf("/etc/krb5.conf")
.setLoginConfig("/etc/kafka_client_jaas.conf")
.setSaslKerberosServiceName("kafka")
.setSaslMechanism("GSSAPI")
.setGroupId(null)
.setKerberosOn(null)
.setSecurityProtocol(null)
.setKrb5Conf(null)
.setLoginConfig(null)
.setSaslKerberosServiceName(null)
.setSaslMechanism(null)
.setHideInternalColumns(true));
}
@ -74,7 +74,7 @@ public class TestKafkaConnectorConfig
.setLoginConfig("/etc/kafka_client_jaas.conf")
.setSaslKerberosServiceName("kafka")
.setSaslMechanism("GSSAPI")
.setKerberosOn(false)
.setKerberosOn("false")
.setSecurityProtocol("SASL_PLAINTEXT")
.setHideInternalColumns(false);