refactor: abstract replicaLvolFilter as a func

Signed-off-by: Shuo Wu <[email protected]>
longhorn · Oct 22, 2024 · 89923d9 · 89923d9
1 parent e3dd9a3
commit 89923d9
Showing 1 changed file with 12 additions and 32 deletions.
diff --git a/pkg/spdk/replica.go b/pkg/spdk/replica.go
@@ -182,20 +182,21 @@ func NewReplica(ctx context.Context, replicaName, lvsName, lvsUUID string, specS
 	}
 }
 
+func (r *Replica) replicaLvolFilter(bdev *spdktypes.BdevInfo) bool {
+	var lvolName string
+	if len(bdev.Aliases) == 1 {
+		lvolName = spdktypes.GetLvolNameFromAlias(bdev.Aliases[0])
+	}
+	return IsReplicaLvol(r.Name, lvolName) || (len(r.ActiveChain) > 0 && r.ActiveChain[0] != nil && r.ActiveChain[0].Name == lvolName)
+}
+
 func (r *Replica) Sync(spdkClient *spdkclient.Client) (err error) {
 	r.Lock()
 	defer r.Unlock()
 	// It's better to let the server send the update signal
 
 	// This lvol and nvmf subsystem fetch should be protected by replica lock, in case of snapshot operations happened during the sync-up.
-	replicaLvolFilter := func(bdev *spdktypes.BdevInfo) bool {
-		var lvolName string
-		if len(bdev.Aliases) == 1 {
-			lvolName = spdktypes.GetLvolNameFromAlias(bdev.Aliases[0])
-		}
-		return IsReplicaLvol(r.Name, lvolName) || (r.ActiveChain[0] != nil && r.ActiveChain[0].Name == lvolName)
-	}
-	bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, replicaLvolFilter)
+	bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, r.replicaLvolFilter)
 	if err != nil {
 		return err
 	}
@@ -802,14 +803,7 @@ func (r *Replica) Delete(spdkClient *spdkclient.Client, cleanupRequired bool, su
 
 	// Clean up the valid snapshot tree
 	if len(r.ActiveChain) > 1 {
-		replicaLvolFilter := func(bdev *spdktypes.BdevInfo) bool {
-			var lvolName string
-			if len(bdev.Aliases) == 1 {
-				lvolName = spdktypes.GetLvolNameFromAlias(bdev.Aliases[0])
-			}
-			return IsReplicaLvol(r.Name, lvolName) || (r.ActiveChain[0] != nil && r.ActiveChain[0].Name == lvolName)
-		}
-		bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, replicaLvolFilter)
+		bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, r.replicaLvolFilter)
 		if err != nil {
 			return err
 		}
@@ -1083,14 +1077,7 @@ func (r *Replica) SnapshotRevert(spdkClient *spdkclient.Client, snapshotName str
 		return nil, err
 	}
 
-	replicaLvolFilter := func(bdev *spdktypes.BdevInfo) bool {
-		var lvolName string
-		if len(bdev.Aliases) == 1 {
-			lvolName = spdktypes.GetLvolNameFromAlias(bdev.Aliases[0])
-		}
-		return IsReplicaLvol(r.Name, lvolName) || (r.ActiveChain[0] != nil && r.ActiveChain[0].Name == lvolName)
-	}
-	bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, replicaLvolFilter)
+	bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, r.replicaLvolFilter)
 	if err != nil {
 		return nil, err
 	}
@@ -1546,14 +1533,7 @@ func (r *Replica) RebuildingDstFinish(spdkClient *spdkclient.Client) (err error)
 
 	_ = r.doCleanupForRebuildingDst(spdkClient, r.rebuildingDstCache.rebuildingState == types.ProgressStateError)
 
-	replicaLvolFilter := func(bdev *spdktypes.BdevInfo) bool {
-		var lvolName string
-		if len(bdev.Aliases) == 1 {
-			lvolName = spdktypes.GetLvolNameFromAlias(bdev.Aliases[0])
-		}
-		return IsReplicaLvol(r.Name, lvolName) || (r.ActiveChain[0] != nil && r.ActiveChain[0].Name == lvolName)
-	}
-	bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, replicaLvolFilter)
+	bdevLvolMap, err := GetBdevLvolMapWithFilter(spdkClient, r.replicaLvolFilter)
 	if err != nil {
 		return err
 	}