Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -222,10 +222,6 @@ public static byte[] extractBytes(InputStream inputStream, long start, int lengt
break;
}
outputStream.write( buffer, 0, amountRead );
if ( amountRead < buffer.length ) {
// we have read up to the end of stream
break;
}
bytesRead += amountRead;
if ( bytesRead >= length ) {
break;
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,119 @@
/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
* See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html
*/
package org.hibernate.orm.test.lob;

import jakarta.persistence.Entity;
import jakarta.persistence.Id;
import jakarta.persistence.Lob;
import org.hibernate.engine.jdbc.NonContextualLobCreator;
import org.hibernate.testing.orm.junit.DomainModel;
import org.hibernate.testing.orm.junit.JiraKey;
import org.hibernate.testing.orm.junit.SessionFactory;
import org.hibernate.testing.orm.junit.SessionFactoryScope;
import org.junit.jupiter.api.Test;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.Blob;
import java.sql.SQLException;
import java.util.Random;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertEquals;

@DomainModel(annotatedClasses = InflaterInputStreamBlobTest.TestEntity.class)
@SessionFactory
@JiraKey("HHH-19464")
class InflaterInputStreamBlobTest {

private static final int RANDOM_SIZE = 32000;

@Test
void hibernate_blob_streaming(SessionFactoryScope scope) throws Exception {
final var randomBytes = getRandomBytes();
final var outputStream = new ByteArrayOutputStream();
try (var zipOutputStream = new GZIPOutputStream( outputStream )) {
zipOutputStream.write( randomBytes );
}

long size = randomBytes.length;
scope.inTransaction( entityManager -> {
try {
InputStream is = new GZIPInputStream( new ByteArrayInputStream( outputStream.toByteArray() ) );
Blob blob = NonContextualLobCreator.INSTANCE.wrap(
NonContextualLobCreator.INSTANCE.createBlob( is, size )
);
TestEntity e = new TestEntity();
e.setId( 1L );
e.setData( blob );

entityManager.persist( e );
}
catch (IOException e) {
throw new RuntimeException( e );
}
}
);

scope.inStatelessSession( session -> {
final var entity = session.get( TestEntity.class, 1L );
try {
final var blob = entity.getData();
assertEquals( size, blob.length() );
assertArrayEquals( randomBytes, blob.getBytes( 1L, (int) blob.length() ) );
}
catch (SQLException e) {
throw new RuntimeException( e );
}
} );
}

private static byte[] getRandomBytes() {

Check warning

Code scanning / CodeQL

Random used only once Warning test

Random object created and used only once.
final var bytes = new byte[RANDOM_SIZE];
new Random().nextBytes( bytes );
return bytes;
}

@Entity
public static class TestEntity {

@Id
Long id;

@Lob
Blob data;

public Long getId() {
return id;
}

public void setId(Long id) {
this.id = id;
}

public Blob getData() {
return data;
}

public InputStream getInputStream() {
try {
return data.getBinaryStream();
}
catch (SQLException e) {
throw new IllegalArgumentException( "Could not obtain requested input stream", e );
}
}

public void setData(Blob data) {
this.data = data;
}
}
}
Loading