Skip to content

Commit e7615d9

Browse files
committed
added threading
1 parent c424826 commit e7615d9

File tree

1 file changed

+17
-3
lines changed
  • iam-user-capability-management

1 file changed

+17
-3
lines changed

iam-user-capability-management/func.py

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import json
33
import logging
44
import oci
5+
from threading import Thread
56
from fdk import response
67

78
# Get Resource Principal Credentials
@@ -48,7 +49,7 @@ def process_users(user,identity_domains_client,tag_namespace,manage_capability,e
4849
)
4950
)
5051
patch_ops.operations = patch_ops_operations
51-
identity_domains_client.patch_user(user_id=user_ocid, patch_op=patch_ops)
52+
identity_domains_client.patch_user(user_id=user_ocid, patch_op=patch_ops,retry_strategy=oci.retry.DEFAULT_RETRY_STRATEGY)
5253

5354

5455
def handler(ctx, data: io.BytesIO=None):
@@ -95,12 +96,25 @@ def handler(ctx, data: io.BytesIO=None):
9596
list_users_response = identity_domains_client.list_users(page=list_users_response.next_page)
9697
users.extend(list_users_response.data.resources)
9798
count = 0
99+
jobs = []
98100

99101
for user in users:
100-
process_users(user, identity_domains_client,tag_namespace,manage_capability,execution_mode)
102+
#process_users(user, identity_domains_client, tag_namespace, manage_capability, execution_mode)
103+
104+
thread = Thread(target=process_users, args=(user, identity_domains_client, tag_namespace, manage_capability, execution_mode))
105+
jobs.append(thread)
101106
count += 1
102107

103-
logging.getLogger().info(f'Processed {str(count)} users....')
108+
# start threads
109+
for job in jobs:
110+
job.start()
111+
112+
# join threads,so we don't quit until all threads have finished
113+
for job in jobs:
114+
job.join()
115+
116+
117+
logging.getLogger().info(f'Processed {str(count)} users for domain - {str(domain_endpoint)}....')
104118

105119

106120
except (Exception, ValueError) as ex:

0 commit comments

Comments
 (0)