Move blkdiscard example into diskwipe

Better off in diskwipe than solo.
metal-toolbox · Jun 10, 2024 · b80f648 · b80f648
1 parent 98f315b
commit b80f648
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 52 deletions.
diff --git a/examples/blkdiscard/main.go b/examples/blkdiscard/main.go
diff --git a/examples/diskwipe/main.go b/examples/diskwipe/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"context"
 	"flag"
+	"strings"
 	"time"
 
 	"github.com/bmc-toolbox/common"
@@ -64,12 +65,23 @@ func main() {
 	case "nvme":
 		wiper = utils.NewNvmeCmd(*verbose)
 	case "sata":
-		wiper = utils.NewFillZeroCmd(*verbose)
-		if timeout == defaultTimeout {
-			timeout = 24 * time.Hour
-			l.WithField("timeout", timeout.String()).Info("increasing timeout")
-			ctx, cancel = context.WithTimeout(context.Background(), timeout)
-			defer cancel()
+		for _, cap := range drive.Capabilities {
+			if strings.HasPrefix(cap.Description, "Data Set Management TRIM supported") {
+				if cap.Enabled {
+					wiper = utils.NewBlkdiscardCmd(*verbose)
+				}
+				break
+			}
+		}
+
+		if wiper == nil {
+			wiper = utils.NewFillZeroCmd(*verbose)
+			if timeout == defaultTimeout {
+				timeout = 24 * time.Hour
+				l.WithField("timeout", timeout.String()).Info("increasing timeout")
+				ctx, cancel = context.WithTimeout(context.Background(), timeout)
+				defer cancel()
+			}
 		}
 	}