从 Go 中的函数返回指向结构的指针

Question

我有两个包含不同数据的公共结构，以及一个包含两个公共结构中的任何一个的私有中间结构。 我还有一个函数可以解组中间结构，确定它包含哪个公共结构，并返回两个公共结构之一。

我面临的问题是最后一个函数的返回值。 在最简单的情况下，我认为我可以返回*struct{}但我的 IDE 中不断出现类型不匹配。

我很抱歉发布了可能不必要的更多代码，但我正在努力使其尽可能接近我正在处理的代码。

package main

import (
    "encoding/json"
    "errors"
)

// These vars are some errors I'll use in the functions later on
var (
    errInvalidBase64     = errors.New("invalid base64")
    errInvalidStructType = errors.New("invalid struct type")
)

// Struct1 public struct
type Struct1 struct {
    FName string `json:"first-name"`
    LName string `json:"last-name"`
}

// Struct2 public struct
type Struct2 struct {
    Date  string `json:"date"`
    Items []int  `json:"items"`
}

// intermediateStruct private struct
// The Type field indicates which kind of struct Data contains (Struct1 or Struct2)
// The Data field contains either Struct1 or Struct2 which was previously marshalled into JSON
type intermediateStruct struct {
    Type structType
    Data []byte
}

// The following type and const are my understanding of an enum in Go

// structType is a private type for the type of struct intermediateStruct contains
type structType int

// These public constants are just to keep my hands out of providing values for the different struct types
const (
    StructType1 structType = iota
    StructType2
)

// unmarshalStruct1 unmarshalls JSON []byte into a new Struct1 and returns a pointer to that struct
func unmarshalStruct1(b []bytes) (*Struct1, error) {
    newStruct1 := new(Struct1)
    err := json.Unmarshal(b, newStruct1)
    if err != nil {
        return nil, errInvalidBase64
    }
    return newStruct1, nil
}

// unmarshalStruct2 unmarshalls JSON []byte into a new Struct2 and returns a pointer to that struct
func unmarshalStruct2(b []bytes) (*Struct2, error) {
    newStruct2 := new(Struct2)
    err := json.Unmarshal(b, newStruct2)
    if err != nil {
        return nil, errInvalidBase64
    }
    return newStruct2, nil
}

// receiveStruct accepts *intermediateStruct who's Data field contains either Struct1 or Struct2
// This function needs to return either *Struct1 or *Struct2 and an error
func receiveStruct(iStruct *intermediateStruct) (*struct{}, error) {
    switch iStruct.Type {
    case StructType1:
        struct1, err := unmarshalStruct1(iStruct.Data)
        if err != nil {
            return nil, err
        }
        // The following line is where I'm getting the type mismatch
        return struct1, nil
    case StructType2:
        struct2, err := unmarshalStruct2(iStruct.Data)
        if err != nil {
            return nil, err
        }
        // The following line is another type mismatch
        return struct2, nil
    default:
        return nil, errInvalidStructType
    }
}

我知道有一种方法可以实现我想要实现的目标——我只是缺乏实现目标的经验/理解。

感谢您的任何和所有输入！

Answer 1

您的unmarshallStruct函数返回一个指向Struct1或Struct2类型的指针（取决于调用的函数版本）。 因此变量struct1和struct2分别是指向Struct1和Struct2类型的指针。 也不是指向 struct 类型的指针（无论如何我必须添加它不是真正的 Go 类型）。 结构是有助于声明包含字段/属性的类型的关键字。

根据您为其余代码考虑的用例，可以尝试以下任何一种：

1. 正如 mkopriva 建议的那样，返回一个 interface{} 对象，但您需要使用类型断言来实际确保该对象
2. 定义一个 Struct1 和 Struct2 都实现的接口，并返回一个指向 this 的指针
3. 制作适用于 Struct1 或 Struct2 的单独函数。 这并不像听起来那么糟糕，因为 Go 允许您以与传递类型相同的方式传递函数（参见sort包中的Less()函数的示例）。