You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.3 KiB
58 lines
1.3 KiB
package gomongo
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
// Client 实例
|
|
type Client struct {
|
|
db *mongo.Client // MongoDB驱动
|
|
ctx context.Context // 上下文
|
|
DatabaseName string // 库名
|
|
collectionName string // 表名
|
|
filterArr []queryFilter // 查询条件数组
|
|
filter bson.D // 查询条件
|
|
}
|
|
|
|
// NewClient 实例化并链接数据库
|
|
func NewClient(dns string) *Client {
|
|
|
|
// 连接到MongoDB
|
|
db, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(dns))
|
|
if err != nil {
|
|
panic(fmt.Sprintf("连接失败:%v", err))
|
|
}
|
|
|
|
// 检查连接
|
|
err = db.Ping(context.TODO(), nil)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("检查连接失败:%v", err))
|
|
}
|
|
|
|
return &Client{db: db, ctx: context.TODO()}
|
|
}
|
|
|
|
// NewClientDb 实例化并传入链接
|
|
func NewClientDb(db *mongo.Client) *Client {
|
|
return &Client{db: db, ctx: context.TODO()}
|
|
}
|
|
|
|
// Close 关闭
|
|
func (c *Client) Close() {
|
|
err := c.db.Disconnect(context.TODO())
|
|
if err != nil {
|
|
panic(errors.New(fmt.Sprintf("关闭失败:%v", err)))
|
|
}
|
|
return
|
|
}
|
|
|
|
// GetDbDriver 获取驱动
|
|
func (c *Client) GetDbDriver() *mongo.Client {
|
|
return c.db
|
|
}
|