/
sort.go
116 lines (98 loc) · 1.95 KB
/
sort.go
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Copyright © 2020 Hedzr Yeh.
package tool
import (
"sort"
"strings"
)
type byDottedSlice []string
func (s byDottedSlice) Len() int { return len(s) }
func (s byDottedSlice) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
func (s byDottedSlice) Less(i, j int) (ret bool) {
sa := strings.Split(s[i], ".")
sb := strings.Split(s[j], ".")
la, lb := len(sa), len(sb)
ll := la
if ll > lb {
ll = lb
}
var lastRet int
for i := 0; i < ll; i++ {
switch rb := strings.Compare(sa[i], sb[i]); {
case rb == -1:
if lastRet != rb {
return true
}
lastRet = rb
continue
case rb == 0:
if lastRet != rb {
return lastRet < 0
}
lastRet = rb
continue
default:
ret = false
return
}
}
rr := strings.Compare(sa[ll-1], sb[ll-1])
if rr < 0 {
ret = true
return
}
if rr == 0 {
ret = la <= lb
return
}
return
}
type byDottedSliceRev []string
func (s byDottedSliceRev) Len() int { return len(s) }
func (s byDottedSliceRev) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
func (s byDottedSliceRev) Less(i, j int) (ret bool) {
sa := strings.Split(s[i], ".")
sb := strings.Split(s[j], ".")
la, lb := len(sa), len(sb)
ll := la
if ll > lb {
ll = lb
}
var lastRet int
for i := 0; i < ll; i++ {
switch rb := strings.Compare(sa[i], sb[i]); {
case rb == -1:
if lastRet != rb {
return false
}
lastRet = rb
continue
case rb == 0:
if lastRet != rb {
return lastRet > 0
}
lastRet = rb
continue
default:
ret = true
return
}
}
rr := strings.Compare(sa[ll-1], sb[ll-1])
if rr < 0 {
ret = false
return
}
if rr == 0 {
ret = la > lb
return
}
return
}
// SortAsDottedSlice sorts a slice by which is treated as a dot-separated path string
func SortAsDottedSlice(ks []string) {
sort.Sort(byDottedSlice(ks))
}
// SortAsDottedSliceReverse sorts a slice by which is treated as a dot-separated path string
func SortAsDottedSliceReverse(ks []string) {
sort.Sort(byDottedSliceRev(ks))
}