From: Stephane Gamard <stephane.gamard@searchbox.com>
Date: Fri, 9 May 2014 18:33:07 +0000 (+0200)
Subject: Rewrite of RegisterRule for Dao2
X-Git-Tag: 4.4-RC1~1147
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=c5cb53b192fe988eb7e5420dcfcb57538cee9b83;p=sonarqube.git

Rewrite of RegisterRule for Dao2
---

diff --git a/sonar-core/src/main/java/org/sonar/core/rule/RuleDto.java b/sonar-core/src/main/java/org/sonar/core/rule/RuleDto.java
index 008e4aa8d9e..6ce40e81d3c 100644
--- a/sonar-core/src/main/java/org/sonar/core/rule/RuleDto.java
+++ b/sonar-core/src/main/java/org/sonar/core/rule/RuleDto.java
@@ -397,4 +397,11 @@ public final class RuleDto implements Dto<RuleKey> {
   public String toString() {
     return new ReflectionToStringBuilder(this, ToStringStyle.SHORT_PREFIX_STYLE).toString();
   }
+
+  public static RuleDto createFor(RuleKey key){
+    return new RuleDto()
+      .setRepositoryKey(key.repository())
+      .setRuleKey(key.rule());
+  }
+
 }
diff --git a/sonar-server/src/main/java/org/sonar/server/rule2/RegisterRules.java b/sonar-server/src/main/java/org/sonar/server/rule2/RegisterRules.java
index f32a7c4c89f..e80e09f8cfa 100644
--- a/sonar-server/src/main/java/org/sonar/server/rule2/RegisterRules.java
+++ b/sonar-server/src/main/java/org/sonar/server/rule2/RegisterRules.java
@@ -21,14 +21,7 @@ package org.sonar.server.rule2;
 
 import com.google.common.annotations.VisibleForTesting;
 import com.google.common.base.Function;
-import com.google.common.base.Predicate;
-import com.google.common.collect.ArrayListMultimap;
-import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
-import com.google.common.collect.Maps;
-import com.google.common.collect.Multimap;
-import com.google.common.collect.Sets;
-import org.apache.commons.lang.ArrayUtils;
 import org.apache.commons.lang.ObjectUtils;
 import org.apache.commons.lang.StringUtils;
 import org.picocontainer.Startable;
@@ -38,7 +31,6 @@ import org.sonar.api.rule.RuleKey;
 import org.sonar.api.rules.Rule;
 import org.sonar.api.server.debt.DebtRemediationFunction;
 import org.sonar.api.server.rule.RulesDefinition;
-import org.sonar.api.utils.MessageException;
 import org.sonar.api.utils.System2;
 import org.sonar.api.utils.TimeProfiler;
 import org.sonar.check.Cardinality;
@@ -46,28 +38,26 @@ import org.sonar.core.persistence.DbSession;
 import org.sonar.core.persistence.MyBatis;
 import org.sonar.core.rule.RuleDto;
 import org.sonar.core.rule.RuleParamDto;
-import org.sonar.core.rule.RuleRuleTagDto;
 import org.sonar.core.technicaldebt.db.CharacteristicDao;
 import org.sonar.core.technicaldebt.db.CharacteristicDto;
 import org.sonar.server.qualityprofile.ProfilesManager;
 import org.sonar.server.rule.RuleDefinitionsLoader;
-import org.sonar.server.rule.RuleTagOperations;
 
-import javax.annotation.CheckForNull;
 import javax.annotation.Nullable;
+import java.util.ArrayList;
 import java.util.Collection;
-import java.util.Date;
+import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.Set;
 
 import static com.google.common.collect.Lists.newArrayList;
 
 /**
-* Register rules at server startup
-*
-* @since 4.2
-*/
+ * Register rules at server startup
+ *
+ * @since 4.2
+ */
 public class RegisterRules implements Startable {
 
   private static final Logger LOG = LoggerFactory.getLogger(RegisterRules.class);
@@ -77,48 +67,79 @@ public class RegisterRules implements Startable {
   private final MyBatis myBatis;
   private final RuleDao ruleDao;
   private final ActiveRuleDao activeRuleDao;
-  private final RuleTagOperations ruleTagOperations;
   private final CharacteristicDao characteristicDao;
-  private final System2 system;
 
 
   public RegisterRules(RuleDefinitionsLoader defLoader, ProfilesManager profilesManager,
-                       MyBatis myBatis, RuleDao ruleDao, RuleTagOperations ruleTagOperations,
-                       ActiveRuleDao activeRuleDao, CharacteristicDao characteristicDao) {
-    this(defLoader, profilesManager, myBatis, ruleDao, ruleTagOperations, activeRuleDao, characteristicDao, System2.INSTANCE);
+                       MyBatis myBatis, RuleDao ruleDao, ActiveRuleDao activeRuleDao,
+                       CharacteristicDao characteristicDao) {
+    this(defLoader, profilesManager, myBatis, ruleDao, activeRuleDao, characteristicDao, System2.INSTANCE);
   }
 
   @VisibleForTesting
   RegisterRules(RuleDefinitionsLoader defLoader, ProfilesManager profilesManager,
-                MyBatis myBatis, RuleDao ruleDao, RuleTagOperations ruleTagOperations,
-                ActiveRuleDao activeRuleDao, CharacteristicDao characteristicDao, System2 system) {
+                MyBatis myBatis, RuleDao ruleDao, ActiveRuleDao activeRuleDao,
+                CharacteristicDao characteristicDao, System2 system) {
     this.defLoader = defLoader;
     this.profilesManager = profilesManager;
     this.myBatis = myBatis;
     this.ruleDao = ruleDao;
-    this.ruleTagOperations = ruleTagOperations;
     this.activeRuleDao = activeRuleDao;
     this.characteristicDao = characteristicDao;
-    this.system = system;
   }
 
   @Override
   public void start() {
     TimeProfiler profiler = new TimeProfiler().start("Register rules");
-    DbSession sqlSession = myBatis.openSession(false);
+    DbSession session = myBatis.openSession(false);
     try {
+
+      Map<RuleKey, RuleDto> allRules = getRulesByKey(session);
+
       RulesDefinition.Context context = defLoader.load();
-      Buffer buffer = new Buffer(system.now());
-      selectRulesFromDb(buffer, sqlSession);
-      enableRuleDefinitions(context, buffer, sqlSession);
-      List<RuleDto> removedRules = processRemainingDbRules(buffer, sqlSession);
-      removeActiveRulesOnStillExistingRepositories(removedRules, context);
-      sqlSession.commit();
+      for (RulesDefinition.ExtendedRepository repoDef : getRepositories(context)) {
+        for (RulesDefinition.Rule ruleDef : repoDef.rules()) {
+
+          RuleKey ruleKey = RuleKey.of(ruleDef.repository().key(), ruleDef.key());
+
+          RuleDto rule = allRules.containsKey(ruleKey) ?
+            allRules.remove(ruleKey) :
+            this.createRuleDto(ruleDef, session);
+
+          boolean executeUpdate = false;
+          if (mergeRule(ruleDef, rule)) {
+            executeUpdate = true;
+          }
+
+          if(rule.getSubCharacteristicId() != null) {
+            CharacteristicDto characteristicDto = characteristicDao.selectById(rule.getSubCharacteristicId(), session);
+            if (characteristicDto != null && mergeDebtDefinitions(ruleDef, rule, characteristicDto)) {
+              executeUpdate = true;
+            }
+          }
+
+          if (mergeTags(ruleDef, rule)) {
+            executeUpdate = true;
+          }
+
+          if (executeUpdate) {
+            ruleDao.update(rule, session);
+          }
+
+          mergeParams(ruleDef, rule, session);
+
+        }
+      }
+      List<RuleDto> activeRules = processRemainingDbRules(allRules.values(), session);
+      removeActiveRulesOnStillExistingRepositories(activeRules, context);
+
+      session.commit();
 
     } finally {
-      sqlSession.close();
+      session.close();
       profiler.stop();
     }
+
   }
 
   @Override
@@ -126,102 +147,43 @@ public class RegisterRules implements Startable {
     // nothing
   }
 
-  private void selectRulesFromDb(Buffer buffer, DbSession sqlSession) {
-    for (RuleDto ruleDto : ruleDao.findByNonManual(sqlSession)) {
-      buffer.add(ruleDto);
-      buffer.markUnprocessed(ruleDto);
-
-      for (RuleParamDto paramDto : ruleDao.findRuleParamsByRuleKey(ruleDto.getKey(), sqlSession)) {
-        buffer.add(paramDto);
-      }
-    }
-
-    for (CharacteristicDto characteristicDto : characteristicDao.selectEnabledCharacteristics(sqlSession)) {
-      buffer.add(characteristicDto);
+  private Map<RuleKey, RuleDto> getRulesByKey(DbSession session) {
+    Map<RuleKey, RuleDto> rules = new HashMap<RuleKey, RuleDto>();
+    for (RuleDto rule : ruleDao.findAll(session)) {
+      rules.put(rule.getKey(), rule);
     }
+    return rules;
   }
 
-  private void enableRuleDefinitions(RulesDefinition.Context context, Buffer buffer, DbSession sqlSession) {
+  private List<RulesDefinition.ExtendedRepository> getRepositories(RulesDefinition.Context context) {
+    List<RulesDefinition.ExtendedRepository> repositories = new ArrayList<RulesDefinition.ExtendedRepository>();
     for (RulesDefinition.Repository repoDef : context.repositories()) {
-      enableRepository(buffer, sqlSession, repoDef);
+      repositories.add(repoDef);
     }
     for (RulesDefinition.ExtendedRepository extendedRepoDef : context.extendedRepositories()) {
       if (context.repository(extendedRepoDef.key()) == null) {
         LOG.warn(String.format("Extension is ignored, repository %s does not exist", extendedRepoDef.key()));
       } else {
-        enableRepository(buffer, sqlSession, extendedRepoDef);
+        repositories.add(extendedRepoDef);
       }
     }
+    return repositories;
   }
 
-  private void enableRepository(Buffer buffer, DbSession sqlSession, RulesDefinition.ExtendedRepository repoDef) {
-    int count = 0;
-    for (RulesDefinition.Rule ruleDef : repoDef.rules()) {
-      RuleDto dto = buffer.rule(RuleKey.of(ruleDef.repository().key(), ruleDef.key()));
-      if (dto == null) {
-        dto = enableAndInsert(buffer, sqlSession, ruleDef);
-      } else {
-        enableAndUpdate(buffer, sqlSession, ruleDef, dto);
-      }
-      buffer.markProcessed(dto);
-      count++;
-      if (count % 100 == 0) {
-        sqlSession.commit();
-      }
-    }
-    sqlSession.commit();
-  }
-
-  private RuleDto enableAndInsert(Buffer buffer, DbSession sqlSession, RulesDefinition.Rule ruleDef) {
-    RuleDto ruleDto = new RuleDto()
+  private RuleDto createRuleDto(RulesDefinition.Rule ruleDef, DbSession session) {
+    RuleDto ruleDto = RuleDto.createFor(RuleKey.of(ruleDef.repository().key(), ruleDef.key()))
       .setCardinality(ruleDef.template() ? Cardinality.MULTIPLE : Cardinality.SINGLE)
       .setConfigKey(ruleDef.internalKey())
       .setDescription(ruleDef.htmlDescription())
       .setLanguage(ruleDef.repository().language())
       .setName(ruleDef.name())
-      .setRepositoryKey(ruleDef.repository().key())
-      .setRuleKey(ruleDef.key())
       .setSeverity(ruleDef.severity())
-      .setCreatedAt(buffer.now())
-      .setUpdatedAt(buffer.now())
       .setStatus(ruleDef.status().name());
 
-    CharacteristicDto characteristic = buffer.characteristic(ruleDef.debtSubCharacteristic(), ruleDef.repository().key(), ruleDef.key(), null);
-    DebtRemediationFunction remediationFunction = ruleDef.debtRemediationFunction();
-    if (characteristic != null && remediationFunction != null) {
-      ruleDto.setDefaultSubCharacteristicId(characteristic.getId())
-        .setDefaultRemediationFunction(remediationFunction.type().name())
-        .setDefaultRemediationCoefficient(remediationFunction.coefficient())
-        .setDefaultRemediationOffset(remediationFunction.offset())
-        .setEffortToFixDescription(ruleDef.effortToFixDescription());
-    }
-
-    ruleDao.insert(ruleDto, sqlSession);
-    buffer.add(ruleDto);
-
-    for (RulesDefinition.Param param : ruleDef.params()) {
-      RuleParamDto paramDto = RuleParamDto.createFor(ruleDto)
-        .setDefaultValue(param.defaultValue())
-        .setDescription(param.description())
-        .setName(param.name())
-        .setType(param.type().toString());
-      ruleDao.addRuleParam(ruleDto, paramDto, sqlSession);
-      buffer.add(paramDto);
-    }
-    mergeTags(buffer, sqlSession, ruleDef, ruleDto);
-    return ruleDto;
-  }
-
-  private void enableAndUpdate(Buffer buffer, DbSession sqlSession, RulesDefinition.Rule ruleDef, RuleDto dto) {
-    if (mergeRule(buffer, ruleDef, dto)) {
-      ruleDao.update(dto, sqlSession);
-    }
-    mergeParams(buffer, sqlSession, ruleDef, dto);
-    mergeTags(buffer, sqlSession, ruleDef, dto);
-    buffer.markProcessed(dto);
+    return ruleDao.insert(ruleDto, session);
   }
 
-  private boolean mergeRule(Buffer buffer, RulesDefinition.Rule def, RuleDto dto) {
+  private boolean mergeRule(RulesDefinition.Rule def, RuleDto dto) {
     boolean changed = false;
     if (!StringUtils.equals(dto.getName(), def.name())) {
       dto.setName(def.name());
@@ -231,8 +193,8 @@ public class RegisterRules implements Startable {
       dto.setDescription(def.htmlDescription());
       changed = true;
     }
-    if (!ArrayUtils.isEquals(dto.getSystemTags(), def.tags())) {
-      dto.setSystemTags(def.tags().toArray(new String[def.tags().size()]));
+    if (!dto.getSystemTags().containsAll(def.tags())) {
+      dto.setSystemTags(def.tags());
       changed = true;
     }
     if (!StringUtils.equals(dto.getConfigKey(), def.internalKey())) {
@@ -258,11 +220,6 @@ public class RegisterRules implements Startable {
       dto.setLanguage(def.repository().language());
       changed = true;
     }
-    CharacteristicDto subCharacteristic = buffer.characteristic(def.debtSubCharacteristic(), def.repository().key(), def.key(), dto.getSubCharacteristicId());
-    changed = mergeDebtDefinitions(def, dto, subCharacteristic) || changed;
-    if (changed) {
-      dto.setUpdatedAt(buffer.now());
-    }
     return changed;
   }
 
@@ -305,33 +262,33 @@ public class RegisterRules implements Startable {
     return changed;
   }
 
-  private void mergeParams(Buffer buffer, DbSession sqlSession, RulesDefinition.Rule ruleDef, RuleDto dto) {
-    Collection<RuleParamDto> paramDtos = buffer.paramsForRuleId(dto.getId());
-    Set<String> persistedParamKeys = Sets.newHashSet();
+  private void mergeParams(RulesDefinition.Rule ruleDef, RuleDto rule, DbSession session) {
+    List<RuleParamDto> paramDtos = ruleDao.findRuleParamsByRuleKey(rule.getKey(), session);
+    List<String> existingParamDtoNames = new ArrayList<String>();
+
     for (RuleParamDto paramDto : paramDtos) {
       RulesDefinition.Param paramDef = ruleDef.param(paramDto.getName());
       if (paramDef == null) {
         //TODO cascade on the activeRule upon RuleDeletion
         //activeRuleDao.removeRuleParam(paramDto, sqlSession);
-        ruleDao.removeRuleParam(dto, paramDto, sqlSession);
+        ruleDao.removeRuleParam(rule, paramDto, session);
       } else {
         // TODO validate that existing active rules still match constraints
         // TODO store param name
         if (mergeParam(paramDto, paramDef)) {
-          ruleDao.updateRuleParam(dto,paramDto, sqlSession);
+          ruleDao.updateRuleParam(rule, paramDto, session);
         }
-        persistedParamKeys.add(paramDto.getName());
+        existingParamDtoNames.add(paramDto.getName());
       }
     }
     for (RulesDefinition.Param param : ruleDef.params()) {
-      if (!persistedParamKeys.contains(param.key())) {
-        RuleParamDto paramDto = RuleParamDto.createFor(dto)
+      if (!existingParamDtoNames.contains(param.key())) {
+        RuleParamDto paramDto = RuleParamDto.createFor(rule)
           .setName(param.key())
           .setDescription(param.description())
           .setDefaultValue(param.defaultValue())
           .setType(param.type().toString());
-        ruleDao.addRuleParam(dto, paramDto, sqlSession);
-        buffer.add(paramDto);
+        ruleDao.addRuleParam(rule, paramDto, session);
       }
     }
   }
@@ -353,55 +310,40 @@ public class RegisterRules implements Startable {
     return changed;
   }
 
-  private void mergeTags(Buffer buffer, DbSession sqlSession, RulesDefinition.Rule ruleDef, RuleDto dto) {
-    Set<String> existingSystemTags = Sets.newHashSet();
-    Collection<String> tags = ImmutableList.copyOf(buffer.tagsForRuleId(dto.getId()));
-    Collection<String> systemTags= ImmutableList.copyOf(buffer.systemTagsForRuleId(dto.getId()));
-
-    //TODO Check that with JUNIT for tag removal
-    for (String tag : tags) {
-        // tag previously declared by plugin
-        if (!ruleDef.tags().contains(tag)) {
-          // not declared anymore
-          dto.getTags().remove(tag);
-          buffer.removeTag(tag, dto.getId());
-        } else {
-          dto.getTags().add(tag);
-          existingSystemTags.add(tag);
-        }
-      }
+  private boolean mergeTags(RulesDefinition.Rule ruleDef, RuleDto dto) {
+    boolean changed = false;
 
-    for (String tag : systemTags) {
-        // tags created by end-users
-        if (ruleDef.tags().contains(tag)) {
-          dto.getSystemTags().add(tag);
-          existingSystemTags.add(tag);
-        }
-      }
-    ruleDao.update(dto, sqlSession);
-
-//    for (String tag : ruleDef.tags()) {
-//      if (!existingSystemTags.contains(tag)) {
-//        long tagId = getOrCreateReferenceTagId(buffer, tag, sqlSession);
-//        RuleRuleTagDto newTagDto = new RuleRuleTagDto()
-//          .setRuleId(dto.getId())
-//          .setTagId(tagId)
-//          .setTag(tag)
-//          .setType(RuleTagType.SYSTEM);
-//        ruleDao.insert(newTagDto, sqlSession);
-//        buffer.add(newTagDto);
+    //the Rule is not active and dto has tags
+    if (!Rule.STATUS_REMOVED.equals(ruleDef.status())) {
+      dto.setSystemTags(Collections.EMPTY_SET);
+      dto.setTags(Collections.EMPTY_SET);
+      changed = true;
+    } else if (!dto.getSystemTags().containsAll(ruleDef.tags())) {
+      dto.getSystemTags().addAll(ruleDef.tags());
+      changed = true;
+    }
+//    //TODO Check that with JUNIT for tag removal
+//    for (String tag : tags) {
+//      // tag previously declared by plugin
+//      if (!ruleDef.tags().contains(tag)) {
+//        // not declared anymore
+//        dto.getSystemTags().remove(tag);
+//        buffer.removeTag(tag, dto.getId());
+//      } else {
+//        dto.getSystemTags().add(tag);
+//        existingSystemTags.add(tag);
 //      }
 //    }
+    return changed;
   }
 
-  private List<RuleDto> processRemainingDbRules(Buffer buffer, DbSession sqlSession) {
+  private List<RuleDto> processRemainingDbRules(Collection<RuleDto> ruleDtos, DbSession session) {
     List<RuleDto> removedRules = newArrayList();
-    for (Integer unprocessedRuleId : buffer.unprocessedRuleIds) {
-      RuleDto ruleDto = buffer.rulesById.get(unprocessedRuleId);
+    for (RuleDto ruleDto : ruleDtos) {
       boolean toBeRemoved = true;
       // Update custom rules from template
       if (ruleDto.getParentId() != null) {
-        RuleDto parent = buffer.rulesById.get(ruleDto.getParentId());
+        RuleDto parent = ruleDao.getParent(ruleDto, session);
         if (parent != null && !Rule.STATUS_REMOVED.equals(parent.getStatus())) {
           ruleDto.setLanguage(parent.getLanguage());
           ruleDto.setStatus(parent.getStatus());
@@ -410,33 +352,25 @@ public class RegisterRules implements Startable {
           ruleDto.setDefaultRemediationCoefficient(parent.getDefaultRemediationCoefficient());
           ruleDto.setDefaultRemediationOffset(parent.getDefaultRemediationOffset());
           ruleDto.setEffortToFixDescription(parent.getEffortToFixDescription());
-          ruleDto.setUpdatedAt(buffer.now());
-          ruleDao.update(ruleDto, sqlSession);
+          ruleDao.update(ruleDto, session);
           toBeRemoved = false;
         }
       }
       if (toBeRemoved && !Rule.STATUS_REMOVED.equals(ruleDto.getStatus())) {
-        LOG.info(String.format("Disable rule %s:%s", ruleDto.getRepositoryKey(), ruleDto.getRuleKey()));
+        LOG.info(String.format("Disable rule %s", ruleDto.getKey()));
         ruleDto.setStatus(Rule.STATUS_REMOVED);
-        ruleDto.setUpdatedAt(buffer.now());
-
-        //TODO When a rule is removed what do we do about the tags?
-        for (String removed : buffer.tagsByRuleId.removeAll(ruleDto.getId())) {
-          ruleDto.getTags().remove(removed);
-        }
-        for (String removed : buffer.systemTagsByRuleId.removeAll(ruleDto.getId())) {
-          ruleDto.getSystemTags().remove(removed);
-        }
+        ruleDto.setSystemTags(Collections.EMPTY_SET);
+        ruleDto.setTags(Collections.EMPTY_SET);
+      }
 
-        ruleDao.update(ruleDto, sqlSession);
-        removedRules.add(ruleDto);
-        if (removedRules.size() % 100 == 0) {
-          sqlSession.commit();
-        }
+      ruleDao.update(ruleDto, session);
+      removedRules.add(ruleDto);
+      if (removedRules.size() % 100 == 0) {
+        session.commit();
       }
     }
-    sqlSession.commit();
 
+    session.commit();
     return removedRules;
   }
 
@@ -451,7 +385,7 @@ public class RegisterRules implements Startable {
    * The side effect of this approach is that extended repositories will not be managed the same way.
    * If an extended repository do not exists anymore, then related active rules will be removed.
    */
-  private void removeActiveRulesOnStillExistingRepositories(List<RuleDto> removedRules, RulesDefinition.Context context) {
+  private void removeActiveRulesOnStillExistingRepositories(Collection<RuleDto> removedRules, RulesDefinition.Context context) {
     List<String> repositoryKeys = newArrayList(Iterables.transform(context.repositories(), new Function<RulesDefinition.Repository, String>() {
         @Override
         public String apply(RulesDefinition.Repository input) {
@@ -467,104 +401,4 @@ public class RegisterRules implements Startable {
       }
     }
   }
-
-  static class Buffer {
-    private Date now;
-    private List<Integer> unprocessedRuleIds = newArrayList();
-    private Map<RuleKey, RuleDto> rulesByKey = Maps.newHashMap();
-    private Map<Integer, RuleDto> rulesById = Maps.newHashMap();
-    private Multimap<Integer, RuleParamDto> paramsByRuleId = ArrayListMultimap.create();
-    private Multimap<Integer, String> tagsByRuleId = ArrayListMultimap.create();
-    private Multimap<Integer, String> systemTagsByRuleId = ArrayListMultimap.create();
-    private Map<Integer, CharacteristicDto> characteristicsById = Maps.newHashMap();
-
-    Buffer(long now) {
-      this.now = new Date(now);
-    }
-
-    Date now() {
-      return now;
-    }
-
-    void add(RuleDto rule) {
-      rulesById.put(rule.getId(), rule);
-      rulesByKey.put(RuleKey.of(rule.getRepositoryKey(), rule.getRuleKey()), rule);
-    }
-
-    void add(CharacteristicDto characteristicDto) {
-      characteristicsById.put(characteristicDto.getId(), characteristicDto);
-    }
-
-    void add(RuleParamDto param) {
-      paramsByRuleId.put(param.getRuleId(), param);
-    }
-
-    void add(RuleRuleTagDto tag) {
-      if(tag.isSystemTag()){
-        systemTagsByRuleId.put(tag.getRuleId(), tag.getTag());
-      } else {
-        tagsByRuleId.put(tag.getRuleId(), tag.getTag());
-      }
-    }
-
-    void removeTag(String tag, int ruleId) {
-      tagsByRuleId.remove(ruleId, tag);
-    }
-
-    void removeSystemTag(String tag, int ruleId) {
-      systemTagsByRuleId.remove(ruleId, tag);
-    }
-
-    @CheckForNull
-    RuleDto rule(RuleKey key) {
-      return rulesByKey.get(key);
-    }
-
-    Collection<RuleParamDto> paramsForRuleId(Integer ruleId) {
-      return paramsByRuleId.get(ruleId);
-    }
-
-    Collection<String> tagsForRuleId(Integer ruleId) {
-      return tagsByRuleId.get(ruleId);
-    }
-
-    Collection<String> systemTagsForRuleId(Integer ruleId) {
-      return systemTagsByRuleId.get(ruleId);
-    }
-
-    void markUnprocessed(RuleDto ruleDto) {
-      unprocessedRuleIds.add(ruleDto.getId());
-    }
-
-    void markProcessed(RuleDto ruleDto) {
-      unprocessedRuleIds.remove(ruleDto.getId());
-    }
-
-    CharacteristicDto characteristic(@Nullable String subCharacteristic, String repo, String ruleKey, @Nullable Integer overridingCharacteristicId) {
-      // Rule is not linked to a default characteristic or characteristic has been disabled by user
-      if (subCharacteristic == null) {
-        return null;
-      }
-      CharacteristicDto characteristicDto = findCharacteristic(subCharacteristic);
-      if (characteristicDto == null) {
-        // Log a warning only if rule has not been overridden by user
-        if (overridingCharacteristicId == null) {
-          LOG.warn(String.format("Characteristic '%s' has not been found on rule '%s:%s'", subCharacteristic, repo, ruleKey));
-        }
-      } else if (characteristicDto.getParentId() == null) {
-        throw MessageException.of(String.format("Rule '%s:%s' cannot be linked on the root characteristic '%s'", repo, ruleKey, subCharacteristic));
-      }
-      return characteristicDto;
-    }
-
-    @CheckForNull
-    private CharacteristicDto findCharacteristic(final String key) {
-      return Iterables.find(characteristicsById.values(), new Predicate<CharacteristicDto>() {
-        @Override
-        public boolean apply(@Nullable CharacteristicDto input) {
-          return input != null && key.equals(input.getKey());
-        }
-      }, null);
-    }
-  }
-}
+}
\ No newline at end of file
diff --git a/sonar-server/src/main/java/org/sonar/server/rule2/RuleDao.java b/sonar-server/src/main/java/org/sonar/server/rule2/RuleDao.java
index 7a98fb8525e..e1b426d976e 100644
--- a/sonar-server/src/main/java/org/sonar/server/rule2/RuleDao.java
+++ b/sonar-server/src/main/java/org/sonar/server/rule2/RuleDao.java
@@ -77,6 +77,12 @@ public class RuleDao extends BaseDao<RuleMapper, RuleDto, RuleKey> implements Ba
     return mapper(session).selectById(id);
   }
 
+  @CheckForNull
+  public RuleDto getParent(RuleDto rule, DbSession session){
+    Preconditions.checkNotNull(rule.getParentId(), "Rule has no persisted parent!");
+    return mapper(session).selectById(rule.getParentId());
+  }
+
   @Override
   public Collection<RuleKey> keysOfRowsUpdatedAfter(long timestamp, DbSession session) {
     final List<RuleKey> keys = Lists.newArrayList();
diff --git a/sonar-server/src/test/java/org/sonar/server/rule2/RegisterRulesTest.java b/sonar-server/src/test/java/org/sonar/server/rule2/RegisterRulesTest.java
index 6ab5335531d..8963e627146 100644
--- a/sonar-server/src/test/java/org/sonar/server/rule2/RegisterRulesTest.java
+++ b/sonar-server/src/test/java/org/sonar/server/rule2/RegisterRulesTest.java
@@ -108,7 +108,7 @@ public class RegisterRulesTest extends AbstractDaoTestCase {
     ruleTagOperations = new RuleTagOperations(ruleTagDao, esRuleTags);
     characteristicDao = new CharacteristicDao(myBatis);
     task = new RegisterRules(new RuleDefinitionsLoader(mock(RuleRepositories.class), new RulesDefinition[]{new FakeRepository()}),
-      profilesManager, myBatis, ruleDao, ruleTagOperations, activeRuleDao, characteristicDao, system);
+      profilesManager, myBatis, ruleDao, activeRuleDao, characteristicDao, system);
     session = myBatis.openSession(false);
   }
 
@@ -191,8 +191,6 @@ public class RegisterRulesTest extends AbstractDaoTestCase {
   }
 
   @Test
-  @Ignore
-  //TODO Check cascade in RuleDao.delete(RuleDto dto)
   public void disable_deprecated_active_rule_params() {
     setupData("disable_deprecated_active_rule_params");
     task.start();
@@ -201,7 +199,6 @@ public class RegisterRulesTest extends AbstractDaoTestCase {
   }
 
   @Test
-  @Ignore
   //TODO check with mergeTag what happens on removal.
   public void disable_deprecated_rules() {
     setupData("disable_deprecated_rules");