mirror of
https://github.com/zeromicro/go-zero.git
synced 2026-05-09 16:10:00 +08:00
105 lines
3.0 KiB
Go
Executable File
105 lines
3.0 KiB
Go
Executable File
package model
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/tal-tech/go-zero/core/stores/cache"
|
|
"github.com/tal-tech/go-zero/core/stores/sqlc"
|
|
"github.com/tal-tech/go-zero/core/stores/sqlx"
|
|
"github.com/tal-tech/go-zero/core/stringx"
|
|
"github.com/tal-tech/go-zero/tools/goctl/model/sql/builderx"
|
|
)
|
|
|
|
var (
|
|
userCamelFieldNames = builderx.FieldNames(&UserCamel{})
|
|
userCamelRows = strings.Join(userCamelFieldNames, ",")
|
|
userCamelRowsExpectAutoSet = strings.Join(stringx.Remove(userCamelFieldNames, "create_time", "update_time"), ",")
|
|
userCamelRowsWithPlaceHolder = strings.Join(stringx.Remove(userCamelFieldNames, "id", "create_time", "update_time"), "=?,") + "=?"
|
|
)
|
|
|
|
type (
|
|
UserCamelModel struct {
|
|
sqlc.CachedConn
|
|
table string
|
|
}
|
|
|
|
UserCamel struct {
|
|
Id int64 `db:"id"`
|
|
Name string `db:"name"` // 用户名称
|
|
Password string `db:"password"` // 用户密码
|
|
Mobile string `db:"mobile"` // 手机号
|
|
Gender string `db:"gender"` // 男｜女｜未公开
|
|
Nickname string `db:"nickname"` // 用户昵称
|
|
CreateTime time.Time `db:"createTime"`
|
|
UpdateTime time.Time `db:"updateTime"`
|
|
}
|
|
)
|
|
|
|
func NewUserCamelModel(conn sqlx.SqlConn, c cache.CacheConf, table string) *UserCamelModel {
|
|
return &UserCamelModel{
|
|
CachedConn: sqlc.NewConn(conn, c),
|
|
table: table,
|
|
}
|
|
}
|
|
|
|
func (m *UserCamelModel) Insert(data UserCamel) error {
|
|
query := `insert into ` + m.table + `(` + userCamelRowsExpectAutoSet + `) value (?, ?, ?, ?, ?, ?)`
|
|
_, err := m.ExecNoCache(query, data.Id, data.Name, data.Password, data.Mobile, data.Gender, data.Nickname)
|
|
return err
|
|
}
|
|
|
|
func (m *UserCamelModel) FindOne(id int64) (*UserCamel, error) {
|
|
query := `select ` + userCamelRows + ` from ` + m.table + ` where id = ? limit 1`
|
|
var resp UserCamel
|
|
err := m.QueryRowNoCache(&resp, query, id)
|
|
switch err {
|
|
case nil:
|
|
return &resp, nil
|
|
case sqlc.ErrNotFound:
|
|
return nil, ErrNotFound
|
|
default:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func (m *UserCamelModel) FindOneByName(name string) (*UserCamel, error) {
|
|
var resp UserCamel
|
|
query := `select ` + userCamelRows + ` from ` + m.table + ` where name limit 1`
|
|
err := m.QueryRowNoCache(&resp, query, name)
|
|
switch err {
|
|
case nil:
|
|
return &resp, nil
|
|
case sqlc.ErrNotFound:
|
|
return nil, ErrNotFound
|
|
default:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func (m *UserCamelModel) FindOneByMobile(mobile string) (*UserCamel, error) {
|
|
var resp UserCamel
|
|
query := `select ` + userCamelRows + ` from ` + m.table + ` where mobile limit 1`
|
|
err := m.QueryRowNoCache(&resp, query, mobile)
|
|
switch err {
|
|
case nil:
|
|
return &resp, nil
|
|
case sqlc.ErrNotFound:
|
|
return nil, ErrNotFound
|
|
default:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func (m *UserCamelModel) Update(data UserCamel) error {
|
|
query := `update ` + m.table + ` set ` + userCamelRowsWithPlaceHolder + ` where id = ?`
|
|
_, err := m.ExecNoCache(query, data.Name, data.Password, data.Mobile, data.Gender, data.Nickname, data.Id)
|
|
return err
|
|
}
|
|
|
|
func (m *UserCamelModel) Delete(id int64) error {
|
|
query := `delete from ` + m.table + ` where id = ?`
|
|
_, err := m.ExecNoCache(query, id)
|
|
return err
|
|
}
|