Fix subscriptions

src/demo_tezos_domains/dipdup.yml

@@ -6,14 +6,14 @@ database:
   path: tezos_domains.sqlite3
 
 contracts:
-  edo_name_registry:
-    address: KT1JJbWfW8CHUY95hG9iq2CEMma1RiKhMHDR
+  mainnet_name_registry:
+    address: KT1GBZmSxmnKJXGMdMLbugPfLyUPmuLSMwKS
     typename: name_registry
 
 datasources:
-  tzkt_edo:
+  tzkt_mainnet:
     kind: tzkt
-    url: ${TZKT_URL:-https://api.edo2net.tzkt.io}
+    url: ${TZKT_URL:-https://api.tzkt.io}
 
 templates:
   tezos_domains:
@@ -34,8 +34,8 @@ templates:
             entrypoint: execute
 
 indexes:
-  tezos_domains_edo:
+  tezos_domains_mainnet:
     template: tezos_domains
     values:
-      datasource: tzkt_edo
-      name_registry: edo_name_registry
+      datasource: tzkt_mainnet
+      name_registry: mainnet_name_registry

src/dipdup/datasources/tzkt/datasource.py

@@ -555,9 +555,12 @@ async def run(self) -> None:
     async def _on_connect(self) -> None:
         """Subscribe to all required channels on established WS connection"""
         self._logger.info('Connected to server')
-        self._subscriptions.reset()
         await self.subscribe()
 
+    async def _on_disconnect(self) -> None:
+        self._logger.info('Disconnected from server')
+        self._subscriptions.reset()
+
     async def subscribe(self) -> None:
         """Subscribe to all required channels"""
         if not self._realtime: