Golang 接口方法链接



我有一个接口Cells有几个方法

type Cells interface{
Len() int
//....
}

具体的实现有StrCellsIntCellsFloatCellsBoolCells,所有这些都实现了上述方法。

例如:

type StrCells []string
func (sC StrCells) Len() int {return len(sC)}
//...
type IntCells []int
func (iC IntCells) Len() int {return len(iC)}
//...
//....

对于两个具体类型 -IntCellsFloatCells- 我想实现仅适用于这些类型的特定功能。

我创建了一个嵌入Cells的新接口NumCells

type NumCells interface{
Cells
Add(NumCells) interface{} // should return either IntCells or FloatCells 
}

以下是我对 IntCellAdd()的实现:

func (iC IntCells) Add(nC NumCells) interface{} {
if iC.Len() != nC.Len() {
// do stuff
}
switch nC.(type) {
case IntCells:
res := make(IntCells, iC.Len())
for i, v := range iC {
res[i] = v + nC.(IntCells)[i]
}
return res
case FloatCells:
res := make(FloatCells, iC.Len())
for i, v := range iC {
res[i] = float64(v) + nC.(FloatCells)[i]
}
return res
default:
// to come
return nil
}
}

这是我的问题/问题

该函数有效,但是,我实际上希望该函数返回NumCells(即 IntCells 或 FloatCells),以便我可以像这样进行方法链接

a := columns.IntCells(1, 2, 4, 2)
b := columns.IntCells{2, 3, 5, 3}
c := columns.FloatCells{3.1, 2, 2.4, 3.2}
d := a.Add(b).Add(c)

如果Add()返回interface{},则无法执行此操作。但是,否则我无法使该功能工作。

如果您以这种方式定义NumCells接口,它将起作用:

type NumCells interface{
Cells
Add(NumCells) NumCells // returns either IntCells or FloatCells
}

然后,您需要IntCellsFloatCells来实现Add并返回其中一种类型。

下面是一个工作游乐场,使用方法链接并打印结果:

https://play.golang.org/p/W7DzcB4A3NH

如注释中所述,在使用接口时,通常希望使每个类型与其余实现无关,并且只使用没有类型开关的接口。

Add的实现中避免这些类型开关的一种方法是在NumCells中添加另一个方法,以将特定位置作为float64返回。

type NumCells interface{
Cells
Add(NumCells) NumCells // returns either IntCells or FloatCells
GetCell(index int) float64
}

这样您就可以获取值,而无需断言特定类型。

由于IntCells无法容纳float64值,因此它仍然需要创建一个FloatCells来返回它,如果我们想避免IntCells这样做,我们需要以某种方式抽象对象的创建,使用工厂模式或类似模式。

最新更新