add keep alive

Author: ohmayr

gapic/utils/cache.py

@@ -98,12 +98,14 @@ def wrapper(self, *, collisions, **kwargs):
         visited = kwargs.get('visited_messages')
         vis_key = tuple(sorted(id(m) for m in visited)) if visited else None
 
-        key = (self, col_key, skip_key, vis_key)
+        key = (id(self), col_key, skip_key, vis_key)
 
         # 3. Check Cache
         if key in context_cache:
             return context_cache[key]
 
+        keep_alive.append(self)
+
         # 4. Execute the actual function
         # We ensure context_cache is passed down to the recursive calls
         result = func(self, collisions=collisions, **kwargs)