Go 函数中返回结构体指针

一分耕耘，一分收获！既然都打开这篇《Go 函数中返回结构体指针》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

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

我面临的问题是最后一个函数的返回值。最简单的情况是，我以为我可以返回 *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
    }
}

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

感谢您的所有意见！

正确答案

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

根据您对其余代码的使用情况，可以尝试以下任一方法：

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

以上就是《Go 函数中返回结构体指针》的详细内容，更多关于的资料请关注golang学习网公众号！