dragonflyoss
/
dragonfly
mirror of https://github.com/dragonflyoss/dragonfly.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: optimize calculate digest (#3343) 

Signed-off-by: Jim Ma <majinjing3@gmail.com>
This commit is contained in:
Jim Ma 2024-06-25 16:59:19 +08:00 committed by GitHub
parent 5dee749657
commit 911efebd63
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 6 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
client/config/peerhost_darwin.go
View File

@ -55,7 +55,7 @@ var peerHostConfig = func() *DaemonOption {
IDC: "",
},
Download: DownloadOption{
CalculateDigest: false,
CalculateDigest: true,
PieceDownloadTimeout: 30 * time.Second,
GRPCDialTimeout: 10 * time.Second,
WatchdogTimeout: 30 * time.Second,

2
client/config/peerhost_linux.go
View File

@ -55,7 +55,7 @@ var peerHostConfig = func() *DaemonOption {
IDC: "",
},
Download: DownloadOption{
CalculateDigest: false,
CalculateDigest: true,
PieceDownloadTimeout: 30 * time.Second,
GRPCDialTimeout: 10 * time.Second,
WatchdogTimeout: 30 * time.Second,

22
client/daemon/peer/peertask_conductor.go
View File

@ -1395,7 +1395,7 @@ func (pt *peerTaskConductor) done() {
// update storage metadata
if err := pt.UpdateStorage(); err == nil {
// validate digest
if err = pt.Validate(); err == nil {
if err = pt.tryStore(); err == nil {
close(pt.successCh)
pt.span.SetAttributes(config.AttributePeerTaskSuccess.Bool(true))
} else {
@ -1408,7 +1408,7 @@ func (pt *peerTaskConductor) done() {
pt.span.SetAttributes(config.AttributePeerTaskSuccess.Bool(false))
pt.span.SetAttributes(config.AttributePeerTaskCode.Int(int(pt.failedCode)))
pt.span.SetAttributes(config.AttributePeerTaskMessage.String(pt.failedReason))
pt.Errorf("validate digest failed: %s", err)
pt.Errorf("store failed: %s", err)
metrics.PeerTaskFailedCount.WithLabelValues(metrics.FailTypeP2P).Add(1)
}
} else {
@ -1580,8 +1580,7 @@ func (pt *peerTaskConductor) fail() {
}
}
// Validate stores metadata and validates digest
func (pt *peerTaskConductor) Validate() error {
func (pt *peerTaskConductor) tryStore() error {
err := pt.GetStorage().Store(pt.ctx,
&storage.StoreRequest{
CommonTaskRequest: storage.CommonTaskRequest{
@ -1595,22 +1594,7 @@ func (pt *peerTaskConductor) Validate() error {
pt.Errorf("store metadata error: %s", err)
return err
}
if !pt.CalculateDigest {
return nil
}
err = pt.GetStorage().ValidateDigest(
&storage.PeerTaskMetadata{
PeerID: pt.GetPeerID(),
TaskID: pt.GetTaskID(),
})
if err != nil {
pt.Errorf("validate digest error: %s", err)
return err
}
pt.Debugf("validate digest ok")
return err
}
func (pt *peerTaskConductor) PublishPieceInfo(pieceNum int32, size uint32) {