diff options
Diffstat (limited to 'java/com/google/gerrit/server/PatchSetUtil.java')
-rw-r--r-- | java/com/google/gerrit/server/PatchSetUtil.java | 225 |
1 files changed, 225 insertions, 0 deletions
diff --git a/java/com/google/gerrit/server/PatchSetUtil.java b/java/com/google/gerrit/server/PatchSetUtil.java new file mode 100644 index 0000000000..f6c7abc09a --- /dev/null +++ b/java/com/google/gerrit/server/PatchSetUtil.java @@ -0,0 +1,225 @@ +// Copyright (C) 2016 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; + +import static com.google.common.base.Preconditions.checkArgument; +import static com.google.common.collect.ImmutableMap.toImmutableMap; +import static com.google.gerrit.server.ChangeUtil.PS_ID_ORDER; +import static com.google.gerrit.server.notedb.PatchSetState.PUBLISHED; +import static java.util.Objects.requireNonNull; +import static java.util.function.Function.identity; + +import com.google.common.collect.ImmutableCollection; +import com.google.common.collect.ImmutableMap; +import com.google.common.collect.Maps; +import com.google.common.collect.Sets; +import com.google.common.collect.Streams; +import com.google.gerrit.common.data.LabelFunction; +import com.google.gerrit.common.data.LabelType; +import com.google.gerrit.extensions.restapi.ResourceConflictException; +import com.google.gerrit.reviewdb.client.Change; +import com.google.gerrit.reviewdb.client.PatchSet; +import com.google.gerrit.reviewdb.client.PatchSetApproval; +import com.google.gerrit.reviewdb.client.Project; +import com.google.gerrit.reviewdb.client.RevId; +import com.google.gerrit.reviewdb.server.ReviewDb; +import com.google.gerrit.server.git.GitRepositoryManager; +import com.google.gerrit.server.notedb.ChangeNotes; +import com.google.gerrit.server.notedb.ChangeUpdate; +import com.google.gerrit.server.notedb.NotesMigration; +import com.google.gerrit.server.project.ProjectCache; +import com.google.gerrit.server.project.ProjectState; +import com.google.gwtorm.server.OrmException; +import com.google.inject.Inject; +import com.google.inject.Provider; +import com.google.inject.Singleton; +import java.io.IOException; +import java.sql.Timestamp; +import java.util.Collections; +import java.util.List; +import java.util.Set; +import org.eclipse.jgit.lib.ObjectId; +import org.eclipse.jgit.lib.Repository; +import org.eclipse.jgit.revwalk.RevCommit; +import org.eclipse.jgit.revwalk.RevWalk; + +/** Utilities for manipulating patch sets. */ +@Singleton +public class PatchSetUtil { + private final NotesMigration migration; + private final Provider<ApprovalsUtil> approvalsUtilProvider; + private final ProjectCache projectCache; + private final Provider<ReviewDb> dbProvider; + private final GitRepositoryManager repoManager; + + @Inject + PatchSetUtil( + NotesMigration migration, + Provider<ApprovalsUtil> approvalsUtilProvider, + ProjectCache projectCache, + Provider<ReviewDb> dbProvider, + GitRepositoryManager repoManager) { + this.migration = migration; + this.approvalsUtilProvider = approvalsUtilProvider; + this.projectCache = projectCache; + this.dbProvider = dbProvider; + this.repoManager = repoManager; + } + + public PatchSet current(ReviewDb db, ChangeNotes notes) throws OrmException { + return get(db, notes, notes.getChange().currentPatchSetId()); + } + + public PatchSet get(ReviewDb db, ChangeNotes notes, PatchSet.Id psId) throws OrmException { + if (!migration.readChanges()) { + return db.patchSets().get(psId); + } + return notes.load().getPatchSets().get(psId); + } + + public ImmutableCollection<PatchSet> byChange(ReviewDb db, ChangeNotes notes) + throws OrmException { + if (!migration.readChanges()) { + return PS_ID_ORDER.immutableSortedCopy(db.patchSets().byChange(notes.getChangeId())); + } + return notes.load().getPatchSets().values(); + } + + public ImmutableMap<PatchSet.Id, PatchSet> byChangeAsMap(ReviewDb db, ChangeNotes notes) + throws OrmException { + if (!migration.readChanges()) { + ImmutableMap.Builder<PatchSet.Id, PatchSet> result = ImmutableMap.builder(); + for (PatchSet ps : PS_ID_ORDER.sortedCopy(db.patchSets().byChange(notes.getChangeId()))) { + result.put(ps.getId(), ps); + } + return result.build(); + } + return notes.load().getPatchSets(); + } + + public ImmutableMap<PatchSet.Id, PatchSet> getAsMap( + ReviewDb db, ChangeNotes notes, Set<PatchSet.Id> patchSetIds) throws OrmException { + if (!migration.readChanges()) { + patchSetIds = Sets.filter(patchSetIds, p -> p.getParentKey().equals(notes.getChangeId())); + return Streams.stream(db.patchSets().get(patchSetIds)) + .sorted(PS_ID_ORDER) + .collect(toImmutableMap(PatchSet::getId, identity())); + } + return ImmutableMap.copyOf(Maps.filterKeys(notes.load().getPatchSets(), patchSetIds::contains)); + } + + public PatchSet insert( + ReviewDb db, + RevWalk rw, + ChangeUpdate update, + PatchSet.Id psId, + ObjectId commit, + List<String> groups, + String pushCertificate, + String description) + throws OrmException, IOException { + requireNonNull(groups, "groups may not be null"); + ensurePatchSetMatches(psId, update); + + PatchSet ps = new PatchSet(psId); + ps.setRevision(new RevId(commit.name())); + ps.setUploader(update.getAccountId()); + ps.setCreatedOn(new Timestamp(update.getWhen().getTime())); + ps.setGroups(groups); + ps.setPushCertificate(pushCertificate); + ps.setDescription(description); + db.patchSets().insert(Collections.singleton(ps)); + + update.setCommit(rw, commit, pushCertificate); + update.setPsDescription(description); + update.setGroups(groups); + + return ps; + } + + public void publish(ReviewDb db, ChangeUpdate update, PatchSet ps) throws OrmException { + ensurePatchSetMatches(ps.getId(), update); + update.setPatchSetState(PUBLISHED); + db.patchSets().update(Collections.singleton(ps)); + } + + private static void ensurePatchSetMatches(PatchSet.Id psId, ChangeUpdate update) { + Change.Id changeId = update.getChange().getId(); + checkArgument( + psId.getParentKey().equals(changeId), + "cannot modify patch set %s on update for change %s", + psId, + changeId); + if (update.getPatchSetId() != null) { + checkArgument( + update.getPatchSetId().equals(psId), + "cannot modify patch set %s on update for %s", + psId, + update.getPatchSetId()); + } else { + update.setPatchSetId(psId); + } + } + + public void setGroups(ReviewDb db, ChangeUpdate update, PatchSet ps, List<String> groups) + throws OrmException { + ps.setGroups(groups); + update.setGroups(groups); + db.patchSets().update(Collections.singleton(ps)); + } + + /** Check if the current patch set of the change is locked. */ + public void checkPatchSetNotLocked(ChangeNotes notes) + throws OrmException, IOException, ResourceConflictException { + if (isPatchSetLocked(notes)) { + throw new ResourceConflictException( + String.format("The current patch set of change %s is locked", notes.getChangeId())); + } + } + + /** Is the current patch set locked against state changes? */ + public boolean isPatchSetLocked(ChangeNotes notes) throws OrmException, IOException { + Change change = notes.getChange(); + if (change.getStatus() == Change.Status.MERGED) { + return false; + } + + ProjectState projectState = projectCache.checkedGet(notes.getProjectName()); + requireNonNull( + projectState, () -> String.format("Failed to load project %s", notes.getProjectName())); + + ApprovalsUtil approvalsUtil = approvalsUtilProvider.get(); + for (PatchSetApproval ap : + approvalsUtil.byPatchSet(dbProvider.get(), notes, change.currentPatchSetId(), null, null)) { + LabelType type = projectState.getLabelTypes(notes).byLabel(ap.getLabel()); + if (type != null + && ap.getValue() == 1 + && type.getFunction() == LabelFunction.PATCH_SET_LOCK) { + return true; + } + } + return false; + } + + /** Returns the commit for the given project at the given patchset revision */ + public RevCommit getRevCommit(Project.NameKey project, PatchSet patchSet) throws IOException { + try (Repository repo = repoManager.openRepository(project); + RevWalk rw = new RevWalk(repo)) { + RevCommit src = rw.parseCommit(ObjectId.fromString(patchSet.getRevision().get())); + rw.parseBody(src); + return src; + } + } +} |