feat(Go-Tool):2020/11/23:修改项目案例(按照一定规则对一组数据进行排序分组)
This commit is contained in:
+81
@@ -0,0 +1,81 @@
|
||||
package linq
|
||||
|
||||
// Group is a type that is used to store the result of GroupBy method.
|
||||
type Group struct {
|
||||
Key interface{}
|
||||
Group []interface{}
|
||||
}
|
||||
|
||||
// GroupBy method groups the elements of a collection according to a specified
|
||||
// key selector function and projects the elements for each group by using a
|
||||
// specified function.
|
||||
func (q Query) GroupBy(keySelector func(interface{}) interface{},
|
||||
elementSelector func(interface{}) interface{}) Query {
|
||||
return Query{
|
||||
func() Iterator {
|
||||
next := q.Iterate()
|
||||
set := make(map[interface{}][]interface{})
|
||||
|
||||
for item, ok := next(); ok; item, ok = next() {
|
||||
key := keySelector(item)
|
||||
set[key] = append(set[key], elementSelector(item))
|
||||
}
|
||||
|
||||
len := len(set)
|
||||
idx := 0
|
||||
groups := make([]Group, len)
|
||||
for k, v := range set {
|
||||
groups[idx] = Group{k, v}
|
||||
idx++
|
||||
}
|
||||
|
||||
index := 0
|
||||
|
||||
return func() (item interface{}, ok bool) {
|
||||
ok = index < len
|
||||
if ok {
|
||||
item = groups[index]
|
||||
index++
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// GroupByT is the typed version of GroupBy.
|
||||
//
|
||||
// - keySelectorFn is of type "func(TSource) TKey"
|
||||
// - elementSelectorFn is of type "func(TSource) TElement"
|
||||
//
|
||||
// NOTE: GroupBy has better performance than GroupByT.
|
||||
func (q Query) GroupByT(keySelectorFn interface{},
|
||||
elementSelectorFn interface{}) Query {
|
||||
keySelectorGenericFunc, err := newGenericFunc(
|
||||
"GroupByT", "keySelectorFn", keySelectorFn,
|
||||
simpleParamValidator(newElemTypeSlice(new(genericType)), newElemTypeSlice(new(genericType))),
|
||||
)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
keySelectorFunc := func(item interface{}) interface{} {
|
||||
return keySelectorGenericFunc.Call(item)
|
||||
}
|
||||
|
||||
elementSelectorGenericFunc, err := newGenericFunc(
|
||||
"GroupByT", "elementSelectorFn", elementSelectorFn,
|
||||
simpleParamValidator(newElemTypeSlice(new(genericType)), newElemTypeSlice(new(genericType))),
|
||||
)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
elementSelectorFunc := func(item interface{}) interface{} {
|
||||
return elementSelectorGenericFunc.Call(item)
|
||||
|
||||
}
|
||||
|
||||
return q.GroupBy(keySelectorFunc, elementSelectorFunc)
|
||||
}
|
||||
Reference in New Issue
Block a user