eamonnmcevoy

defer func() {
    session.DropDatabase(dbName)
    session.Close()
}()

eamonnmcevoy

func(s *Session) DropDatabase(db string) error {
  if(s.session != nil) {
    return s.session.DB(db).DropDatabase()
  }
  return nil
}

eamonnmcevoy

package mongo

import (
  "gopkg.in/mgo.v2"
)

type Session struct {
  session *mgo.Session
}

eamonnmcevoy

package mongo_test

import (
  "log"
  "testing"
  "go_rest_api/pkg"
  "go_rest_api/pkg/mongo"
)

eamonnmcevoy

func(p *UserService) Create(u *root.User) error {
  user := newUserModel(u)
  return p.collection.Insert(&user)
}

func(p *UserService) GetByUsername(username string) (*root.User,error) {
  model := userModel{}
  err := p.collection.Find(bson.M{"username": username}).One(&model)
  return model.toRootUser(), err
}

eamonnmcevoy

package root

type MongoConfig struct {
  Ip string `json:"ip"`
  DbName string `json:"dbName"`
}

type ServerConfig struct {
   Port string `json:"port"`
}

eamonnmcevoy

package config

import (
  "os"
  "fmt"
  "encoding/json"
  "path/filepath"
  "go_rest_api/pkg"
)

eamonnmcevoy

package mongo

import (
  "go_rest_api/pkg"
  "gopkg.in/mgo.v2/bson"
  "gopkg.in/mgo.v2"
)

type userModel struct {
  Id           bson.ObjectId `bson:"_id,omitempty"`

eamonnmcevoy

package mongo

import (
  "gopkg.in/mgo.v2"
  "gopkg.in/mgo.v2/bson"
  "go_rest_api/pkg"
)

type UserService struct {
 collection *mgo.Collection

eamonnmcevoy

package mongo

import (
  "gopkg.in/mgo.v2"
)

type Session struct {
  session *mgo.Session
}