根据 Golang 中的不同切片过滤结构切片

Question

在下面的示例代码中，我有几个用户manySimpleUsers ，我想从删除manyFullUsers基于用户名。

如果我使用嵌套的几个for... range循环来执行此操作，则将需要许多循环来过滤所有元素，尤其是当两个 Slices 中都有大量元素时。

在 Go 中实现这一目标的最佳方法是什么？

package main

import "fmt"

func main() {
    fmt.Println("Hello, playground")

    type FullUser struct {
        UserName  string
        UserEmail string
    }

    manyFullUsers := []FullUser{{"foo", "foo@jawohl.com"},
        {"bar", "bar@jawohl.com"},
        {"baz", "baz@jawohl.com"}}

    type SimpleUser struct {
        UserName string
    }

    manySimpleUsers := []SimpleUser{{"foo"}, {"bar"}}

    fmt.Println(manyFullUsers)
    fmt.Println(manySimpleUsers)
}

Answer 1

创建地图，然后使用它进行过滤。

func filterByUserName(fu []FullUser, su []SimpleUser) (out []FullUser) {
    f := make(map[string]struct{}, len(su))
    for _, u := range su {
        f[u.UserName] = struct{}{}
    }
    for _, u := range fu {
        if _, ok := f[u.UserName]; ok {
            out = append(out, u)
        }
    }
    return
}

操场

Answer 2

您可以使用来自https://github.com/ledongthuc/goterators 的Filter() 和 Exist()

    filteredFullUsers := goterators.Filter(manyFullUsers, func(item FullUser) bool {
        return !goterators.Exist(manySimpleUsers, SimpleUser{item.UserName})
    })

您还可以从 manySimpleUsers 构建一个地图，这将有助于优化搜索用户名。

这个库需要 Go 1.18 来使用你想要使用的支持泛型 + 动态类型。