mig_0015_modify_cluster_rule

mig_0015_modify_cluster_rule_toggle.go
	/* Copyright © 2021 Red Hat, Inc. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. / package ocpmigrations import ( "database/sql" "fmt" "github.com/RedHatInsights/insights-results-aggregator/migration" "github.com/RedHatInsights/insights-results-aggregator/types" "github.com/rs/zerolog/log" ) / migration15 adds error_key and set as a new primary key This make us available to disable rule results instead of whole rules for rules with more than one possible error key as result */
mig0015ClusterRuleToggle is a helper for update the clusterruletoggle table	var mig0015ClusterRuleToggle = migration . Migration { StepUp : func ( tx * sql . Tx , driver types . DBDriver ) error { var err error if _ , err = tx . Exec ( ` ALTER TABLE cluster_rule_toggle ADD error_key VARCHAR NOT NULL DEFAULT ''; ` ) ; err != nil { return err } _ , err = tx . Exec ( ` ALTER TABLE cluster_rule_toggle DROP CONSTRAINT cluster_rule_toggle_pkey, ADD CONSTRAINT cluster_rule_toggle_pkey PRIMARY KEY (cluster_id, rule_id, error_key); ` ) return err } , StepDown : func ( tx * sql . Tx , driver types . DBDriver ) error { _ , err := tx . Exec ( ` ALTER TABLE cluster_rule_toggle DROP CONSTRAINT cluster_rule_toggle_pkey, ADD CONSTRAINT cluster_rule_toggle_pkey PRIMARY KEY (cluster_id, rule_id); ALTER TABLE cluster_rule_toggle DROP COLUMN error_key; ` ) return err } , }
mig0015ClusterRuleUserFeedback is a helper for update the clusterruleuser_feedback table	var mig0015ClusterRuleUserFeedback = migration . Migration { StepUp : func ( tx * sql . Tx , driver types . DBDriver ) error { var err error if _ , err = tx . Exec ( ` ALTER TABLE cluster_rule_user_feedback ADD error_key VARCHAR NOT NULL DEFAULT ''; ` ) ; err != nil { return err } _ , err = tx . Exec ( ` ALTER TABLE cluster_rule_user_feedback DROP CONSTRAINT cluster_rule_user_feedback_pkey1, ADD CONSTRAINT cluster_rule_user_feedback_pkey PRIMARY KEY (cluster_id, rule_id, user_id, error_key); ` ) return err } , StepDown : func ( tx * sql . Tx , driver types . DBDriver ) error { _ , err := tx . Exec ( ` ALTER TABLE cluster_rule_user_feedback DROP CONSTRAINT cluster_rule_user_feedback_pkey, ADD CONSTRAINT cluster_rule_user_feedback_pkey1 PRIMARY KEY (cluster_id, rule_id, user_id); ALTER TABLE cluster_rule_user_feedback DROP COLUMN error_key; ` ) return err } , }
mig0015ClusterUserRuleDisableFeedback is a helper for update the clusteruserruledisablefeedback	var mig0015ClusterUserRuleDisableFeedback = migration . Migration { StepUp : func ( tx * sql . Tx , driver types . DBDriver ) error { var err error if _ , err := tx . Exec ( ` ALTER TABLE cluster_user_rule_disable_feedback ADD error_key VARCHAR NOT NULL DEFAULT ''; ` ) ; err != nil { return err } _ , err = tx . Exec ( ` ALTER TABLE cluster_user_rule_disable_feedback DROP CONSTRAINT cluster_user_rule_disable_feedback_pkey, ADD CONSTRAINT cluster_user_rule_disable_feedback_pkey PRIMARY KEY (cluster_id, user_id, rule_id, error_key); ` ) return err } , StepDown : func ( tx * sql . Tx , driver types . DBDriver ) error { _ , err := tx . Exec ( ` ALTER TABLE cluster_user_rule_disable_feedback DROP CONSTRAINT cluster_user_rule_disable_feedback_pkey, ADD CONSTRAINT cluster_user_rule_disable_feedback_pkey PRIMARY KEY (cluster_id, user_id, rule_id); ALTER TABLE cluster_user_rule_disable_feedback DROP COLUMN error_key; ` ) return err } , }
migrateClusterRoleToggleData is a helper to update the current data with default values It takes the only possible value for error_key on the rules that only has one possible error key	func migrateClusterRoleToggleData ( tx * sql . Tx ) error { updateClusterRuleToggleQuery := ` UPDATE cluster_rule_toggle SET error_key=$1 WHERE rule_id LIKE $2 ` updateClusterRuleUserFeedbackQuery := ` UPDATE cluster_rule_user_feedback SET error_key=$1 WHERE rule_id LIKE $2 ` updateClusterUserRuleDisableFeedbackQuery := ` UPDATE cluster_user_rule_disable_feedback SET error_key=$1 WHERE rule_id LIKE $2 ` for ruleID , errorKey := range defaultErrorKeysPerRuleID { ruleIDWildcard := fmt . Sprintf ( "%s%%" , ruleID ) log . Info ( ) . Str ( "rule_id" , ruleIDWildcard ) . Str ( "errorKey" , errorKey ) . Msg ( "Updating DB data" ) err := migration . UpdateTableData ( tx , "cluster_rule_toggle" , updateClusterRuleToggleQuery , errorKey , ruleIDWildcard ) if err != nil { return err } err = migration . UpdateTableData ( tx , "cluster_rule_user_feedback" , updateClusterRuleUserFeedbackQuery , errorKey , ruleIDWildcard ) if err != nil { return err } err = migration . UpdateTableData ( tx , "cluster_user_rule_disable_feedback" , updateClusterUserRuleDisableFeedbackQuery , errorKey , ruleIDWildcard ) if err != nil { return err } } return nil }
mig0015ModifyClusterRuleTables migrates the tables related to user toggle and feedback with error_key	var mig0015ModifyFeedbackTables = migration . Migration { StepUp : func ( tx * sql . Tx , driver types . DBDriver ) error { if err := mig0015ClusterRuleToggle . StepUp ( tx , driver ) ; err != nil { return err } if err := mig0015ClusterRuleUserFeedback . StepUp ( tx , driver ) ; err != nil { return err } if err := mig0015ClusterUserRuleDisableFeedback . StepUp ( tx , driver ) ; err != nil { return err } return migrateClusterRoleToggleData ( tx ) } , StepDown : func ( tx * sql . Tx , driver types . DBDriver ) error { if err := mig0015ClusterRuleToggle . StepDown ( tx , driver ) ; err != nil { return err } if err := mig0015ClusterRuleUserFeedback . StepDown ( tx , driver ) ; err != nil { return err } return mig0015ClusterUserRuleDisableFeedback . StepDown ( tx , driver ) } , } var defaultErrorKeysPerRuleID = map [ string ] string { "ccx_rules_ocp.external.bug_rules.bug_1765280" : "BUGZILLA_BUG_1765280" , "ccx_rules_ocp.external.bug_rules.bug_1766907" : "BUGZILLA_BUG_1766907" , "ccx_rules_ocp.external.bug_rules.bug_1798049" : "BUGZILLA_BUG_1798049" , "ccx_rules_ocp.external.bug_rules.bug_1821905" : "BUGZILLA_BUG_1821905" , "ccx_rules_ocp.external.bug_rules.bug_1832986" : "BUGZILLA_BUG_1832986" , "ccx_rules_ocp.external.bug_rules.bug_1893386" : "BUGZILLA_BUG_1893386" , "ccx_rules_ocp.external.rules.ccxdev_auxiliary_rule" : "CCXDEV_E2E_TEST_RULE" , "ccx_rules_ocp.external.rules.check_sap_sdi_observer_pods" : "SAP_SDI_OBSERVER_POD_ERROR" , "ccx_rules_ocp.external.rules.check_sdi_preload_kernel_modules" : "SDI_PRELOAD_KERNEL_MODULES_ERROR" , "ccx_rules_ocp.external.rules.cluster_wide_proxy_auth_check" : "AUTH_OPERATOR_PROXY_ERROR" , "ccx_rules_ocp.external.rules.cmo_container_run_as_non_root" : "CMO_CONTAINER_RUN_AS_NON_ROOT" , "ccx_rules_ocp.external.rules.container_max_root_partition_size" : "CONTAINER_ROOT_PARTITION_SIZE" , "ccx_rules_ocp.external.rules.control_plane_replicas" : "CONTROL_PLANE_NODE_REPLICAS" , "ccx_rules_ocp.external.rules.empty_prometheus_db_volume" : "PROMETHEUS_DB_VOLUME_IS_EMPTY" , "ccx_rules_ocp.external.rules.image_registry_multiple_storage_types" : "IMAGE_REGISTRY_MULTIPLE_STORAGE_TYPES" , "ccx_rules_ocp.external.rules.lib_bucket_provisioner_check" : "LIB_BUCKET_PROVISIONER_INSTALL_PLANS_ISSUE" , "ccx_rules_ocp.external.rules.machineconfig_stuck_by_node_taints" : "NODE_HAS_TAINTS_APPLIED" , "ccx_rules_ocp.external.rules.master_defined_as_machinesets" : "MASTER_DEFINED_AS_MACHINESETS" , "ccx_rules_ocp.external.rules.node_installer_degraded" : "NODE_INSTALLER_DEGRADED" , "ccx_rules_ocp.external.rules.nodes_container_runtime_version_check" : "NODES_CONTAINER_RUNTIME_VERSION" , "ccx_rules_ocp.external.rules.nodes_kubelet_version_check" : "NODE_KUBELET_VERSION" , "ccx_rules_ocp.external.rules.nodes_requirements_check" : "NODES_MINIMUM_REQUIREMENTS_NOT_MET" , "ccx_rules_ocp.external.rules.openshift_sdn_egress_ip_in_no_hostsubnet" : "OPENSHIFT_SDN_EGRESS_IP_IN_NO_HOSTSUBNET" , "ccx_rules_ocp.external.rules.operator_unmanaged" : "OPERATOR_UNMANAGED" , "ccx_rules_ocp.external.rules.prometheus_backed_by_pvc" : "PROMETHEUS_BACKED_BY_PVC" , "ccx_rules_ocp.external.rules.prometheus_rule_evaluation_fail" : "PROMETHEUS_RULE_EVALUATION_FAIL" , "ccx_rules_ocp.external.rules.same_egress_ip_in_multiple_netnamespaces" : "SAME_EGRESS_IP_IN_MULTIPLE_NETNAMESPACES" , "ccx_rules_ocp.external.rules.samples_op_failed_image_import_check" : "SAMPLES_FAILED_IMAGE_IMPORT_ERR" , "ccx_rules_ocp.external.rules.sap_data_intelligence_permissions" : "SAP_DATA_INTELLIGENCE_PERMISSIONS" , "ccx_rules_ocp.external.rules.subnets_migration_failure_massive_egressip" : "SUBNETS_MIGRATION_FAILURE_MASSIVE_EGRESSIP" , "ccx_rules_ocp.external.rules.tls_handshake_fails_in_azure" : "TLS_HANDSHAKE_FAILS_IN_AZURE" , "ccx_rules_ocp.external.rules.unsupported_cni_plugin" : "UNSUPPORTED_CNI_PLUGIN" , "ccx_rules_ocp.external.rules.vsphere_upi_machine_is_in_phase" : "VSPHERE_UPI_MACHINE_WITH_NO_RUNNING_PHASE" , "ccx_rules_ocp.external.security.CVE_2020_8555_kubernetes" : "CVE_2020_8555_KUBERNETES" , "ccx_rules_ocp.external.security.CVE_2021_30465_runc" : "CVE_2021_30465_RUNC_VULN" ,
Several ERRORKEY "ccxrulesocp.external.rules.imageregistrystorage" "ccxrulesocp.external.rules.ocpversionendoflifeeus" "ccxrulesocp.external.rules.ocpversionendoflife" "ccxrulesocp.external.rules.openshiftsdnegressipinmultiplehostsubnets" "ccxrulesocp.external.rules.upgradetoocp47failson_vsphere"	`}`

mig_0015_modify_cluster_rule_toggle.go