helper/struct_test.go

package helper

import (
	"testing"
)

type User struct {
	ID   int
	NAME string
}

type UserString struct {
	ID   string
	NAME string
}

func Test_structColumn(t *testing.T) {
	user1 := UserString{
		ID:   "2",
		NAME: "zwk",
	}
	user2 := UserString{
		ID:   "3",
		NAME: "zzz",
	}
	var list3 []UserString
	list3 = append(list3, user1)
	list3 = append(list3, user2)

	t.Log(list3)
	t.Log("---------------------")

	var userMap map[string]string
	StructColumn(&userMap, list3, "NAME", "ID")
	t.Logf("%#v\n", userMap)
	t.Log("---------------------")

	var userMap1 map[int]UserString
	StructColumn(&userMap1, list3, "", "ID")
	t.Logf("%#v\n", userMap1)
	t.Log("---------------------")

	var userSlice []string
	StructColumn(&userSlice, list3, "ID", "")
	t.Logf("%#v\n", userSlice)
	t.Log("---------------------")
}