Examples with Operation org.opensearch.repositories.hdfs.HdfsBlobStore.Operation used on opensource projects

Example 1 with Operation

use of org.opensearch.repositories.hdfs.HdfsBlobStore.Operation in project OpenSearch by opensearch-project.

the class HdfsBlobContainer method writeBlobAtomic.

@Override
public void writeBlobAtomic(String blobName, InputStream inputStream, long blobSize, boolean failIfAlreadyExists) throws IOException {
    final String tempBlob = FsBlobContainer.tempBlobName(blobName);
    final Path tempBlobPath = new Path(path, tempBlob);
    final Path blob = new Path(path, blobName);
    store.execute((Operation<Void>) fileContext -> {
        writeToPath(inputStream, blobSize, fileContext, tempBlobPath, EnumSet.of(CreateFlag.CREATE, CreateFlag.SYNC_BLOCK));
        try {
            fileContext.rename(tempBlobPath, blob, failIfAlreadyExists ? Options.Rename.NONE : Options.Rename.OVERWRITE);
        } catch (org.apache.hadoop.fs.FileAlreadyExistsException faee) {
            throw new FileAlreadyExistsException(blob.toString(), null, faee.getMessage());
        }
        return null;
    });
}

Example 2 with Operation

use of org.opensearch.repositories.hdfs.HdfsBlobStore.Operation in project OpenSearch by opensearch-project.

the class HdfsBlobContainer method writeBlob.

@Override
public void writeBlob(String blobName, InputStream inputStream, long blobSize, boolean failIfAlreadyExists) throws IOException {
    Path blob = new Path(path, blobName);
    // we pass CREATE, which means it fails if a blob already exists.
    final EnumSet<CreateFlag> flags = failIfAlreadyExists ? EnumSet.of(CreateFlag.CREATE, CreateFlag.SYNC_BLOCK) : EnumSet.of(CreateFlag.CREATE, CreateFlag.OVERWRITE, CreateFlag.SYNC_BLOCK);
    store.execute((Operation<Void>) fileContext -> {
        try {
            writeToPath(inputStream, blobSize, fileContext, blob, flags);
        } catch (org.apache.hadoop.fs.FileAlreadyExistsException faee) {
            throw new FileAlreadyExistsException(blob.toString(), null, faee.getMessage());
        }
        return null;
    });
}