Java：如何实时计算正在保存的流中的sha1摘要？

Question

我有一个用Java编写的servlet，它接受多部分形式的发布文件，该文件需要保存在MongoDb / GridFS中。我已经有为此工作的代码。

这是一个代码片段，显示了如何使用org.apache.commons.fileupload包完成此操作。它几乎不占用内存，因为它不会在内存中保留太多数据。

        ServletFileUpload upload = new ServletFileUpload();
        FileItemIterator iter = upload.getItemIterator(req);
        while (iter.hasNext()) {
            FileItemStream item = iter.next();
            String name = item.getFieldName();
            InputStream stream = item.openStream();
            if (item.isFormField()) {
                toProcess.put(name, Streams.asString(stream));
            } else {
                String fileName = item.getName();
                String contentType = item.getHeaders().getHeader("Content-Type");
                GridFSUploadOptions options = new GridFSUploadOptions()
                        // .chunkSizeBytes(358400)
                        .metadata(new Document("content_type", contentType));
                ObjectId fileId = gridFSFilesBucket.uploadFromStream(fileName, stream, options);
                fileIds.add(fileId);
                fileNames.add(fileName);
            }

我还需要计算所有文件的sha1哈希值。阿帕奇digestutils可以用于此目的。它具有一种可以计算流中的sha1的方法：

https://commons.apache.org/proper/commons-codec/apidocs/org/apache/commons/codec/digest/DigestUtils.html#sha1-java.io.InputStream-

我的问题是此方法完全消耗了流。我需要将输入流分为两部分。将一部分输入到SHA-1计算中，另一部分输入到GridFS存储桶中。

我怎样才能做到这一点？我当时正在考虑创建自己的“管道”，该管道具有输入和输出流，可以转发所有数据，但可以实时更新摘要。

我只是不知道如何开始写这样的管道。

Answer 1

您可以使用Java API类DigestInputStream

如Javadoc所述，

一个透明的流，它使用流中经过的位来更新关联的消息摘要。

要完成消息摘要计算，请在调用此摘要输入流的read方法之一之后，在关联的消息摘要上调用其中一种摘要方法。

在您的代码中，您可以执行以下操作：

InputStream stream = item.openStream();
MessageDigest digest = MessageDigest.getInstance("SHA-256");
stream = new DigestInputStream(stream, digest);

最后，您可以获得以下内容的摘要：

byte[] hash = digest.digest();