Add posibility to define default quotas on user && client (#153)

StephenSorriaux · Mar 30, 2023 · 846cc6d · 846cc6d
1 parent bb7fff8
commit 846cc6d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -297,6 +297,20 @@ Here some examples on how to use this library:
 - name: Ensure quota for user test and test2
   kafka_quotas:
     entries:
+    # Applied on default-user
+    - entity:
+        user: '<default>'
+      quotas:
+        producer_byte_rate: 1044050
+        consumer_byte_rate: 1044050
+        request_percentage: 65
+    # Applied on default-client
+    - entity:
+        client: '<default>'
+      quotas:
+        producer_byte_rate: 1044050
+        consumer_byte_rate: 1044050
+        request_percentage: 65
     - entity:
         user: test
       quotas:

diff --git a/module_utils/kafka_lib_quotas.py b/module_utils/kafka_lib_quotas.py
@@ -10,18 +10,22 @@
     maybe_clean_zk_ssl_files
 )
 
+DEFAULT_ENTITY = '<default>'
+
 
 def _map_entity(entity):
     e = []
     if 'user' in entity and entity['user']:
         e.append({
             'entity_type': 'user',
             'entity_name': entity['user']
+            if entity['user'] != DEFAULT_ENTITY else None
         })
     if 'client' in entity and entity['client']:
         e.append({
             'entity_type': 'client-id',
             'entity_name': entity['client']
+            if entity['client'] != DEFAULT_ENTITY else None
         })
     return e