Changes for code review
This commit is contained in:
parent
3438c1bfad
commit
ed84835d23
4 changed files with 25 additions and 15 deletions
|
@ -7,7 +7,7 @@ from peewee import JOIN_LEFT_OUTER
|
|||
import features
|
||||
from data.database import (BuildTriggerService, RepositoryBuildTrigger, Repository, Namespace, User,
|
||||
RepositoryBuild, BUILD_PHASE, db_random_func, UseThenDisconnect,
|
||||
DisableReason)
|
||||
DisableReason, TRIGGER_DISABLE_REASON)
|
||||
from data.model import (InvalidBuildTriggerException, InvalidRepositoryBuildException,
|
||||
db_transaction, user as user_model, config)
|
||||
|
||||
|
@ -258,12 +258,13 @@ def mark_build_archived(build_uuid):
|
|||
.execute()) > 0
|
||||
|
||||
|
||||
def toggle_build_trigger(trigger, enabled, reason='user_toggled'):
|
||||
def toggle_build_trigger(trigger, enabled, reason=TRIGGER_DISABLE_REASON.USER_TOGGLED):
|
||||
""" Toggles the enabled status of a build trigger. """
|
||||
trigger.enabled = enabled
|
||||
|
||||
if not enabled:
|
||||
trigger.disabled_reason = RepositoryBuildTrigger.disabled_reason.get_id(reason)
|
||||
trigger.disabled_datetime = datetime.utcnow()
|
||||
|
||||
trigger.save()
|
||||
|
||||
|
@ -275,6 +276,12 @@ def update_trigger_disable_status(trigger, final_phase):
|
|||
ensures that triggers that continue to error are eventually automatically disabled.
|
||||
"""
|
||||
with db_transaction():
|
||||
try:
|
||||
trigger = RepositoryBuildTrigger.get(id=trigger.id)
|
||||
except RepositoryBuildTrigger.DoesNotExist:
|
||||
# Already deleted.
|
||||
return
|
||||
|
||||
# If the build completed successfully, then reset the successive counters.
|
||||
if final_phase == BUILD_PHASE.COMPLETE:
|
||||
trigger.successive_failure_count = 0
|
||||
|
@ -287,7 +294,6 @@ def update_trigger_disable_status(trigger, final_phase):
|
|||
trigger.successive_failure_count = trigger.successive_failure_count + 1
|
||||
trigger.successive_internal_error_count = 0
|
||||
elif final_phase == BUILD_PHASE.INTERNAL_ERROR:
|
||||
trigger.successive_failure_count = 0
|
||||
trigger.successive_internal_error_count = trigger.successive_internal_error_count + 1
|
||||
|
||||
# Check if we need to disable the trigger.
|
||||
|
@ -295,15 +301,11 @@ def update_trigger_disable_status(trigger, final_phase):
|
|||
error_threshold = config.app_config.get('SUCCESSIVE_TRIGGER_INTERNAL_ERROR_DISABLE_THRESHOLD')
|
||||
|
||||
if failure_threshold and trigger.successive_failure_count >= failure_threshold:
|
||||
trigger.disabled_datetime = datetime.utcnow()
|
||||
trigger.enabled = False
|
||||
trigger.disabled_reason = RepositoryBuildTrigger.disabled_reason.get_id('successive_build_failures')
|
||||
toggle_build_trigger(trigger, False, TRIGGER_DISABLE_REASON.BUILD_FALURES)
|
||||
elif (error_threshold and
|
||||
trigger.successive_internal_error_count >= error_threshold):
|
||||
trigger.disabled_datetime = datetime.utcnow()
|
||||
trigger.enabled = False
|
||||
trigger.disabled_reason = RepositoryBuildTrigger.disabled_reason.get_id('successive_build_internal_errors')
|
||||
|
||||
# Save the trigger changes.
|
||||
trigger.save()
|
||||
toggle_build_trigger(trigger, False, TRIGGER_DISABLE_REASON.INTERNAL_ERRORS)
|
||||
else:
|
||||
# Save the trigger changes.
|
||||
trigger.save()
|
||||
|
Reference in a new issue