added blog example

examples/blog/database/database.go

@@ -0,0 +1,20 @@
+package database
+
+import "gopkg.in/mgo.v2"
+
+type Database struct {
+	Session *mgo.Session
+}
+
+func New(uri string) *Database {
+	session, err := mgo.Dial(uri)
+	if err != nil {
+		panic(err)
+	}
+
+	session.SetMode(mgo.Monotonic, true)
+
+	return &Database{
+		Session: session,
+	}
+}

examples/blog/database/post.go

@@ -0,0 +1,73 @@
+package database
+
+import (
+	"errors"
+
+	"github.com/iron-io/functions/examples/blog/models"
+	"gopkg.in/mgo.v2"
+	"gopkg.in/mgo.v2/bson"
+)
+
+var (
+	ErrNotObjectIdHex = errors.New("Invalid ID")
+)
+
+func (db *Database) SavePost(post *models.Post) (*models.Post, error) {
+	s := db.Session.Copy()
+	defer s.Close()
+
+	c := s.DB("").C("post")
+
+	if post.ID.Hex() == "" {
+		post.ID = bson.NewObjectId()
+	}
+	id := post.ID
+
+	change := mgo.Change{
+		Update:    bson.M{"$set": post},
+		ReturnNew: true,
+		Upsert:    true,
+	}
+
+	_, err := c.Find(bson.M{"_id": id}).Apply(change, &post)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return post, nil
+}
+
+func (db *Database) GetPost(id string) (*models.Post, error) {
+	s := db.Session.Copy()
+	defer s.Close()
+
+	c := s.DB("").C("post")
+
+	var post models.Post
+	if !bson.IsObjectIdHex(id) {
+		return nil, ErrNotObjectIdHex
+	}
+
+	err := c.Find(bson.M{"_id": bson.ObjectIdHex(id)}).One(&post)
+	if err != nil {
+		return nil, err
+	}
+
+	return &post, nil
+}
+
+func (db *Database) GetPosts(query []bson.M) ([]*models.Post, error) {
+	s := db.Session.Copy()
+	defer s.Close()
+
+	c := s.DB("").C("post")
+
+	var posts []*models.Post
+
+	err := c.Pipe(query).All(&posts)
+	if err != nil {
+		return nil, err
+	}
+	return posts, nil
+}

examples/blog/database/user.go

@@ -0,0 +1,65 @@
+package database
+
+import (
+	"github.com/iron-io/functions/examples/blog/models"
+	"gopkg.in/mgo.v2"
+	"gopkg.in/mgo.v2/bson"
+)
+
+func (db *Database) SaveUser(user *models.User) (*models.User, error) {
+	s := db.Session.Copy()
+	defer s.Close()
+
+	c := s.DB("").C("user")
+	id := user.Username
+	user.Username = ""
+
+	if len(user.Password) > 0 {
+		user.Password = models.UserPasswordEncrypt(user.Password)
+	}
+
+	change := mgo.Change{
+		Update:    bson.M{"$set": user},
+		ReturnNew: true,
+		Upsert:    true,
+	}
+
+	_, err := c.Find(bson.M{"_id": id}).Apply(change, &user)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return user, nil
+}
+
+func (db *Database) GetUser(id string) (*models.User, error) {
+	s := db.Session.Copy()
+	defer s.Close()
+
+	c := s.DB("").C("user")
+
+	var user models.User
+
+	err := c.Find(bson.M{"_id": id}).One(&user)
+	if err != nil {
+		return nil, err
+	}
+
+	return &user, nil
+}
+
+func (db *Database) GetUsers(query []bson.M) ([]*models.User, error) {
+	s := db.Session.Copy()
+	defer s.Close()
+
+	c := s.DB("").C("user")
+
+	var users []*models.User
+
+	err := c.Pipe(query).All(&users)
+	if err != nil {
+		return nil, err
+	}
+	return users, nil
+}