From 2a1abb852b1066f8d0172fe55bd6eb11c9331a8c Mon Sep 17 00:00:00 2001 From: Derrick Hammer Date: Fri, 9 Feb 2024 15:22:46 -0500 Subject: [PATCH] refactor: have bao hash return totalReadSize --- bao/bao.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/bao/bao.go b/bao/bao.go index ac9c72f..5885fd3 100644 --- a/bao/bao.go +++ b/bao/bao.go @@ -73,7 +73,7 @@ func Shutdown() { client.Kill() } -func Hash(r io.Reader) (*Result, error) { +func Hash(r io.Reader) (*Result, int, error) { hasherId := bao.NewHasher() initialSize := 4 * units.KiB maxSize := 3.5 * units.MiB @@ -89,12 +89,12 @@ func Hash(r io.Reader) (*Result, error) { if err == io.EOF { break } - return nil, err + return nil, 0, err } totalReadSize += n if !bao.Hash(hasherId, buf[:n]) { - return nil, errors.New("hashing failed") + return nil, 0, errors.New("hashing failed") } // Adaptively adjust buffer size based on read patterns @@ -107,5 +107,5 @@ func Hash(r io.Reader) (*Result, error) { result := bao.Finish(hasherId) - return &result, nil + return &result, totalReadSize, nil }