Simplify listCPURange implementation

Continue early for single element range
tklauser · May 23, 2024 · 9824f86 · 9824f86
1 parent 9ff7f79
commit 9824f86
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/numcpus_linux.go b/numcpus_linux.go
@@ -99,14 +99,14 @@ func listCPURange(cpus string) ([]int, error) {
 		if err != nil {
 			return nil, err
 		}
-		var last uint64
-		if found {
-			last, err = strconv.ParseUint(to, 10, 32)
-			if err != nil {
-				return nil, err
-			}
-		} else {
-			last = first
+		if !found {
+			// range containing a single element
+			list = append(list, int(first))
+			continue
+		}
+		last, err := strconv.ParseUint(to, 10, 32)
+		if err != nil {
+			return nil, err
 		}
 		if last < first {
 			return nil, fmt.Errorf("last CPU in range (%d) less than first (%d)", last, first)