model命令

goctl model 为go-zero下的工具模块中的组件之一，目前支持识别mysql ddl进行model层代码生成，通过命令行或者idea插件（即将支持）可以有选择地生成带redis cache或者不带redis cache的代码逻辑。

快速开始

  • 通过ddl生成

    1.   $ goctl model mysql ddl -src="./*.sql" -dir="./sql/model" -c

    执行上述命令后即可快速生成CURD代码。

    1.   model
    2.   │   ├── error.go
    3.   │   └── usermodel.go

  • 通过datasource生成

    1.   $ goctl model mysql datasource -url="user:password@tcp(127.0.0.1:3306)/database" -table="*"  -dir="./model"

  • 生成代码示例

    1.   package model
    3.   import (
    4.       "database/sql"
    5.       "fmt"
    6.       "strings"
    7.       "time"
    9.       "github.com/tal-tech/go-zero/core/stores/cache"
    10.       "github.com/tal-tech/go-zero/core/stores/sqlc"
    11.       "github.com/tal-tech/go-zero/core/stores/sqlx"
    12.       "github.com/tal-tech/go-zero/core/stringx"
    13.       "github.com/tal-tech/go-zero/tools/goctl/model/sql/builderx"
    14.   )
    16.   var (
    17.       userFieldNames          = builderx.RawFieldNames(&User{})
    18.       userRows                = strings.Join(userFieldNames, ",")
    19.       userRowsExpectAutoSet   = strings.Join(stringx.Remove(userFieldNames, "`id`", "`create_time`", "`update_time`"), ",")
    20.       userRowsWithPlaceHolder = strings.Join(stringx.Remove(userFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?"
    22.       cacheUserNamePrefix   = "cache#User#name#"
    23.       cacheUserMobilePrefix = "cache#User#mobile#"
    24.       cacheUserIdPrefix     = "cache#User#id#"
    25.       cacheUserPrefix       = "cache#User#user#"
    26.   )
    28.   type (
    29.       UserModel interface {
    30.           Insert(data User) (sql.Result, error)
    31.           FindOne(id int64) (*User, error)
    32.           FindOneByUser(user string) (*User, error)
    33.           FindOneByName(name string) (*User, error)
    34.           FindOneByMobile(mobile string) (*User, error)
    35.           Update(data User) error
    36.           Delete(id int64) error
    37.       }
    39.       defaultUserModel struct {
    40.           sqlc.CachedConn
    41.           table string
    42.       }
    44.       User struct {
    45.           Id         int64     `db:"id"`
    46.           User       string    `db:"user"`     // 用户
    47.           Name       string    `db:"name"`     // 用户名称
    48.           Password   string    `db:"password"` // 用户密码
    49.           Mobile     string    `db:"mobile"`   // 手机号
    50.           Gender     string    `db:"gender"`   // 男｜女｜未公开
    51.           Nickname   string    `db:"nickname"` // 用户昵称
    52.           CreateTime time.Time `db:"create_time"`
    53.           UpdateTime time.Time `db:"update_time"`
    54.       }
    55.   )
    57.   func NewUserModel(conn sqlx.SqlConn, c cache.CacheConf) UserModel {
    58.       return &defaultUserModel{
    59.           CachedConn: sqlc.NewConn(conn, c),
    60.           table:      "`user`",
    61.       }
    62.   }
    64.   func (m *defaultUserModel) Insert(data User) (sql.Result, error) {
    65.       userNameKey := fmt.Sprintf("%s%v", cacheUserNamePrefix, data.Name)
    66.       userMobileKey := fmt.Sprintf("%s%v", cacheUserMobilePrefix, data.Mobile)
    67.       userKey := fmt.Sprintf("%s%v", cacheUserPrefix, data.User)
    68.       ret, err := m.Exec(func(conn sqlx.SqlConn) (result sql.Result, err error) {
    69.           query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?)", m.table, userRowsExpectAutoSet)
    70.           return conn.Exec(query, data.User, data.Name, data.Password, data.Mobile, data.Gender, data.Nickname)
    71.       }, userNameKey, userMobileKey, userKey)
    72.       return ret, err
    73.   }
    75.   func (m *defaultUserModel) FindOne(id int64) (*User, error) {
    76.       userIdKey := fmt.Sprintf("%s%v", cacheUserIdPrefix, id)
    77.       var resp User
    78.       err := m.QueryRow(&resp, userIdKey, func(conn sqlx.SqlConn, v interface{}) error {
    79.           query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", userRows, m.table)
    80.           return conn.QueryRow(v, query, id)
    81.       })
    82.       switch err {
    83.       case nil:
    84.           return &resp, nil
    85.       case sqlc.ErrNotFound:
    86.           return nil, ErrNotFound
    87.       default:
    88.           return nil, err
    89.       }
    90.   }
    92.   func (m *defaultUserModel) FindOneByUser(user string) (*User, error) {
    93.       userKey := fmt.Sprintf("%s%v", cacheUserPrefix, user)
    94.       var resp User
    95.       err := m.QueryRowIndex(&resp, userKey, m.formatPrimary, func(conn sqlx.SqlConn, v interface{}) (i interface{}, e error) {
    96.           query := fmt.Sprintf("select %s from %s where `user` = ? limit 1", userRows, m.table)
    97.           if err := conn.QueryRow(&resp, query, user); err != nil {
    98.               return nil, err
    99.           }
    100.           return resp.Id, nil
    101.       }, m.queryPrimary)
    102.       switch err {
    103.       case nil:
    104.           return &resp, nil
    105.       case sqlc.ErrNotFound:
    106.           return nil, ErrNotFound
    107.       default:
    108.           return nil, err
    109.       }
    110.   }
    112.   func (m *defaultUserModel) FindOneByName(name string) (*User, error) {
    113.       userNameKey := fmt.Sprintf("%s%v", cacheUserNamePrefix, name)
    114.       var resp User
    115.       err := m.QueryRowIndex(&resp, userNameKey, m.formatPrimary, func(conn sqlx.SqlConn, v interface{}) (i interface{}, e error) {
    116.           query := fmt.Sprintf("select %s from %s where `name` = ? limit 1", userRows, m.table)
    117.           if err := conn.QueryRow(&resp, query, name); err != nil {
    118.               return nil, err
    119.           }
    120.           return resp.Id, nil
    121.       }, m.queryPrimary)
    122.       switch err {
    123.       case nil:
    124.           return &resp, nil
    125.       case sqlc.ErrNotFound:
    126.           return nil, ErrNotFound
    127.       default:
    128.           return nil, err
    129.       }
    130.   }
    132.   func (m *defaultUserModel) FindOneByMobile(mobile string) (*User, error) {
    133.       userMobileKey := fmt.Sprintf("%s%v", cacheUserMobilePrefix, mobile)
    134.       var resp User
    135.       err := m.QueryRowIndex(&resp, userMobileKey, m.formatPrimary, func(conn sqlx.SqlConn, v interface{}) (i interface{}, e error) {
    136.           query := fmt.Sprintf("select %s from %s where `mobile` = ? limit 1", userRows, m.table)
    137.           if err := conn.QueryRow(&resp, query, mobile); err != nil {
    138.               return nil, err
    139.           }
    140.           return resp.Id, nil
    141.       }, m.queryPrimary)
    142.       switch err {
    143.       case nil:
    144.           return &resp, nil
    145.       case sqlc.ErrNotFound:
    146.           return nil, ErrNotFound
    147.       default:
    148.           return nil, err
    149.       }
    150.   }
    152.   func (m *defaultUserModel) Update(data User) error {
    153.       userIdKey := fmt.Sprintf("%s%v", cacheUserIdPrefix, data.Id)
    154.       _, err := m.Exec(func(conn sqlx.SqlConn) (result sql.Result, err error) {
    155.           query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, userRowsWithPlaceHolder)
    156.           return conn.Exec(query, data.User, data.Name, data.Password, data.Mobile, data.Gender, data.Nickname, data.Id)
    157.       }, userIdKey)
    158.       return err
    159.   }
    161.   func (m *defaultUserModel) Delete(id int64) error {
    162.       data, err := m.FindOne(id)
    163.       if err != nil {
    164.           return err
    165.       }
    167.       userNameKey := fmt.Sprintf("%s%v", cacheUserNamePrefix, data.Name)
    168.       userMobileKey := fmt.Sprintf("%s%v", cacheUserMobilePrefix, data.Mobile)
    169.       userIdKey := fmt.Sprintf("%s%v", cacheUserIdPrefix, id)
    170.       userKey := fmt.Sprintf("%s%v", cacheUserPrefix, data.User)
    171.       _, err = m.Exec(func(conn sqlx.SqlConn) (result sql.Result, err error) {
    172.           query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
    173.           return conn.Exec(query, id)
    174.       }, userNameKey, userMobileKey, userIdKey, userKey)
    175.       return err
    176.   }
    178.   func (m *defaultUserModel) formatPrimary(primary interface{}) string {
    179.       return fmt.Sprintf("%s%v", cacheUserIdPrefix, primary)
    180.   }
    182.   func (m *defaultUserModel) queryPrimary(conn sqlx.SqlConn, v, primary interface{}) error {
    183.       query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", userRows, m.table)
    184.       return conn.QueryRow(v, query, primary)
    185.   }

