Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

OP-19040: Bugfix for rollback the charts table level updation #437

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

OP-19040: Bugfix for rollback the charts table level updation #437

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 35 additions & 1 deletion scripts/oes-data-migration-scripts/migration_v3.12.x_to_v4.0.x.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -192,6 +192,7 @@ def update_db(version): # pre-upgrade DB Update
logging.error("Exception occurred while updating databases from v3.12.x to v4.0.x:", exc_info=True)
logging.critical(e.__str__(), exc_info=True)
rollback_transactions()
rollback_audit_db_charts()
exit(1)
finally:
close_connections()
Expand Down Expand Up @@ -578,6 +579,16 @@ def rollback_transactions():
raise e


def rollback_audit_db_charts():
try:
updatePrimaryKeyAreaCharts()
updatePrimaryKeyDeliveryInsights()
audit_conn.commit()
except Exception as e:
logging.critical("Exception occurred while rolling back the audit charts primary key transactions : ", exc_info=True)
raise e


def get_distinct_source():
try:
cur_audit.execute("select distinct(source) from audit_events")
Expand Down Expand Up @@ -859,6 +870,30 @@ def createAreaCharts():
raise e


def updatePrimaryKeyDeliveryInsights():
try:
cur_audit.execute("ALTER TABLE IF EXISTS public.delivery_insights_chart_counts DROP CONSTRAINT delivery_insights_chart_counts_pkey")
cur_audit.execute("ALTER TABLE IF EXISTS public.delivery_insights_chart_counts ADD CONSTRAINT delivery_insights_chart_counts_pkey(app_pipeline_name, days))");
logging.info("Successfully updated the PRIMARY KEY area_chart_counts table")
print("Successfully updated the PRIMARY KEY delivery_insights_chart_counts table")
except Exception as e:
print("Exception occurred in creating delivery_insights_chart_counts table while updating script : ", e)
logging.error("Exception occurred in creating delivery_insights_chart_counts table while updating script: ", exc_info=True)
raise e


def updatePrimaryKeyAreaCharts():
try:
cur_audit.execute("ALTER TABLE IF EXISTS public.area_chart_counts DROP CONSTRAINT area_chart_counts_pkey")
cur_audit.execute("ALTER TABLE IF EXISTS public.area_chart_counts ADD CONSTRAINT area_chart_counts_pkey (application_name, days, pipeline_name, status))");
logging.info("Successfully updated the PRIMARY KEY area_chart_counts table")
print("Successfully updated the PRIMARY KEY area_chart_counts table")
except Exception as e:
print("Exception occurred in creating area_chart_counts table while updating script : ", e)
logging.error("Exception occurred in creating area_chart_counts table while updating script: ", exc_info=True)
raise e


def alterServiceGateTable():
try:
cur_platform.execute("ALTER TABLE service_gate ADD COLUMN IF NOT EXISTS ref_id int")
Expand Down Expand Up @@ -1652,4 +1687,3 @@ def addDBVersion(version):
elif migrate_data_flag == 'true':
perform_migration("4.0.2") # pass the ISD version we are performing data migration for (Note: version here should be updated for each ISD release)