123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146 |
- /*
- * SonarQube
- * Copyright (C) 2009-2019 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.db.metric;
-
- import com.google.common.base.Predicate;
- import com.google.common.collect.Collections2;
- import com.google.common.collect.Lists;
- import java.util.ArrayList;
- import java.util.Collection;
- import java.util.Collections;
- import java.util.List;
- import java.util.Map;
- import java.util.Set;
- import javax.annotation.CheckForNull;
- import javax.annotation.Nonnull;
- import javax.annotation.Nullable;
- import org.apache.ibatis.session.RowBounds;
- import org.sonar.db.Dao;
- import org.sonar.db.DbSession;
- import org.sonar.db.RowNotFoundException;
-
- import static com.google.common.collect.Lists.newArrayList;
- import static org.sonar.db.DatabaseUtils.executeLargeInputs;
- import static org.sonar.db.DatabaseUtils.executeLargeInputsWithoutOutput;
-
- public class MetricDao implements Dao {
-
- @CheckForNull
- public MetricDto selectByKey(DbSession session, String key) {
- return mapper(session).selectByKey(key);
- }
-
- public List<MetricDto> selectByKeys(final DbSession session, Collection<String> keys) {
- return executeLargeInputs(keys, mapper(session)::selectByKeys);
- }
-
- public MetricDto selectOrFailByKey(DbSession session, String key) {
- MetricDto metric = selectByKey(session, key);
- if (metric == null) {
- throw new RowNotFoundException(String.format("Metric key '%s' not found", key));
- }
- return metric;
- }
-
- public List<MetricDto> selectAll(DbSession session) {
- return mapper(session).selectAll();
- }
-
- public List<MetricDto> selectEnabled(DbSession session) {
- return mapper(session).selectAllEnabled();
- }
-
- public List<MetricDto> selectEnabled(DbSession session, @Nullable Boolean isCustom, int offset, int limit) {
- Map<String, Object> properties;
- if (isCustom != null) {
- properties = Collections.singletonMap("isCustom", isCustom);
- } else {
- properties = Collections.emptyMap();
- }
-
- return mapper(session).selectAllEnabled(properties, new RowBounds(offset, limit));
- }
-
- public int countEnabled(DbSession session, @Nullable Boolean isCustom) {
- return mapper(session).countEnabled(isCustom);
- }
-
- public MetricDto insert(DbSession session, MetricDto dto) {
- mapper(session).insert(dto);
-
- return dto;
- }
-
- public void insert(DbSession session, Collection<MetricDto> items) {
- for (MetricDto item : items) {
- insert(session, item);
- }
- }
-
- public void insert(DbSession session, MetricDto item, MetricDto... others) {
- insert(session, Lists.asList(item, others));
- }
-
- public List<String> selectEnabledDomains(DbSession session) {
- return newArrayList(Collections2.filter(mapper(session).selectDomains(), new NotEmptyPredicate()));
- }
-
- public List<MetricDto> selectAvailableCustomMetricsByComponentUuid(DbSession session, String projectUuid) {
- return mapper(session).selectAvailableCustomMetricsByComponentUuid(projectUuid);
- }
-
- public List<MetricDto> selectByIds(DbSession session, Set<Integer> idsSet) {
- return executeLargeInputs(new ArrayList<>(idsSet), mapper(session)::selectByIds);
- }
-
- private static class NotEmptyPredicate implements Predicate<String> {
-
- @Override
- public boolean apply(@Nonnull String input) {
- return !input.isEmpty();
- }
- }
-
- private static MetricMapper mapper(DbSession session) {
- return session.getMapper(MetricMapper.class);
- }
-
- public void disableCustomByIds(final DbSession session, List<Integer> ids) {
- executeLargeInputsWithoutOutput(ids, input -> mapper(session).disableByIds(input));
- }
-
- /**
- * Disable a metric and return {@code false} if the metric does not exist
- * or is already disabled.
- */
- public boolean disableCustomByKey(DbSession session, String key) {
- return mapper(session).disableByKey(key) == 1;
- }
-
- public void update(DbSession session, MetricDto metric) {
- mapper(session).update(metric);
- }
-
- @CheckForNull
- public MetricDto selectById(DbSession session, long id) {
- return mapper(session).selectById(id);
- }
-
- }
|