GO学习-(26) Go语言操作mongoDB

今年五年级 發表於 2021-2-20 09:24:00

<h1 class="post-title">Go语言操作mongoDB</h1>
 
<div class="cnblogs_code"><img src="https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif" id="code_img_closed_a321bcbe-e526-4570-906f-8a22c356ce69" class="code_img_closed"><img src="https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif" id="code_img_opened_a321bcbe-e526-4570-906f-8a22c356ce69" class="code_img_opened" style="display: none">
<div id="cnblogs_code_open_a321bcbe-e526-4570-906f-8a22c356ce69" class="cnblogs_code_hide">
<pre>package main

import (
"context"
"fmt"
"log"
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)

func ConnectToDB(uri, name string, timeout time.Duration, num uint64) (*mongo.Database, error) {
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
o := options.Client().ApplyURI(uri)
o.SetMaxPoolSize(num)
client, err := mongo.Connect(ctx, o)
if err != nil {
 return nil, err
}
return client.Database(name), nil
}

func main(){

mgClient, err := ConnectToDB("mongodb://192.168.10.10:27017", "bigdata", 10, 100)
if err != nil {
 //fmt.Println("链接失败")
 log.Print(err)
}
// 指定获取要操作的数据集
collection := mgClient.Collection("bigdata")
//fmt.Println(collection)


type Student struct {
 Name string
 Age int
}
//接下来，创建一些Student类型的值，准备插入到数据库中：
s1 := Student{"小红", 12}
//插入文档
//使用collection.InsertOne()方法插入一条文档记录：
insertResult, err := collection.InsertOne(context.TODO(), s1)
if err != nil {
 log.Print(err)
}

fmt.Println("Inserted a single document: ", insertResult.InsertedID)
log.Println("Inserted a single document: ", insertResult.InsertedID)

s2 := Student{"小兰", 10}
s3 := Student{"小黄", 11}
students := []interface{}{s2, s3}
insertManyResult, err := collection.InsertMany(context.TODO(), students)
if err != nil {
 log.Print(err)
}
fmt.Println("Inserted multiple documents: ", insertManyResult.InsertedIDs)
log.Println("Inserted multiple documents: ", insertManyResult.InsertedIDs)

filter := bson.D{{"name", "小兰"}}

update := bson.D{
 {"$inc", bson.D{
 {"age", 1},
 }},
}
//接下来，就可以通过下面的语句找到小兰，给他增加一岁了：

updateResult, err := collection.UpdateOne(context.TODO(), filter, update)
if err != nil {
 log.Print(err)
}
fmt.Printf("Matched %v documents and updated %v documents.\n", updateResult.MatchedCount, updateResult.ModifiedCount)
log.Printf("Matched %v documents and updated %v documents.\n", updateResult.MatchedCount, updateResult.ModifiedCount)

var result Student
err = collection.FindOne(context.TODO(), filter).Decode(&result)
if err != nil {
 log.Fatal(err)
}
fmt.Printf("Found a single document: %+v\n", result)

// 查询多个
// 将选项传递给Find()
findOptions := options.Find()
findOptions.SetLimit(2)

// 定义一个切片用来存储查询结果
var results []*Student

// 把bson.D{{}}作为一个filter来匹配所有文档
cur, err := collection.Find(context.TODO(), filter, findOptions)
if err != nil {
 log.Fatal(err)
}

// 查找多个文档返回一个光标
// 遍历游标允许我们一次解码一个文档
for cur.Next(context.TODO()) {
 // 创建一个值，将单个文档解码为该值
 var elem Student
 err := cur.Decode(&elem)
 if err != nil {
 log.Fatal(err)
 }
 results = append(results, &elem)
}

if err := cur.Err(); err != nil {
 log.Fatal(err)
}

fmt.Printf("Found all document: %+v\n", results)

for k, v := range results{
 fmt.Printf("key:%v value:%v value type:%T\n", k, v, v)
}

// 删除名字是小黄的那个
deleteResult1, err := collection.DeleteOne(context.TODO(), bson.D{{"name","小黄"}})
if err != nil {
 log.Fatal(err)
}
fmt.Printf("Deleted %v documents in the trainers collection\n", deleteResult1.DeletedCount)
// 删除所有
deleteResult2, err := collection.DeleteMany(context.TODO(), bson.D{{}})
if err != nil {
 log.Fatal(err)
}
fmt.Printf("Deleted %v documents in the trainers collection\n", deleteResult2.DeletedCount)
}</pre>
</div>
例子</div>
 
