mongo_orm.gogo

{{define "mongo_orm"}}package {{.GoPackage}}
{{$obj := .}}
{{if ($obj.DbSwitch "mongo")}}
import (
	"time"

	//3rd party libs
	"gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"

	//Own libs
	"github.com/ezbuy/ezorm/db"
	. "github.com/ezbuy/ezorm/orm"

	{{range .GetOrmImports}}
	"{{.}}"
	{{end}}
)

var _ time.Time

func init() {
	{{if .NeedIndex}}
	db.SetOnEnsureIndex(init{{.Name}}Index)
	addMgoIndexFunc(init{{.Name}}Index)
	{{end}}
	{{if .NeedMapping}}
	db.SetOnFinishInit(init{{.Name}}Mapping)
	{{end}}
	RegisterEzOrmObjByID("{{.Package}}", "{{.Name}}", new{{.Name}}FindByID)
	RegisterEzOrmObjRemove("{{.Package}}", "{{.Name}}", {{.Name}}Mgr.RemoveByID)
	{{if .NeedSearch}}
	RegisterIndexer("{{.Package}}", "{{.Name}}", {{.Name}}Mgr.SearchRebuild)
	{{end}}
}

{{if .NeedMapping}}
func init{{.Name}}Mapping() {
	opt := indices.MappingOptions{
		Properties: map[string]interface{}{
			{{range .GetStringFilterFields}}
			"{{.Name}}": map[string]string{"index": "not_analyzed", "type": "string"},
			{{end}}
		},
	}

	indices.CreateIndex("{{.SearchIndex}}")
	indices.PutMapping("{{.SearchIndex}}", "{{.SearchType}}", {{.Name}}{}, opt)
}
{{end}}

{{if .NeedIndex}}
func init{{.Name}}Index() {
	session, collection := {{.Name}}Mgr.GetCol()
	defer session.Close()
	{{range $index := $obj.Indexes}}
	if err := collection.EnsureIndex(mgo.Index{
		Key: []string{"{{$index.GetFieldList}}"},
		{{- if $index.IsUnique }}
		Unique: true,
		{{- end}}
		Background: true,
		{{- if $index.IsSparse }}
		Sparse: true,
		{{- end}}
	}); err != nil {
		panic("ensureIndex {{$obj.DbName}}.{{$obj.Name}} {{$index.Name}} error:" + err.Error())
	}
	{{end}}
}
{{end}}

func new{{.Name}}FindByID(id string) (result EzOrmObj, err error) {
	return {{.Name}}Mgr.FindByID(id)
}

{{ .LoadTpl "mongo_mongo" }}

{{ .LoadTpl "mongo_foreign_key" }}

{{ .LoadTpl "mongo_collection" }}

{{ .LoadTpl "mongo_search" }}
{{end}}
{{end}}