org.ojai.store.DocumentMutation Java Examples

/**
 * {@inheritDoc}
 *
 * @param id         identifier of document, which will be updated.
 * @param artistRate artist rate.
 * @return updated artist rate.
 */
@Override
public ArtistRate update(String id, ArtistRate artistRate) {
    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Create a DocumentMutation to update non-null fields
        DocumentMutation mutation = connection.newMutation();

        // Update only non-null fields
        if (artistRate.getRating() != null) {
            mutation.set("rating", artistRate.getRating());
        }

        // Update the OJAI Document with specified identifier
        store.update(id, mutation);

        Document updatedOjaiDoc = store.findById(id);

        log.debug("Update document from table '{}' with id: '{}'. Elapsed time: {}", tablePath, id, stopwatch);

        // Map Ojai document to the actual instance of model class
        return mapOjaiDocument(updatedOjaiDoc);
    });
}
 

/**
 * {@inheritDoc}
 *
 * @param id        identifier of document, which will be updated.
 * @param id        identifier of document, which will be updated.
 * @param statistic statistic.
 * @return updated statistic.
 */
@Override
public Statistic update(String id, Statistic statistic) {
    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Create a DocumentMutation to update non-null fields
        DocumentMutation mutation = connection.newMutation();

        // Update only non-null fields
        if (statistic.getDocumentNumber() != null) {
            mutation.set("document_number", statistic.getDocumentNumber());
        }

        // Update the OJAI Document with specified identifier
        store.update(id, mutation);

        Document updatedOjaiDoc = store.findById(id);

        log.debug("Update document from table '{}' with id: '{}'. Elapsed time: {}", tablePath, id, stopwatch);

        // Map Ojai document to the actual instance of model class
        return mapOjaiDocument(updatedOjaiDoc);
    });
}
 

/**
 * {@inheritDoc}
 *
 * @param id        identifier of document, which will be updated.
 * @param albumRate album rate.
 * @return updated album rate.
 */
@Override
public AlbumRate update(String id, AlbumRate albumRate) {
    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Create a DocumentMutation to update non-null fields
        DocumentMutation mutation = connection.newMutation();

        // Update only non-null fields
        if (albumRate.getRating() != null) {
            mutation.set("rating", albumRate.getRating());
        }

        // Update the OJAI Document with specified identifier
        store.update(id, mutation);

        Document updatedOjaiDoc = store.findById(id);

        log.debug("Update document from table '{}' with id: '{}'. Elapsed time: {}", tablePath, id, stopwatch);

        // Map Ojai document to the actual instance of model class
        return mapOjaiDocument(updatedOjaiDoc);
    });
}
 

/**
 * Deletes single track according to the specified album identifier and track identifier.
 *
 * @param albumId identifier of album, for which track will be deleted.
 * @param trackId identifier of track, which will be deleted.
 * @return <code>true</code> if track is successfully deleted, <code>false</code> otherwise.
 */
public boolean deleteTrack(String albumId, String trackId) {

    List<Track> existingAlbumTracks = getTracksList(albumId);
    if (existingAlbumTracks == null) {
        return false;
    }

    int trackIndex = getTrackIndexById(existingAlbumTracks, trackId);
    if (trackIndex < 0) {
        return false;
    }

    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Delete single track
        DocumentMutation mutation = AlbumMutationBuilder.forConnection(connection)
                .deleteTrack(trackIndex).build();

        // Set update info if available
        getUpdateInfo().ifPresent(updateInfo -> mutation.set("update_info", updateInfo));

        // Update the OJAI Document with specified identifier
        store.update(albumId, mutation);

        log.debug("Deleting album's track with id: '{}' for albumId: '{}' took {}", trackId, albumId, stopwatch);

        return true;
    });

}
 

@Override
public User update(String id, User user) {
    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Create a DocumentMutation to update non-null fields
        DocumentMutation mutation = connection.newMutation();

        // Update only non-null fields
        if (user.getFirstName() != null) {
            mutation.set("first_name", user.getFirstName());
        }

        if (user.getLastName() != null) {
            mutation.set("last_name", user.getLastName());
        }

        // Update the OJAI Document with specified identifier
        store.update(id, mutation);

        Document updatedOjaiDoc = store.findById(id);

        log.debug("Update document from table '{}' with id: '{}'. Elapsed time: {}", tablePath, id, stopwatch);

        // Map Ojai document to the actual instance of model class
        return mapOjaiDocument(updatedOjaiDoc);
    });
}
 

/**
 * {@inheritDoc}
 *
 * @param id    identifier of album, which will be updated.
 * @param album contains album info that will be updated.
 * @return updated album.
 */