用法

  1. $ goctl model mysql -h
  1. NAME:
  2.    goctl model mysql - generate mysql model"
  4. USAGE:
  5.    goctl model mysql command [command options] [arguments...]
  7. COMMANDS:
  8.    ddl         generate mysql model from ddl"
  9.    datasource  generate model from datasource"
  11. OPTIONS:
  12.    --help, -h  show help

生成规则

  • 默认规则

    我们默认用户在建表时会创建createTime、updateTime字段(忽略大小写、下划线命名风格)且默认值均为CURRENT_TIMESTAMP，而updateTime支持ON UPDATE CURRENT_TIMESTAMP，对于这两个字段生成insert、update时会被移除，不在赋值范畴内，当然，如果你不需要这两个字段那也无大碍。

  • 带缓存模式

    • ddl

      1. $ goctl model mysql -src={patterns} -dir={dir} -cache

      help

      1. NAME:
      2.    goctl model mysql ddl - generate mysql model from ddl
      4. USAGE:
      5.    goctl model mysql ddl [command options] [arguments...]
      7. OPTIONS:
      8.    --src value, -s value  the path or path globbing patterns of the ddl
      9.    --dir value, -d value  the target dir
      10.    --style value          the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]
      11.    --cache, -c            generate code with cache [optional]
      12.    --idea                 for idea plugin [optional]

    • datasource

      1. $ goctl model mysql datasource -url={datasource} -table={patterns}  -dir={dir} -cache=true

      help

      1. NAME:
      2.    goctl model mysql datasource - generate model from datasource
      4. USAGE:
      5.    goctl model mysql datasource [command options] [arguments...]
      7. OPTIONS:
      8.    --url value              the data source of database,like "root:password@tcp(127.0.0.1:3306)/database
      9.    --table value, -t value  the table or table globbing patterns in the database
      10.    --cache, -c              generate code with cache [optional]
      11.    --dir value, -d value    the target dir
      12.    --style value            the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]
      13.    --idea                   for idea plugin [optional]

      [!TIP] goctl model mysql ddl/datasource 均新增了一个--style参数，用于标记文件命名风格。

    目前仅支持redis缓存，如果选择带缓存模式，即生成的FindOne(ByXxx)&Delete代码会生成带缓存逻辑的代码，目前仅支持单索引字段（除全文索引外），对于联合索引我们默认认为不需要带缓存，且不属于通用型代码，因此没有放在代码生成行列，如example中user表中的id、name、mobile字段均属于单字段索引。

  • 不带缓存模式

    • ddl

      1.   $  goctl model -src={patterns} -dir={dir}

    • datasource

      1.   $  goctl model mysql datasource -url={datasource} -table={patterns}  -dir={dir}

    or

    • ddl

      1.   $  goctl model -src={patterns} -dir={dir}

    • datasource

      1.   $  goctl model mysql datasource -url={datasource} -table={patterns}  -dir={dir}

