去 : model ( or interface ) function with input of different types

Question

现在我有一个用并行算法计算统计数据的小应用程序。 现在我在扩展某些功能时遇到了问题。我会尽快解释。 应用程序是建立在 revel 框架上的。 “stat” Controller 的一项操作采用传入的 POST json。解析它。并为任务和结果生成两个 channel (goroutines)。所有这一切都像一个魅力。但是我在模型方面遇到了麻烦。我编写了能够线性扩展模型数量的代码，但目前只有一个在工作。

而且并不是所有的方法都用于这种扩展。

在代码的某些部分我有这个:

for t := range in {
        for sourceName, charts := range t.Request.Charts {

            var cacheData []byte
            var deserializedData models.StatModel

            //determine the model type
            switch sourceName {
            case "noagg":
                deserializedData = new(models.NoaggModel)
            case "acsi":
                deserializedData = new(models.AcsiModel)
            }

            cache_err := cache.Get(string(string(sourceName) + "_" + string(t.Date)), &cacheData);
            if cache_err != nil {
                panic("the cache is empty")
            }

            marshal_error := json.Unmarshal([]byte(cacheData), &deserializedData)
            if marshal_error == nil {

            }

            deserializedData.FilterData(t.Request.Filters)

            deserializedData.ClusterData(t.Request.Filters)

            w := Work{}
            for _, chart := range charts {
                countedData := ChartElements{}

                if marshal_error == nil {
                    countedData = deserializedData.CountDataForChart(string(chart.Name))
                }else {
                    panic("some is bad")
                }

                w.Name, w.Data = chart.Name, countedData
                out <- w
            }
        }
    }

Noagg 模型和 Asci 模型正在实现“stat”模型的相同接口(interface):

type StatModel interface {
    FilterData(Filter)
    ClusterData(Filter)
    CountDataForChart(string)[]ChartElement
    GroupByTreeGroups(Filter)[]OrgPack
}

但现在我必须添加一些具有相同界面的新模型，但是有代码，我无法扩展。我不记得如何做到这一点..

func statCount(model NoaggRow, f func(NoaggRow) float64) float64 {
    countedStat := f(model)
    return countedStat
}

func Count(model NoaggRow, name string) float64{

    m := map[string]func(NoaggRow) float64 {
        "HOLD" : HOLD,
        "INB"  : INB,
        "AHT"  : AHT,
        "RING" : RING,
        "TALK" : TALK,
        "ACW"  : ACW,
        "OCC"  : OCC,
    }
    countedStat := statCount(model, m[name])
    return countedStat
}

我需要用于 ASCI 模型的相同方法。对于 AcsiRow 而不是 NoaggRow。如何使此输入参数类型动态化或如何使方法对所有模型通用。只有数组和“map[string]func(......Row)”的名称在这个地方会有所不同。谁能帮我解决这个问题？

Answer 1

如果映射中的函数(即 HOLD()、INB() 等)不需要比 StatModel 接口(interface)提供的更多的模型访问权限，那么您的要求应该可以通过更改这些函数来接收 StatModel 而不是 NoaggRow 来实现:

func HOLD(s StatModel) float64 {
    ...
}

然后你要添加的AcsiRow函数可以有相同的签名，statCount和Count可以重写如下:

func statCount(model StatModel, f func(StatModel) float64) float64 {
    countedStat := f(model)
    return countedStat
}

func Count(model StatModel, name string) float64 {
    m := map[string]func(StatModel) float64 {
        "HOLD" : HOLD,
        "INB"  : INB,
        "AHT"  : AHT,
        "RING" : RING,
        "TALK" : TALK,
        "ACW"  : ACW,
        "OCC"  : OCC,
        // Map AcsiModel functions here, let's call them ACSIn here
        // (you mentioned that the names would be different, so
        // I assume they don't get in the way of the functions
        // above):
        "ACSI1": ACSI1,
        "ACSI2": ACSI2,
        "ACSI3": ACSI3,
    }
    countedStat := statCount(model, m[name])
    return countedStat
}

免责声明:这只是一个基于我在帖子中看到的内容的想法。我对 Revel 一无所知，也许因此我可能错过了一些由 Revel 上下文引起的重要细节，这些细节会阻止实现此方法。

根据评论更新:

使用type assertions启用 HOLD、OCC 等函数来访问结构特定的属性，如下所示:

func HOLD (s StatModel) float64 {
    noagg, ok := s.(NoaggRow) // typecast from StateModel to NoaggRow
    if !ok {
        // Looks like the caller did not pass a matching model/name pair into Count().
        // Error handling...
    }
    attr := noagg.AttributeSpecificToNoaggRow
    ...
}

这在运行时可能看起来有点冒险，但如果调用者总是将匹配的模型/名称值对传递给 Count()(我假设它这样做)，则类型转换应该是安全的。