大家都知道框架自带的开发工具可以生成do对象代码，该do对象主要用于查询、修改、写入等操作时对操作字段的自动nil过滤。

今天教给大家一个新的玩法，通过指针结合do对象快速实现灵活、便捷的修改操作API实现。

数据结构

以下是我们使用的用户表数据结构：

CREATE TABLE `user`(
    `id`        int(10) unsigned NOT NULL AUTO_INCREMENT COMMENT '用户ID',
    `passport`  varchar(32) NOT NULL COMMENT '账号',
    `password`  varchar(32) NOT NULL COMMENT '密码',
    `nickname`  varchar(32) NOT NULL COMMENT '昵称',
    `status`    varchar(32) NOT NULL COMMENT '状态',
    `brief`     varchar(512) NOT NULL COMMENT '备注信息',
    `create_at` datetime DEFAULT NULL COMMENT '创建时间',
    `update_at` datetime DEFAULT NULL COMMENT '修改时间',
    PRIMARY KEY (`id`),
    UNIQUE KEY `uniq_passport` (`passport`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

其中的用户状态，我们使用了单独的类型定义，用以实现枚举值：

type Status string

const (
	StatusEnabled  = "enabled"
	StatusDisabled = "disabled"
)

通过gf gen dao命令，我们自动生成的do对象如下：

type User struct {
	g.Meta    `table:"uer" orm:"do:true"`
	Id        interface{}
	Passport  interface{}
	Password  interface{}
	Nickname  interface{}
	Status    interface{}
	Brief     interface{}
	CreatedAt interface{}
	UpdatedAt interface{}
}

请求API定义

我们来实现一个用户信息修改的API接口，这是一个运维管理接口，可以通过用户账号名称来修改用户信息。该API的定义如下：

type UpdateReq struct {
	g.Meta   `path:"/user/{Id}" method:"post" summary:"修改用户信息"`
	Passport string  `v:"required" dc:"用户账号"`
	Password *string `dc:"修改用户密码"`
	Nickname *string `dc:"修改用户昵称"`
	Status   *Status `dc:"修改用户状态"`
	Brief    *string `dc:"修改用户描述"`
}

其中，用户的可修改信息为密码、昵称、状态和描述，可能同时修改一项或者多项。这里使用了指针类型的属性参数，用于实现：当传递该参数时执行修改，不传递时不修改。

业务逻辑实现

为了简化实例，我们这里直接在控制器中将指针参数传递给do对象。我们知道当调用端没有传递该参数时，该参数为nil，那么传递给do对象的字段时，仍然是nil，这个时候执行数据库更新操作时，do对象中的nil字段将会被自动过滤掉。

func (c *Controller) Update(ctx context.Context, req *v1.UpdateReq) (res *v1.UpdateRes, err error) {
	_, err = dao.User.Ctx(ctx).Data(do.User{
		Password: req.Password,
		Nickname: req.Nickname,
		Status:   req.Status,
		Brief:    req.Brief,
	}).Where(do.User{
		Passport: req.Passport,
	}).Update()
	return
}












  • No labels