mirrors/acp
1
0
Fork 0
mirror of https://github.com/samuelncui/acp.git synced 2026-01-03 01:55:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: disable hash when all target is bad

Browse Source
This commit is contained in:
崔竞宁
2023-09-18 12:30:59 +08:00
parent df6caee408
commit b094d98c63
1 changed files with 20 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
copy.go
View File

@@ -97,26 +97,6 @@ func (c *Copyer) write(ctx context.Context, job *writeJob, ch chan<- *baseJob, c
} }
}() }()
if c.withHash {
sha := sha256Pool.Get().(hash.Hash)
sha.Reset()
ch := make(chan []byte, 4)
chans = append(chans, ch)
wg.Add(1)
go wrap(ctx, func() {
defer wg.Done()
defer sha256Pool.Put(sha)
for buf := range ch {
sha.Write(buf)
}
job.setHash(sha.Sum(nil))
})
}
var readErr error var readErr error
for _, target := range job.targets { for _, target := range job.targets {
dev := c.getDevice(target) dev := c.getDevice(target)
@@ -195,10 +175,29 @@ func (c *Copyer) write(ctx context.Context, job *writeJob, ch chan<- *baseJob, c
} }
}) })
} }
if len(chans) == 0 { if len(chans) == 0 {
return return
} }
if c.withHash {
sha := sha256Pool.Get().(hash.Hash)
sha.Reset()
ch := make(chan []byte, 4)
chans = append(chans, ch)
wg.Add(1)
go wrap(ctx, func() {
defer wg.Done()
defer sha256Pool.Put(sha)
for buf := range ch {
sha.Write(buf)
}
job.setHash(sha.Sum(nil))
})
}
readErr = c.streamCopy(ctx, chans, job.src, &cntr.bytes) readErr = c.streamCopy(ctx, chans, job.src, &cntr.bytes)
} }