生成代码仅基本的CURD结构。

缓存

对于缓存这一块我选择用一问一答的形式进行罗列。我想这样能够更清晰的描述model中缓存的功能。

  • 缓存会缓存哪些信息？

    对于主键字段缓存，会缓存整个结构体信息，而对于单索引字段（除全文索引）则缓存主键字段值。

  • 数据有更新（update）操作会清空缓存吗？

    会，但仅清空主键缓存的信息，why？这里就不做详细赘述了。

  • 为什么不按照单索引字段生成updateByXxx和deleteByXxx的代码？

    理论上是没任何问题，但是我们认为，对于model层的数据操作均是以整个结构体为单位，包括查询，我不建议只查询某部分字段（不反对），否则我们的缓存就没有意义了。

  • 为什么不支持findPageLimit、findAll这么模式代码生层？

    目前，我认为除了基本的CURD外，其他的代码均属于业务型代码，这个我觉得开发人员根据业务需要进行编写更好。

类型转换规则

mysql dataType golang dataType golang dataType(if null&&default null)
bool int64 sql.NullInt64
boolean int64 sql.NullInt64
tinyint int64 sql.NullInt64
smallint int64 sql.NullInt64
mediumint int64 sql.NullInt64
int int64 sql.NullInt64
integer int64 sql.NullInt64
bigint int64 sql.NullInt64
float float64 sql.NullFloat64
double float64 sql.NullFloat64
decimal float64 sql.NullFloat64
date time.Time sql.NullTime
datetime time.Time sql.NullTime
timestamp time.Time sql.NullTime
time string sql.NullString
year time.Time sql.NullInt64
char string sql.NullString
varchar string sql.NullString
binary string sql.NullString
varbinary string sql.NullString
tinytext string sql.NullString
text string sql.NullString
mediumtext string sql.NullString
longtext string sql.NullString
enum string sql.NullString
set string sql.NullString
json string sql.NullString