// Get a list of entities by the selector. Should pass the pointer of an empty list in
// Usage:
// var list []Entity
// ok := testAccessor.FindAll(id, &list)
func (m *MgoAccessor) FindAll(query interface{}, selector interface{}, result interface{}) {
err := m.coll().Find(query).Select(selector).All(result)
fiplog.GetLogger().Debug("query:", query, ",result", result)
if err != nil {
fiplog.GetLogger().Error("Find all failed. query:", query, "error:", err)
}
}
func NewMgoConnWithAuth(url, db, user, pwd string) *MgoConn {
//m := &MgoManager{url,db}
conn, err := mgo.Dial(url)
if err != nil {
fiplog.GetLogger().Error("Connect to database failed, url:", url, ",error:", err)
}
database := conn.DB(db)
err = database.Login(user, pwd)
if err != nil {
fiplog.GetLogger().Error("Database auth error:", err)
}
return &MgoConn{database}
}
func NewMgoConnWithUrl(connStr string) *MgoConn {
conn, err := mgo.Dial(connStr)
if err != nil {
fiplog.GetLogger().Error("Connect to database failed, conn:", connStr, ",error:", err)
return nil
}
return &MgoConn{conn.DB("")}
}
func (m *MgoAccessor) DeleteAll(selector interface{}) int {
info, err := m.coll().RemoveAll(selector)
if err != nil {
fiplog.GetLogger().Error("Delete all failed. selector:", selector, ",error:", err)
return 0
}
return info.Removed
}
func (m *MgoAccessor) Delete(id interface{}) bool {
err := m.coll().RemoveId(id)
if err != nil {
fiplog.GetLogger().Error("Delete failed. Id:", id, ",error:", err)
return false
}
return true
}
func (m *MgoAccessor) UpdatePartialById(id, update interface{}) bool {
err := m.coll().UpdateId(id, update)
if err != nil {
fiplog.GetLogger().Error("Update partial by id failed. id:", id, "error:", err)
return false
}
return true
}
func (m *MgoAccessor) Update(entity MgoEntity) bool {
err := m.coll().UpdateId(entity.GetId(), entity)
if err != nil {
fiplog.GetLogger().Error("Update entity failed. id:", entity.GetId(), "error:", err)
return false
}
return true
}
func (m *MgoAccessor) GetById(id interface{}, selector interface{}, entity interface{}) bool {
err := m.coll().FindId(id).Select(selector).One(entity)
if err != nil {
fiplog.GetLogger().Error("Get partial entity failed. id:", id, "error:", err)
return false
}
return true
}
func (m *MgoAccessor) IdExists(id interface{}) (exists bool, err error) {
count, err := m.coll().FindId(id).Count()
if err != nil {
fiplog.GetLogger().Error("Check if id exists failed. id:", id, ",error:", err)
return
}
exists = count != 0
return
}
func (m *MgoAccessor) UpdatePartial(selector interface{}, update interface{}) bool {
err := m.coll().Update(selector, update)
if err != nil {
fiplog.GetLogger().Error("Partial update failed.", selector, update, "error:", err)
return false
}
return true
}
func (m *MgoAccessor) Insert(entity ...interface{}) bool {
err := m.coll().Insert(entity...)
if err != nil {
fiplog.GetLogger().Error("Insert entity failed. error:", err)
return false
}
//entity.SetId(id)
return true
}
func NewMgoConn(url, db string) *MgoConn {
//m := &MgoManager{url,db}
conn, err := mgo.Dial(url)
if err != nil {
fiplog.GetLogger().Error("Connect to database failed, url:", url, ",error:", err)
return nil
}
database := conn.DB(db)
return &MgoConn{database}
}
func (m *MgoAccessor) GetPartial(id interface{}, entity interface{}, fields ...string) bool {
if len(fields) == 0 {
fiplog.GetLogger().Error("Get partial entity, no field specified")
return false
}
selector := M{}
for _, field := range fields {
selector[field] = 1
}
return m.GetById(id, selector, entity)
}
//pageIndex starts with 1
func (m *MgoAccessor) FindPage(query, selector, result interface{}, pageSize, pageIndex int, sortFields ...string) (pageCount int) {
if pageSize == 0 {
pageSize = defaultPageSize
}
if pageIndex < 1 {
pageIndex = 1
}
q := m.coll().Find(query).Select(selector)
total, err := q.Count()
if err != nil {
fiplog.GetLogger().Error("Find page failed. query:", query, "error:", err)
return
}
skip := pageSize * (pageIndex - 1)
if total <= skip {
return
}
pageCount = total / pageSize
/*if pageCount == 0 || pageCount < pageIndex {
fiplog.GetLogger().Error("Find page failed, pageCound:",pageCount,", pageIndex:",pageIndex)
return
}*/
if len(sortFields) != 0 {
q = q.Sort(sortFields...)
}
err = q.Skip(skip).Limit(pageSize).All(result)
if err != nil {
fiplog.GetLogger().Error("Find all failed. query:", query, "error:", err)
return
}
return
}
func (m *MgoAccessor) FindOne(query interface{}, selector interface{}, result interface{}, sortFields ...string) bool {
q := m.coll().Find(query).Select(selector)
if len(sortFields) != 0 {
q = q.Sort(sortFields...)
}
err := q.One(result)
if err != nil {
fiplog.GetLogger().Error("Find one failed. query:", query, "error:", err)
return false
}
return true
}