<div class="post-content">
mongoDB是目前比较流行的一个基于分布式文件存储的数据库，它是一个介于关系数据库和非关系数据库(NoSQL)之间的产品，是非关系数据库当中功能最丰富，最像关系数据库的。
<h2 id="autoid-0-0-0">mongoDB介绍</h2>
mongoDB是目前比较流行的一个基于分布式文件存储的数据库，它是一个介于关系数据库和非关系数据库(NoSQL)之间的产品，是非关系数据库当中功能最丰富，最像关系数据库的。
mongoDB中将一条数据存储为一个文档（document），数据结构由键值（key-value）对组成。其中文档类似于我们平常编程中用到的JSON对象。文档中的字段值可以包含其他文档，数组及文档数组。
<h3 id="autoid-0-1-0">mongoDB相关概念</h3>
mongoDB中相关概念与我们熟悉的SQL概念对比如下：
<table>
<thead>
<tr><th>MongoDB术语/概念</th><th>说明</th><th>对比SQL术语/概念</th></tr>
</thead>
<tbody>
<tr>
<td>database</td>
<td>数据库</td>
<td>database</td>
</tr>
<tr>
<td>collection</td>
<td>集合</td>
<td>table</td>
</tr>
<tr>
<td>document</td>
<td>文档</td>
<td>row</td>
</tr>
<tr>
<td>field</td>
<td>字段</td>
<td>column</td>
</tr>
<tr>
<td>index</td>
<td>index</td>
<td>索引</td>
</tr>
<tr>
<td>primary key</td>
<td>主键 MongoDB自动将_id字段设置为主键</td>
<td>primary key</td>
</tr>
</tbody>
</table>
<h2 id="autoid-0-1-1">mongoDB安装</h2>
我们这里下载和安装社区版，官网下载地址。打开上述连接后，选择对应的版本、操作系统平台（常见的平台均支持）和包类型，点击Download按钮下载即可。
这里补充说明下，Windows平台有<code>ZIP</code>和<code>MSI</code>两种包类型: * ZIP：压缩文件版本 * MSI：可执行文件版本，点击”下一步”安装即可。
macOS平台除了在该网页下载<code>TGZ</code>文件外，还可以使用<code>Homebrew</code>安装。
更多安装细节可以参考官方安装教程，里面有<code>Linux</code>、<code>macOS</code>和<code>Windows</code>三大主流平台的安装教程。
<h2 id="autoid-0-2-0">mongoDB基本使用</h2>
<h3 id="autoid-0-3-0">启动mongoDB数据库</h3>
<h4 id="windows">Windows</h4>
<pre class="line-numbers language-bash"><code class=" language-bash">"C:\Program Files\MongoDB\Server\4.2\bin\mongod.exe" --dbpath="c:\data\db"
</code></pre>
<h4 id="mac">Mac</h4>
<pre class="line-numbers language-bash"><code class=" language-bash">mongod --config /usr/local/etc/mongod.conf
</code></pre>
或
<pre class="line-numbers language-bash"><code class=" language-bash">brew services start mongodb-community@4.2
</code></pre>
<h3 id="autoid-0-3-1">启动client</h3>
<h4 id="windows-1">Windows</h4>
<pre class="line-numbers language-bash"><code class=" language-bash">"C:\Program Files\MongoDB\Server\4.2\bin\mongo.exe"
</code></pre>
<h4 id="mac-1">Mac</h4>
<pre class="line-numbers language-bash"><code class=" language-bash">mongo
</code></pre>
<h3 id="autoid-0-3-2">数据库常用命令</h3>
<code>show dbs;</code>：查看数据库
<pre class="line-numbers language-bash"><code class=" language-bash">> show dbs;
admin 0.000GB
config0.000GB
local 0.000GB
test 0.000GB
</code></pre>
<code>use q1mi;</code>：切换到指定数据库，如果不存在该数据库就创建。
<pre class="line-numbers language-bash"><code class=" language-bash">> use q1mi;
switched to db q1mi
</code></pre>
<code>db;</code>：显示当前所在数据库。
<pre class="line-numbers language-bash"><code class=" language-bash">> db;
q1mi
</code></pre>
<code>db.dropDatabase()</code>：删除当前数据库
<pre class="line-numbers language-bash"><code class=" language-bash">> db.dropDatabase();
{ "ok" : 1 }
</code></pre>
<h3 id="autoid-0-3-3">数据集常用命令</h3>
<code>db.createCollection(name,options)</code>：创建数据集
<ul>
<li>name：数据集名称</li>
<li>options：可选参数，指定内存大小和索引。</li>
</ul>
<pre class="line-numbers language-bash"><code class=" language-bash">> db.createCollection("student");
{ "ok" : 1 }
</code></pre>
<code>show collections;</code>：查看当前数据库中所有集合。
<pre class="line-numbers language-bash"><code class=" language-bash">> show collections;
student
</code></pre>
<code>db.student.drop()</code>：删除指定数据集
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.drop()
true
</code></pre>
<h3 id="autoid-0-3-4">文档常用命令</h3>
插入一条文档：
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.insertOne({name:"小王子",age:18});
{
"acknowledged" : true,
"insertedId" : ObjectId("5db149e904b33457f8c02509")
}
</code></pre>
插入多条文档：
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.insertMany([
... {name:"张三",age:20},
... {name:"李四",age:25}
... ]);
{
"acknowledged" : true,
"insertedIds" : [
ObjectId("5db14c4704b33457f8c0250a"),
ObjectId("5db14c4704b33457f8c0250b")
]
}
</code></pre>
查询所有文档：
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.find();
{ "_id" : ObjectId("5db149e904b33457f8c02509"), "name" : "小王子", "age" : 18 }
{ "_id" : ObjectId("5db14c4704b33457f8c0250a"), "name" : "张三", "age" : 20 }
{ "_id" : ObjectId("5db14c4704b33457f8c0250b"), "name" : "李四", "age" : 25 }
</code></pre>
查询age>20岁的文档：
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.find(
... {age:{$gt:20}}
... )
{ "_id" : ObjectId("5db14c4704b33457f8c0250b"), "name" : "李四", "age" : 25 }
</code></pre>
更新文档：
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.update(
... {name:"小王子"},
... {name:"老王子",age:98}
... );
WriteResult({ "nMatched" : 1, "nUpserted" : 0, "nModified" : 1 })
> db.student.find()
{ "_id" : ObjectId("5db149e904b33457f8c02509"), "name" : "老王子", "age" : 98 }
{ "_id" : ObjectId("5db14c4704b33457f8c0250a"), "name" : "张三", "age" : 20 }
{ "_id" : ObjectId("5db14c4704b33457f8c0250b"), "name" : "李四", "age" : 25 }
</code></pre>
删除文档：
<pre class="line-numbers language-bash"><code class=" language-bash">> db.student.deleteOne({name:"李四"});
{ "acknowledged" : true, "deletedCount" : 1 }
> db.student.find()
{ "_id" : ObjectId("5db149e904b33457f8c02509"), "name" : "老王子", "age" : 98 }
{ "_id" : ObjectId("5db14c4704b33457f8c0250a"), "name" : "张三", "age" : 20 }
</code></pre>
命令实在太多，更多命令请参阅官方文档：shell命令和官方文档：CRUD操作。
<h2 id="autoid-0-3-5">Go语言操作mongoDB</h2>
我们这里使用的是官方的驱动包，当然你也可以使用第三方的驱动包（如mgo等）。 mongoDB官方版的Go驱动发布的比较晚（2018年12月13号）。
<h3 id="autoid-0-4-0">安装mongoDB Go驱动包</h3>
<pre class="line-numbers language-bash"><code class=" language-bash">go get github.com/mongodb/mongo-go-driver
</code></pre>
<h3 id="autoid-0-4-1">通过Go代码连接mongoDB</h3>
<pre class="line-numbers language-go"><code class=" language-go">package main

