1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
package gleak
import (
"fmt"
"sort"
"strconv"
"strings"
"github.com/onsi/gomega/format"
)
// G takes an actual "any" untyped value and returns it as a typed Goroutine, if
// possible. It returns an error if actual isn't of either type Goroutine or a
// pointer to it. G is intended to be mainly used by goroutine-related Gomega
// matchers, such as IgnoringTopFunction, et cetera.
func G(actual interface{}, matchername string) (Goroutine, error) {
if actual != nil {
switch actual := actual.(type) {
case Goroutine:
return actual, nil
case *Goroutine:
return *actual, nil
}
}
return Goroutine{},
fmt.Errorf("%s matcher expects a Goroutine or *Goroutine. Got:\n%s",
matchername, format.Object(actual, 1))
}
// goids returns a (sorted) list of Goroutine IDs in textual format.
func goids(gs []Goroutine) string {
ids := make([]uint64, len(gs))
for idx, g := range gs {
ids[idx] = g.ID
}
sort.Sort(Uint64Slice(ids))
var buff strings.Builder
for idx, id := range ids {
if idx > 0 {
buff.WriteString(", ")
}
buff.WriteString(strconv.FormatInt(int64(id), 10))
}
return buff.String()
}
// Uint64Slice implements the sort.Interface for a []uint64 to sort in
// increasing order.
type Uint64Slice []uint64
func (s Uint64Slice) Len() int { return len(s) }
func (s Uint64Slice) Less(a, b int) bool { return s[a] < s[b] }
func (s Uint64Slice) Swap(a, b int) { s[a], s[b] = s[b], s[a] }
|
