71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
// Package op_array ...
|
|
//
|
|
// Description : op_array ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2025-11-25 11:30
|
|
package op_array
|
|
|
|
import "fmt"
|
|
|
|
// ToMap 数组转map
|
|
func ToMap[Key comparable, Value any](dataList []Value, keyFormat func(item Value) Key) map[Key]Value {
|
|
res := make(map[Key]Value)
|
|
for _, item := range dataList {
|
|
key := keyFormat(item)
|
|
res[key] = item
|
|
}
|
|
return res
|
|
}
|
|
|
|
// ToCustomMap 数组转map
|
|
func ToCustomMap[Key comparable, Value any, CustomValue any](dataList []Value, keyFormat func(item Value) Key, formatCustomValue func(item Value) CustomValue) map[Key]CustomValue {
|
|
res := make(map[Key]CustomValue)
|
|
for _, item := range dataList {
|
|
key := keyFormat(item)
|
|
res[key] = formatCustomValue(item)
|
|
}
|
|
return res
|
|
}
|
|
|
|
// ExtractField 提取数组指定字段, 并构建成一个新的数组
|
|
func ExtractField[FieldValue any, Value any](dataList []Value, fieldValue func(item Value) FieldValue) []FieldValue {
|
|
res := make([]FieldValue, 0)
|
|
for _, item := range dataList {
|
|
v := fieldValue(item)
|
|
res = append(res, v)
|
|
}
|
|
return res
|
|
}
|
|
|
|
// Filter 过滤列表数据
|
|
func Filter[Value any](dataList []Value, filterValue func(item Value) bool) []Value {
|
|
res := make([]Value, 0)
|
|
for _, item := range dataList {
|
|
if filterValue(item) {
|
|
res = append(res, item)
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
// Group 按照指定字段进行分组
|
|
func Group[Key comparable, Value any](dataList []Value, keyFormat func(item Value) Key) [][]Value {
|
|
keyList := make([]string, 0)
|
|
dataTable := make(map[string][]Value)
|
|
res := make([][]Value, 0)
|
|
for _, item := range dataList {
|
|
key := fmt.Sprintf("%v", keyFormat(item))
|
|
if _, ok := dataTable[key]; !ok {
|
|
keyList = append(keyList, key)
|
|
}
|
|
dataTable[key] = append(dataTable[key], item)
|
|
}
|
|
formatList := make([][]Value, 0)
|
|
for _, key := range keyList {
|
|
formatList = append(formatList, dataTable[key])
|
|
}
|
|
return res
|
|
}
|