Get method on promise store (#203)

Author: Tomperez98

resonate/models/store.py

@@ -21,6 +21,7 @@ def tasks(self) -> TaskStore: ...
 
 
 class PromiseStore(Protocol):
+    def get(self, *, id: str) -> DurablePromise: ...
     def create(
         self,
         *,

resonate/resonate.py

@@ -41,7 +41,6 @@ def __init__(
         url: str | None = None,
         pid: str | None = None,
         ttl: int = 10,
-        opts: Options | None = None,
         anycast: str | None = None,
         unicast: str | None = None,
         store: Store | None = None,
@@ -53,7 +52,7 @@ def __init__(
         self._started = False
 
         self._pid = pid or uuid.uuid4().hex
-        self._opts = opts or Options()
+        self._opts = Options()
 
         self._registry = registry or Registry()
         self._dependencies = dependencies or Dependencies()

resonate/stores/local.py

@@ -113,6 +113,16 @@ def __init__(self, store: LocalStore, encoder: Encoder[Any, str | None], promise
         self._routers = routers
         self._clock = clock
 
+    def get(self, *, id: str) -> DurablePromise:
+        record = self._promises.get(id)
+        if record is None:
+            msg = "Not found"
+            raise ResonateStoreError(msg, "STORE_NOT_FOUND")
+        return DurablePromise.from_dict(
+            self._store,
+            record.to_dict(),
+        )
+
     def create(
         self,
         *,

resonate/stores/remote.py

@@ -50,6 +50,17 @@ def _headers(self, *, strict: bool, ikey: str | None) -> dict[str, str]:
             headers["idempotency-Key"] = ikey
         return headers
 
+    def get(self, *, id: str) -> DurablePromise:
+        req = Request(
+            method="post",
+            url=f"{self._store.url}/promises/{id}",
+        )
+        res = _call(req.prepare()).json()
+        return DurablePromise.from_dict(
+            self._store,
+            res,
+        )
+
     def create(
         self,
         *,