mongo_middleware_example.go

package models

import (
	"net/http"

	"labix.org/v2/mgo"
)

type DB struct {
	Database string
	Session  *mgo.Session
}

func NewDB(server string, database string) *DB {
	session, err := mgo.Dial(server)
	if err != nil {
		panic(err)
	}
	return &DB{
		Database: database,
		Session:  session,
	}
}

// Default pages
func (db *DB) Pages() *Pages {
	return NewPages(db.Session.DB(db.Database).C("pages"))
}

func (db *DB) Events() *Events {
	return NewEvents(db.Session.DB(db.Database).C("events"))
}

func (db *DB) ServeHTTP(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
	session := db.Session.Clone()
	defer session.Close()

	database := session.DB(db.Database)

	// Map Dependencies
	SetPages(r, NewPages(database.C("pages")))
	SetEvents(r, NewEvents(database.C("events")))

	next(rw, r)
}