Implement missing request builder implementation

Author: lolski

dependencies/vaticle/artifacts.bzl

@@ -39,5 +39,5 @@ def vaticle_typedb_cluster_artifacts():
         artifact_name = "typedb-cluster-all-{platform}-{version}.{ext}",
         tag_source = deployment_private["artifact.release"],
         commit_source = deployment_private["artifact.snapshot"],
-        commit = "5a423f70c416d52f49ba541fe6b10b0ff2c89602"
+        commit = "f6d9eae58c83c165d932c0433a3a0d858e206fab"
     )

requirements.txt

@@ -24,14 +24,13 @@
 
 ## Configuration options
 
-
 # Allow importing of snapshots
 --extra-index-url https://repo.vaticle.com/repository/pypi-snapshot/simple
 
 
 ## Dependencies
 
 # IMPORTANT: Any changes to these dependencies should be copied to requirements_dev.txt.
-typedb-protocol==2.4.0
+typedb-protocol==0.0.0-ec89440dadb449f223ad2c925b238aad06b80c81
 grpcio==1.38.0
 protobuf==3.15.5

requirements_dev.txt

@@ -30,7 +30,7 @@
 
 ## Dependencies
 
-typedb-protocol==2.4.0
+typedb-protocol==0.0.0-ec89440dadb449f223ad2c925b238aad06b80c81
 grpcio==1.38.0
 protobuf==3.15.5
 

typedb/common/rpc/request_builder.py

@@ -111,6 +111,11 @@ def cluster_user_password_req(username: str, password: str):
     req.password = password
     return req
 
+def cluster_user_token_req(username: str):
+    req = cluster_user_proto.ClusterUser.Token.Req()
+    req.username = username
+    return req
+
 def cluster_user_delete_req(username: str):
     req = cluster_user_proto.ClusterUser.Delete.Req()
     req.username = username

typedb/connection/cluster/stub.py

@@ -33,6 +33,7 @@
 
 from typedb.api.connection.credential import TypeDBCredential
 from typedb.common.exception import CLUSTER_TOKEN_CREDENTIAL_INVALID, TypeDBClientException, UNABLE_TO_CONNECT
+from typedb.common.rpc.request_builder import cluster_user_token_req
 from typedb.common.rpc.stub import TypeDBStub
 
 T = TypeVar('T')
@@ -48,7 +49,7 @@ def __init__(self, channel: Channel, credential: TypeDBCredential):
         self._cluster_stub = cluster_service_proto.TypeDBClusterStub(channel)
         self._token = None
         try:
-            self._token = self._cluster_stub.user_token_renew(self._credential.username())
+            self._token = self._cluster_stub.user_token(cluster_user_token_req(self._credential.username()))
         except RpcError as e:
             e2 = TypeDBClientException.of_rpc(e)
             if e2.error_message is not None and e2.error_message is not UNABLE_TO_CONNECT:
@@ -120,7 +121,7 @@ def may_renew_token(self, function: Callable[[], T]) -> T:
         except TypeDBClientException as e:
             if e.error_message is not None and e.error_message is CLUSTER_TOKEN_CREDENTIAL_INVALID:
                 self._token = None
-                res = self._cluster_stub.user_token_renew(self._credential.username())
+                res = self._cluster_stub.user_token(cluster_user_token_req(self._credential.username()))
                 self._token = res.token
                 try:
                     return self.resilient_call(function)