另外一种添加选项的方法是实现BeanPostProcessor类。它提供了两个方法:
postProcessAfterInitialization
和postProcessBeforeInitialization
工厂钩子,允许自定义修改新的bean实例 - 例如,检查标记接口或使用代理包装bean。通常,通过标记接口或类似方式填充bean的后处理器将实现postProcessBeforeInitialization(java.lang.Object, java.lang.String),而将bean包装为代理的后处理器通常会实现postProcessAfterInitialization(java.lang.Object, java.lang.String)。
我正在使用Spring Boot与Spring Kafka,并且只想更改本地配置文件。
在我的代码示例中,我使用它来覆盖Kafka位置属性,因为对于SSL,它不会从类路径中读取。
所以这就是代码:
import io.confluent.kafka.schemaregistry.client.SchemaRegistryClientConfig;
import java.io.IOException;
import java.util.Arrays;
import lombok.RequiredArgsConstructor;
import lombok.SneakyThrows;
import org.apache.kafka.common.config.SslConfigs;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.boot.autoconfigure.kafka.KafkaProperties;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
@Configuration
@RequiredArgsConstructor
public class KafkaConfiguration implements BeanPostProcessor {
@Value("${spring.kafka.ssl.key-store-location:}")
private Resource keyStoreResource;
@Value("${spring.kafka.properties.schema.registry.ssl.truststore.location:}")
private Resource trustStoreResource;
private final Environment environment;
@SneakyThrows
@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
if (bean instanceof KafkaProperties) {
KafkaProperties kafkaProperties = (KafkaProperties) bean;
if(isLocalProfileActive()) {
configureStoreLocation(kafkaProperties);
}
}
return BeanPostProcessor.super.postProcessAfterInitialization(bean, beanName);
}
private boolean isLocalProfileActive() {
return Arrays.stream(environment.getActiveProfiles()).anyMatch(profile -> "local".equals(profile));
}
private void configureStoreLocation(KafkaProperties kafkaProperties) throws IOException {
kafkaProperties.getSsl().setKeyStoreLocation(new FileSystemResource(keyStoreResource.getFile().getAbsolutePath()));
kafkaProperties.getProperties().put(SchemaRegistryClientConfig.CLIENT_NAMESPACE + SslConfigs.SSL_KEYSTORE_LOCATION_CONFIG, keyStoreResource.getFile().getAbsolutePath());
kafkaProperties.getSsl().setTrustStoreLocation(new FileSystemResource(trustStoreResource.getFile().getAbsolutePath()));
kafkaProperties.getProperties().put(SchemaRegistryClientConfig.CLIENT_NAMESPACE + SslConfigs.SSL_TRUSTSTORE_LOCATION_CONFIG, trustStoreResource.getFile().getAbsolutePath());
}
}
这样我就可以在我的属性文件中写入:
spring.kafka.ssl.key-store-location=classpath:mykeystore.jks
代码将从中获取绝对路径并进行设置。还可以根据配置文件进行过滤。
需要注意的是,BeanPostProcessor会对每个bean运行,所以请确保过滤您想要的内容。
Properties
类来自哪里? - Ben Creasyjava.util
有一个 Properties 类。 - Ajit Soman