我正在使用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
}
像这样我只能有一个帮助器来返回数据类型而不是 对于不同的模型,但是相同的数据类型,一遍又一遍地做同样的事情。
我的问题是:
答案 0 :(得分:25)
是的,但是您的功能可能会返回interface{}
而不是[]*interface
。
func (c Helper) ReturnModels(modelName string) interface{} {}
在这种情况下,您可以使用Type Switches and/or Type Assertions将返回值强制转换为原始类型。
示例强>
注意:我从未使用过Revel,但以下代码段应该为您提供一个大致的想法:
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]
}