External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-12 18:40:01 -03:30
Code Issues Packages Projects Releases Wiki Activity

add option to create partitions that span a single minute

Browse Source 
* turned on by default currently for testing
This commit is contained in:
Jim Ladd 2021-02-19 16:58:19 -08:00
parent c87d7b0d79
commit 82df3ebddb
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
awx/main/utils/common.py
View File

@ -1026,21 +1026,33 @@ def deepmerge(a, b):
return b
def create_partition(start=None, end=None, partition_label=None):
def create_partition(start=None, end=None, partition_label=None, minutely=True):
"""Creates new partition tables for events.
- start defaults to beginning of current hour
- end defaults to end of current hour
- partition_label defaults to YYYYMMDD_HH"""
- partition_label defaults to YYYYMMDD_HH
- minutely will create partitions that span _a single minute_ for testing purposes
"""
current_time = now()
if not start:
start = now().replace(microsecond=0, second=0, minute=0)
if minutely:
start = current_time.replace(microsecond=0, second=0)
else:
start = current_time.replace(microsecond=0, second=0, minute=0)
if not end:
end = start.replace(microsecond=0, second=0, minute=0) + timedelta(hours=1)
if minutely:
end = start.replace(microsecond=0, second=0) + timedelta(minutes=1)
else:
end = start.replace(microsecond=0, second=0, minute=0) + timedelta(hours=1)
start_timestamp = str(start)
end_timestamp = str(end)
if not partition_label:
partition_label = start.strftime('%Y%m%d_%H')
if minutely:
partition_label = start.strftime('%Y%m%d_%H%M')
else:
partition_label = start.strftime('%Y%m%d_%H')
with connection.cursor() as cursor:
# Only partitioning main_jobevent on first pass