public Collection<EsQueueDto> selectForRecovery(DbSession dbSession, long beforeDate, long limit) { return mapper(dbSession).selectForRecovery(beforeDate, limit); }
public static EsQueueDto create(String docType, String docUuid) { return new EsQueueDto().setDocType(docType).setDocId(docUuid); }
public static EsQueueDto create(String docType, String docId, @Nullable String docIdType, @Nullable String docRouting) { return new EsQueueDto().setDocType(docType) .setDocId(docId).setDocIdType(docIdType).setDocRouting(docRouting); } }
private Collection<EsQueueDto> insertIntoEsQueue(DbSession dbSession, Collection<String> projectUuids) { List<EsQueueDto> items = indexTypes.stream() .flatMap(indexType -> projectUuids.stream().map(projectUuid -> EsQueueDto.create(indexType.format(), projectUuid, null, projectUuid))) .collect(toArrayList()); dbClient.esQueueDao().insert(dbSession, items); return items; }
private static RuleExtensionId explodeRuleExtensionDocId(EsQueueDto esQueueDto) { checkArgument(Objects.equals(esQueueDto.getDocType(), "rules/ruleExtension")); return new RuleExtensionId(esQueueDto.getDocId()); }
private static EsQueueDto createQueueDtoForRule(int ruleId) { String docId = String.valueOf(ruleId); return EsQueueDto.create("rules/rule", docId, null, docId); }
public EsQueueDto insert(DbSession dbSession, EsQueueDto item) { insert(dbSession, singletonList(item)); return item; }
public void delete(DbSession dbSession, EsQueueDto item) { delete(dbSession, singletonList(item)); }
private static DocId toDocId(EsQueueDto dto) { return new DocId(IndexType.parse(dto.getDocType()), dto.getDocId()); }
@Override public IndexingResult index(DbSession dbSession, Collection<EsQueueDto> items) { IndexingResult result = new IndexingResult(); items.forEach(item -> { result.incrementRequests(); if (!item.getUuid().equals(failing.getUuid())) { result.incrementSuccess(); db.getDbClient().esQueueDao().delete(dbSession, item); dbSession.commit(); } }); return result; }
public IndexingResult commitAndIndex(DbSession dbSession, Collection<String> projectUuids) { List<EsQueueDto> items = projectUuids.stream() .map(projectUuid -> EsQueueDto.create(INDEX_TYPE_PROJECT_MEASURES.format(), projectUuid, null, projectUuid)) .collect(MoreCollectors.toArrayList(projectUuids.size())); dbClient.esQueueDao().insert(dbSession, items); dbSession.commit(); return index(dbSession, items); }
private static EsQueueDto createQueueDto(String docId, String docIdType, String projectUuid) { return EsQueueDto.create(INDEX_TYPE_ISSUE.format(), docId, docIdType, projectUuid); }
private void deleteQueueDto(DbSession dbSession, EsQueueDto item) { dbClient.esQueueDao().delete(dbSession, item); dbSession.commit(); }
public void commitAndIndex(DbSession dbSession, Collection<UserDto> users) { List<String> uuids = users.stream().map(UserDto::getUuid).collect(toList()); List<EsQueueDto> items = uuids.stream() .map(uuid -> EsQueueDto.create(INDEX_TYPE_USER.format(), uuid)) .collect(MoreCollectors.toArrayList()); dbClient.esQueueDao().insert(dbSession, items); dbSession.commit(); postCommit(dbSession, users.stream().map(UserDto::getLogin).collect(toList()), items); }
private static EsQueueDto newQueueDto(String docId, String docIdType, @Nullable String routing) { return EsQueueDto.create(INDEX_TYPE_ACTIVE_RULE.format(), docId, docIdType, routing); } }
@Override public void onFinish(IndexingResult result) { if (result.isSuccess()) { dbClient.esQueueDao().delete(dbSession, items); dbSession.commit(); } } }