[DPE-3039] Update rock (#336)

* Update rock

* Check for streaming status

Author: dragomirp

metadata.yaml

@@ -28,7 +28,7 @@ resources:
   postgresql-image:
     type: oci-image
     description: OCI image for PostgreSQL
-    upstream-source: ghcr.io/canonical/charmed-postgresql@sha256:a6aa592506aa4cda85b63f66e1c9d079088ca7c9d84ed4bba9442dea36ec3f17
+    upstream-source: ghcr.io/canonical/charmed-postgresql@sha256:d3041e50c0d2f613db72228df912e510786644e57c9d2c22cde99db8ce83a5a9
 
 peers:
   database-peers:

src/patroni.py

@@ -25,6 +25,8 @@
 
 from constants import REWIND_USER, TLS_CA_FILE
 
+RUNNING_STATES = ["running", "streaming"]
+
 logger = logging.getLogger(__name__)
 
 
@@ -162,7 +164,7 @@ def are_all_members_ready(self) -> bool:
         except RetryError:
             return False
 
-        return all(member["state"] == "running" for member in r.json()["members"])
+        return all(member["state"] in RUNNING_STATES for member in r.json()["members"])
 
     @property
     def is_creating_backup(self) -> bool:
@@ -222,7 +224,7 @@ def primary_endpoint_ready(self) -> bool:
                         f"{'https' if self._tls_enabled else 'http'}://{self._primary_endpoint}:8008/health",
                         verify=self._verify,
                     )
-                    if r.json()["state"] != "running":
+                    if r.json()["state"] not in RUNNING_STATES:
                         raise EndpointNotReadyError
         except RetryError:
             return False
@@ -264,7 +266,7 @@ def member_started(self) -> bool:
         except RetryError:
             return False
 
-        return r.json()["state"] == "running"
+        return r.json()["state"] in RUNNING_STATES
 
     @property
     def is_database_running(self) -> bool: