fix 代码整理

app/api/api_outfit_matcher.py

@@ -29,16 +29,15 @@ def outfit_matcher(request_item: OutfitMatcher):
     service = OutfitMaterTypeAware()
     all_items = request_item["query"] + request_item["database"]
     prepared_feature = {}
+    have_features_data = []
+    no_have_features_data = []
+    temp_data = deepcopy(all_items)
 
     # 连接milvus
     client = MilvusClient(uri=MILVUS_URL, token="root:Milvus", db_name="mixi")
+    try:
         search_data = client.get(collection_name='mixi_outfit', ids=[item['item_name'] for item in all_items])
-
         # 查询数据库，分成两批 需要过模型推理的和不需要的
-    have_features_data = []
-    no_have_features_data = []
-
-    temp_data = deepcopy(all_items)
         for td in temp_data:
             for sd in search_data:
                 if td['item_name'] == sd['item_name']:
@@ -61,9 +60,10 @@ def outfit_matcher(request_item: OutfitMatcher):
             # 存入数据
             res = client.insert(collection_name="mixi_outfit", data=data)
             # 断开连接
-        client.close()
             for d in data:
                 prepared_feature[d['item_name']] = d['features']
+    finally:
+        client.close()
 
     for hfd in have_features_data:
         prepared_feature[hfd['item_name']] = hfd['features']

app/service/similar_match/service.py

@@ -86,7 +86,6 @@ class SimilarMatch:
 
     @RunTime
     def match_features(self):
-        # 连接milvus
         # 连接milvus
         client = MilvusClient(uri=MILVUS_URL, db_name="mixi")
         try: