001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2011 SonarSource 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020 package org.sonar.core.components; 021 022 import com.google.common.collect.BiMap; 023 import com.google.common.collect.HashBiMap; 024 import com.google.common.collect.Maps; 025 import org.sonar.api.rules.Rule; 026 import org.sonar.api.rules.RuleQuery; 027 import org.sonar.jpa.session.DatabaseSessionFactory; 028 029 import java.util.Map; 030 031 public final class CacheRuleFinder extends DefaultRuleFinder { 032 033 private BiMap<Integer, Rule> rulesById = HashBiMap.create(); 034 private Map<String, Map<String, Rule>> rulesByKey = Maps.newHashMap(); 035 036 public CacheRuleFinder(DatabaseSessionFactory sessionFactory) { 037 super(sessionFactory); 038 } 039 040 @Override 041 public Rule findById(int ruleId) { 042 Rule rule = rulesById.get(ruleId); 043 if (rule == null) { 044 rule = doFindById(ruleId); 045 if (rule != null) { 046 loadRepository(rule.getRepositoryKey()); 047 } 048 } 049 return rule; 050 } 051 052 @Override 053 public Rule findByKey(String repositoryKey, String ruleKey) { 054 Map<String, Rule> repository = loadRepository(repositoryKey); 055 return repository.get(ruleKey); 056 } 057 058 private Map<String, Rule> loadRepository(String repositoryKey) { 059 Map<String, Rule> repository = rulesByKey.get(repositoryKey); 060 if (repository == null) { 061 repository = Maps.newHashMap(); 062 rulesByKey.put(repositoryKey, repository); 063 064 for (Rule rule : findAll(RuleQuery.create().withRepositoryKey(repositoryKey))) { 065 repository.put(rule.getKey(), rule); 066 rulesById.put(rule.getId(), rule); 067 } 068 } 069 return repository; 070 } 071 }