使用JPA / Hibernate从SQL Server流式传输二进制数据

Question

我在SQL Server上存储了大文件，我需要将它们作为InputStream获得，以流式传输到客户端而不会耗尽内存。 我正在使用Hibernate 5.2.11，WildFly 10.1和Microsoft JDBC驱动程序6.2.1（支持到SQL Server的流传输）。

要将InputStream映射到我的实体，我需要创建一个自定义的Hibernate类型，因为不幸的是Hibernate没有提供这种映射。

import java.io.InputStream;
import java.io.Serializable;
import java.sql.Blob;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.Objects;

import org.hibernate.HibernateException;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.type.BlobType;
import org.hibernate.type.SerializationException;
import org.hibernate.usertype.UserType;

public class InputStreamType implements UserType {

    private static final int[] SQL_TYPES = { Types.LONGVARBINARY };

    @Override
    public int[] sqlTypes() {
        return SQL_TYPES;
    }

    @Override
    public Class<?> returnedClass() {
        return InputStream.class;
    }

    @Override
    public boolean equals(Object x, Object y) throws HibernateException {
        return Objects.equals(x, y);
    }

    @Override
    public int hashCode(Object x) throws HibernateException {
        return Objects.hashCode(x);
    }

    @Override
    public Object nullSafeGet(ResultSet rs, String[] names, SharedSessionContractImplementor session, Object owner)
            throws HibernateException, SQLException {
        Blob blob = (Blob) BlobType.INSTANCE.nullSafeGet(rs, names, session, owner);
        return blob == null ? null : blob.getBinaryStream();
    }

    @Override
    public void nullSafeSet(PreparedStatement st, Object value, int index, SharedSessionContractImplementor session)
            throws HibernateException, SQLException {
        if (value == null) {
            st.setNull(index, SQL_TYPES[0]);
        } else {
            st.setBinaryStream(index, (InputStream) value);
        }
    }

    @Override
    public Object deepCopy(Object value) throws HibernateException {
        return value;
    }

    @Override
    public boolean isMutable() {
        return false;
    }

    @Override
    public Serializable disassemble(Object value) throws HibernateException {
        throw new SerializationException("Cannot serialize " + InputStream.class.getName(), null);
    }

    @Override
    public Object assemble(Serializable cached, Object owner) throws HibernateException {
        return cached;
    }

    @Override
    public Object replace(Object original, Object target, Object owner) throws HibernateException {
        return original;
    }

}

然后，我注释我的实体字段以使用此类型：

@Entity
public class User {

    @Id
    Long id;

    @Lob
    @Type(type = "InputStreamType")
    InputStream picture;

    // ...

    InputStream getPicture() {
        return this.picture;
    }

    // ...

}

但是当我尝试读取流时，出现异常：

@Service
public class UserService {

    @PersistenceContext
    EntityManager em;

    @Transactional
    public void testReadInputStream() {
        InputStream picture = em.find(User.class, 1L).getPicture();
        System.out.println(picture.getClass().getName());
        // prints: com.microsoft.sqlserver.jdbc.PLPInputStream
        picture.read();
        // throws: IOException The TDS protocol stream is not valid.
        //         at com.microsoft.sqlserver.jdbc.PLPInputStream.readBytes(PLPInputStream.java:304)
        //         at com.microsoft.sqlserver.jdbc.PLPInputStream.read(PLPInputStream.java:244)
    }

}

我试图读取InputStreamType.nullSafeGet的流，并且那里没有抛出任何异常。

那么，从InputStreamType.nullSafeGet返回后的流将如何处理？ 我如何仍然可以使用它？

更新1

我将情况简化为：

import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

public class Main {

    public static void main(String[] args) throws Exception {
        Class.forName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
        Connection connection = DriverManager.getConnection(
                "jdbc:sqlserver://localhost:1433;DatabaseName=test");
        Statement statement = connection.createStatement();
        ResultSet resultSet = statement.executeQuery("select picture from [User] where id = 1");
        resultSet.next();
        InputStream inputStream = resultSet.getBlob(1).getBinaryStream();
        System.out.println(inputStream.getClass().getName());
        // prints: com.microsoft.sqlserver.jdbc.PLPInputStream

        inputStream.read();
        // no exception

        statement.close();
        inputStream.read();
        // throws: IOException: The TDS protocol stream is not valid.
    }

}

从CustomType返回的语句可能是关闭的，并且流变得不可访问吗？ 如果是这样，如何在JPA中克服呢？

更新2

我的最后发现：如果我返回Blob的流，则在会话关闭时它将关闭。 但是，如果我返回Blob本身，则在会话关闭时，流会以某种方式加载到内存中，从而耗尽大型流（我认为此行为是由Blob上的Hibernate代理触发的）。

从存储库方法返回时，是否有办法让会话打开，以使流可通过HTTP访问并刷新数据而不会耗尽内存？

Answer 1

如果我没记错的话，SQL设计根本就没有数据流。 要么所有数据都已返回到SQL客户端，要么根本没有。 这是一件原子的事情。

您可能能够做的就是将二进制数据拆分为多个较小的块，然后将它们一个接一个地拉。 然后，可以在将每个数据块发送到HTTP客户端时关闭与数据库的网络连接。

或者，您可以将二进制数据作为文件存储在文件系统中，然后从那里进行流传输。 但是，这意味着您必须自己注意文件系统和数据库之间的一致性。