数据库驱动
go语言中mongodb 有一个社区驱动 mgo
mgo文档
但是最近mongo官方推出了Go驱动，所以我们用官方的。
相关链接 ：
MongoDB 文档
MongoDB Go Driver
链接数据库
package db
var (
dataBase *mongo.Database //DB实例
ctx context.Context //DB context
client mongo.Client
)
func init(){
client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
err = client.Connect(Ctx)
if err != nil {
log.Fatal(err.Error())
}
dataBase = client.Database(DATABASE_NAME) //DATABASE_NAME即数据库名称
}
//在这下面定义两个函数来获取DB 和 context
func GetDB() *mongo.Database {
return dataBase
}
func GetCtx() *context.Context{
return ctx
}
数据库结构
假设有数据库结构如下:
Collection "user"
username string
password string
age int
created date
对应go的model如下：
package userModel
type User struct{
Username string `bson:"username"`
Password string `bson:"passowrd"`
Age int `bson:"age"`
Created time.Time `bson:"created"`
}
插入操作
func AddOneUser () error{
//获取collection
con := db.GetDB().Collection("user")
// 插入的数据
insertData := userModel.User{
Username:"123",
Password:"321",
Age : 18,
Created:Time.Now(),
}
//下面的方法也可以
/*
insertData := bson.M{
Username:"123",
Password:"321",
Age : 18,
Created:Time.Now(),
}
*/
//插入
result, err := con.InsertOne(db.GetCtx(), &insertData)
log.PrintLn(result) //result 是插入后的结果，/插入/匹配/修改/删除/Upsert/ 的document数量
return err
}
批量插入操作
func AddOneUser (users *[]User) error{
//获取collection
con := db.GetDB().Collection("user")
// users 即插入的数据
//插入
result, err := con.InsertMany(db.GetCtx(), users)
log.PrintLn(result) //result 是插入后的结果，/插入/匹配/修改/删除/Upsert/ 的document数量
return err
}
删除操作
func DeleteUser (name string) error{
//通过name来删除user
//获取collection
con := db.GetDB().Collection("user")
result, err := con.DeleteOne(db.GetCtx(),bson.M{"name":name})
log.PrintLn(result)
return err
}
修改操作
func UpdateUser(name string,newUserInfo User) error{
//通过name来更新 ，newUserInfo 是需要更新的user信息
//获取collection
con := db.GetDB().Collection("user")
//匹配名称为name的文档记录 并更新为 newUserInfo
result, err := con.UpdateOne(db.GetCtx(), bson.M{"name":name},bson.M{"$set":newUserInfo})
log.PrintLn(result)
return err
}
查询操作
func FindUserByName(name string) error{
//通过name寻找user信息
//获取collection
con := db.GetDB().Collection("user")
//将获取到的user储存在user变量中
user := User{}
//查询到结果后解码到user变量中
err := con.FindOne(db.GetCtx(), bson.M{"name":name}).Decode(&user)
return err
}
