Commit 4a9dcfdc authored by James Lopez's avatar James Lopez

enable AR caching, refactor tree hash loop

parent a54c50d1
...@@ -23,8 +23,10 @@ module Gitlab ...@@ -23,8 +23,10 @@ module Gitlab
@project_members = @tree_hash.delete('project_members') @project_members = @tree_hash.delete('project_members')
ActiveRecord::Base.no_touching do ActiveRecord::Base.uncached do
create_relations ActiveRecord::Base.no_touching do
create_relations
end
end end
rescue => e rescue => e
@shared.error(e) @shared.error(e)
...@@ -52,6 +54,7 @@ module Gitlab ...@@ -52,6 +54,7 @@ module Gitlab
@saved = [] @saved = []
default_relation_list.each do |relation| default_relation_list.each do |relation|
next unless relation.is_a?(Hash) || @tree_hash[relation.to_s].present? next unless relation.is_a?(Hash) || @tree_hash[relation.to_s].present?
if relation.is_a?(Hash) if relation.is_a?(Hash)
create_sub_relations(relation, @tree_hash) create_sub_relations(relation, @tree_hash)
else else
...@@ -82,6 +85,10 @@ module Gitlab ...@@ -82,6 +85,10 @@ module Gitlab
relation_hash = create_relation(relation_key, relation_hash_batch) relation_hash = create_relation(relation_key, relation_hash_batch)
@saved << restored_project.append_or_update_attribute(relation_key, relation_hash) @saved << restored_project.append_or_update_attribute(relation_key, relation_hash)
@restored_project = nil
@project = nil
relation_hash = nil
relation_hash_batch = nil
@restored_project = Project.find_by_id(@project_id) @restored_project = Project.find_by_id(@project_id)
end end
...@@ -117,20 +124,28 @@ module Gitlab ...@@ -117,20 +124,28 @@ module Gitlab
relation_key = relation.keys.first.to_s relation_key = relation.keys.first.to_s
return if tree_hash[relation_key].blank? return if tree_hash[relation_key].blank?
[tree_hash[relation_key]].flatten.each do |relation_item| tree_array = [tree_hash[relation_key]].flatten
relation.values.flatten.each do |sub_relation|
# We just use author to get the user ID, do not attempt to create an instance.
next if sub_relation == :author
create_sub_relations(sub_relation, relation_item, false) if sub_relation.is_a?(Hash) while relation_item = tree_array.shift
relation.values.flatten.each do |sub_relation|
# We just use author to get the user ID, do not attempt to create an instance.
next if sub_relation == :author
relation_hash, sub_relation = assign_relation_hash(relation_item, sub_relation) create_sub_relations(sub_relation, relation_item, false) if sub_relation.is_a?(Hash)
relation_item[sub_relation.to_s] = create_relation(sub_relation, relation_hash) unless relation_hash.blank?
end
save_relation_hash([relation_item], relation_key) if save relation_hash, sub_relation = assign_relation_hash(relation_item, sub_relation)
tree_hash.delete(relation_key) if save relation_item[sub_relation.to_s] = create_relation(sub_relation, relation_hash) unless relation_hash.blank?
end
if save
save_relation_hash([relation_item], relation_key)
end
tree_hash[relation_key].delete(relation_item) if save
relation_item = nil
end end
tree_hash.delete(relation_key) if save
end end
def assign_relation_hash(relation_item, sub_relation) def assign_relation_hash(relation_item, sub_relation)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment