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

环境声明:
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

go go Go

2 声望
0 粉丝
0 条评论
推荐阅读
一文搞懂秒杀系统,欢迎参与开源,提交PR,提高竞争力。早日上岸,升职加薪。
前言秒杀和高并发是面试的高频考点,也是我们做电商项目必知必会的场景。欢迎大家参与我们的开源项目,提交PR,提高竞争力。早日上岸,升职加薪。知识点详解秒杀系统架构图秒杀流程图秒杀系统设计这篇文章一万多...

王中阳Go32阅读 2.4k评论 1

封面图
前端如何入门 Go 语言
类比法是一种学习方法,它是通过将新知识与已知知识进行比较,从而加深对新知识的理解。在学习 Go 语言的过程中,我发现,通过类比已有的前端知识,可以更好地理解 Go 语言的特性。

robin22阅读 3.2k评论 4

封面图
Golang 中 []byte 与 string 转换
string 类型和 []byte 类型是我们编程时最常使用到的数据结构。本文将探讨两者之间的转换方式,通过分析它们之间的内在联系来拨开迷雾。

机器铃砍菜刀24阅读 57.8k评论 2

年度最佳【golang】map详解
这篇文章主要讲 map 的赋值、删除、查询、扩容的具体执行过程,仍然是从底层的角度展开。结合源码,看完本文一定会彻底明白 map 底层原理。

去去100216阅读 11.5k评论 2

年度最佳【golang】GMP调度详解
Golang最大的特色可以说是协程(goroutine)了, 协程让本来很复杂的异步编程变得简单, 让程序员不再需要面对回调地狱, 虽然现在引入了协程的语言越来越多, 但go中的协程仍然是实现的是最彻底的. 这篇文章将通过分析...

去去100215阅读 11.9k评论 4

万字详解,吃透 MongoDB!
MongoDB 是一个基于 分布式文件存储 的开源 NoSQL 数据库系统,由 C++ 编写的。MongoDB 提供了 面向文档 的存储方式,操作起来比较简单和容易,支持“无模式”的数据建模,可以存储比较复杂的数据类型,是一款非常...

JavaGuide8阅读 1.6k

封面图
数据结构与算法:二分查找
一、常见数据结构简单数据结构(必须理解和掌握)有序数据结构:栈、队列、链表。有序数据结构省空间(储存空间小)无序数据结构:集合、字典、散列表,无序数据结构省时间(读取时间快)复杂数据结构树、 堆图二...

白鲸鱼9阅读 5.2k

go go Go

2 声望
0 粉丝
宣传栏