用Go语言借助mgo实现一个对MongoDB进行增删改查的demo

Jingbin_ 发布于2019-06-26 17:52 / 3659人阅读

摘要：准备数据结构这个以对一个简单的通讯录进行插入查询更新删除记录为例，中包含和两个字段下载还没出官方的驱动包，是现在比较流行的第三方包，能找到的相关资料也比较多。准备实现对的增删改查操作，先想了下怎么写测试案例。查找，通过查找该条记录。

环境声明：
go version go1.11
MongoDB server version 4.0.3

背景

这是我第一次接触golang和MongoDB，正在在参照他们的官方手册学习。想试着用“测试驱动开发”(TDD)的模式来做这个demo。

写这个demo的目的是为了让自己开始写一些golang的代码，和练习一下MongoDB简单的CRUD操作。

准备 数据结构

这个demo以对一个简单的通讯录进行插入、查询、更新、删除记录为例，collection中包含name和phone两个字段：

{ name : "Jack_Green", phone : "9987650" }

下载mgo

MongoDB还没出官方的golang驱动包，mgo是现在比较流行的第三方包，能找到的相关资料也比较多。

go get gopkg.in/mgo.v2

开始

参照MongoDB官方手册在本地搭了一套环境，练习了在mongo shell中使用命令的CRUD操作，熟悉了一下这个数据库。

准备实现对MongoDB的增、删、改、查操作，先想了下怎么写测试案例。测试用了比较简单的方法来实现，而且没有考虑可能出现的全部情况，而是只写了各个操作的某一种情况。

编写测试案例

文件名 gomd_test.go

可以自定义不同的db、collection名称，且测试时方便使对象统一修改，声明两个struct：

type Person struct {
    Name  string
    Phone string
}
type Collection struct {
    DB string 
    Name string
}

在测试代码文件中声明两个变量：

var col = Collection{"testDB", "contacts"}
var p = Person{"Jack_Green", "9987650"}

“C” 插入，测试插入方法，往通讯录中添加记录。不过要首先确保查询方法是可用的，我自己刚开始写的时候是先实现了查询方法，在插入之后，用查询方法把数据取出来打印。

// C create/insert
func TestInsert(t *testing.T) {
    insert(p, col)
    pb := findByName(p.Name, col)
    if pb.Name != p.Name || pb.Phone != p.Phone {
        t.Error("insert failed")
    }
    fmt.Println("Insert Result")
    fmt.Println(pb)
}

“R” 查找，通过name查找该条记录。

// R read/find
func TestFindByName(t *testing.T) {
    p := findByName(p.Name, col)
    if p.Name == "" || p.Phone == "" {
        t.Error("find by name test failed")
    }
    fmt.Println("Find Result")
    fmt.Println(p)
}

“U” 更新，修改通讯录中某位的手机号，根据name修改phone

// U update
func TestUpdate(t *testing.T) {
    p := Person{"WEW", "121212122"}
    update(p, col)
    ub := findByName(p.Name, col)
    fmt.Println("Update Result")
    fmt.Println(ub)
}

“D” 删除，根据name删除已存在的某条记录

// D delete
func TestDeleteData(t *testing.T) {
    deleteData(p, col)
}

golang代码实现对MongoDB的CRUD

文件名 godm.go

刚刚开始写golang，对于创建session这部分有很多的冗余，等对它和mgo有了更深的了解之后，再来优化下。

导入必要的包

import (
    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
    "log"
)

数据库地址

const (
    URL = "127.0.0.1:27017"
)

往通讯录中添加一条记录

func insert(p Person, col Collection) {
    session, err := mgo.Dial(URL)
    if err != nil {
        panic(err)
    }
    defer session.Close()
    c := session.DB(col.DB).C(col.Name)
    err = c.Insert(p)
    if err != nil {
        log.Fatal(err)
    }
}

根据姓名从通讯录中查找联系方式

func findByName(name string, col Collection) Person {
    session, err := mgo.Dial(URL)
    if err != nil {
        panic(err)
    }
    defer session.Close()
    session.SetMode(mgo.Monotonic, true)
    collection := session.DB(col.DB).C(col.Name)
    result := Person{}
    err = collection.Find(bson.M{"name": name}).One(&result)
    if err != nil {
        log.Fatal(err)
    }
    return result
}

更新通讯录中某人的联系方式

func update(p Person, col Collection) {
    session, err := mgo.Dial(URL)
    if err != nil {
        panic(err)
    }
    defer session.Close()
    collection := session.DB(col.DB).C(col.Name)
    err = collection.Update(bson.M{"name": p.Name}, bson.M{"$set": bson.M{"phone": p.Phone}})
    if err != nil {
        log.Fatal(err)
    }
}

将某人的信息从通讯录中删除

func deleteData(p Person, col Collection) {
    session, err := mgo.Dial(URL)
    if err != nil {
        panic(err)
    }
    defer session.Close()
    collection := session.DB(col.DB).C(col.Name)
    err = collection.Remove(bson.M{"name": p.Name})
}

测试结果

运行：

go test -v

输出：

=== RUN   TestInsert
Insert Result
{Jack_Green 9987650}
--- PASS: TestInsert (0.03s)
=== RUN   TestFindByName
Find Result
{Jack_Green 9987650}
--- PASS: TestFindByName (0.01s)
=== RUN   TestUpdate
Update Result
{Jack_Green 121212122}
--- PASS: TestUpdate (0.02s)
=== RUN   TestDeleteData
--- PASS: TestDeleteData (0.01s)
PASS
ok      gomd    0.240s

附录 完整代码

Golang operation MongoDB demo : gomd

参考资料

Golang连接MongoDB数据库

How To Build Microservice With MongoDB In Golang

godoc : labix mgo v2

文章版权归作者所有，未经允许请勿转载,若此文章存在违规行为，您可以联系管理员删除。

转载请注明本文地址：https://www.ucloud.cn/yun/19408.html

Go与MongoDB

摘要：没有类这一概念，对于变量的捆绑都是由键值对形式的结构体实现的。在做记录的查询时，需要新建一个同类型的结构体或结构体数组，然后执行操作，如果参数为空，则返回全部记录如果你熟悉的操作，可以看出，这样的操作与直接在内操作几乎相同。 go没有类这一概念，对于变量的捆绑都是由键值对形式的结构体（struct）实现的。这对同样是以键值对形式存储的mongodb以及json都非常友好。所以在go中，...

Baaaan 2019-06-26 17:37 评论0 收藏0
利用mongo的findAndModify原子性操作实现auto increment ID

摘要：实际情况默认情况下，使用自动生成，而且在自带的命令里，无法指定一个自增字段。自增字段在多线程时必须是原子性的，这在大数据情况下很难实现伸缩性。而且，在里面，有一个命令是原子性的。代码如下无法打开创建个模拟多线程环境输出查询结果实际情况默认情况下，mongo使用_id自动生成uniq id，而且在mongo自带的命令里，无法指定一个自增字段。自增字段在多线程时必须是原子性的，这在大数...

genedna 2019-06-26 17:00 评论0 收藏0
golang基础学习-MongoDB使用

摘要：如果删除所有用替换其他写了一个和结合的数据查询服务细节可点击链接到查看 1.系统环境 Golang:go version go1.10.3 darwin/amd64 OS:MacOS MongoDB: version: 3.4.4 2.Golang使用MongoDB 使用：gopkg.in/mgo.v2 获取包：go get gopkg.in/mgo.v2 引入：impor...

LinkedME2016 2019-06-26 17:49 评论0 收藏0
使用express+mongoose对mongodb实现增删改查操作

摘要：项目地址写在开头本文主要分享我如何使用对实现增删改查操作，感谢社区所有精品文章的帮助，以及的开源项目对我的启发。我们这个项目是建立一个班级学生管理系统，能够对学生的姓名及学号进行增删改查的操作。项目地址：https://github.com/jrainlau/mongoose_cru... 写在开头本文主要分享我如何使用express+mongoose对mongodb实现增删改查...

Markxu 2019-06-26 17:01 评论0 收藏0
mongodb操作基本语法（增删改查）

摘要：启动启动服务输入命令进入操作终端，需要新开窗口基础语法查看当前所有的数据库查看当前所有的数据库新建数据库创建的数据库这个时候是看不到的，因为数据库里没有任何东西。查找集合所有信息班创建一个集合，并插入一条数据，数据里是带有对象的。 1、启动mongodb mongod -f /usr/local/etc/mongod.conf //启动服务 mongo //输入命令进入mongo操作...

DirtyMind 2019-06-26 17:19 评论0 收藏0