Skip to content

Commit

Permalink
Improve Ample logging for rejected conditional mutations (#5219)
Browse files Browse the repository at this point in the history
* Improve Ample logging for rejected conditional mutations

---------

Co-authored-by: Keith Turner <[email protected]>
  • Loading branch information
DomGarguilo and keith-turner authored Jan 6, 2025
1 parent be70c2a commit c06fa70
Show file tree
Hide file tree
Showing 13 changed files with 83 additions and 41 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import java.util.Set;
import java.util.function.Consumer;
import java.util.function.Predicate;
import java.util.function.Supplier;

import org.apache.accumulo.core.client.ConditionalWriter;
import org.apache.accumulo.core.client.admin.TabletAvailability;
Expand Down Expand Up @@ -623,6 +624,16 @@ ConditionalTabletMutator requireSame(TabletMetadata tabletMetadata, ColumnType t
* let the rejected status carry forward in this case.
*/
void submit(RejectionHandler rejectionHandler);

/**
* Overloaded version of {@link #submit(RejectionHandler)} that takes a short description of the
* operation to assist with debugging.
*
* @param rejectionHandler The rejection handler
* @param description A short description of the operation (e.g., "bulk import", "compaction")
*/
void submit(RejectionHandler rejectionHandler, Supplier<String> description);

}

/**
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
import java.util.function.BiConsumer;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;

import org.apache.accumulo.core.client.IteratorSetting;
import org.apache.accumulo.core.client.admin.TabletAvailability;
Expand Down Expand Up @@ -86,17 +87,20 @@ public class ConditionalTabletMutatorImpl extends TabletMutatorBase<Ample.Condit
private final ServerContext context;
private final ServiceLock lock;
private final KeyExtent extent;
private final BiConsumer<KeyExtent,Supplier<String>> descriptionConsumer;

private boolean sawOperationRequirement = false;
private boolean checkPrevEndRow = true;

protected ConditionalTabletMutatorImpl(ServerContext context, KeyExtent extent,
Consumer<ConditionalMutation> mutationConsumer,
BiConsumer<KeyExtent,Ample.RejectionHandler> rejectionHandlerConsumer) {
BiConsumer<KeyExtent,Ample.RejectionHandler> rejectionHandlerConsumer,
BiConsumer<KeyExtent,Supplier<String>> descriptionConsumer) {
super(new ConditionalMutation(extent.toMetaRow()));
this.mutation = (ConditionalMutation) super.mutation;
this.mutationConsumer = mutationConsumer;
this.rejectionHandlerConsumer = rejectionHandlerConsumer;
this.descriptionConsumer = descriptionConsumer;
this.extent = extent;
this.context = context;
this.lock = this.context.getServiceLock();
Expand Down Expand Up @@ -390,4 +394,10 @@ public void submit(Ample.RejectionHandler rejectionCheck) {
mutationConsumer.accept(mutation);
rejectionHandlerConsumer.accept(extent, rejectionCheck);
}

@Override
public void submit(Ample.RejectionHandler rejectionHandler, Supplier<String> description) {
descriptionConsumer.accept(extent, description);
this.submit(rejectionHandler);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
import java.util.Optional;
import java.util.Set;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collectors;

import org.apache.accumulo.core.client.AccumuloException;
Expand Down Expand Up @@ -66,6 +67,7 @@ public class ConditionalTabletsMutatorImpl implements Ample.ConditionalTabletsMu
private boolean active = true;

final Map<KeyExtent,Ample.RejectionHandler> rejectedHandlers = new HashMap<>();
private final Map<KeyExtent,Supplier<String>> operationDescriptions = new HashMap<>();
private final Function<DataLevel,String> tableMapper;

public ConditionalTabletsMutatorImpl(ServerContext context) {
Expand Down Expand Up @@ -93,7 +95,8 @@ public Ample.OperationRequirements mutateTablet(KeyExtent extent) {

Preconditions.checkState(extents.putIfAbsent(extent.toMetaRow(), extent) == null,
"Duplicate extents not handled %s", extent);
return new ConditionalTabletMutatorImpl(context, extent, mutations::add, rejectedHandlers::put);
return new ConditionalTabletMutatorImpl(context, extent, mutations::add, rejectedHandlers::put,
operationDescriptions::put);
}

protected ConditionalWriter createConditionalWriter(Ample.DataLevel dataLevel)
Expand Down Expand Up @@ -262,16 +265,20 @@ public Status getStatus() {
status = Status.ACCEPTED;
}

Supplier<String> descSupplier = operationDescriptions.get(extent);
String desc = (descSupplier == null) ? null : descSupplier.get();

if (log.isTraceEnabled()) {
// log detailed info about tablet metadata and mutation
log.trace("Mutation was rejected, status:{} {} {}", status, tabletMetadata,
result.getMutation().prettyPrint());
log.trace("Mutation was rejected, status:{}. Operation description: {} {} {}",
status, desc, tabletMetadata, result.getMutation().prettyPrint());
} else if (log.isDebugEnabled()) {
// log a single line of info that makes it apparent this happened and gives enough
// information to investigate
log.debug("Mutation was rejected, status:{} extent:{} row:{}", status,
tabletMetadata == null ? null : tabletMetadata.getExtent(),
new String(result.getMutation().getRow(), UTF_8));
log.debug(
"Mutation was rejected, status:{} extent:{} row:{} operation description: {}",
status, tabletMetadata == null ? null : tabletMetadata.getExtent(),
new String(result.getMutation().getRow(), UTF_8), desc);
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -350,12 +350,10 @@ public void hostOndemand(Collection<KeyExtent> extents) {
// Do not add any code here, it may interfere with the finally block removing extents from
// hostingRequestInProgress
try (var mutator = manager.getContext().getAmple().conditionallyMutateTablets()) {
inProgress.forEach(ke -> {
mutator.mutateTablet(ke).requireAbsentOperation()
.requireTabletAvailability(TabletAvailability.ONDEMAND).requireAbsentLocation()
.setHostingRequested().submit(TabletMetadata::getHostingRequested);

});
inProgress.forEach(ke -> mutator.mutateTablet(ke).requireAbsentOperation()
.requireTabletAvailability(TabletAvailability.ONDEMAND).requireAbsentLocation()
.setHostingRequested()
.submit(TabletMetadata::getHostingRequested, () -> "host ondemand"));

List<Range> ranges = new ArrayList<>();

Expand Down Expand Up @@ -1094,7 +1092,7 @@ private void replaceVolumes(List<VolumeUtil.VolumeReplacements> volumeReplacemen
"replaceVolume conditional mutation rejection check {} logsRemoved:{} filesRemoved:{}",
tm.getExtent(), logsRemoved, filesRemoved);
return logsRemoved && filesRemoved;
});
}, () -> "replace volume");
}

tabletsMutator.process().forEach((extent, result) -> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -635,8 +635,9 @@ public CompactionMetadata get() {
}

tabletMutator.putExternalCompaction(externalCompactionId, ecm);
tabletMutator
.submit(tm -> tm.getExternalCompactions().containsKey(externalCompactionId));
tabletMutator.submit(
tm -> tm.getExternalCompactions().containsKey(externalCompactionId),
() -> "compaction reservation");

var result = tabletsMutator.process().get(extent);

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -128,8 +128,9 @@ private TabletMetadata commitCompaction(ServerContext ctx, ExternalCompactionId
// make the needed updates to the tablet
updateTabletForCompaction(commitData.stats, ecid, tablet, newDatafile, ecm, tabletMutator);

tabletMutator
.submit(tabletMetadata -> !tabletMetadata.getExternalCompactions().containsKey(ecid));
tabletMutator.submit(
tabletMetadata -> !tabletMetadata.getExternalCompactions().containsKey(ecid),
() -> "commit compaction " + ecid);

if (LOG.isDebugEnabled()) {
LOG.debug("Compaction completed {} added {} removed {}", tablet.getExtent(), newDatafile,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,8 @@ public long isReady(FateId fateId, Manager manager) throws Exception {
tabletExtent);
mutator.mutateTablet(tabletExtent).requireAbsentOperation()
.putTabletAvailability(tabletAvailability)
.submit(tabletMeta -> tabletMeta.getTabletAvailability() == tabletAvailability);
.submit(tabletMeta -> tabletMeta.getTabletAvailability() == tabletAvailability,
() -> "set tablet availability");
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ private static void removeBulkLoadEntries(Ample ample, TableId tableId, FateId f
tabletsMutator.mutateTablet(tablet.getExtent()).requireAbsentOperation();
tablet.getLoaded().entrySet().stream().filter(entry -> entry.getValue().equals(fateId))
.map(Map.Entry::getKey).forEach(tabletMutator::deleteBulkFile);
tabletMutator.submit(tm -> false);
tabletMutator.submit(tm -> false, () -> "remove bulk load entries " + fateId);
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,7 @@ void load(List<TabletMetadata> tablets, Files files) {
Preconditions.checkState(
loadingFiles.put(tablet.getExtent(), List.copyOf(filesToLoad.keySet())) == null);

tabletMutator.submit(tm -> false);
tabletMutator.submit(tm -> false, () -> "bulk load files " + fateId);
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,8 @@ public int updateAndCheckTablets(Manager manager, FateId fateId)
// this tablet has no files try to mark it as done
tabletsMutator.mutateTablet(tablet.getExtent()).requireAbsentOperation()
.requireSame(tablet, FILES, COMPACTED).putCompacted(fateId)
.submit(tabletMetadata -> tabletMetadata.getCompacted().contains(fateId));
.submit(tabletMetadata -> tabletMetadata.getCompacted().contains(fateId),
() -> "no files, attempting to mark as compacted. " + fateId);
noFiles++;
} else if (tablet.getSelectedFiles() == null && tablet.getExternalCompactions().isEmpty()) {
// there are no selected files
Expand Down Expand Up @@ -242,7 +243,8 @@ public int updateAndCheckTablets(Manager manager, FateId fateId)
// no files were selected so mark the tablet as compacted
tabletsMutator.mutateTablet(tablet.getExtent()).requireAbsentOperation()
.requireSame(tablet, FILES, SELECTED, ECOMP, COMPACTED).putCompacted(fateId)
.submit(tabletMetadata -> tabletMetadata.getCompacted().contains(fateId));
.submit(tabletMetadata -> tabletMetadata.getCompacted().contains(fateId),
() -> "no files, attempting to mark as compacted. " + fateId);

noneSelected++;
} else {
Expand All @@ -260,9 +262,11 @@ public int updateAndCheckTablets(Manager manager, FateId fateId)

selectionsSubmitted.put(tablet.getExtent(), filesToCompact);

mutator.submit(tabletMetadata -> tabletMetadata.getSelectedFiles() != null
&& tabletMetadata.getSelectedFiles().getFateId().equals(fateId)
|| tabletMetadata.getCompacted().contains(fateId));
mutator.submit(
tabletMetadata -> tabletMetadata.getSelectedFiles() != null
&& tabletMetadata.getSelectedFiles().getFateId().equals(fateId)
|| tabletMetadata.getCompacted().contains(fateId),
() -> "selecting files for compaction. " + fateId);

if (minSelected == null || tablet.getExtent().compareTo(minSelected) < 0) {
minSelected = tablet.getExtent();
Expand Down Expand Up @@ -298,7 +302,8 @@ public int updateAndCheckTablets(Manager manager, FateId fateId)
var mutator = tabletsMutator.mutateTablet(tablet.getExtent()).requireAbsentOperation()
.requireSame(tablet, ECOMP, USER_COMPACTION_REQUESTED)
.putUserCompactionRequested(fateId);
mutator.submit(tm -> tm.getUserCompactionsRequested().contains(fateId));
mutator.submit(tm -> tm.getUserCompactionsRequested().contains(fateId),
() -> "marking as needing a user requested compaction. " + fateId);
userCompactionRequested++;
} else {
// Marker was already added and we are waiting
Expand Down Expand Up @@ -400,7 +405,8 @@ private void cleanupTabletMetadata(FateId fateId, Manager manager) throws Except
mutator.deleteUserCompactionRequested(fateId);
}

mutator.submit(needsNoUpdate::test);
mutator.submit(needsNoUpdate::test,
() -> "cleanup metadata for failed compaction. " + fateId);
}
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,8 @@ public long isReady(FateId fateId, Manager manager) throws Exception {
// must wait for the tablet to have no location before proceeding to actually delete. See
// the documentation about the opid column in the MetadataSchema class for more details.
conditionalMutator.mutateTablet(tabletMeta.getExtent()).requireAbsentOperation()
.putOperation(opid).submit(tm -> opid.equals(tm.getOperationId()));
.putOperation(opid)
.submit(tm -> opid.equals(tm.getOperationId()), () -> "put opid " + opid);
submitted++;
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -178,16 +178,20 @@ private Optional<KeyExtent> deleteTabletFiles(Manager manager, FateId fateId) {
}
}

filesToDelete.forEach(file -> log.debug("{} deleting file {} for {}", fateId, file,
tabletMetadata.getExtent()));
filesToAddMap.forEach((file, dfv) -> log.debug("{} adding file {} {} for {}", fateId, file,
dfv, tabletMetadata.getExtent()));

filesToDelete.forEach(tabletMutator::deleteFile);
filesToAddMap.forEach(tabletMutator::putFile);

tabletMutator.submit(tm -> tm.getFiles().containsAll(filesToAddMap.keySet())
&& Collections.disjoint(tm.getFiles(), filesToDelete));
filesToDelete.forEach(file -> {
log.debug("{} deleting file {} for {}", fateId, file, tabletMetadata.getExtent());
tabletMutator.deleteFile(file);
});

filesToAddMap.forEach((file, dfv) -> {
log.debug("{} adding file {} {} for {}", fateId, file, dfv, tabletMetadata.getExtent());
tabletMutator.putFile(file, dfv);
});

tabletMutator.submit(
tm -> tm.getFiles().containsAll(filesToAddMap.keySet())
&& Collections.disjoint(tm.getFiles(), filesToDelete),
() -> "delete tablet files (as part of merge or deleterow operation) " + fateId);
}

var results = tabletsMutator.process();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,8 @@ public void testLocations() throws Exception {
// test require absent with a future location set
try (var ctmi = new ConditionalTabletsMutatorImpl(context)) {
ctmi.mutateTablet(e1).requireAbsentOperation().requireAbsentLocation()
.putLocation(Location.future(ts2)).submit(tm -> false);
.putLocation(Location.future(ts2)).submit(tm -> false,
() -> "Testing that requireAbsentLocation() fails when a future location is set");
assertEquals(Status.REJECTED, ctmi.process().get(e1).getStatus());
}
assertEquals(Location.future(ts1), context.getAmple().readTablet(e1).getLocation());
Expand All @@ -196,7 +197,8 @@ public void testLocations() throws Exception {
try (var ctmi = new ConditionalTabletsMutatorImpl(context)) {
ctmi.mutateTablet(e1).requireAbsentOperation().requireAbsentLocation()
.putLocation(Location.future(ts2)).submit(tm -> false);
assertEquals(Status.REJECTED, ctmi.process().get(e1).getStatus());
assertEquals(Status.REJECTED, ctmi.process().get(e1).getStatus(),
() -> "Testing that requireAbsentLocation() fails when a current location is set");
}
assertEquals(Location.current(ts1), context.getAmple().readTablet(e1).getLocation());

Expand Down

0 comments on commit c06fa70

Please sign in to comment.