如何重构Go函数以适配存储相似结构体的自定义切片类型

问题根因 #

你之前尝试定义上层接口不生效，是因为Go语言中[]结构体类型和[]接口类型是完全独立的两种类型，无法直接隐式转换，所以不能直接把自定义切片类型传入参数为接口切片的函数。 #

方案1：泛型实现（推荐，Go 1.18及以上版本适用） #

泛型可以直接适配不同元素类型的切片，无需做类型转换，是目前最优的解决方案：

1. 定义公共接口约束所有Activity类型的行为

import "time"

type Activity interface {
    GetStartTime() time.Time
}

2. 为两种Activity结构体实现上述接口

// 假设你的两种Activity结构体定义如下
type WorkActivity struct {
    StartTime time.Time
    // 其他字段...
}
type LifeActivity struct {
    StartTime time.Time
    // 其他字段...
}

func (w WorkActivity) GetStartTime() time.Time {
    return w.StartTime
}
func (l LifeActivity) GetStartTime() time.Time {
    return l.StartTime
}

3. 实现统一的泛型处理函数

// T为切片元素类型，约束为实现了Activity接口的任意类型
func ProcessActivities[T Activity](activities []T) {
    // 此处编写全局统一的处理逻辑
    for _, act := range activities {
        s := act.GetStartTime()
        // 示例逻辑：格式化输出StartTime，可替换为你自己的处理逻辑
        println(s.Format("2006-01-02 15:04:05"))
    }
}

调用时直接传入两种自定义切片类型的实例即可，无需额外转换。 #

方案2：非泛型实现（兼容Go 1.18以下版本） #

如果无法升级Go版本，可以将核心处理逻辑抽离，仅保留最小化的类型适配代码：

1. 抽离单条记录的公共处理逻辑

import "time"

func processSingleStartTime(s time.Time) {
    // 此处编写全局统一的处理逻辑
    println(s.Format("2006-01-02 15:04:05"))
}

2. 为两种自定义切片类型添加适配方法

// 假设你的自定义切片类型定义如下
type WorkActivities []WorkActivity
type LifeActivities []LifeActivity

func (wa WorkActivities) Process() {
    for _, act := range wa {
        processSingleStartTime(act.StartTime)
    }
}
func (la LifeActivities) Process() {
    for _, act := range la {
        processSingleStartTime(act.StartTime)
    }
}

该方案核心逻辑仅需维护一份，适配代码只有简单的遍历调用，后续迭代成本极低。 #

内容的提问来源于stack exchange，提问作者digitalbadger