pushpkg/db/mongodb.go

package db

import (
	. "app.yhyue.com/moapp/jybase/mongodb"

	"github.com/donnie4w/go-logger/logger"
)

var MyMgo = &myMgo{}

type myMgo struct {
}

func (m *myMgo) SaveBulk(mgo *MongodbSim, sess *MgoSess, dbName, collName string, array *[]map[string]interface{}) error {
	if sess == nil {
		sess = mgo.GetMgoConn()
		defer mgo.DestoryMongoConn(sess)
	}
	bulk := sess.DB(dbName).C(collName).Bulk()
	for _, v := range *array {
		bulk.Insert(v)
	}
	_, err := bulk.Run()
	if nil != err {
		logger.Info("SaveBulkError", err)
	}
	*array = []map[string]interface{}{}
	return err
}
func (m *myMgo) DelBulk(mgo *MongodbSim, sess *MgoSess, dbName, collName string, array *[]interface{}) error {
	if sess == nil {
		sess = mgo.GetMgoConn()
		defer mgo.DestoryMongoConn(sess)
	}
	bulk := sess.DB(dbName).C(collName).Bulk()
	for _, v := range *array {
		bulk.Remove(map[string]interface{}{"_id": v})
	}
	_, err := bulk.Run()
	if nil != err {
		logger.Info("DelBulkError", err)
	}
	*array = []interface{}{}
	return err
}