@Override
@SuppressWarnings("unchecked")
public Album update(String id, Album album) {

    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Update basic fields
        DocumentMutation albumMutation = AlbumMutationBuilder.forConnection(connection)
                .setName(album.getName())
                .setBarcode(album.getBarcode())
                .setCountry(album.getCountry())
                .setLanguage(album.getLanguage())
                .setPackaging(album.getPackaging())
                .setTrackList(album.getTrackList())
                .setArtists(album.getArtists())
                .setFormat(album.getFormat())
                .setDateDay(album.getReleasedDate())
                .setRating(album.getRating())
                .build();

        // Set update info if available
        getUpdateInfo().ifPresent(updateInfo -> albumMutation.set("update_info", updateInfo));

        // Update the OJAI Document with specified identifier
        store.update(id, albumMutation);

        Document updatedOjaiDoc = store.findById(id);

        log.debug("Update document from table '{}' with id: '{}'. Elapsed time: {}", tablePath, id, stopwatch);

        // Map Ojai document to the actual instance of model class
        return mapOjaiDocument(updatedOjaiDoc);
    });
}
 

public static void commitMutation(String tableName, String fieldPath, DocumentMutation documentMutation) throws MapRJsonDocumentLoaderException {
  if(isTest) {
    Preconditions.checkNotNull(testDelegate);
    testDelegate.commitMutationInternal(tableName, fieldPath, documentMutation);
  } else {
    Preconditions.checkNotNull(delegate);
    delegate.commitMutationInternal(tableName, fieldPath, documentMutation);
  }
}
 

public static DocumentMutation createDocumentMutation() {
  if(isTest) {
    Preconditions.checkNotNull(testDelegate);
    return testDelegate.createDocumentMutationInternal();
  } else {
    Preconditions.checkNotNull(delegate);
    return delegate.createDocumentMutationInternal();
  }
}
 

@Override
protected DocumentMutation createDocumentMutationInternal() {
  if(connection == null) {
    startConnection();
  }

  return connection.newMutation();
}
 

private DocumentMutation populateDocumentMutation(Record rec) throws OnRecordErrorException {
  DocumentMutation documentMutation = MapRJsonDocumentLoader.createDocumentMutation();
  boolean replace = mapRJsonConfigBean.setOrReplace == SetOrReplace.REPLACE;

  if(rec != null && (rec.get().getType() == Field.Type.LIST_MAP || rec.get().getType() == Field.Type.MAP)) {
    Map<String, Field> fields = rec.get().getValueAsMap();
    for(Map.Entry<String, Field> entry : fields.entrySet()) {
      String path = entry.getKey();
      Field field = entry.getValue();

      //don't add the keyField to the document mutation set. that gets added later
      if(entry.getKey().equals(mapRJsonConfigBean.keyField)) {
        continue;
      }
      switch(field.getType()) {
        case DOUBLE:
          double d = field.getValueAsDouble();
          documentMutation = (replace ? documentMutation.setOrReplace(path, d) : documentMutation.set(path, d));
          break;
        case FLOAT:
          float f = field.getValueAsFloat();
          documentMutation = (replace ? documentMutation.setOrReplace(path, f) : documentMutation.set(path, f));
          break;
        case INTEGER:
          int i = field.getValueAsInteger();
          documentMutation = (replace ? documentMutation.setOrReplace(path, i) : documentMutation.set(path, i));
          break;
        case SHORT:
          short s = field.getValueAsShort();
          documentMutation = (replace ? documentMutation.setOrReplace(path, s) : documentMutation.set(path, s));
          break;
        case LONG:
        case DATE:
        case TIME:
        case DATETIME:
          long l = field.getValueAsLong();
          documentMutation = (replace ? documentMutation.setOrReplace(path, l) : documentMutation.set(path, l));
          break;
        case STRING:
          String st = field.getValueAsString();
          documentMutation = (replace ? documentMutation.setOrReplace(path, st) : documentMutation.set(path, st));
          break;
        case BYTE_ARRAY:
          byte[] ba = field.getValueAsByteArray();
          documentMutation = (replace ? documentMutation.setOrReplace(path, ByteBuffer.wrap(ba)) : documentMutation.set(path, ByteBuffer.wrap(ba)));
          break;
        case BOOLEAN:
        case MAP:
        case LIST:
        case LIST_MAP:
        case CHAR:
        case BYTE:
        default:
          throw new OnRecordErrorException(rec, Errors.MAPR_JSON_14, field.getType().name());
      }
    }
  }

  return documentMutation;
}
 

private void performUpdateOperation(Record rec) throws StageException {
  os.reset();
  createJson(os, rec);
  DocumentMutation document = populateDocumentMutation(rec);
  doUpdate(document, rec);
}
 

