Add Set.SortedSliceFunc

Author: adracus

set/set.go

@@ -183,3 +183,15 @@ func SortedSlice[E constraints.Ordered](set Set[E]) []E {
 	})
 	return res
 }
+
+// SortedSliceFunc takes a Set and converts it to a slice, sorting it by less.
+func SortedSliceFunc[E comparable](set Set[E], less func(e1, e2 E) bool) []E {
+	res := make([]E, 0, len(set))
+	for item := range set {
+		res = append(res, item)
+	}
+	sort.Slice(res, func(i, j int) bool {
+		return less(res[i], res[j])
+	})
+	return res
+}

set/set_test.go

@@ -168,4 +168,12 @@ var _ = Describe("Set", func() {
 			Expect(SortedSlice(s)).To(Equal([]int{1, 3, 5, 7, 10}))
 		})
 	})
+
+	Describe("SortedSliceFunc", func() {
+		It("should return an ordered slice of the items depending on the less function", func() {
+			s := New[int](5, 1, 10, 7, 3)
+
+			Expect(SortedSliceFunc(s, func(v1, v2 int) bool { return v1 > v2 })).To(Equal([]int{10, 7, 5, 3, 1}))
+		})
+	})
 })