This commit is contained in:
孙玉朝
2024-11-26 11:08:48 +08:00
committed by GitHub

View File

@@ -38,6 +38,8 @@ import static org.apache.rocketmq.client.ClientConfig.SEND_MESSAGE_WITH_VIP_CHAN
@Configuration @Configuration
@ConfigurationProperties(prefix = "rocketmq.config") @ConfigurationProperties(prefix = "rocketmq.config")
public class RMQConfigure { public class RMQConfigure {
private static final String ROCKET_MQ_AK_PROPERTY = "rocketmq.config.accessKey";
private static final String ROCKET_MQ_SK_PROPERTY = "rocketmq.config.secretKey";
private Logger logger = LoggerFactory.getLogger(RMQConfigure.class); private Logger logger = LoggerFactory.getLogger(RMQConfigure.class);
//use rocketmq.namesrv.addr first,if it is empty,than use system proerty or system env //use rocketmq.namesrv.addr first,if it is empty,than use system proerty or system env
@@ -54,9 +56,9 @@ public class RMQConfigure {
private boolean loginRequired = false; private boolean loginRequired = false;
private String accessKey; private String accessKey = System.getProperty(ROCKET_MQ_AK_PROPERTY, System.getenv("ROCKET_MQ_AK"));
private String secretKey; private String secretKey = System.getProperty(ROCKET_MQ_SK_PROPERTY, System.getenv("ROCKET_MQ_SK"));
private boolean useTLS = false; private boolean useTLS = false;
@@ -71,7 +73,11 @@ public class RMQConfigure {
} }
public void setAccessKey(String accessKey) { public void setAccessKey(String accessKey) {
this.accessKey = accessKey; if (StringUtils.isNotBlank(accessKey)) {
this.accessKey = accessKey;
System.setProperty(ROCKET_MQ_AK_PROPERTY, accessKey);
logger.info("setAccessKey accessKey={}", accessKey);
}
} }
public String getSecretKey() { public String getSecretKey() {
@@ -79,7 +85,11 @@ public class RMQConfigure {
} }
public void setSecretKey(String secretKey) { public void setSecretKey(String secretKey) {
this.secretKey = secretKey; if (StringUtils.isNotBlank(secretKey)) {
this.secretKey = secretKey;
System.setProperty(ROCKET_MQ_SK_PROPERTY, secretKey);
logger.info("setAccessKey accessKey={}", secretKey);
}
} }
public String getNamesrvAddr() { public String getNamesrvAddr() {