From b966600f00dd24085c5ad2cc21f60bbee3c00278 Mon Sep 17 00:00:00 2001 From: santong Date: Thu, 28 Oct 2021 12:00:36 +0800 Subject: [PATCH] add peer to task failed because InnerBucketMaxLength is small Signed-off-by: santong --- pkg/structure/sortedlist/sorted_list.go | 4 ++-- scheduler/supervisor/task.go | 16 +++++++++++++--- 2 files changed, 15 insertions(+), 5 deletions(-) diff --git a/pkg/structure/sortedlist/sorted_list.go b/pkg/structure/sortedlist/sorted_list.go index 606a5557bf8..575cc1a517b 100644 --- a/pkg/structure/sortedlist/sorted_list.go +++ b/pkg/structure/sortedlist/sorted_list.go @@ -21,8 +21,8 @@ import ( "sync" ) -const BucketMaxLength = 10000 -const InnerBucketMaxLength = 100 +const BucketMaxLength = 100000 +const InnerBucketMaxLength = 10000 type Item interface { GetSortKeys() (key1 int, key2 int) diff --git a/scheduler/supervisor/task.go b/scheduler/supervisor/task.go index 566777bf700..415ee5e447d 100644 --- a/scheduler/supervisor/task.go +++ b/scheduler/supervisor/task.go @@ -261,16 +261,26 @@ func (task *Task) UpdateSuccess(pieceCount int32, contentLength int64) { } func (task *Task) AddPeer(peer *Peer) { - task.peers.UpdateOrAdd(peer) + err := task.peers.UpdateOrAdd(peer) + if err != nil { + task.logger.Errorf("add peer %s failed: %v", peer.ID, err) + } task.logger.Debugf("peer %s has been added, current total peer count is %d", peer.ID, task.peers.Size()) } func (task *Task) UpdatePeer(peer *Peer) { - task.peers.Update(peer) + err := task.peers.Update(peer) + if err != nil { + task.logger.Errorf("update peer %s failed: %v", peer.ID, err) + } + task.logger.Debugf("peer %s has been updated, current total peer count is %d", peer.ID, task.peers.Size()) } func (task *Task) DeletePeer(peer *Peer) { - task.peers.Delete(peer) + err := task.peers.Delete(peer) + if err != nil { + task.logger.Errorf("delete peer %s failed: %v", peer.ID, err) + } task.logger.Debugf("peer %s has been deleted, current total peer count is %d", peer.ID, task.peers.Size()) }