在Golang中返回接口

Question

我正在尝试在一个采用接口类型并返回该接口类型但转换为适当类型的结构上编写一个方法。

type Model interface {
    GetEntity()
}

type TrueFalseQuestions struct {
   //some stuff
}

func (q *TrueFalseQuestions) GetEntity() {
   //some stuff
}

type MultiQuestions struct {
    //some stuff
}

func (q *MultiQuestions) GetEntity() {
    //some stuff
}


type Manager struct {
}


func (man *Manager) GetModel(mod Model) Model {
    mod.GetEntity()
    return mod
}

func main() {
    var man Manager

    q := TrueFalseQuestions {}
    q = man.GetModel(&TrueFalseQuestions {})
}

所以，当我打电话GetModel()型TrueFalseQuestions我想自动返回TrueFalseQuestions类型。 我认为这意味着我的GetModel()方法应该返回Model类型。 这样，如果我传递MultiQuestion类型，则返回MultiQuestion结构。

Answer 1

当返回类型为Model时，您不能直接返回TrueFalseQuestions 。 它将始终隐式包装在Model接口中。

为了找回TrueFalseQuestions ，您需要使用类型断言。 （您还需要注意指针与值）

// this should be a pointer, because the interface methods all have pointer receivers
q := &TrueFalseQuestions{}
q = man.GetModel(&TrueFalseQuestions{}).(*TrueFalseQuestions)

如果您遇到MultiQuestions ，那当然会感到恐慌，因此您应该检查ok值，或使用类型开关

switch q := man.GetModel(&TrueFalseQuestions{}).(type) {
case *TrueFalseQuestions:
    // q isTrueFalseQuestions
case *MultiQuestions:
    // q is MultiQuestions
default:
    // unexpected type
}

Answer 2

您不能，但是可以对返回值使用类型断言。

func main() {
    var man Manager

    tfq := &TrueFalseQuestions{}
    q := man.GetModel(tfq)
    if v, ok := q.(*TrueFalseQuestions); ok {
        fmt.Println("v is true/false", v)
    } else if v, ok := q.(*MultiQuestions); ok {
        fmt.Println("v is mq", v)
    } else {
        fmt.Println("unknown", q)
    }

}

操场