admin/MilvusPlus
1
0
Fork 0
mirror of https://gitee.com/dromara/MilvusPlus.git synced 2025-12-07 09:28:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

处理初始化client兼容Boot3问题

Browse Source
This commit is contained in:
xgc 2024-08-01 11:04:56 +08:00
parent 9d24794117
commit 65e89609a2
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java
View File

@ -5,30 +5,36 @@ import org.dromara.milvus.plus.config.MilvusPropertiesConfiguration;
import org.dromara.milvus.plus.log.LogLevelController; import org.dromara.milvus.plus.log.LogLevelController;
import org.dromara.milvus.plus.model.MilvusProperties; import org.dromara.milvus.plus.model.MilvusProperties;
import org.springframework.beans.BeanUtils; import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
@Service @Service
public class MilvusInit extends AbstractMilvusClientBuilder { public class MilvusInit extends AbstractMilvusClientBuilder implements InitializingBean, DisposableBean {
@Autowired private final MilvusPropertiesConfiguration milvusPropertiesConfiguration;
private MilvusPropertiesConfiguration milvusPropertiesConfiguration;
private MilvusClientV2 client; private MilvusClientV2 client;
// Spring会调用这个方法来初始化client public MilvusInit(MilvusPropertiesConfiguration milvusPropertiesConfiguration) {
@PostConstruct this.milvusPropertiesConfiguration = milvusPropertiesConfiguration;
}
@Override
public void afterPropertiesSet() {
initialize();
}
@Override
public void destroy() throws Exception {
super.close();
}
public void initialize() { public void initialize() {
printBanner(); printBanner();
LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus", LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus",
milvusPropertiesConfiguration.isOpenLog(), milvusPropertiesConfiguration.isOpenLog(),
milvusPropertiesConfiguration.getLogLevel()); milvusPropertiesConfiguration.getLogLevel());
MilvusProperties milvusProperties = new MilvusProperties(); MilvusProperties milvusProperties = new MilvusProperties();
BeanUtils.copyProperties(milvusPropertiesConfiguration, milvusProperties); BeanUtils.copyProperties(milvusPropertiesConfiguration, milvusProperties);
super.setProperties(milvusProperties); super.setProperties(milvusProperties);
@ -36,12 +42,6 @@ public class MilvusInit extends AbstractMilvusClientBuilder {
client = getClient(); client = getClient();
} }
// Spring会调用这个方法来关闭client
@PreDestroy
public void close() throws InterruptedException {
super.close();
}
@Bean @Bean
public MilvusClientV2 milvusClientV2() { public MilvusClientV2 milvusClientV2() {
return client; return client;