| // Copyright (C) 2021 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or |
| // implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package com.googlesource.gerrit.owners.common; |
| |
| import static com.google.gerrit.extensions.client.InheritableBoolean.TRUE; |
| import static com.googlesource.gerrit.owners.common.AutoassignConfigModule.PROJECT_CONFIG_AUTOASSIGN_FIELD; |
| import static com.googlesource.gerrit.owners.common.AutoassignConfigModule.PROJECT_CONFIG_AUTOASSIGN_WIP_CHANGES; |
| import static java.util.concurrent.TimeUnit.MILLISECONDS; |
| |
| import com.google.common.annotations.VisibleForTesting; |
| import com.google.gerrit.entities.Project; |
| import com.google.gerrit.extensions.annotations.PluginName; |
| import com.google.gerrit.extensions.client.ReviewerState; |
| import com.google.gerrit.server.config.ConfigUtil; |
| import com.google.gerrit.server.config.PluginConfig; |
| import com.google.gerrit.server.config.PluginConfigFactory; |
| import com.google.gerrit.server.project.NoSuchProjectException; |
| import com.google.inject.Inject; |
| import com.google.inject.Singleton; |
| import org.eclipse.jgit.lib.Config; |
| |
| @Singleton |
| public class AutoAssignConfig { |
| public static final String REVIEWERS_SECTION = "reviewers"; |
| public static final String ASYNC = "async"; |
| public static final boolean ASYNC_DEF = false; |
| public static final String DELAY = "delay"; |
| public static final long DELAY_MSEC_DEF = 1500L; |
| public static final String RETRY_COUNT = "retryCount"; |
| public static final int RETRY_COUNT_DEF = 2; |
| public static final String RETRY_INTERVAL = "retryInterval"; |
| public static final long RETRY_INTERVAL_MSEC_DEF = DELAY_MSEC_DEF; |
| public static final String THREADS = "threads"; |
| public static final int THREADS_DEF = 1; |
| |
| private final boolean asyncReviewers; |
| private final int asyncThreads; |
| private final int retryCount; |
| private final long retryInterval; |
| private final long asyncDelay; |
| |
| private final PluginConfigFactory cfgFactory; |
| private final String pluginName; |
| |
| private final PluginSettings settings; |
| |
| @Inject |
| AutoAssignConfig( |
| PluginSettings settings, PluginConfigFactory configFactory, @PluginName String pluginName) { |
| this.settings = settings; |
| this.pluginName = pluginName; |
| this.cfgFactory = configFactory; |
| Config config = configFactory.getGlobalPluginConfig(pluginName); |
| asyncReviewers = config.getBoolean(REVIEWERS_SECTION, ASYNC, ASYNC_DEF); |
| asyncThreads = config.getInt(REVIEWERS_SECTION, THREADS, THREADS_DEF); |
| asyncDelay = |
| ConfigUtil.getTimeUnit( |
| config, REVIEWERS_SECTION, null, DELAY, DELAY_MSEC_DEF, MILLISECONDS); |
| |
| retryCount = config.getInt(REVIEWERS_SECTION, RETRY_COUNT, RETRY_COUNT_DEF); |
| retryInterval = |
| ConfigUtil.getTimeUnit( |
| config, REVIEWERS_SECTION, null, RETRY_INTERVAL, asyncDelay, MILLISECONDS); |
| } |
| |
| @VisibleForTesting |
| public AutoAssignConfig() { |
| this.settings = null; |
| this.pluginName = "owners-autoassign"; |
| cfgFactory = null; |
| Config config = new Config(); |
| asyncReviewers = config.getBoolean(REVIEWERS_SECTION, ASYNC, ASYNC_DEF); |
| asyncThreads = config.getInt(REVIEWERS_SECTION, THREADS, THREADS_DEF); |
| asyncDelay = |
| ConfigUtil.getTimeUnit( |
| config, REVIEWERS_SECTION, null, DELAY, DELAY_MSEC_DEF, MILLISECONDS); |
| |
| retryCount = config.getInt(REVIEWERS_SECTION, RETRY_COUNT, RETRY_COUNT_DEF); |
| retryInterval = |
| ConfigUtil.getTimeUnit( |
| config, REVIEWERS_SECTION, null, RETRY_INTERVAL, asyncDelay, MILLISECONDS); |
| } |
| |
| public boolean isAsyncReviewers() { |
| return asyncReviewers; |
| } |
| |
| public int asyncThreads() { |
| return asyncThreads; |
| } |
| |
| public int retryCount() { |
| return retryCount; |
| } |
| |
| public long retryInterval() { |
| return retryInterval; |
| } |
| |
| public long asyncDelay() { |
| return asyncDelay; |
| } |
| |
| private PluginConfig cfg(Project.NameKey projectKey) throws NoSuchProjectException { |
| return cfgFactory.getFromProjectConfigWithInheritance(projectKey, pluginName); |
| } |
| |
| public boolean autoAssignWip(Project.NameKey projectKey) throws NoSuchProjectException { |
| return settings |
| .projectSpecificConfig(projectKey) |
| .getEnum(PROJECT_CONFIG_AUTOASSIGN_WIP_CHANGES, TRUE) |
| .equals(TRUE); |
| } |
| |
| public ReviewerState autoassignedReviewerState(Project.NameKey projectKey) |
| throws NoSuchProjectException { |
| return settings |
| .projectSpecificConfig(projectKey) |
| .getEnum(PROJECT_CONFIG_AUTOASSIGN_FIELD, ReviewerState.REVIEWER); |
| } |
| |
| public boolean isBranchDisabled(String branch) { |
| return settings.isBranchDisabled(branch); |
| } |
| |
| public boolean expandGroups() { |
| return settings.expandGroups(); |
| } |
| } |