49 lines
1.7 KiB
Ruby
49 lines
1.7 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class AddOkrHierarchyRestrictions < Gitlab::Database::Migration[2.0]
|
||
|
|
class WorkItemType < MigrationRecord
|
||
|
|
self.table_name = 'work_item_types'
|
||
|
|
end
|
||
|
|
|
||
|
|
class HierarchyRestriction < MigrationRecord
|
||
|
|
self.table_name = 'work_item_hierarchy_restrictions'
|
||
|
|
end
|
||
|
|
|
||
|
|
restrict_gitlab_migration gitlab_schema: :gitlab_main
|
||
|
|
disable_ddl_transaction!
|
||
|
|
|
||
|
|
def up
|
||
|
|
objective = WorkItemType.find_by_name_and_namespace_id('Objective', nil)
|
||
|
|
key_result = WorkItemType.find_by_name_and_namespace_id('Key Result', nil)
|
||
|
|
issue = WorkItemType.find_by_name_and_namespace_id('Issue', nil)
|
||
|
|
task = WorkItemType.find_by_name_and_namespace_id('Task', nil)
|
||
|
|
incident = WorkItemType.find_by_name_and_namespace_id('Incident', nil)
|
||
|
|
|
||
|
|
# work item default types should be filled, if this is not the case
|
||
|
|
# then restrictions will be created together with work item types
|
||
|
|
unless objective && key_result && issue && task && incident
|
||
|
|
Gitlab::AppLogger.warn('default types are missing, not adding restrictions')
|
||
|
|
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
restrictions = [
|
||
|
|
{ parent_type_id: objective.id, child_type_id: objective.id, maximum_depth: 9 },
|
||
|
|
{ parent_type_id: objective.id, child_type_id: key_result.id, maximum_depth: 1 },
|
||
|
|
{ parent_type_id: issue.id, child_type_id: task.id, maximum_depth: 1 },
|
||
|
|
{ parent_type_id: incident.id, child_type_id: task.id, maximum_depth: 1 }
|
||
|
|
]
|
||
|
|
|
||
|
|
HierarchyRestriction.upsert_all(
|
||
|
|
restrictions,
|
||
|
|
unique_by: :index_work_item_hierarchy_restrictions_on_parent_and_child
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
# so far restrictions table was empty so we can delete all records when
|
||
|
|
# migrating down
|
||
|
|
HierarchyRestriction.delete_all
|
||
|
|
end
|
||
|
|
end
|