add_function can support both async and sync; add await enforce.enforce()

Author: Chimou0

casbin/async_enforcer.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from functools import partial
+import asyncio
 
 from casbin.async_management_enforcer import AsyncManagementEnforcer
 from casbin.util import join_slice, array_remove_duplicates, set_subtract
@@ -22,6 +23,15 @@ class AsyncEnforcer(AsyncManagementEnforcer):
     """
     AsyncEnforcer = AsyncManagementEnforcer + RBAC_API + RBAC_WITH_DOMAIN_API
     """
+    async def enforce(self, *rvals):
+        loop = asyncio.get_running_loop()
+        result = await loop.run_in_executor(None, super().enforce, *rvals)
+        return result
+        
+    async def batch_enforce(self, rvals):
+        """batch_enforce enforce in batches"""
+        tasks = [self.enforce(*request) for request in rvals]
+        return await asyncio.gather(*tasks)
 
     async def get_roles_for_user(self, name):
         """gets the roles that a user has."""

casbin/core_enforcer.py

@@ -15,6 +15,7 @@
 import copy
 import logging
 import re
+import asyncio
 
 from casbin.effect import Effector, get_effector, effect_to_bool
 from casbin.model import Model, FunctionMap
@@ -457,6 +458,8 @@ def enforce_ex(self, *rvals):
                     expression = self._get_expression(exp_with_rule, functions)
 
                 result = expression.eval(parameters)
+                if asyncio.iscoroutine(result):
+                    result = asyncio.run(result)
 
                 if isinstance(result, bool):
                     if not result: