goのmgoでfindするときの処理

はじめに

go言語でmongoDBにアクセスするためのライブラリにmgoがあります。mgoでデータベースからfindしてくるときの処理について、書きたいと思います。

サンプル

サンプルとして以下のようなコードで2件データを挿入しておきます。

package main

import (
	"log"

	mgo "gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"
)

type Man struct {
	Id   bson.ObjectId `bson:"_id"`
	Name string        `bson:"name"`
}

func main() {
	session, err := mgo.Dial("mongodb://localhost/")
	if err != nil {
		log.Fatal(err)
	}
	defer session.Close()
	c := session.DB("some-db-name").C("man")

	if err := c.Insert(Man{
		Id:   bson.NewObjectId(),
		Name: "Taro",
	}, Man{
		Id:   bson.NewObjectId(),
		Name: "Jiro",
	}); err != nil {
		log.Fatal(err)
	}

}

全件取得

Findメソッドにnilあるいはbson.M{}を渡すと全件取得できます。

	var men []Man
	if err := c.Find(nil).All(&men); err != nil {
		log.Fatal(err)
	}

	spew.Dump(men)
	/*
		([]main.Man) (len=2 cap=2) {
		 (main.Man) {
		  Id: (bson.ObjectId) (len=12) ObjectIdHex("565edd868bc93d268a13bc01"),
		  Name: (string) (len=4) "Taro"
		 },
		 (main.Man) {
		  Id: (bson.ObjectId) (len=12) ObjectIdHex("565edd868bc93d268a13bc02"),
		  Name: (string) (len=4) "Jiro"
		 }
		}
	*/

絞り込み

上述のFindメソッドに条件を指定できます。
以下のサイトに記載されているmongoのオペレータは同様に使えます。
https://docs.mongodb.org/manual/reference/operator/query/

	//例 name が Taro か Hanako
	var men []Man
	if err := c.Find(bson.M{
		"name": bson.M{
			"$in": []string{"Taro", "Hanako"},
		},
	}).All(&men); err != nil {
		log.Fatal(err)
	}

ObjectIdで1件取得

"565edd868bc93d268a13bc02" というようなId形式の文字列でfindするときは、ObjectId型に変換しなければいけません。
bson.ObjectIdHex(string)で変換するのですが、ObjectIdの形式の文字列ではない場合、panicします（panic: Invalid input to ObjectIdHex:）。
そのため、bson.IsObjectIdHex(string)でチェックしてから変換するのが安全です。

	idStr := "565edd868bc93d268a13bc02"
	if !bson.IsObjectIdHex(idStr) {
		log.Fatal("not objectId")
	}
	id := bson.ObjectIdHex(idStr)

	var man Man
	if err := c.FindId(id).One(&man); err != nil {
		log.Fatal(err)
	}
	spew.Dump(man)
	/*
		(main.Man) {
		 Id: (bson.ObjectId) (len=12) ObjectIdHex("565edd868bc93d268a13bc02"),
		 Name: (string) (len=4) "Jiro"
		}
	*/

注意として、FindIdでマッチするデータが存在しない場合、エラーとして返されます。
そのため、存在しない場合を別途処理するときは、以下のように記述する必要があります。

	if err := c.FindId(id).One(&man); err != nil {
		if err == mgo.ErrNotFound {
			fmt.Println("Not found")
		} else {
			log.Fatal(err)
		}
	}