revert some handles of cancellederror

Author: rekby

ydb/_topic_writer/topic_writer_asyncio.py

@@ -539,7 +539,8 @@ async def _send_loop(self, writer: "WriterAsyncIOStream"):
                     writer.write([m])
         except asyncio.CancelledError:
             # the loop task cancelled be parent code, for example for reconnection
-            pass
+            # no need to stop all work.
+            raise
         except BaseException as e:
             self._stop(e)
             raise

ydb/aio/credentials.py

@@ -18,7 +18,7 @@ async def consume(self, timeout=3):
             if self._value is None:
                 try:
                     await asyncio.wait_for(self._condition.wait(), timeout=timeout)
-                except BaseException:
+                except Exception:
                     return self._value
             return self._value
 
@@ -41,7 +41,7 @@ async def _wrapped_execution(self, callback):
             res = callback()
             if asyncio.iscoroutine(res):
                 await res
-        except BaseException:
+        except Exception:
             pass
 
         finally:
@@ -81,11 +81,15 @@ async def _refresh(self):
         except (KeyboardInterrupt, SystemExit):
             return
 
-        except BaseException as e:
+        except Exception as e:
             self.last_error = str(e)
             await asyncio.sleep(1)
             self._tp.submit(self._refresh)
 
+        except BaseException as e:
+            self.last_error = str(e)
+            raise
+
     async def token(self):
         current_time = time.time()
         if current_time > self._refresh_in:

ydb/aio/pool.py

@@ -170,7 +170,7 @@ async def run(self):
         while True:
             try:
                 successful = await self.execute_discovery()
-            except BaseException:
+            except Exception:
                 successful = False
             if successful:
                 self._cache.complete_discovery(None)