fix: add peer to task failed because InnerBucketMaxLength is small (#765

) Signed-off-by: santong <weipeng.swp@alibaba-inc.com>
dragonflyoss · Jun 28, 2023 · fe8befb · fe8befb
1 parent e21c76b
commit fe8befb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git 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
@@ -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())
 }