fix: atomic nil

Signed-off-by: Gaius <[email protected]>

Author: gaius-qi

scheduler/supervisor/cdn.go

@@ -124,7 +124,7 @@ func (c *cdn) receivePiece(ctx context.Context, task *Task, stream *client.Piece
 		if err != nil {
 			if err == io.EOF {
 				logger.Infof("task %s connection closed", task.ID)
-				if task.GetStatus() == TaskStatusSuccess && cdnPeer != nil {
+				if cdnPeer != nil && task.GetStatus() == TaskStatusSuccess {
 					span.SetAttributes(config.AttributePeerDownloadSuccess.Bool(true))
 					return cdnPeer, nil
 				}

scheduler/supervisor/peer.go

@@ -329,6 +329,10 @@ func (peer *Peer) GetChildren() *sync.Map {
 }
 
 func (peer *Peer) SetParent(parent *Peer) {
+	if parent == nil {
+		return
+	}
+
 	peer.parent.Store(parent)
 }
 
@@ -457,6 +461,10 @@ func (peer *Peer) BindNewConn(stream scheduler.Scheduler_ReportPieceResultServer
 }
 
 func (peer *Peer) setConn(conn *Channel) {
+	if conn == nil {
+		return
+	}
+
 	peer.conn.Store(conn)
 }