Go语言中实现通用数据访问功能

type Person struct {
    FirstName string
    LastName  string
}

type Company struct {
    Industry string
    Name     string
}

// 假设我们希望实现一个通用的函数，能够根据类型、字段和值来获取数据
// var persons []Person
// persons = getItems("Person", "FirstName", "John")

// var companies []Company
// companies = getItems("Company", "Industry", "Software")

// 模拟一个数据库和getItems函数
var database []interface{}

func init() {
    // 填充一些模拟数据
    database = append(database, Person{FirstName: "John", LastName: "Doe"})
    database = append(database, Company{Industry: "Software", Name: "Tech Solutions"})
    database = append(database, Person{FirstName: "Jane", LastName: "Smith"})
    database = append(database, Company{Industry: "Finance", Name: "Global Investments"})
}

// getItems 模拟从数据库中获取所有符合条件的项，返回 []interface{}
// 注意：这里简化了实际的查询逻辑，仅为演示类型处理
func getItems(typ string, field string, val string) []interface{} {
    var results []interface{}
    // 实际的数据库查询逻辑会在这里，根据typ, field, val进行过滤
    // 这里我们简单地返回所有模拟数据，后续通过类型断言进行筛选
    for _, item := range database {
        results = append(results, item)
    }
    return results
}

// getTypedItems 接收一个通用接口切片，并通过类型断言筛选并返回指定类型的切片
func getTypedItems[T any](items []interface{}) []T {
    output := make([]T, 0)
    for _, item := range items {
        // 类型断言：尝试将 item 转换为类型 T
        // thing 是转换后的值，ok 表示断言是否成功
        thing, ok := item.(T)
        if ok {
            output = append(output, thing)
        }
    }
    return output
}

// 示例用法
func main() {
    // 假设我们已经从数据库获取了所有潜在的Person和Company数据
    allPotentialItems := getItems("Person", "FirstName", "John") // 这里的参数现在可能只是一个占位符

    // 使用类型断言筛选出 Person 类型
    persons := getTypedItems[Person](allPotentialItems)
    fmt.Println("Filtered Persons:", persons)

    // 使用类型断言筛选出 Company 类型
    companies := getTypedItems[Company](allPotentialItems)
    fmt.Println("Filtered Companies:", companies)
}

// getItemByCriteria 接收一个判别函数，根据该函数筛选数据库中的项
func getItemByCriteria(criteria func(item interface{}) bool) []interface{} {
    output := make([]interface{}, 0)
    for _, item := range database { // 遍历模拟数据库中的所有项
        if criteria(item) { // 如果判别函数返回 true，则添加到结果中
            output = append(output, item)
        }
    }
    return output
}

// 示例用法
func main() {
    // 查找 FirstName 为 "John" 的 Person
    johns := getItemByCriteria(func(item interface{}) bool {
        if p, ok := item.(Person); ok {
            return p.FirstName == "John"
        }
        return false
    })
    fmt.Println("Persons named John:", getTypedItems[Person](johns))

    // 查找 Industry 为 "Software" 的 Company
    softwareCompanies := getItemByCriteria(func(item interface{}) bool {
        if c, ok := item.(Company); ok {
            return c.Industry == "Software"
        }
        return false
    })
    fmt.Println("Software Companies:", getTypedItems[Company](softwareCompanies))
}