init
This commit is contained in:
life
276
app/db/Mgo.go
Normal file
276
app/db/Mgo.go
Normal file
@ -0,0 +1,276 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"labix.org/v2/mgo"
|
||||
"labix.org/v2/mgo/bson"
|
||||
"github.com/revel/revel"
|
||||
)
|
||||
|
||||
// Init mgo and the common DAO
|
||||
|
||||
// 数据连接
|
||||
var Session *mgo.Session
|
||||
|
||||
// 各个表的Collection对象
|
||||
var Notebooks *mgo.Collection
|
||||
var Notes *mgo.Collection
|
||||
var NoteContents *mgo.Collection
|
||||
var NoteContentHistories *mgo.Collection
|
||||
|
||||
var ShareNotes *mgo.Collection
|
||||
var ShareNotebooks *mgo.Collection
|
||||
var HasShareNotes *mgo.Collection
|
||||
var Blogs *mgo.Collection
|
||||
var Users *mgo.Collection
|
||||
|
||||
var Tags *mgo.Collection
|
||||
var TagNotes *mgo.Collection
|
||||
|
||||
var UserBlogs *mgo.Collection
|
||||
|
||||
var Tokens *mgo.Collection
|
||||
|
||||
var Suggestions *mgo.Collection
|
||||
|
||||
// 初始化时连接数据库
|
||||
func Init() {
|
||||
var url string
|
||||
var ok bool
|
||||
config := revel.Config;
|
||||
url, ok = config.String("db.url")
|
||||
dbname, _ := config.String("db.dbname")
|
||||
if !ok {
|
||||
host, _ := revel.Config.String("db.host")
|
||||
port, _ := revel.Config.String("db.port")
|
||||
username, _ := revel.Config.String("db.username")
|
||||
password, _ := revel.Config.String("db.password")
|
||||
url = "mongodb://" + username + ":" + password + "@" + host + ":" + port + "/" + dbname
|
||||
}
|
||||
|
||||
// [mongodb://][user:pass@]host1[:port1][,host2[:port2],...][/database][?options]
|
||||
// mongodb://myuser:mypass@localhost:40001,otherhost:40001/mydb
|
||||
var err error
|
||||
Session, err = mgo.Dial(url)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Optional. Switch the session to a monotonic behavior.
|
||||
Session.SetMode(mgo.Monotonic, true)
|
||||
|
||||
// notebook
|
||||
Notebooks = Session.DB(dbname).C("notebooks")
|
||||
|
||||
// notes
|
||||
Notes = Session.DB(dbname).C("notes")
|
||||
|
||||
// noteContents
|
||||
NoteContents = Session.DB(dbname).C("note_contents")
|
||||
NoteContentHistories = Session.DB(dbname).C("note_content_histories")
|
||||
|
||||
// share
|
||||
ShareNotes = Session.DB(dbname).C("share_notes")
|
||||
ShareNotebooks = Session.DB(dbname).C("share_notebooks")
|
||||
HasShareNotes = Session.DB(dbname).C("has_share_notes")
|
||||
|
||||
// user
|
||||
Users = Session.DB(dbname).C("users")
|
||||
|
||||
// blog
|
||||
Blogs = Session.DB(dbname).C("blogs")
|
||||
|
||||
// tag
|
||||
Tags = Session.DB(dbname).C("tags")
|
||||
TagNotes = Session.DB(dbname).C("tag_notes")
|
||||
|
||||
// blog
|
||||
UserBlogs = Session.DB(dbname).C("user_blogs")
|
||||
|
||||
// find password
|
||||
Tokens = Session.DB(dbname).C("tokens")
|
||||
|
||||
//
|
||||
Suggestions = Session.DB(dbname).C("suggestions")
|
||||
}
|
||||
|
||||
func init() {
|
||||
revel.OnAppStart(func() {
|
||||
Init()
|
||||
})
|
||||
}
|
||||
|
||||
func close() {
|
||||
Session.Close()
|
||||
}
|
||||
|
||||
// common DAO
|
||||
// 公用方法
|
||||
|
||||
//----------------------
|
||||
|
||||
func Insert(collection *mgo.Collection, i interface{}) bool {
|
||||
err := collection.Insert(i)
|
||||
return Err(err)
|
||||
}
|
||||
|
||||
//----------------------
|
||||
|
||||
// 适合一条记录全部更新
|
||||
func Update(collection *mgo.Collection, query interface{}, i interface{}) bool {
|
||||
err := collection.Update(query, i)
|
||||
return Err(err)
|
||||
}
|
||||
func Upsert(collection *mgo.Collection, query interface{}, i interface{}) bool {
|
||||
_, err := collection.Upsert(query, i)
|
||||
return Err(err)
|
||||
}
|
||||
func UpdateAll(collection *mgo.Collection, query interface{}, i interface{}) bool {
|
||||
_, err := collection.UpdateAll(query, i)
|
||||
return Err(err)
|
||||
}
|
||||
func UpdateByIdAndUserId(collection *mgo.Collection, id, userId string, i interface{}) bool {
|
||||
err := collection.Update(GetIdAndUserIdQ(id, userId), i)
|
||||
return Err(err)
|
||||
}
|
||||
|
||||
func UpdateByIdAndUserId2(collection *mgo.Collection, id, userId bson.ObjectId, i interface{}) bool {
|
||||
err := collection.Update(GetIdAndUserIdBsonQ(id, userId), i)
|
||||
return Err(err)
|
||||
}
|
||||
func UpdateByIdAndUserIdField(collection *mgo.Collection, id, userId, field string, value interface{}) bool {
|
||||
return UpdateByIdAndUserId(collection, id, userId, bson.M{"$set": bson.M{field:value}})
|
||||
}
|
||||
func UpdateByIdAndUserIdMap(collection *mgo.Collection, id, userId string, v bson.M) bool {
|
||||
return UpdateByIdAndUserId(collection, id, userId, bson.M{"$set": v})
|
||||
}
|
||||
func UpdateByIdAndUserIdField2(collection *mgo.Collection, id, userId bson.ObjectId, field string, value interface{}) bool {
|
||||
return UpdateByIdAndUserId2(collection, id, userId, bson.M{"$set": bson.M{field:value}})
|
||||
}
|
||||
func UpdateByIdAndUserIdMap2(collection *mgo.Collection, id, userId bson.ObjectId, v bson.M) bool {
|
||||
return UpdateByIdAndUserId2(collection, id, userId, bson.M{"$set": v})
|
||||
}
|
||||
//
|
||||
func UpdateByQField(collection *mgo.Collection, q interface{}, field string, value interface{}) bool {
|
||||
_, err := collection.UpdateAll(q, bson.M{"$set": bson.M{field: value}})
|
||||
return Err(err)
|
||||
}
|
||||
// 查询条件和值
|
||||
func UpdateByQMap(collection *mgo.Collection, q interface{}, v interface{}) bool {
|
||||
_, err := collection.UpdateAll(q, bson.M{"$set": v})
|
||||
return Err(err)
|
||||
}
|
||||
|
||||
//------------------------
|
||||
|
||||
// 删除一条
|
||||
func Delete(collection *mgo.Collection, q interface{}) bool {
|
||||
err := collection.Remove(q)
|
||||
return Err(err)
|
||||
}
|
||||
func DeleteByIdAndUserId(collection *mgo.Collection, id, userId string) bool {
|
||||
err := collection.Remove(GetIdAndUserIdQ(id, userId))
|
||||
return Err(err)
|
||||
}
|
||||
func DeleteByIdAndUserId2(collection *mgo.Collection, id, userId bson.ObjectId) bool {
|
||||
err := collection.Remove(GetIdAndUserIdBsonQ(id, userId))
|
||||
return Err(err)
|
||||
}
|
||||
|
||||
// 删除所有
|
||||
func DeleteAllByIdAndUserId(collection *mgo.Collection, id, userId string) bool {
|
||||
_, err := collection.RemoveAll(GetIdAndUserIdQ(id, userId))
|
||||
return Err(err)
|
||||
}
|
||||
func DeleteAllByIdAndUserId2(collection *mgo.Collection, id, userId bson.ObjectId) bool {
|
||||
_, err := collection.RemoveAll(GetIdAndUserIdBsonQ(id, userId))
|
||||
return Err(err)
|
||||
}
|
||||
|
||||
func DeleteAll(collection *mgo.Collection, q interface{}) bool {
|
||||
_, err := collection.RemoveAll(q)
|
||||
return Err(err)
|
||||
}
|
||||
|
||||
//-------------------------
|
||||
|
||||
func Get(collection *mgo.Collection, id string, i interface{}) {
|
||||
collection.FindId(bson.ObjectIdHex(id)).One(i)
|
||||
}
|
||||
func Get2(collection *mgo.Collection, id bson.ObjectId, i interface{}) {
|
||||
collection.FindId(id).One(i)
|
||||
}
|
||||
|
||||
func GetByQ(collection *mgo.Collection, q interface{}, i interface{}) {
|
||||
collection.Find(q).One(i)
|
||||
}
|
||||
func ListByQ(collection *mgo.Collection, q interface{}, i interface{}) {
|
||||
collection.Find(q).All(i)
|
||||
}
|
||||
|
||||
func ListByQLimit(collection *mgo.Collection, q interface{}, i interface{}, limit int) {
|
||||
collection.Find(q).Limit(limit).All(i)
|
||||
}
|
||||
|
||||
// 查询某些字段, q是查询条件, fields是字段名列表
|
||||
func GetByQWithFields(collection *mgo.Collection, q bson.M, fields []string, i interface{}) {
|
||||
selector := make(bson.M, len(fields))
|
||||
for _, field := range fields {
|
||||
selector[field] = true
|
||||
}
|
||||
collection.Find(q).Select(selector).One(i)
|
||||
}
|
||||
// 查询某些字段, q是查询条件, fields是字段名列表
|
||||
func ListByQWithFields(collection *mgo.Collection, q bson.M, fields []string, i interface{}) {
|
||||
selector := make(bson.M, len(fields))
|
||||
for _, field := range fields {
|
||||
selector[field] = true
|
||||
}
|
||||
collection.Find(q).Select(selector).All(i)
|
||||
}
|
||||
func GetByIdAndUserId(collection *mgo.Collection, id, userId string, i interface{}) {
|
||||
collection.Find(GetIdAndUserIdQ(id, userId)).One(i)
|
||||
}
|
||||
func GetByIdAndUserId2(collection *mgo.Collection, id, userId bson.ObjectId, i interface{}) {
|
||||
collection.Find(GetIdAndUserIdBsonQ(id, userId)).One(i)
|
||||
}
|
||||
|
||||
//----------------------
|
||||
|
||||
func Count(collection *mgo.Collection, q interface{}) int {
|
||||
cnt, err := collection.Find(q).Count()
|
||||
if err != nil {
|
||||
Err(err)
|
||||
}
|
||||
return cnt
|
||||
}
|
||||
|
||||
func Has(collection *mgo.Collection, q interface{}) bool {
|
||||
if Count(collection, q) > 0 {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//-----------------
|
||||
|
||||
// 得到主键和userId的复合查询条件
|
||||
func GetIdAndUserIdQ(id, userId string) bson.M {
|
||||
return bson.M{"_id": bson.ObjectIdHex(id), "UserId": bson.ObjectIdHex(userId)}
|
||||
}
|
||||
func GetIdAndUserIdBsonQ(id, userId bson.ObjectId) bson.M {
|
||||
return bson.M{"_id": id, "UserId": userId}
|
||||
}
|
||||
|
||||
// DB处理错误
|
||||
func Err(err error) bool {
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
// 删除时, 查找
|
||||
if err.Error() == "not found" {
|
||||
return true;
|
||||
}
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
Reference in New Issue
Block a user