From 65e89609a2b015fe5915fe384d2843e58ea5072e Mon Sep 17 00:00:00 2001
From: xgc <javpower@163.com>
Date: Thu, 1 Aug 2024 11:04:56 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=84=E7=90=86=E5=88=9D=E5=A7=8B=E5=8C=96cl?=
 =?UTF-8?q?ient=E5=85=BC=E5=AE=B9Boot3=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../milvus/plus/service/MilvusInit.java       | 34 +++++++++----------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java b/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java
index 8abae15..40452b1 100644
--- a/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java
+++ b/milvus-plus-boot-starter/src/main/java/org/dromara/milvus/plus/service/MilvusInit.java
@@ -5,30 +5,36 @@ import org.dromara.milvus.plus.config.MilvusPropertiesConfiguration;
 import org.dromara.milvus.plus.log.LogLevelController;
 import org.dromara.milvus.plus.model.MilvusProperties;
 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.stereotype.Service;
 
-import javax.annotation.PostConstruct;
-import javax.annotation.PreDestroy;
-
 @Service
-public class MilvusInit extends AbstractMilvusClientBuilder {
+public class MilvusInit extends AbstractMilvusClientBuilder implements InitializingBean, DisposableBean {
 
-    @Autowired
-    private MilvusPropertiesConfiguration milvusPropertiesConfiguration;
+    private final MilvusPropertiesConfiguration milvusPropertiesConfiguration;
 
     private MilvusClientV2 client;
 
-    // Spring会调用这个方法来初始化client
-    @PostConstruct
+    public MilvusInit(MilvusPropertiesConfiguration milvusPropertiesConfiguration) {
+        this.milvusPropertiesConfiguration = milvusPropertiesConfiguration;
+    }
+
+    @Override
+    public void afterPropertiesSet() {
+        initialize();
+    }
+    @Override
+    public void destroy() throws Exception {
+        super.close();
+    }
+
     public void initialize() {
         printBanner();
-
         LogLevelController.setLoggingEnabledForPackage("org.dromara.milvus.plus",
                 milvusPropertiesConfiguration.isOpenLog(),
                 milvusPropertiesConfiguration.getLogLevel());
-
         MilvusProperties milvusProperties = new MilvusProperties();
         BeanUtils.copyProperties(milvusPropertiesConfiguration, milvusProperties);
         super.setProperties(milvusProperties);
@@ -36,12 +42,6 @@ public class MilvusInit extends AbstractMilvusClientBuilder {
         client = getClient();
     }
 
-    // Spring会调用这个方法来关闭client
-    @PreDestroy
-    public void close() throws InterruptedException {
-        super.close();
-    }
-
     @Bean
     public MilvusClientV2 milvusClientV2() {
         return client;