-
Notifications
You must be signed in to change notification settings - Fork 1
/
rows.go
70 lines (60 loc) · 1.03 KB
/
rows.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
package row
import (
"sort"
"github.com/gotodb/gotodb/datatype"
)
// Rows for sort rows
type Rows struct {
Data []*Row
Order []datatype.OrderType
}
func NewRows(order []datatype.OrderType) *Rows {
return &Rows{
Data: []*Row{},
Order: order,
}
}
func (r *Rows) Min() int {
res := -1
ln := len(r.Data)
for i := 0; i < ln; i++ {
if r.Data[i] == nil {
continue
}
if res < 0 {
res = i
} else {
if r.Less(i, res) {
res = i
}
}
}
return res
}
func (r *Rows) Less(i, j int) bool {
rowi, rowj := r.Data[i], r.Data[j]
for k := 0; k < len(r.Order); k++ {
vi, vj := rowi.Keys[k], rowj.Keys[k]
if datatype.EQFunc(vi, vj).(bool) {
continue
}
res := datatype.LTFunc(vi, vj).(bool)
if r.Order[k] == datatype.DESC {
res = !res
}
return res
}
return false
}
func (r *Rows) Swap(i, j int) {
r.Data[i], r.Data[j] = r.Data[j], r.Data[i]
}
func (r *Rows) Len() int {
return len(r.Data)
}
func (r *Rows) Sort() {
sort.Sort(r)
}
func (r *Rows) Append(row *Row) {
r.Data = append(r.Data, row)
}