fix: check collection id match for possible alias switch

Signed-off-by: MrPresent-Han <[email protected]>

pymilvus/orm/iterator.py

@@ -101,6 +101,7 @@ def __init__(
     ) -> QueryIterator:
         self._conn = connection
         self._collection_name = collection_name
+        self.__set_up_collection_id()
         self._output_fields = output_fields
         self._partition_names = partition_names
         self._schema = schema
@@ -120,6 +121,10 @@ def __init__(
         self.__set_up_ts_cp()
         self.__seek_to_offset()
 
+    def __set_up_collection_id(self):
+        col_desc = self._conn.describe_collection(self._collection_name, timeout=60.0)
+        self._collection_id = col_desc.get("collection_id")
+
     def __seek_to_offset(self):
         # read pk cursor from cp file, no need to seek offset
         if self._next_id is not None:
@@ -142,6 +147,7 @@ def seek_offset_by_batch(batch: int, expr: str) -> int:
                     timeout=self._timeout,
                     **seek_params,
                 )
+                self.__check_collection_match(res)
                 self.__update_cursor(res)
                 return len(res)
 
@@ -334,6 +340,7 @@ def next(self):
             ret = res[0 : min(self._kwargs[BATCH_SIZE], len(res))]
 
         ret = self.__check_reached_limit(ret)
+        self.__check_collection_match(res)
         self.__update_cursor(ret)
         io_operation(self.__save_pk_cursor, "failed to save pk cursor")
         self._returned_count += len(ret)
@@ -374,6 +381,19 @@ def __setup_next_expr(self) -> str:
             return filtered_pk_str
         return "(" + current_expr + ")" + " and " + filtered_pk_str
 
+    def __check_collection_match(self, res: List):
+        res_collection_id = res[-1]["collection_id"]
+        if (
+            res_collection_id is not None
+            and res_collection_id > 0
+            and res_collection_id != self._collection_id
+        ):
+            raise MilvusException(
+                message="collection_id in the result is not the "
+                "same as the inited collection id, the alias may be changed, cut off"
+                "iterator connection"
+            )
+
     def __update_cursor(self, res: List) -> None:
         if len(res) == 0:
             return