Question

我需要为REST API创建一个HTTP Handler。

这个REST API有许多不同的对象存储在数据库中（在我的例子中是MongoDB）。

目前，我需要为每个对象的每个操作编写一个处理程序。

我想找到一种像Generics这样可以编写一个通用处理程序的方法，该处理程序可以处理特定的操作但是对于任何类型的对象（基本上它只是在大多数情况下的CRUD）情况下）

我该怎么做？

以下是我想要转换为通用处理程序的处理程序示例：

func IngredientIndex(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "application/json; charset=UTF-8")
    db := r.Context().Value("Database").(*mgo.Database)
    ingredients := []data.Ingredient{}
    err := db.C("ingredients").Find(bson.M{}).All(&ingredients)
    if err != nil {
        panic(err)
    }
    json.NewEncoder(w).Encode(ingredients)
}

func IngredientGet(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "application/json; charset=UTF-8")
    vars := mux.Vars(r)
    logger := r.Context().Value("Logger").(zap.Logger)
    db := r.Context().Value("Database").(*mgo.Database)
    ingredient := data.Ingredient{}
    err := db.C("ingredients").Find(bson.M{"_id": bson.ObjectIdHex(vars["id"])}).One(&ingredient)
    if err != nil {
        w.WriteHeader(404)
        logger.Info("Fail to find entity", zap.Error(err))
    } else {
        json.NewEncoder(w).Encode(ingredient)
    }
}

基本上我需要一个像这样的处理程序（这是我的暂时不起作用）：

func ObjectIndex(collection string, container interface{}) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Content-Type", "application/json; charset=UTF-8")
        db := r.Context().Value("Database").(*mgo.Database)
        objects := container
        err := db.C(collection).Find(bson.M{}).All(&objects)
        if err != nil {
            panic(err)
        }
        json.NewEncoder(w).Encode(objects)
    }

我该怎么做？

Answer 1

使用reflect包在每次调用时创建一个新容器：

func ObjectIndex(collection string, container interface{}) func(http.ResponseWriter, *http.Request) {
  t := reflect.TypeOf(container)
  return func(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "application/json; charset=UTF-8")
    db := r.Context().Value("Database").(*mgo.Database)
    objects := reflect.New(t).Interface()
    err := db.C(collection).Find(bson.M{}).All(objects)
    if err != nil {
        panic(err)
    }
    json.NewEncoder(w).Encode(objects)
}

这样称呼：

h := ObjectIndex("ingredients", data.Ingredient{})

假设data.Indgredient是切片类型。

如何在Go中编写通用处理程序？

1 个答案: