Examples with BlobContainer io.crate.blob.BlobContainer used on opensource projects

Example 1 with BlobContainer

use of io.crate.blob.BlobContainer in project crate by crate.

the class DigestBlobTests method testResumeDigestBlobAddHeadAfterContent.

@Test
public void testResumeDigestBlobAddHeadAfterContent() throws IOException {
    UUID transferId = UUID.randomUUID();
    BlobContainer container = new BlobContainer(tmpFolder.newFolder().toPath());
    DigestBlob digestBlob = DigestBlob.resumeTransfer(container, "417de3231e23dcd6d224ff60918024bc6c59aa58", transferId, 2);
    BytesArray contentTail = new BytesArray("CDEFGHIJKLMN".getBytes(StandardCharsets.UTF_8));
    digestBlob.addContent(contentTail, false);
    BytesArray contentHead = new BytesArray("AB".getBytes(StandardCharsets.UTF_8));
    digestBlob.addToHead(contentHead);
    contentTail = new BytesArray("O".getBytes(StandardCharsets.UTF_8));
    digestBlob.addContent(contentTail, true);
    // check if tmp file's content is correct
    byte[] buffer = new byte[15];
    try (FileInputStream stream = new FileInputStream(digestBlob.file())) {
        assertThat(stream.read(buffer, 0, 15), is(15));
        assertThat(new BytesArray(buffer).toUtf8().trim(), is("ABCDEFGHIJKLMNO"));
    }
    File file = digestBlob.commit();
    // check if final file's content is correct
    buffer = new byte[15];
    try (FileInputStream stream = new FileInputStream(file)) {
        assertThat(stream.read(buffer, 0, 15), is(15));
        assertThat(new BytesArray(buffer).toUtf8().trim(), is("ABCDEFGHIJKLMNO"));
    }
    // assert file created
    assertThat(file.exists(), is(true));
    // just in case any references to file left
    assertThat(file.delete(), is(true));
}

Example 2 with BlobContainer

use of io.crate.blob.BlobContainer in project crate by crate.

the class DigestBlobTests method testDigestBlobResumeHeadAndAddContent.

@Test
public void testDigestBlobResumeHeadAndAddContent() throws IOException {
    String digest = "417de3231e23dcd6d224ff60918024bc6c59aa58";
    UUID transferId = UUID.randomUUID();
    int currentPos = 2;
    BlobContainer container = new BlobContainer(tmpFolder.newFolder().toPath());
    File filePath = new File(container.getTmpDirectory().toFile(), String.format(Locale.ENGLISH, "%s.%s", digest, transferId.toString()));
    if (filePath.exists()) {
        assertThat(filePath.delete(), is(true));
    }
    DigestBlob digestBlob = DigestBlob.resumeTransfer(container, digest, transferId, currentPos);
    BytesArray contentHead = new BytesArray("A".getBytes(StandardCharsets.UTF_8));
    digestBlob.addToHead(contentHead);
    BytesArray contentTail = new BytesArray("CDEFGHIJKL".getBytes(StandardCharsets.UTF_8));
    digestBlob.addContent(contentTail, false);
    contentHead = new BytesArray("B".getBytes(StandardCharsets.UTF_8));
    digestBlob.addToHead(contentHead);
    contentTail = new BytesArray("MNO".getBytes(StandardCharsets.UTF_8));
    digestBlob.addContent(contentTail, true);
    // check if tmp file's content is correct
    byte[] buffer = new byte[15];
    try (FileInputStream stream = new FileInputStream(digestBlob.file())) {
        assertThat(stream.read(buffer, 0, 15), is(15));
        assertThat(new BytesArray(buffer).toUtf8().trim(), is("ABCDEFGHIJKLMNO"));
    }
    File file = digestBlob.commit();
    // check if final file's content is correct
    buffer = new byte[15];
    try (FileInputStream stream = new FileInputStream(file)) {
        assertThat(stream.read(buffer, 0, 15), is(15));
        assertThat(new BytesArray(buffer).toUtf8().trim(), is("ABCDEFGHIJKLMNO"));
    }
    // assert file created
    assertTrue(file.exists());
    // just in case any references to file left
    assertTrue(file.delete());
}