You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
313 lines
7.0 KiB
313 lines
7.0 KiB
package app
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/go-chi/chi/v5"
|
|
"gorm.io/gorm"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type userInfo struct {
|
|
id uint
|
|
Name string
|
|
PhoneNumber string
|
|
Password string
|
|
}
|
|
|
|
func (u *userInfo) Bind(r *Request) error {
|
|
if name, ok := r.Get("name"); ok && len(name) > 0 {
|
|
u.Name = name
|
|
} else {
|
|
return NewError(1, "缺少用户名称")
|
|
}
|
|
|
|
if phoneNumber, ok := r.Get("phone_number"); ok && len(phoneNumber) > 0 {
|
|
if len(phoneNumber) != 11 {
|
|
return NewError(2, "手机号码格式错误")
|
|
}
|
|
u.PhoneNumber = phoneNumber
|
|
} else {
|
|
return NewError(2, "缺少手机号码")
|
|
}
|
|
|
|
if password, ok := r.Get("password"); ok && len(password) > 0 {
|
|
if len(password) < 6 {
|
|
return NewError(2, "密码太短")
|
|
}
|
|
u.Password = password
|
|
} else {
|
|
return NewError(2, "缺少密码")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func userInfoFromRequest(r *Request) (*User, error) {
|
|
var user User
|
|
|
|
if name, ok := r.Get("name"); ok && len(name) > 0 {
|
|
user.Name = name
|
|
} else {
|
|
return nil, NewError(1, "缺少用户名称")
|
|
}
|
|
|
|
if phoneNumber, ok := r.Get("phone_number"); ok && len(phoneNumber) > 0 {
|
|
if len(phoneNumber) != 11 {
|
|
return nil, NewError(2, "手机号码格式错误")
|
|
}
|
|
user.PhoneNumber = phoneNumber
|
|
} else {
|
|
return nil, NewError(2, "缺少手机号码")
|
|
}
|
|
|
|
if password, ok := r.Get("password"); ok && len(password) > 0 {
|
|
if len(password) < 6 {
|
|
return nil, NewError(2, "密码太短")
|
|
}
|
|
user.Password = password
|
|
} else {
|
|
return nil, NewError(2, "缺少密码")
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
// CreateUser 创建用户
|
|
func CreateUser(w *ResponseWriter, r *Request) {
|
|
var user userInfo
|
|
if err := user.Bind(r); err != nil {
|
|
w.Error(err)
|
|
return
|
|
}
|
|
|
|
var count int64
|
|
if err := DB.Model(&User{}).Where("phone_number = ?", user.PhoneNumber).Count(&count).Error; err != nil {
|
|
w.Error(err)
|
|
return
|
|
}
|
|
if count > 0 {
|
|
w.Error(NewError(2, "手机号码已经被使用了"))
|
|
return
|
|
}
|
|
|
|
if err := DB.Create(&user).Error; err != nil {
|
|
w.Error(err)
|
|
} else {
|
|
w.Ok(user, "创建用户成功")
|
|
}
|
|
}
|
|
|
|
// UpdateUser 更新用户
|
|
func UpdateUser(w *ResponseWriter, r *Request) {
|
|
id := chi.URLParam(r.Request, "id")
|
|
if len(id) == 0 {
|
|
w.Error(NewError(1, "缺少用户ID"))
|
|
}
|
|
user, err := userInfoFromRequest(r)
|
|
if err != nil {
|
|
w.Error(err)
|
|
return
|
|
}
|
|
|
|
// 查询用户信息
|
|
var u User
|
|
err = DB.First(&u, "id = ?", id).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
err = NewError(1, "用户不存在")
|
|
}
|
|
w.Error(err)
|
|
return
|
|
}
|
|
|
|
// 用户信息未发生变化
|
|
if u.Name == user.Name && u.PhoneNumber == u.PhoneNumber && u.Password == user.Password {
|
|
w.Ok(nil, "操作成功")
|
|
return
|
|
}
|
|
|
|
// 检查手机号码是否被使用
|
|
var count int64
|
|
err = DB.Model(&User{}).Where("phone_number = ? AND id != ?", user.PhoneNumber, id).Count(&count).Error
|
|
if err != nil {
|
|
w.Error(err)
|
|
return
|
|
}
|
|
if count > 0 {
|
|
w.Error(NewError(1, "手机号码已被使用"))
|
|
return
|
|
}
|
|
|
|
// 保存用户信息
|
|
u.Name = user.Name
|
|
u.PhoneNumber = user.PhoneNumber
|
|
u.Password = user.Password
|
|
if err := DB.Save(&u).Error; err != nil {
|
|
w.Error(err)
|
|
} else {
|
|
w.Ok(nil, "操作成功")
|
|
}
|
|
}
|
|
|
|
// DeleteUser 删除用户
|
|
func DeleteUser(w *ResponseWriter, r *Request) {
|
|
id := chi.URLParam(r.Request, "id")
|
|
if len(id) == 0 {
|
|
w.Error(NewError(1, "缺少用户ID"))
|
|
}
|
|
|
|
// 查询用户信息
|
|
var u User
|
|
if err := DB.First(&u, "id = ?", id).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
err = NewError(1, "用户不存在")
|
|
}
|
|
w.Error(err)
|
|
return
|
|
}
|
|
|
|
// 删除用户
|
|
if err := DB.Delete(&u).Error; err != nil {
|
|
LogError(err)
|
|
w.Error(NewError(1, "删除用户失败"))
|
|
} else {
|
|
w.Ok(nil, "删除用户成功")
|
|
}
|
|
}
|
|
|
|
// ListUser 用户列表
|
|
func ListUser(w *ResponseWriter, r *Request) {
|
|
|
|
}
|
|
|
|
// Login 用户登录
|
|
func Login(w *ResponseWriter, r *Request) {
|
|
|
|
}
|
|
|
|
// CreateGoods 创建商品
|
|
func CreateGoods(w *ResponseWriter, r *Request) {
|
|
name := r.Value("name")
|
|
price := r.Float32("price", 0)
|
|
if len(name) == 0 {
|
|
w.Fail(http.StatusBadRequest, 1, "商品名称错误")
|
|
return
|
|
}
|
|
if price <= 0 {
|
|
w.Fail(http.StatusBadRequest, 2, "商品价格错误")
|
|
return
|
|
}
|
|
var goods Goods
|
|
err := DB.First(&goods, "name = ?", name).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
goods = Goods{
|
|
Name: name,
|
|
Price: price,
|
|
}
|
|
err = DB.Create(&goods).Error
|
|
if err != nil {
|
|
w.Fail(http.StatusBadRequest, 3, "创建商品失败")
|
|
} else {
|
|
w.Ok(goods)
|
|
}
|
|
} else if err != nil {
|
|
LogError(err)
|
|
w.Fail(http.StatusBadRequest, 4, "商品价格错误")
|
|
} else {
|
|
w.Fail(http.StatusBadRequest, 5, "商品已经存在")
|
|
}
|
|
}
|
|
|
|
// UpdateGoods 更新产品信息
|
|
func UpdateGoods(w *ResponseWriter, r *Request) {
|
|
name := r.Value("name")
|
|
price := r.Float32("price", 0)
|
|
id := uint(r.Uint64("id", 0))
|
|
if len(name) == 0 {
|
|
w.Fail(http.StatusBadRequest, 1, "商品名称错误")
|
|
return
|
|
}
|
|
if price <= 0 {
|
|
w.Fail(http.StatusBadRequest, 2, "商品价格错误")
|
|
return
|
|
}
|
|
var goods Goods
|
|
err := DB.First(&goods, "id = ?", id).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
w.Fail(http.StatusBadRequest, 2, "商品不存在")
|
|
} else if err != nil {
|
|
LogError(err)
|
|
w.Fail(http.StatusBadRequest, 3, err.Error())
|
|
} else {
|
|
// 商品名称不能重复
|
|
err = DB.Where("id <> ?", id).First(&Goods{}, "name = ?", name).Error
|
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
LogError(err)
|
|
w.Fail(http.StatusBadRequest, 4, err.Error())
|
|
return
|
|
}
|
|
if goods.Name != name || goods.Price != price {
|
|
goods.Name = name
|
|
goods.Price = price
|
|
err = DB.Save(&goods).Error
|
|
if err != nil {
|
|
LogError(err)
|
|
w.Fail(http.StatusBadRequest, 5, err.Error())
|
|
return
|
|
}
|
|
}
|
|
w.Ok(goods, "修改成功")
|
|
}
|
|
}
|
|
|
|
// GetGoodsList 查询商品列表
|
|
func GetGoodsList(w *ResponseWriter, r *Request) {
|
|
search := func(db *gorm.DB) *gorm.DB {
|
|
return db.
|
|
Model(&Goods{}).
|
|
Scopes(TimeRange(r, "created_at")).
|
|
Scopes(Paginate(r)).
|
|
Scopes(Search(r, "name", "name LIKE ?"))
|
|
}
|
|
var goodsList []Goods
|
|
var total int64
|
|
var err error
|
|
if err = DB.Scopes(search).Count(&total).Error; err == nil {
|
|
err = DB.Scopes(search).Find(&goodsList).Error
|
|
}
|
|
if err != nil {
|
|
w.Fail(http.StatusInternalServerError, 1, err.Error())
|
|
} else {
|
|
w.Ok(map[string]any{
|
|
"list": goodsList,
|
|
"total": total,
|
|
})
|
|
}
|
|
}
|
|
|
|
func GetGoodsPrices(w *ResponseWriter, r *Request) {
|
|
id := uint(r.Uint64("id", 0))
|
|
var goods Goods
|
|
if err := DB.Scopes(func(db *gorm.DB) *gorm.DB {
|
|
var queries []string
|
|
var args []any
|
|
if val, ok := r.Get("start_time"); ok {
|
|
queries = append(queries, "created_at >= ?")
|
|
args = append(args, val)
|
|
}
|
|
if val, ok := r.Get("end_time"); ok {
|
|
queries = append(queries, "created_at <= ?")
|
|
args = append(args, val)
|
|
}
|
|
if len(queries) == 0 {
|
|
return db.Preload("Prices")
|
|
}
|
|
args = append([]any{strings.Join(queries, " AND ")}, args...)
|
|
return db.Preload("Prices", args...)
|
|
}).First(&goods, "id = ?", id).Error; err != nil {
|
|
w.Fail(http.StatusInternalServerError, 1, err.Error())
|
|
} else {
|
|
w.Ok(goods)
|
|
}
|
|
}
|
|
|