| // Copyright (C) 2015 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.plugins.verifystatus.server; |
| |
| import com.google.common.collect.Maps; |
| import com.google.gerrit.extensions.restapi.RestReadView; |
| import com.google.gerrit.server.change.RevisionResource; |
| import com.google.gwtorm.server.OrmException; |
| import com.google.gwtorm.server.ResultSet; |
| import com.google.gwtorm.server.SchemaFactory; |
| import com.google.inject.Inject; |
| |
| import com.googlesource.gerrit.plugins.verifystatus.common.VerificationInfo; |
| |
| import org.apache.commons.lang.builder.CompareToBuilder; |
| import org.kohsuke.args4j.Option; |
| |
| import java.io.IOException; |
| import java.sql.Timestamp; |
| import java.util.Collections; |
| import java.util.Comparator; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| |
| public class GetVerifications implements RestReadView<RevisionResource> { |
| private final SchemaFactory<CiDb> schemaFactory; |
| |
| @Inject |
| GetVerifications(SchemaFactory<CiDb> schemaFactory) { |
| this.schemaFactory = schemaFactory; |
| } |
| |
| @Option(name = "--current", aliases = {"-c"}, |
| usage = "List only the current report") |
| private boolean current; |
| |
| @Override |
| public Map<String, VerificationInfo> apply(RevisionResource rsrc) |
| throws IOException, OrmException { |
| Map<String, VerificationInfo> out = Maps.newLinkedHashMap(); |
| try (CiDb db = schemaFactory.open()) { |
| ResultSet<PatchSetVerification> rs = |
| db.patchSetVerifications().byPatchSet(rsrc.getPatchSet().getId()); |
| List<PatchSetVerification> jobs = rs.toList(); |
| // sort the jobs list by reporter (acending) then reported date (decending) |
| Collections.sort(jobs, new Comparator<PatchSetVerification>() { |
| @Override |
| public int compare(PatchSetVerification a, PatchSetVerification b) { |
| return new CompareToBuilder() |
| .append(a.getReporter(),b.getReporter()) |
| .append(b.getGranted(),a.getGranted()) |
| .toComparison(); |
| } |
| }); |
| |
| if (current) { |
| Map<String, Timestamp> reported = new HashMap<>(); |
| for (PatchSetVerification v : jobs) { |
| if (!reported.containsKey(v.getReporter())) { |
| reported.put(v.getReporter(), v.getGranted()); |
| } |
| } |
| for (PatchSetVerification v : jobs) { |
| Timestamp ts = v.getGranted(); |
| if (reported.values().contains(ts)) { |
| VerificationInfo info = new VerificationInfo(); |
| info.value = v.getValue(); |
| info.abstain = v.getAbstain(); |
| info.url = v.getUrl(); |
| info.name = v.getName(); |
| info.reporter = v.getReporter(); |
| info.comment = v.getComment(); |
| info.granted = v.getGranted(); |
| info.category = v.getCategory(); |
| info.duration = v.getDuration(); |
| out.put(v.getJobId().get(), info); |
| } |
| } |
| } else { |
| // show all reports |
| for (PatchSetVerification v : jobs) { |
| VerificationInfo info = new VerificationInfo(); |
| info.value = v.getValue(); |
| info.abstain = v.getAbstain(); |
| info.url = v.getUrl(); |
| info.name = v.getName(); |
| info.reporter = v.getReporter(); |
| info.comment = v.getComment(); |
| info.granted = v.getGranted(); |
| info.category = v.getCategory(); |
| info.duration = v.getDuration(); |
| out.put(v.getJobId().get(), info); |
| } |
| } |
| } |
| return out; |
| } |
| } |