工具函数——对任意类型的切片去重

最新推荐文章于 2024-10-27 19:55:47 发布

小小小熊猫5

最新推荐文章于 2024-10-27 19:55:47 发布

阅读量254

点赞数 3

分类专栏：工具函数文章标签：开发语言

本文链接：https://blog.csdn.net/trinityleo5/article/details/136847849

版权

工具函数专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文介绍了如何在Go语言中使用`reflect`包创建一个名为`RemoveDuplicates`的工具函数，用于对任意类型的切片进行去重，并提供了对应的单元测试示例。

摘要由CSDN通过智能技术生成

工具函数——对任意类型的切片去重

- 函数
- 单元测试

函数

func RemoveDuplicates(slice interface{}) interface{} {
	encountered := make(map[interface{}]bool)
	resultSlice := reflect.MakeSlice(reflect.TypeOf(slice), 0, 0)

	valSlice := reflect.ValueOf(slice)
	for i := 0; i < valSlice.Len(); i++ {
		val := valSlice.Index(i)
		if !encountered[val.Interface()] {
			encountered[val.Interface()] = true
			resultSlice = reflect.Append(resultSlice, val)
		}
	}

	return resultSlice.Interface()
}

单元测试

func TestRemoveDuplicates(t *testing.T) {
	intSlice := []int{1, 2, 3, 4, 3, 2, 1}
	uniqueInts := RemoveDuplicates(intSlice).([]int)
	fmt.Println(intSlice)
	fmt.Println(uniqueInts) // Output: [1 2 3 4]

	stringSlice := []string{"apple", "banana", "apple", "orange", "banana"}
	uniqueStrings := RemoveDuplicates(stringSlice).([]string)
	fmt.Println(stringSlice)
	fmt.Println(uniqueStrings) // Output: [apple banana orange]
}