| // 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.List; |
| import java.util.Map; |
| |
| public class GetVerifications implements RestReadView<RevisionResource> { |
| private final SchemaFactory<CiDb> schemaFactory; |
| |
| @Inject |
| GetVerifications(SchemaFactory<CiDb> schemaFactory) { |
| this.schemaFactory = schemaFactory; |
| } |
| |
| private String sort; |
| private String filter; |
| |
| @Option(name = "--sort", aliases = {"-s"}, metaVar = "SORT", |
| usage = "Sort the list by an entry") |
| public void setSort(String sort) { |
| this.sort = sort.toUpperCase(); |
| } |
| |
| @Option(name = "--filter", aliases = {"-f"}, metaVar = "FILTER", |
| usage = "filter the results") |
| public void setFilter(String filter) { |
| this.filter = filter.toUpperCase(); |
| } |
| |
| @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(); |
| if (sort != null && !sort.isEmpty()) { |
| // sort the jobs list by reporter (ascending) then reported date (descending) |
| if (sort.equals("REPORTER")) { |
| 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(); |
| } |
| }); |
| } else if (sort.equals("NAME")) { |
| // sort the jobs list by name (ascending) then reported date (descending) |
| Collections.sort(jobs, new Comparator<PatchSetVerification>() { |
| @Override |
| public int compare(PatchSetVerification a, PatchSetVerification b) { |
| return new CompareToBuilder() |
| .append(a.getName(),b.getName()) |
| .append(b.getGranted(),a.getGranted()) |
| .toComparison(); |
| } |
| }); |
| } |
| } |
| |
| if (filter != null && !filter.isEmpty()) { |
| if (filter.equals("CURRENT") ) { |
| Map<String, Timestamp> reported = Maps.newHashMap(); |
| 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 if (filter.equals("FAILED") ) { |
| for (PatchSetVerification v : jobs) { |
| if (v.getValue() < 0) { |
| 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; |
| } |
| } |