Files
wrapper/op_array/util_test.go

35 lines
828 B
Go

// Package op_array ...
//
// Description : op_array ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2025-11-27 12:40
package op_array
import (
"testing"
"git.zhangdeman.cn/zhangdeman/serialize"
)
func TestTree(t *testing.T) {
type Data struct {
ID int `json:"id" dc:"数据ID"`
ParentID int `json:"parent_id" dc:"父级ID"`
Value string `json:"value" dc:"数据值"`
}
dataList := []Data{
{ID: 1, ParentID: 0, Value: "1"},
{ID: 2, ParentID: 1, Value: "2"},
{ID: 3, ParentID: 1, Value: "3"},
{ID: 4, ParentID: 2, Value: "4"},
{ID: 5, ParentID: 2, Value: "5"},
{ID: 6, ParentID: 3, Value: "6"},
{ID: 7, ParentID: 3, Value: "7"},
}
res := Tree(dataList, func(v Data) int { return v.ParentID }, func(v Data) int { return v.ID })
serialize.JSON.ConsoleOutput(res)
}