dongseshu0698
dongseshu0698
2017-07-24 17:47

错误使用go-gin和mgo从mongoDB按ID获取民意调查

已采纳

How do i query polls by id with go-gin and MongoDB, i have tried several methods but i still get errors (not found), can't seem to find a walk around below is my code, with my database on mongoDB:

type Poll struct {
    //ID        string `json:"_id,omitempty"`
    ID        bson.ObjectId     `json:"id,omitempty" bson:"_id,omitempty"`
    Firstname string            `json:"firstname,omitempty"`
    Lastname  string            `json:"lastname,omitempty"`
    Poll      string            `json:"poll,omitempty"`
    //  Address   *Address `json:"address,omitempty"`
}

var (
    // Session stores mongo session
    Session *mgo.Session

    // Mongo stores the mongodb connection string information
    Mongo *mgo.DialInfo
)

const (
    // MongoDBUrl is the default mongodb url that will be used to connect to the
    // database.
    MongoDBUrl = "mongodb://localhost:27017/smartpoll"

        // CollectionPoll holds the name of the poll collection
    CollectionPoll = "polls"
)

// Connect connects to mongodb
func Connect() {
    uri := os.Getenv("MONGODB_URL")

    if len(uri) == 0 {
        uri = MongoDBUrl
    }

    mongo, err := mgo.ParseURL(uri)
    s, err := mgo.Dial(uri)
    if err != nil {
        fmt.Printf("Can't connect to mongo, go error %v
", err)
        panic(err.Error())
    }
    s.SetSafe(&mgo.Safe{})
    fmt.Println("Connected to", uri)
    Session = s
    Mongo = mongo
}


func init() {
    Connect()
}

func main() {
    port := os.Getenv("PORT")

    if port == "" {
        log.Fatal("$PORT must be set")
    }



    router := gin.Default()
    router.Use(ConnectMiddleware)
    router.GET("/", func (c *gin.Context) {
        c.JSON(http.StatusOK, gin.H{"message": "OK"})

    })

    router.GET("/polls/:_id", pollsByID)
    router.Run(":" + port)
}




func ConnectMiddleware(c * gin.Context) {
    c.Set("db", Session.DB(Mongo.Database))
    c.Next()
}

func pollsByID(c * gin.Context) {
    db := c.MustGet("db").(*mgo.Database)
    id := c.Param("id")
    poll := []Poll{}
//  err := db.C(CollectionPoll).Find(id).One(&poll)
    err := db.C(CollectionPoll).Find(bson.M{"_id": id}).One(&poll)



    if err != nil {
        //c.Error(err)
        //panic(err)
      log.Println(err)
    }
    result := gin.H{"payload": poll}
  c.Writer.Header().Set("Content-Type", "application/json")
    c.JSON(200, result)

}

my DB is as follows:

{
    "_id" : ObjectId("58d9cf1cdf353f3d2f5951b4"),
    "id" : "1",
    "firstname" : "Sam",
    "lastname" : "Smith",
    "poll" : "Who is the Richest in the World"
}
  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享
  • 邀请回答

2条回答

  • dr897777 dr897777 4年前

    Change polls from an array:

     polls := []Poll{}
    

    TO:

    polls := Poll{}
    
    点赞 评论 复制链接分享
  • duanqing2209 duanqing2209 4年前

    Your ID is an ObjectId, but your input is a string. You need to use bson.ObjectIdHex to parse the string into an ObjectId:

    err := db.C(CollectionPoll).FindId(bson.ObjectIdHex(id)).One(&poll)
    
    点赞 评论 复制链接分享