我想要一个功能来检查不同地图中的重复键。
这就是我所拥有的
ma := map[string]typea
mb := map[string]typeb
mc := map[string]typec
dup := map[string]bool{}
for k := range ma{
if !dup[k] {
dup[k] = true
} else {
return fmt.errorf("duplicate key[%v]", k)
}
}
for k := range mb{
if !dup[k] {
dup[k] = true
} else {
return fmt.errorf("duplicate key[%v]", k)
}
}
for k := range mc {
if !dup[k] {
dup[k] = true
} else {
return fmt.errorf("duplicate key[%v]", k)
}
}
return nil
我想重构它并编写一个函数
func checkDupKeys[M ~map[K]V, K comparable, V any](maps ...M) error {
dup := map[K]bool{}
for _, m := range maps {
for k := range m {
if !dup[k] {
dup[k] = true
} else {
return fmt.Errorf("duplicate key[%v]", k)
}
}
}
return nil
}
但它只能接受相同类型的映射,不能接受 typea、typeb 和 typec。
正确答案
您可以尝试使用 any
类型和反射
func checkDupKeys(maps ...any) error {
dup := map[any]bool{}
for i, m := range maps {
t := reflect.TypeOf(m)
if t.Kind() != reflect.Map {
return fmt.Errorf("not a map at index: %d", i)
}
keys := reflect.ValueOf(m).MapKeys()
for _, k := range keys {
v := k.Interface()
if !dup[v] {
dup[v] = true
} else {
return fmt.Errorf("duplicate key[%v]", v)
}
}
}
return nil
}
这种方法的缺点是该函数还将接受非映射参数,并且编译器不会对此发出警告。