Merge pull request #86 from RelationalAI/ss-txn-cancel

Support transaction cancellation

Author: salamehsameera

examples/cancel_transaction.py

@@ -0,0 +1,39 @@
+# Copyright 2021-2022 RelationalAI, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License
+
+
+import json
+from argparse import ArgumentParser
+from urllib.request import HTTPError
+
+from railib import api, config, show
+
+
+def run(id: str, profile: str):
+    cfg = config.read(profile=profile)
+    ctx = api.Context(**cfg)
+    rsp = api.cancel_transaction(ctx, id)
+    print(json.dumps(rsp, indent=2))
+
+
+if __name__ == "__main__":
+    p = ArgumentParser()
+    p.add_argument("id", type=str, help="transaction id")
+    p.add_argument("-p", "--profile", type=str,
+                   help="profile name", default="default")
+    args = p.parse_args()
+    try:
+        run(args.id, args.profile)
+    except HTTPError as e:
+        show.http_error(e)

railib/api.py

@@ -118,6 +118,7 @@ class Permission(str, Enum):
     "get_transaction",
     "get_transaction_metadata",
     "get_transaction_results",
+    "cancel_transaction",
     "get_user",
     "list_databases",
     "list_edbs",
@@ -303,6 +304,11 @@ def get_transaction_results(ctx: Context, id: str) -> list:
     return _parse_multipart(content_type, rsp.read())
 
 
+def cancel_transaction(ctx: Context, id: str) -> dict:
+    rsp = rest.post(ctx, _mkurl(ctx, f"{PATH_TRANSACTIONS}/{id}/cancel"), {})
+    return json.loads(rsp.read())
+
+
 def get_user(ctx: Context, userid: str) -> dict:
     return _get_resource(ctx, f"{PATH_USER}/{userid}", name=userid)