douke7431 2016-12-16 14:46
浏览 30
已采纳

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

I need to create a HTTP Handler for a REST API.

This REST API have many different objects that are stored in a database (MongoDB in my case).

Currently, I need to write one handler per action per object.

I would like to find a way like it's possible with Generics to write a generic handler that could handle a specific action but for any kind of object (As basically it's just CRUD in most of the case)

How can I do this ?

Here is examples of Handlers I would like to transform into a generic one :

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)
    }
}

Basically I would need a handler like this (This is my tentative that doesn't work) :

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)
    }

How can I do that ?

  • 写回答

1条回答 默认 最新

  • dswwuo1223 2016-12-16 15:14
    关注

    Use the reflect package to create a new container on each invocation:

    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)
    }
    

    Call it like this:

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

    assuming that data.Indgredient is a slice type.

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 用前端向数据库插入数据,通过debug发现数据能走到后端,但是放行之后就会提示错误
  • ¥15 python天天向上类似问题,但没有清零
  • ¥30 3天&7天&&15天&销量如何统计同一行
  • ¥30 帮我写一段可以读取LD2450数据并计算距离的Arduino代码
  • ¥15 C#调用python代码(python带有库)
  • ¥15 矩阵加法的规则是两个矩阵中对应位置的数的绝对值进行加和
  • ¥15 活动选择题。最多可以参加几个项目?
  • ¥15 飞机曲面部件如机翼,壁板等具体的孔位模型
  • ¥15 vs2019中数据导出问题
  • ¥20 云服务Linux系统TCP-MSS值修改?