Go的ORM也太拉跨了吧，赶紧给他封装一下

宅哥聊构架后端 2023-01-16

背景

去年慢慢开始接触了Go语言，也在公司写了几个Go的生产项目。我是从Java转过来的。（其实也不算转，公司用啥，我用啥）在这个过程中，老是想用Java的思维写Go，在开始的一两个月，那是边写边吐槽。

丑陋的错误处理，没有流式处理，还竟然没有泛型，框架生态链不成熟，没有一家独大的类似Spring的框架。（其实现在写了快一年的Go，Go还是挺香的，哈哈）

今天，我来聊一下，我在我在写Go过程中用的最多orm框架gorm。

Java的orm

写过Java的基本都知道Mybatis，Mybatis-plus。

在Mybatis-plus中操作单表非常方便，通过QueryWrapper，对于单表的操作非常的丝滑，没有任何的思维负担。

类似下面这样：

        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(User::getUsername,"zhangsan").eq(User::getAge,18);
        userMapper.selectList(queryWrapper);
复制代码

Go的orm

这里的条件查询都需要开发手动来拼接字符串，如果项目比较大的话，就会看到漫天飞的SQL字段，维护起来非常麻烦。

  var users []*User
  sqlResult := db.Where("username = ? and age = ?", "zhangsan", 18).Find(&users)
复制代码

解决方式

写了一段时间的Go之后，实在不想每次都写这些字符串拼接了，于是我给gorm封装了一个gorm-plus。

这里我使用到了go 1.18的泛型。泛型出了这么久，也该使用上了。

其实就是把Mybatis-plus的那套语法借鉴了一下。（好吧，就是抄他的）

Mybatis-plus对于单表操作提供了非常多的CRUD操作。

Go的ORM也太拉跨了吧，赶紧给他封装一下

我给gorm-plus 也提供了类似的操作

Go的ORM也太拉跨了吧，赶紧给他封装一下

下面是具体用法

下载：

go get github.com/acmestack/gorm-plus
复制代码

初始化sql

DROP TABLE IF EXISTS `users`;
CREATE TABLE `users`  (
                          `id` int(0) NOT NULL AUTO_INCREMENT,
                          `username` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
                          `password` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
                          `age` int(0) NULL DEFAULT NULL,
                          `phone` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
                          `score` int(0) NULL DEFAULT NULL,
                          `address` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
                          `dept` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
                          `created_at` datetime(0) NULL DEFAULT NULL,
                          `updated_at` datetime(0) NULL DEFAULT NULL,
                          PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;

SET FOREIGN_KEY_CHECKS = 1;
复制代码

连接数据库

var GormDb *gorm.DB

func init() {
  dsn := "root:123456@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
  var err error
  GormDb, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
    Logger: logger.Default.LogMode(logger.Info),
  })
  if err != nil {
    log.Fatalln(err)
  }
  gplus.Init(GormDb)
}
复制代码

插入语句

  user := &User{Username: "zhangsan", Password: "123456", Age: 18, Score: 100, Dept: "A部门"}
  result := gplus.Insert(user)
  fmt.Println(result.RowsAffected)
复制代码

查询语句

根据id查询：

注意这里需要传入泛型User

  user, resultDb := gplus.SelectById[User](1)
  fmt.Println(user, resultDb.RowsAffected)
复制代码

根据ids查询：

  var ids = []int{1,2}
  users, resultDb := gplus.SelectByIds[User](ids)
  fmt.Println(users, resultDb.RowsAffected)
复制代码

条件查询：

  q := gplus.NewQuery[User]()
  q.Eq("username", "zhangsan").Eq("age",18)
  users, resultDb := gplus.SelectList(q)
  fmt.Println(users,resultDb.RowsAffected)
复制代码

对比一下Mybatis-plus写法基本一致了。

        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(User::getUsername,"zhangsan").eq(User::getAge,18);
        userMapper.selectList(queryWrapper);
复制代码

更多操作请查看：

github.com/acmestack/g…

gplus工具

其实上面的写法还是需要写数据库的字段名，如果数据库的字段名很多，我们很容易写错，导致不必要的bug产生。

一旦名称长，非常容易误写，而且如果有字段名称修改的话，还需要全局搜索一个个地修改，比较麻烦。

Go没有提供类似Java的lambad表达式或者C#中 nameof 方式直接获取某个对象的字段名称的操作，但是我们可以通过生成代码的方式生成字段名。

所有就有了gplus，它作用就是自动识别结构体，把结构体的字段名生成出来。

下载使用：

go install github.com/acmestack/gorm-plus/cmd/gplus@latest
复制代码

通过 gplus gen paths=路径，gplus 会自动识别带有// +gplus:column=true注释的结构体，给这个结构体生成字段。

gplus 会在输入的路径下面生成 zz_gen.column.go文件。

例如：

在example目录下创建了了一个users.go 目录，执行 gplus gen paths=./eample

users.go

// +gplus:column=true

type User struct {
  ID        int64
  Username  string `gorm:"column:username"`
  Password  string
  Address   string
  Age       int
  Phone     string
  Score     int
  Dept      string
  CreatedAt time.Time
  UpdatedAt time.Time
}
复制代码

zz_gen.column.go （自动生成的）

var UserColumn = struct {
  ID        string
  Username  string
  Password  string
  Address   string
  Age       string
  Phone     string
  Score     string
  Dept      string
  CreatedAt string
  UpdatedAt string
}{
  ID:        "id",
  Username:  "username",
  Password:  "password",
  Address:   "address",
  Age:       "age",
  Phone:     "phone",
  Score:     "score",
  Dept:      "dept",
  CreatedAt: "created_at",
  UpdatedAt: "updated_at",
}
复制代码

其实你自己也可以手写这个文件，只不过通过工具生成更加方便而已。

有了这个文件，我们的查询就变成这样：

  q := gplus.NewQuery[User]()
  q.Eq(UserColumn.Username, "zhangsan").Eq(UserColumn.Age,18)
  users, resultDb := gplus.SelectList(q)
  fmt.Println(users,resultDb.RowsAffected)
复制代码

这样维护起来就非常方便了。