java - 如何计算S3文件内容的SHA-256校验和

Question

开箱即用的 S3 提供 S3 对象内容的 MD5 校验和。但我需要计算文件内容的 SHA-256 校验和。该文件可能足够大，因此我不想将文件加载到内存中并计算校验和，而是需要一个解决方案来计算校验和而不将整个文件加载到内存中。

Answer 1

在Java中可以通过以下步骤实现:

1. 获取S3对象的InputStream
2. 使用 MessageDigest 和 DigestInputStream 类进行 SHA-256 哈希(或 SHA-1 或 MD5)

以下是有关如何执行此操作的代码片段:

String getS3FileHash(AmazonS3 amazonS3, String s3bucket, String filePath) {
    try {
        InputStream inputStream = amazonS3.getObject(s3bucket, filePath).getObjectContent();
        MessageDigest messageDigest = MessageDigest.getInstance("SHA-256");
        DigestInputStream digestInputStream = new DigestInputStream(inputStream, messageDigest);
        byte[] buffer = new byte[4096];
        int count = 0;
        while (digestInputStream.read(buffer) > -1) {
            count++;
        }
        log.info("total read: " + count);
        MessageDigest digest = digestInputStream.getMessageDigest();
        digestInputStream.close();
        byte[] md5 = digest.digest();
        StringBuilder sb = new StringBuilder();
        for (byte b: md5) {
            sb.append(String.format("%02X", b));
        }
        return sb.toString().toLowerCase();
    } catch (Exception e) {
        log.error(e);
    }
    return null; 
}