用接口干掉我的 Go 函数答案

【问题标题】：DRY out my Go function with interfaces用接口干掉我的 Go 函数
【发布时间】：2013-07-06 22:35:03
【问题描述】：

我有以下功能：

func (r *Resource) Create(kind string, data io.ReadCloser) (err error) {
  decoder := json.NewDecoder(data)
  r.Kind = kind

  switch kind {
  case "user":
    var user User
    if err = decoder.Decode(&user); err != nil {
      panic(err)
    }

    if err = user.Save(r.Context); err != nil {
      panic(err)
    }

    r.Data = user
    break

  case "space":
    var space Space
    if err = decoder.Decode(&space); err != nil {
      panic(err)
    }

    if err = space.Save(r.Context); err != nil {
      panic(err)
    }

    r.Data = space
    break

  case "room":
    var room Room
    if err = decoder.Decode(&room); err != nil {
      panic(err)
    }

    if err = room.Save(r.Context); err != nil {
      panic(err)
    }

    r.Data = room
    break

  case "element":
    var element Element
    if err = decoder.Decode(&element); err != nil {
      panic(err)
    }

    if err = element.Save(r.Context); err != nil {
      panic(err)
    }

    r.Data = element
    break

  default:
    break
  }
  return
}

如您所见，除了接收 JSON 数据的结构的类型外，switch 中的每个 case 都是相同的。

我怀疑接口和类型断言中有答案。

编辑：

我能够将保存部分分解为一个单独的方法，但我仍然无法找到一种在没有 switch 语句的情况下将 JSON 对象解码为适当结构的好方法。

func (r *Resource) Create(kind string, data io.ReadCloser) (err error) {
  decoder := json.NewDecoder(data)
  r.Kind = kind

  switch kind {
  case "user":
    var user User
    if err = decoder.Decode(&user); err != nil {
      panic(err)
    }
    r.saveEntity(&user)
    break

  case "space":
    var space Space
    if err = decoder.Decode(&space); err != nil {
      panic(err)
    }
    r.saveEntity(&space)
    break

  case "room":
    var room Room
    if err = decoder.Decode(&room); err != nil {
      panic(err)
    }
    r.saveEntity(&room)
    break

  case "element":
    var element Element
    if err = decoder.Decode(&element); err != nil {
      panic(err)
    }
    r.saveEntity(&element)
    break

  default:
    break
  }
  return
}

func (r *Resource) saveEntity(e Entity) {
  if err := e.Save(r.Context); err != nil {
    panic(err)
  }

  r.Data = e
}

【问题讨论】：

标签： interface struct go

【解决方案1】：

您可以将实例化移动到单行函数并创建一个映射来映射对应于各自的实例化函数。其余代码应该是可重用的。

例子：

kinds := map[string]func() Entity {
    "user": func() Entity { return &User{} },
    "space": func() Entity { return &Space{} },
    "room": func() Entity { return &Room{} },
}

func Create(kind string) {
    instance := kinds[kind]()

    decoder.Decode(instance)

    saveEntity(instance)
}

【讨论】：