import (
"context"
"fmt"
"log"

"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)

func main() {
// 设置客户端连接配置
clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

// 连接到MongoDB
client, err := mongo.Connect(context.TODO(), clientOptions)
if err != nil {
log.Fatal(err)
}

// 检查连接
err = client.Ping(context.TODO(), nil)
if err != nil {
log.Fatal(err)
}
fmt.Println("Connected to MongoDB!")
}
</code></pre>
连接上MongoDB之后，可以通过下面的语句处理我们上面的q1mi数据库中的student数据集了：
<pre class="line-numbers language-go"><code class=" language-go">// 指定获取要操作的数据集
collection := client.Database("q1mi").Collection("student")
</code></pre>
处理完任务之后可以通过下面的命令断开与MongoDB的连接：
<pre class="line-numbers language-go"><code class=" language-go">// 断开连接
err = client.Disconnect(context.TODO())
if err != nil {
log.Fatal(err)
}
fmt.Println("Connection to MongoDB closed.")
</code></pre>
<h3 id="autoid-0-4-2">连接池模式</h3>
<pre class="line-numbers language-go"><code class=" language-go">import (
"context"
"time"

"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)

func ConnectToDB(uri, name string, timeout time.Duration, num uint64) (*mongo.Database, error) {
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
o := options.Client().ApplyURI(uri)
o.SetMaxPoolSize(num)
client, err := mongo.Connect(ctx, o)
if err != nil {
return nil, err
}

return client.Database(name), nil
}
</code></pre>
<h3 id="autoid-0-4-3">BSON</h3>
MongoDB中的JSON文档存储在名为BSON(二进制编码的JSON)的二进制表示中。与其他将JSON数据存储为简单字符串和数字的数据库不同，BSON编码扩展了JSON表示，使其包含额外的类型，如int、long、date、浮点数和decimal128。这使得应用程序更容易可靠地处理、排序和比较数据。
连接MongoDB的Go驱动程序中有两大类型表示BSON数据：<code>D</code>和<code>Raw</code>。
类型<code>D</code>家族被用来简洁地构建使用本地Go类型的BSON对象。这对于构造传递给MongoDB的命令特别有用。<code>D</code>家族包括四类:
<ul>
<li>D：一个BSON文档。这种类型应该在顺序重要的情况下使用，比如MongoDB命令。</li>
<li>M：一张无序的map。它和D是一样的，只是它不保持顺序。</li>
<li>A：一个BSON数组。</li>
<li>E：D里面的一个元素。</li>
</ul>
要使用BSON，需要先导入下面的包：
<pre class="line-numbers language-go"><code class=" language-go">import "go.mongodb.org/mongo-driver/bson"
</code></pre>
下面是一个使用D类型构建的过滤器文档的例子，它可以用来查找name字段与’张三’或’李四’匹配的文档:
<pre class="line-numbers language-go"><code class=" language-go">bson.D{{
"name",
bson.D{{
"$in",
bson.A{"张三", "李四"},
}},
}}
</code></pre>
<code>Raw</code>类型家族用于验证字节切片。你还可以使用<code>Lookup()</code>从原始类型检索单个元素。如果你不想要将BSON反序列化成另一种类型的开销，那么这是非常有用的。这个教程我们将只使用D类型。
<h3 id="autoid-0-4-4">CRUD</h3>
我们现在Go代码中定义一个<code>Studet</code>类型如下：
<pre class="line-numbers language-go"><code class=" language-go">type Student struct {
Name string
Age int
}
</code></pre>
接下来，创建一些<code>Student</code>类型的值，准备插入到数据库中：
<pre class="line-numbers language-go"><code class=" language-go">s1 := Student{"小红", 12}
s2 := Student{"小兰", 10}
s3 := Student{"小黄", 11}
</code></pre>
<h4 id="插入文档">插入文档</h4>
使用<code>collection.InsertOne()</code>方法插入一条文档记录：
<pre class="line-numbers language-go"><code class=" language-go">insertResult, err := collection.InsertOne(context.TODO(), s1)
if err != nil {
log.Fatal(err)
}

fmt.Println("Inserted a single document: ", insertResult.InsertedID)
</code></pre>
使用<code>collection.InsertMany()</code>方法插入多条文档记录：
<pre class="line-numbers language-go"><code class=" language-go">students := []interface{}{s2, s3}
insertManyResult, err := collection.InsertMany(context.TODO(), students)
if err != nil {
log.Fatal(err)
}
fmt.Println("Inserted multiple documents: ", insertManyResult.InsertedIDs)
</code></pre>
<h4 id="更新文档">更新文档</h4>
<code>updateone()</code>方法允许你更新单个文档。它需要一个筛选器文档来匹配数据库中的文档，并需要一个更新文档来描述更新操作。你可以使用<code>bson.D</code>类型来构建筛选文档和更新文档:
<pre class="line-numbers language-go"><code class=" language-go">filter := bson.D{{"name", "小兰"}}

update := bson.D{
{"$inc", bson.D{
{"age", 1},
}},
}
</code></pre>
接下来，就可以通过下面的语句找到小兰，给他增加一岁了：
<pre class="line-numbers language-go"><code class=" language-go">updateResult, err := collection.UpdateOne(context.TODO(), filter, update)
if err != nil {
log.Fatal(err)
}
fmt.Printf("Matched %v documents and updated %v documents.\n", updateResult.MatchedCount, updateResult.ModifiedCount)
</code></pre>
<h4 id="查找文档">查找文档</h4>
要找到一个文档，你需要一个filter文档，以及一个指向可以将结果解码为其值的指针。要查找单个文档，使用<code>collection.FindOne()</code>。这个方法返回一个可以解码为值的结果。
我们使用上面定义过的那个filter来查找姓名为’小兰’的文档。
<pre class="line-numbers language-go"><code class=" language-go">// 创建一个Student变量用来接收查询的结果
var result Student
err = collection.FindOne(context.TODO(), filter).Decode(&result)
if err != nil {
log.Fatal(err)
}
fmt.Printf("Found a single document: %+v\n", result)
</code></pre>
要查找多个文档，请使用<code>collection.Find()</code>。此方法返回一个游标。游标提供了一个文档流，你可以通过它一次迭代和解码一个文档。当游标用完之后，应该关闭游标。下面的示例将使用<code>options</code>包设置一个限制以便只返回两个文档。
<pre class="line-numbers language-go"><code class=" language-go">// 查询多个
// 将选项传递给Find()
findOptions := options.Find()
findOptions.SetLimit(2)

// 定义一个切片用来存储查询结果
var results []*Student

// 把bson.D{{}}作为一个filter来匹配所有文档
cur, err := collection.Find(context.TODO(), bson.D{{}}, findOptions)
if err != nil {
log.Fatal(err)
}

// 查找多个文档返回一个光标
// 遍历游标允许我们一次解码一个文档
for cur.Next(context.TODO()) {
// 创建一个值，将单个文档解码为该值
var elem Student
err := cur.Decode(&elem)
if err != nil {
log.Fatal(err)
}
results = append(results, &elem)
}

if err := cur.Err(); err != nil {
log.Fatal(err)
}

// 完成后关闭游标
cur.Close(context.TODO())
fmt.Printf("Found multiple documents (array of pointers): %#v\n", results)
</code></pre>
<h4 id="删除文档">删除文档</h4>
最后，可以使用<code>collection.DeleteOne()</code>或<code>collection.DeleteMany()</code>删除文档。如果你传递<code>bson.D{{}}</code>作为过滤器参数，它将匹配数据集中的所有文档。还可以使用<code>collection. drop()</code>删除整个数据集。
<pre class="line-numbers language-go"><code class=" language-go">// 删除名字是小黄的那个
deleteResult1, err := collection.DeleteOne(context.TODO(), bson.D{{"name","小黄"}})
if err != nil {
log.Fatal(err)
}
fmt.Printf("Deleted %v documents in the trainers collection\n", deleteResult1.DeletedCount)
// 删除所有
deleteResult2, err := collection.DeleteMany(context.TODO(), bson.D{{}})
if err != nil {
log.Fatal(err)
}
fmt.Printf("Deleted %v documents in the trainers collection\n", deleteResult2.DeletedCount)
</code></pre>
更多方法请查阅官方文档。
<h2 id="autoid-0-4-5">参考链接</h2>
https://docs.mongodb.com/manual/mongo/
https://www.mongodb.com/blog/post/mongodb-go-driver-tutorial
</div> 
来源：https://www.cnblogs.com/xiao-xue-di/p/14419126.html

頁: [1]

琼殿技术论坛's Archiver

GO学习-(26) Go语言操作mongoDB