Do a full re-hydrate on reconnect (#5980)

It turns out that just rehydrating and not also re-running the on_load handlers
can leave applications in a bad state; particularly when their background tasks
exiting are predicated on the `sid`, that will change on reconnect, but without
an on_load, no mechanism is there to restart the tasks.

Author: masenf

reflex/.templates/web/utils/state.js

@@ -618,19 +618,13 @@ export const connect = async (
     window.addEventListener("unload", disconnectTrigger);
     if (socket.current.rehydrate) {
       socket.current.rehydrate = false;
-      const events = initialEvents();
-      if (events.length > 0) {
-        // On reconnect, we only hydrate, do not re-run on_load events.
-        const hydrate_event = initialEvents()[0];
-        hydrate_event.payload.is_reconnect = true;
-        queueEvents(
-          [hydrate_event],
-          socket,
-          true,
-          navigate,
-          () => params.current,
-        );
-      }
+      queueEvents(
+        initialEvents(),
+        socket,
+        true,
+        navigate,
+        () => params.current,
+      );
     }
     // Drain any initial events from the queue.
     while (event_queue.length > 0 && !event_processing) {

reflex/middleware/hydrate_middleware.py

@@ -35,14 +35,11 @@ async def preprocess(
         if event.name != get_hydrate_event(state):
             return None
 
-        # In reconnect mode, don't reset client storage or call on_load.
-        is_reconnect = event.payload.get("is_reconnect", False)
-        if not is_reconnect:
-            # Clear client storage, to respect clearing cookies
-            state._reset_client_storage()
-
-            # Mark state as not hydrated (until on_loads are complete)
-            setattr(state, constants.CompileVars.IS_HYDRATED, False)
+        # Clear client storage, to respect clearing cookies
+        state._reset_client_storage()
+
+        # Mark state as not hydrated (until on_loads are complete)
+        setattr(state, constants.CompileVars.IS_HYDRATED, False)
 
         # Get the initial state.
         delta = await _resolve_delta(state.dict())