Compare commits

...

5 Commits

Author SHA1 Message Date
熊二 bf7064f992 fix: 🐛 路由必须在中间件之后注册 2 years ago
熊二 963d4aabf3 refactor: ♻️ 响应体中添加 success 字段 2 years ago
熊二 a5c6cc4045 refactor: ♻️ 手写 gorm.Model 中的字段 2 years ago
熊二 a5e311c337 fix: 🐛 向前端暴露用户角色 2 years ago
熊二 f58abf7425 feat: 初始化管理员账号 2 years ago
  1. 62
      app/db.go
  2. 7
      app/net.go
  3. 8
      main.go

@ -17,43 +17,67 @@ func ConfigGormDB() {
if err != nil {
panic(err)
}
if err = DB.AutoMigrate(&User{}, &Goods{}, &Price{}); err != nil {
if err = DB.AutoMigrate(&User{}, &UserToken{}, &Goods{}, &Price{}); err != nil {
panic(err)
}
hash, err := HashPassword("123456")
if err != nil {
panic(err)
}
admin := User{
ID: 1,
Name: "超级管理员",
PhoneNumber: "18982052224",
Password: hash,
Admin: true,
}
err = DB.Save(&admin).Error
if err != nil {
panic(err)
}
}
// User 用户
type User struct {
gorm.Model
Name string `json:"name"` // 用户名称
PhoneNumber string `json:"phone_number"` // 用户手机
Password string `json:"-"` // 登录密码
Admin bool `json:"-"` // 是不是管理员
ID uint `json:"id" gorm:"primarykey"`
Name string `json:"name"` // 用户名称
PhoneNumber string `json:"phone_number"` // 用户手机
Password string `json:"-"` // 登录密码
Admin bool `json:"admin"` // 是不是管理员
CreatedAt time.Time `json:"-"`
UpdatedAt time.Time `json:"-"`
DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
}
// UserToken 用户令牌
type UserToken struct {
Code string `gorm:"primarykey"` // 主键
UserID uint // 用户ID
User *User // 关联用户
AccessToken string // 授权令牌
RefreshToken string // 刷新令牌
CreatedAt time.Time // 创建时间
Code string `json:"code" gorm:"primarykey"` // 主键
UserID uint `json:"user_id"` // 用户ID
User *User `json:"user,omitempty"` // 关联用户
AccessToken string `json:"access_token"` // 授权令牌
RefreshToken string `json:"refresh_token"` // 刷新令牌
CreatedAt time.Time `json:"-"` // 创建时间
}
// Goods 商品
type Goods struct {
gorm.Model
Name string `json:"name"` // 商品名称
Price float32 `json:"price"` // 商品当前价格
Prices []Price `json:"prices,omitempty"` // 商品价格列表
ID uint `json:"id" gorm:"primarykey"`
Name string `json:"name"` // 商品名称
Price float32 `json:"price"` // 商品当前价格
Prices []Price `json:"prices,omitempty"` // 商品价格列表
CreatedAt time.Time `json:"-"`
UpdatedAt time.Time `json:"-"`
DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
}
// Price 商品价格
type Price struct {
gorm.Model
GoodsID uint `json:"goods_id"` // 管理商品
Price float32 `json:"price"` // 商品价格
ID uint `json:"id" gorm:"primarykey"`
GoodsID uint `json:"goods_id"` // 管理商品
Price float32 `json:"price"` // 商品价格
CreatedAt time.Time `json:"-"`
UpdatedAt time.Time `json:"-"`
DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
}
// Paginate 分页查询作用域

@ -248,13 +248,18 @@ func (w *ResponseWriter) Error(err error) {
if len(message) == 0 {
message = http.StatusText(status)
}
w.Send(status, map[string]any{"code": code, "message": message})
w.Send(status, map[string]any{
"code": code,
"message": message,
"success": code == 0,
})
}
func (w *ResponseWriter) Ok(data any, message ...string) {
info := map[string]any{
"code": 0,
"message": "ok",
"success": true,
"data": data,
}
if len(message) > 0 {

@ -12,7 +12,7 @@ import (
)
func main() {
app.ConfigLogger("debug.log", app.LogWhenMinute)
app.ConfigLogger("./tmp/debug.log", app.LogWhenMinute)
app.ConfigGormDB()
r := chi.NewRouter()
@ -38,9 +38,6 @@ func main() {
// Please see _example/main.go for other more, or read the library code.
r.Use(httprate.LimitByIP(100, 1*time.Minute))
// mounting net/http/pprof
r.Mount("/debug", middleware.Profiler())
// 允许跨域
// see: https://developer.github.com/v3/#cross-origin-resource-sharing
r.Use(cors.Handler(cors.Options{
@ -54,6 +51,9 @@ func main() {
MaxAge: 300, // Maximum value not ignored by any of major browsers
}))
// mounting net/http/pprof
r.Mount("/debug", middleware.Profiler())
app.RegisterRoutes(r)
if err := http.ListenAndServe(":3000", r); err != nil {

Loading…
Cancel
Save