How to use

updateIf

method

in

org.sonar.db.ce.CeQueueMapper

private Optional<CeQueueDto> tryToPeek(DbSession session, String eligibleTaskUuid, String workerUuid) {
 long now = system2.now();
 int touchedRows = mapper(session).updateIf(eligibleTaskUuid,
  new UpdateIf.NewProperties(IN_PROGRESS, workerUuid, now, now),
  new UpdateIf.OldProperties(PENDING));
 if (touchedRows != 1) {
  return Optional.empty();
 }
 CeQueueDto result = mapper(session).selectByUuid(eligibleTaskUuid);
 session.commit();
 return Optional.ofNullable(result);
}

public static void makeInProgress(DbSession dbSession, String workerUuid, long now, CeQueueDto... ceQueueDtos) {
 Stream.of(ceQueueDtos).forEach(ceQueueDto -> {
  CeQueueMapper mapper = dbSession.getMapper(CeQueueMapper.class);
  int touchedRows = mapper.updateIf(ceQueueDto.getUuid(),
   new UpdateIf.NewProperties(IN_PROGRESS, workerUuid, now, now),
   new UpdateIf.OldProperties(PENDING));
  assertThat(touchedRows).isEqualTo(1);
 });
}

 public static void reset(DbSession dbSession, long now, CeQueueDto... ceQueueDtos) {
  Stream.of(ceQueueDtos).forEach(ceQueueDto -> {
   checkArgument(ceQueueDto.getStatus() == IN_PROGRESS);
   checkArgument(ceQueueDto.getWorkerUuid() != null);

   CeQueueMapper mapper = dbSession.getMapper(CeQueueMapper.class);
   int touchedRows = mapper.updateIf(ceQueueDto.getUuid(),
    new UpdateIf.NewProperties(PENDING, ceQueueDto.getUuid(), now, now),
    new UpdateIf.OldProperties(IN_PROGRESS));
   assertThat(touchedRows).isEqualTo(1);
  });
 }
}