blob: db21f1178053ec9695b1bd9074d31024e69f49c5 [file] [log] [blame]
// Copyright (C) 2019 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.google.gerrit.server.change;
import static java.util.concurrent.TimeUnit.MINUTES;
import com.google.common.collect.ImmutableListMultimap;
import com.google.common.flogger.FluentLogger;
import com.google.gerrit.entities.Change;
import com.google.gerrit.extensions.common.PluginDefinedInfo;
import com.google.gerrit.extensions.registration.Extension;
import com.google.gerrit.server.DynamicOptions.BeanProvider;
import com.google.gerrit.server.query.change.ChangeData;
import java.util.Collection;
import java.util.stream.Stream;
/** Static helpers for use by {@link PluginDefinedInfosFactory} implementations. */
public class PluginDefinedAttributesFactories {
private static final FluentLogger logger = FluentLogger.forEnclosingClass();
public static ImmutableListMultimap<Change.Id, PluginDefinedInfo> createAll(
Collection<ChangeData> cds,
BeanProvider beanProvider,
Stream<Extension<ChangePluginDefinedInfoFactory>> infoFactories) {
ImmutableListMultimap.Builder<Change.Id, PluginDefinedInfo> pluginInfosByChangeBuilder =
ImmutableListMultimap.builder();
infoFactories.forEach(
e -> tryCreate(cds, beanProvider, e.getPluginName(), e.get(), pluginInfosByChangeBuilder));
ImmutableListMultimap<Change.Id, PluginDefinedInfo> result = pluginInfosByChangeBuilder.build();
return result;
}
private static void tryCreate(
Collection<ChangeData> cds,
BeanProvider beanProvider,
String plugin,
ChangePluginDefinedInfoFactory infoFactory,
ImmutableListMultimap.Builder<Change.Id, PluginDefinedInfo> pluginInfosByChangeBuilder) {
try {
infoFactory
.createPluginDefinedInfos(cds, beanProvider, plugin)
.forEach(
(id, pdi) -> {
if (pdi != null) {
pdi.name = plugin;
pluginInfosByChangeBuilder.put(id, pdi);
}
});
} catch (RuntimeException ex) {
/* Propagate runtime exceptions as structured API data types so that queries don't fail. */
logger.atWarning().atMostEvery(1, MINUTES).withCause(ex).log(
"error populating attribute on changes from plugin %s", plugin);
PluginDefinedInfo errorInfo = new PluginDefinedInfo();
errorInfo.name = plugin;
errorInfo.message = "Something went wrong in plugin: " + plugin;
cds.forEach(cd -> pluginInfosByChangeBuilder.put(cd.getId(), errorInfo));
}
}
private PluginDefinedAttributesFactories() {}
}