如何在Golang中返回动态类型结构?

时间:2016-02-26 16:51:09

标签: go revel

我正在使用Golang Revel进行一些网络项目,到目前为止,我确实喜欢了12个项目。在所有这些中,由于返回类型,我有很多代码冗余。看看这两个功能:

func (c Helper) Brands() []*models.Brand{

    //do some select on rethinkdb and populate correct model
    var brands []*models.Brand
    rows.All(&brands)

    return brands

}

func (c Helper) BlogPosts() []*models.Post{

    //do some select on rethinkdb and populate correct model
    var posts []*models.Post
    rows.All(&posts)

    return posts

}

正如您所看到的,它们都返回相同类型的数据(类型struct)。 我的想法就是像这样传递字符串var:

func (c Helper) ReturnModels(modelName string) []*interface{} {

    //do rethinkdb select with modelName and return []*interface{} for modelName
}

像这样我只能有一个帮助器来返回数据类型而不是 对于不同的模型,但是相同的数据类型,一遍又一遍地做同样的事情。

我的问题是:

  1. 这是否可行
  2. 如果是,你能指出我正确的文件
  3. 如果不是,我会更乐意回复你的答案:)

1 个答案:

答案 0 :(得分:25)

是的,但是您的功能可能会返回interface{}而不是[]*interface

func (c Helper) ReturnModels(modelName string) interface{} {}

在这种情况下,您可以使用Type Switches and/or Type Assertions将返回值强制转换为原始类型。

示例

注意:我从未使用过Revel,但以下代码段应该为您提供一个大致的想法:

Playground

package main

import "fmt"

type Post struct {
    Author  string
    Content string
}

type Brand struct {
    Name string
}

var database map[string]interface{}

func init() {
    database = make(map[string]interface{})

    brands := make([]Brand, 2)
    brands[0] = Brand{Name: "Gucci"}
    brands[1] = Brand{Name: "LV"}

    database["brands"] = brands

    posts := make([]Post, 1)
    posts[0] = Post{Author: "J.K.R", Content: "Whatever"}

    database["posts"] = posts
}

func main() {
    fmt.Println("List of Brands: ")
    if brands, ok := ReturnModels("brands").([]Brand); ok {
        fmt.Printf("%v", brands)
    }

    fmt.Println("\nList of Posts: ")
    if posts, ok := ReturnModels("posts").([]Post); ok {
        fmt.Printf("%v", posts)
    }

}

func ReturnModels(modelName string) interface{} {

    return database[modelName]
}