blob: bb5759960df7fa7d97745bb3b85208da1b85eb34 (
plain)
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
|
package main
import (
"math/rand"
"reflect"
"time"
)
// firstN returns the first n items of a slice.
func firstN(n int, items interface{}) interface{} {
v := reflect.ValueOf(items)
if v.Kind() != reflect.Slice {
panic("firstN: not a slice")
}
if v.Len() < n {
return items
}
return v.Slice(0, n).Interface()
}
// lastN returns the last n items of any slice.
func lastN(n int, items interface{}) interface{} {
v := reflect.ValueOf(items)
if v.Kind() != reflect.Slice {
panic("lastN: not a slice")
}
l := v.Len()
if l < n {
return items
}
return v.Slice(l-n, l).Interface()
}
// randomN returns n random items of any slice.
func randomN(n int, items interface{}) interface{} {
v := reflect.ValueOf(items)
if v.Kind() != reflect.Slice {
panic("randomN: not a slice")
}
l := v.Len()
if l < n {
return items
}
rand.Seed(time.Now().UnixNano())
indices := rand.Perm(l)[:n]
result := reflect.MakeSlice(v.Type(), n, n)
for i, idx := range indices {
result.Index(i).Set(v.Index(idx))
}
return result.Interface()
}
// filterByType filters pages by their type.
func filterByType(pageType string, pages interface{}) interface{} {
v := reflect.ValueOf(pages)
if v.Kind() != reflect.Slice {
panic("filterByType: not a slice")
}
var filtered []interface{}
for i := 0; i < v.Len(); i++ {
page := v.Index(i).Interface().(Page)
if page.Type == pageType {
filtered = append(filtered, page)
}
}
return filtered
}
|