@Override
protected void commitMutationInternal(String tableName, String fieldPath, DocumentMutation documentMutation) throws MapRJsonDocumentLoaderException {
  initTable(tableName, false);
  theStores.get(tableName).update(fieldPath, documentMutation);
}
 

@Override
protected void commitMutationInternal(String tableName, String fieldPath, DocumentMutation documentMutation) throws MapRJsonDocumentLoaderException {
  initTable(tableName, false);
  theTables.get(tableName).update(fieldPath, documentMutation);
}
 

@Override
protected DocumentMutation createDocumentMutationInternal() {
  return MapRDB.newMutation();
}
 

@Override
public boolean checkAndMutate(final String id, final QueryCondition condition, final DocumentMutation m)
    throws StoreException {
  return store.checkAndMutate(id, condition, m);
}
 

@Override
public void update(final String id, final DocumentMutation m) throws StoreException {
  store.update(id, m);
}
 

@Override
public boolean checkAndMutate(final Value _id, final QueryCondition condition, final DocumentMutation mutation)
    throws StoreException {
  return store.checkAndUpdate(_id, condition, mutation);
}
 

@Override
public void update(final Value _id, final DocumentMutation mutation) throws StoreException {
  store.update(_id, mutation);
}
 

@Override
public DocumentMutation newMutation() {
  throw new UnsupportedOperationException();
}
 

@Override
public DocumentMutation newMutation() {
  return driver.newMutation();
}
 

public DocumentMutation build() {
    return this.mutation;
}
 

/**
 * {@inheritDoc}
 *
 * @param id     identifier of document, which will be updated.
 * @param artist contains artist info that will be updated.
 * @return updated artist.
 */
@Override
public Artist update(String id, Artist artist) {
    return processStore((connection, store) -> {

        Stopwatch stopwatch = Stopwatch.createStarted();

        // Create a DocumentMutation to update non-null fields
        DocumentMutation mutation = connection.newMutation();

        // Update only non-null fields
        if (artist.getName() != null) {
            mutation.set("name", artist.getName());
        }

        if (artist.getGender() != null) {
            mutation.set("gender", artist.getGender());
        }

        if (artist.getBeginDate() != null) {
            mutation.set("begin_date", artist.getBeginDate());
        }

        if (artist.getEndDate() != null) {
            mutation.set("end_date", artist.getEndDate());
        }

        if (artist.getIpi() != null) {
            mutation.set("IPI", artist.getIpi());
        }

        if (artist.getIsni() != null) {
            mutation.set("ISNI", artist.getIsni());
        }

        if (artist.getArea() != null) {
            mutation.set("area", artist.getArea());
        }

        if (artist.getAlbums() != null) {

            List<Map> albumsMapList = artist.getAlbums().stream()
                    .map(album -> mapper.convertValue(album, Map.class))
                    .collect(Collectors.toList());

            mutation.set("albums", albumsMapList);
        }

        if (artist.getProfileImageUrl() != null) {
            mutation.set("profile_image_url", artist.getProfileImageUrl());
        }

        if (artist.getDeleted() != null) {
            mutation.set("deleted", artist.getDeleted());
        }

        if (artist.getRating() != null) {
            mutation.set("rating", artist.getRating());
        }

        // Set update info if available
        getUpdateInfo().ifPresent(updateInfo -> mutation.set("update_info", updateInfo));

        // Update the OJAI Document with specified identifier
        store.update(id, mutation);

        Document updatedOjaiDoc = store.findById(id);

        log.debug("Update document from table '{}' with id: '{}'. Elapsed time: {}", tablePath, id, stopwatch);

        // Map Ojai document to the actual instance of model class
        return mapOjaiDocument(updatedOjaiDoc);
    });
}
 

/**
 * Decodes an instance of <T> into an OJAI {@link DocumentMutation}
 *
 * @param encodedMutation an instance of <T> containing encoded OJAI DocumentMutation
 * @return an OJAI {@link DocumentMutation}
 * @throws DecodingException
 */
public DocumentMutation decodeMutation(@NonNullable T encodedMutation) throws DecodingException;
 

/**
 * Encodes an OJAI {@link DocumentMutation} into an instance of <T>
 *
 * @param condition OJAI DocumentMutation to encode
 * @return an instance of <T>
 * @throws EncodingException
 */
public T encodeMutation(@NonNullable DocumentMutation mutation) throws EncodingException;
 

protected abstract DocumentMutation createDocumentMutationInternal(); 

protected abstract void commitMutationInternal(String tableName, String fieldPath, DocumentMutation documentMutation) throws MapRJsonDocumentLoaderException;