Exemplo n.º 1
0
// 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)
	}
}
Exemplo n.º 2
0
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}
}
Exemplo n.º 3
0
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("")}
}
Exemplo n.º 4
0
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
}
Exemplo n.º 5
0
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
}
Exemplo n.º 6
0
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
}
Exemplo n.º 7
0
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
}
Exemplo n.º 8
0
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
}
Exemplo n.º 9
0
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
}
Exemplo n.º 10
0
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
}
Exemplo n.º 11
0
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
}
Exemplo n.º 12
0
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}
}
Exemplo n.º 13
0
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)
}
Exemplo n.º 14
0
//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
}
Exemplo n.º 15
0
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
}