首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Go Web编程--应用ORM

Go Web编程--应用ORM

作者头像
KevinYan
发布2020-02-26 14:35:44
9790
发布2020-02-26 14:35:44
举报
文章被收录于专栏:网管叨bi叨网管叨bi叨

上篇文章中我们在使用的开发环境中增加了 MySQL容器,然后介绍了使用 database/sql标准库结合数据库驱动包进行数据库操作的方法。不过它们是相对偏底层的软件包。实际开发经常会使用一些在它的基础上封装的 ORM库。ORM的查询使用起来更简单些,语法更富表达力。这篇文章我们主要探究下面这些内容。

  • gorm的基本用法
  • 如何管理 ORM的使用
  • 如何合理规划项目目录结构

安装gorm包

gorm是一个出色的,对开发人员友好的 Golang ORM 库,其支持的特性包括:

  • 全特性 ORM (几乎包含所有特性)
  • 模型关联 (一对一, 一对多,一对多(反向), 多对多, 多态关联)
  • 钩子 (Before/After Create/Save/Update/Delete/Find)
  • 预加载
  • 事务
  • 复合主键
  • SQL 构造器
  • 自动迁移
  • 日志

使用如下命令进行安装:

go get -u github.com/jinzhu/gorm

将gorm加入项目中

规划数据模型目录结构

我们在项目根目录下新建如下目录:

http_demo
|
└───model
│   └───dao
│   │   init.go
│   └───────table
│           │   user.go

在 Go 中包以目录的形式来组织,所以 model包中存放所有数据模型, dao代表数据访问对象,存放数据库 CRUD方法的封装,其中的 init.go存放 dao包的初始化函数主要是用来在加载包后连接上数据库。table包里放与数据表对应的模型定义(使用 ORM 之前要先定义模型与数据库中的表对应),在示例里我们会定义一个 User模型放在 user.go文件中。

规划完目录后就可以在各部分写相应的代码了,首先来看使用 gorm连接数据库。

连接数据库

我们在 dao包的 init.go中加入包的初始化逻辑进行数据库连接,初始化函数会在 dao包第一次被导入时执行,由于 gorm文档连接数据库的例子太简单,参考价值不大,我们根据项目需要做些简单封装, init.go中的代码如下所示:

package dao

import (
   _ "github.com/go-sql-driver/mysql"
   "github.com/jinzhu/gorm"
   "time"
)

var _DB *gorm.DB

func DB() *gorm.DB {
   return _DB
}

func init() {
   _DB = initDB()
}

func initDB() *gorm.DB {   

   db, err := gorm.Open("mysql", "go_web:go_web@tcp(localhost:33063)/go_web?charset=utf8&parseTime=True&loc=Local")
   if err != nil {
      panic(err)
   }
   db.DB().SetMaxOpenConns(100)
   db.DB().SetMaxIdleConns(10)
   db.DB().SetConnMaxLifetime(time.Second * 300)
   if err = db.DB().Ping(); err != nil {
      panic(err)
   }
   return db
}

代码很简单,大家实操的时候根据自己的 MySQL配置更改代码里面的配置就行了。唯一说明一点的是,如果使用了我们提供的 Docker环境,在连接数据库时 host要改为 database:3306,因为我在容器环境里将 MySQL容器的服务名定义成了 database,在运行了 Goapp容器需要用服务名访问容器网络中的其他容器。关于容器环境的详细配置请大家查看Go Web编程--应用数据库 中的描述。

定义模型

使用模型访问数据库的表之前我们需要先定义对应的模型。我们示例中现在只有一个 users表,接下来我们在 table包中添加 users表的模型定义并放置在 user.go文件中。

package table

import "time"

type User struct {
   Id        int64     `gorm:"column:id;primary_key"`
   UserName  string    `gorm:"column:username"`
   Secret    string    `gorm:"column:secret;type:varchar(1000)"`
   CreatedAt time.Time `gorm:"column:created_at"`
   UpdatedAt time.Time `gorm:"column:updated_at"`
}


func (m *User) TableName() string {
   return "users"
}

模型 CRUD

关于模型的 CRUD,建议将单个模型的 CRUD放在 dao包的单个文件中,这样方便以后代码的管理。这里多说一点,建议不要直接用 controller或者叫 handler包直接访问 dao包,而是在中间加一层 logic包,把逻辑放在这一层。这样对代码的管理、复用性都有帮助。

因为数据库的 CRUD 有很多种操作,本文的目的是帮助大家快速开始使用 gorm所以我就只放简单的 CRUD 做演示了。大家按照这里步骤引入 gorm后用到其他的数据库操作了直接去官方文档里查一查就好。

dao包中新建 user.go用来存放 User模型的操作方法。

package dao

import "example.com/http_demo/model/dao/table"

func CreateUser(user *table.User) (err error) {
    err = DB().Create(user).Error

    return
}

func GetUserById(userId int64) (user *table.User, err error) {
    user = new(table.User)
    err = DB().Where("id = ?", userId).First(user).Error

    return
}

func GetAllUser() (users []*table.User, err error) {
    err = DB().Find(&users).Error
    return
}

func UpdateUserNameById(userName string, userId int64) (err error) {
    user := new(table.User)
    err = DB().Where("id = ?", userId).First(user).Error
    if err != nil {
        return
    }

    user.UserName = userName
    err = DB().Save(user).Error

    return
}

func DeleteUserById(userId int64) (err error) {
    user := new(table.User)
    err = DB().Where("id = ?", userId).First(user).Error
    if err != nil {
        return
    }
    err = DB().Delete(user).Error

    return
}

验证ORM 方法

经过上面几步的设置后我们就可以在项目里使用 gorm访问数据库了,由于我们项目的 main goroutine中运行了 http服务,所以我们使用测试用例对上面 dao包中定义的几个方法进行一下测试。

篇幅原因我就只贴一个 GetAllUsers方法的测试用例了:

func TestGetAllUser(t *testing.T) {
    tests := []struct {
        name      string
        wantErr   bool
    }{
        {
            name: "test",
            wantErr: false,
        },
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            gotUsers, err := GetAllUser()
            if (err != nil) != tt.wantErr {
                t.Errorf("GetAllUser() error = %v, wantErr %v", err, tt.wantErr)
                return
            }
            for _, user := range gotUsers {
                log.Info("user: %v", user)
            }

        })
    }
}

运行测试后,可以看到结果:

INFO user: &{1   2020-02-15 14:14:46 +0800 CST 2020-02-15 06:44:17 +0800 CST}
--- PASS: TestGetAllUsers (0.00s)
    --- PASS: TestGetAllUsers/test (0.00s)
PASS

Process finished with exit code 0

在公众号里回复 gohttp05可以得到完整的测试用例代码,建议这些 CRUD都要写好测试用例进行自测,使用 GoLand可以很容易的生成测试函数和运行测试。

重新规划项目目录

引入 ORM后,我们项目中的代码就比较多了,都放在根目录下的 main包中有点杂乱,所以我们根据各部分的功能和职责对项目目录进行了简单的划分,划分后的目录结构如下:

http_demo
|
└───handler//route handler
|
└───logic//business logic
|
└───middleware
│
└───model
│   └───dao
│   │   init.go
│   └───table
│       │   user.go
└───router// router
|
|   main.go

感觉今天的内容还是挺多的,尤其对于刚入门 Go的同学们一定要把今天的代码下载下来实操一遍才能掌握。gorm提供的功能还是很多的,每个功能在官方文档里都有讲解,我们这里就不做过多介绍了。这篇文章的目的主要是让大家能快速入门,同时把 ORM相关的代码管理和初始化流程做的规范些,这些组织方式完全可以应用到生产级别的项目中的。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-02-16,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 网管叨bi叨 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 安装gorm包
  • 将gorm加入项目中
    • 规划数据模型目录结构
      • 连接数据库
        • 定义模型
          • 模型 CRUD
            • 验证ORM 方法
            • 重新规划项目目录
            相关产品与服务
            云数据库 SQL Server
            腾讯云数据库 SQL Server (TencentDB for SQL Server)是业界最常用的商用数据库之一,对基于 Windows 架构的应用程序具有完美的支持。TencentDB for SQL Server 拥有微软正版授权,可持续为用户提供最新的功能,避免未授权使用软件的风险。具有即开即用、稳定可靠、安全运行、弹性扩缩等特点。
            领券
            问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档