123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- /*
- * SonarQube
- * Copyright (C) 2009-2024 SonarSource SA
- * mailto:info AT sonarsource DOT com
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 3 of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public License
- * along with this program; if not, write to the Free Software Foundation,
- * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
- */
- package org.sonar.ce.task.projectanalysis.issue;
-
- import java.util.HashMap;
- import java.util.Map;
- import java.util.Optional;
- import java.util.function.Supplier;
- import org.junit.jupiter.api.extension.AfterEachCallback;
- import org.junit.jupiter.api.extension.ExtensionContext;
- import org.junit.rules.ExternalResource;
- import org.sonar.api.rule.RuleKey;
- import org.sonar.db.DbSession;
-
- import static com.google.common.base.Preconditions.checkArgument;
- import static java.util.Objects.requireNonNull;
-
- public class RuleRepositoryRule extends ExternalResource implements RuleRepository, AfterEachCallback {
-
- private final Map<RuleKey, Rule> rulesByKey = new HashMap<>();
- private final Map<String, Rule> rulesByUuid = new HashMap<>();
- private final Map<RuleKey, NewAdHocRule> newExternalRulesById = new HashMap<>();
-
- @Override
- protected void after() {
- rulesByKey.clear();
- rulesByUuid.clear();
- }
-
- @Override
- public Rule getByKey(RuleKey key) {
- Rule rule = rulesByKey.get(requireNonNull(key));
- checkArgument(rule != null);
- return rule;
- }
-
- @Override
- public Rule getByUuid(String uuid) {
- Rule rule = rulesByUuid.get(uuid);
- checkArgument(rule != null);
- return rule;
- }
-
- @Override
- public Optional<Rule> findByKey(RuleKey key) {
- return Optional.ofNullable(rulesByKey.get(requireNonNull(key)));
- }
-
- @Override
- public Optional<Rule> findByUuid(String uuid) {
- return Optional.ofNullable(rulesByUuid.get(uuid));
- }
-
- @Override
- public void saveOrUpdateAddHocRules(DbSession dbSession) {
- throw new UnsupportedOperationException();
- }
-
- public DumbRule add(RuleKey key) {
- DumbRule rule = new DumbRule(key);
- rule.setUuid(key.rule());
- rulesByKey.put(key, rule);
- rulesByUuid.put(rule.getUuid(), rule);
- return rule;
- }
-
- public RuleRepositoryRule add(DumbRule rule) {
- rulesByKey.put(requireNonNull(rule.getKey()), rule);
- rulesByUuid.put(rule.getUuid(), rule);
- return this;
- }
-
- @Override
- public void addOrUpdateAddHocRuleIfNeeded(RuleKey ruleKey, Supplier<NewAdHocRule> ruleSupplier) {
- newExternalRulesById.computeIfAbsent(ruleKey, k -> ruleSupplier.get());
- }
-
- @Override
- public void afterEach(ExtensionContext context) {
- after();
- }
- }
|