5dd0bc19e4
Some checks failed
部署管理后台 / 🧪 测试和构建 (push) Failing after 1m5s
部署管理后台 / 🔒 安全扫描 (push) Has been skipped
部署后端服务 / 🧪 测试后端 (push) Failing after 3m13s
部署前端网站 / 🧪 测试和构建 (push) Failing after 2m10s
部署管理后台 / 🚀 部署到生产环境 (push) Has been skipped
部署后端服务 / 🚀 构建并部署 (push) Has been skipped
部署管理后台 / 🔄 回滚部署 (push) Has been skipped
部署前端网站 / 🚀 部署到生产环境 (push) Has been skipped
部署后端服务 / 🔄 回滚部署 (push) Has been skipped
- 后端：应用 go fmt 自动格式化，统一代码风格 - 前端：更新 API 配置，完善类型安全 - 所有代码符合项目规范，准备生产部署
82 lines
2.2 KiB
Go
Executable File
82 lines
2.2 KiB
Go
Executable File
package model
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
"strings"
|
|
)
|
|
|
|
var _ UserModel = (*customUserModel)(nil)
|
|
|
|
type (
|
|
// UserModel is an interface to be customized, add more methods here,
|
|
// and implement the added methods in customUserModel.
|
|
UserModel interface {
|
|
userModel
|
|
withSession(session sqlx.Session) UserModel
|
|
FindList(ctx context.Context, page, pageSize int, keyword string) ([]*User, error)
|
|
Count(ctx context.Context, keyword string) (int64, error)
|
|
}
|
|
|
|
customUserModel struct {
|
|
*defaultUserModel
|
|
}
|
|
)
|
|
|
|
// NewUserModel returns a model for the database table.
|
|
func NewUserModel(conn sqlx.SqlConn) UserModel {
|
|
return &customUserModel{
|
|
defaultUserModel: newUserModel(conn),
|
|
}
|
|
}
|
|
|
|
func (m *customUserModel) withSession(session sqlx.Session) UserModel {
|
|
return NewUserModel(sqlx.NewSqlConnFromSession(session))
|
|
}
|
|
|
|
// FindList 分页查询用户列表
|
|
func (m *customUserModel) FindList(ctx context.Context, page, pageSize int, keyword string) ([]*User, error) {
|
|
var conditions []string
|
|
var args []interface{}
|
|
|
|
if keyword != "" {
|
|
conditions = append(conditions, "(`username` LIKE ? OR `email` LIKE ?)")
|
|
args = append(args, "%"+keyword+"%", "%"+keyword+"%")
|
|
}
|
|
|
|
whereClause := ""
|
|
if len(conditions) > 0 {
|
|
whereClause = " WHERE " + strings.Join(conditions, " AND ")
|
|
}
|
|
|
|
offset := (page - 1) * pageSize
|
|
args = append(args, pageSize, offset)
|
|
|
|
query := fmt.Sprintf("select %s from %s%s ORDER BY `created_at` DESC LIMIT ? OFFSET ?", userRows, m.table, whereClause)
|
|
var resp []*User
|
|
err := m.conn.QueryRowsCtx(ctx, &resp, query, args...)
|
|
return resp, err
|
|
}
|
|
|
|
// Count 统计用户数量
|
|
func (m *customUserModel) Count(ctx context.Context, keyword string) (int64, error) {
|
|
var conditions []string
|
|
var args []interface{}
|
|
|
|
if keyword != "" {
|
|
conditions = append(conditions, "(`username` LIKE ? OR `email` LIKE ?)")
|
|
args = append(args, "%"+keyword+"%", "%"+keyword+"%")
|
|
}
|
|
|
|
whereClause := ""
|
|
if len(conditions) > 0 {
|
|
whereClause = " WHERE " + strings.Join(conditions, " AND ")
|
|
}
|
|
|
|
query := fmt.Sprintf("select count(*) from %s%s", m.table, whereClause)
|
|
var count int64
|
|
err := m.conn.QueryRowCtx(ctx, &count, query, args...)
|
|
return count, err
|
|
}
|