Refactor employee management

This commit is contained in:
Frudrax Cheng
2026-05-28 10:05:59 +08:00
parent f394d3a8bd
commit d3ee215f61
17 changed files with 5391 additions and 174 deletions
+12 -4
View File
@@ -50,7 +50,7 @@ backend-go/
│ ├── employees_controller.go # Employee serials: generate, query, update, revoke, qrcode
│ ├── helper.go # Helper functions (GetCurrentUser, BindJSON, Response)
│ ├── serials_controller.go # Company serials: generate, query, update, revoke, qrcode
│ └── users_controller.go # User management (admin): create, list, update, reset password, delete
│ └── users_controller.go # Employee master data / backend account management
├── database/ # Database connection and migrations
│ └── database.go # GORM init, AutoMigrate
├── docs/ # Swagger documentation (auto-generated)
@@ -70,8 +70,8 @@ backend-go/
│ ├── serials_service.go # Company serials: generate, query, update, revoke, qrcode
│ ├── aftersales_service_test.go # Aftersales unit tests
│ ├── services_test.go # Auth / Serials / Employees / Companies unit tests
│ ├── users_service.go # User CRUD, role management, password reset (admin)
│ └── users_service_test.go # Users unit tests
│ ├── users_service.go # Employee CRUD, role management, password reset (admin)
│ └── users_service_test.go # Employee/user account unit tests
├── tests/ # Integration tests
│ └── main_test.go # End-to-end tests
├── data/ # SQLite data directory
@@ -97,7 +97,15 @@ backend-go/
- **Aftersales** (技术员+管理员): `POST /api/aftersales`, `GET /api/aftersales`, `GET /api/aftersales/:serialNumber`, `PATCH /api/aftersales/:serialNumber`, `POST /api/aftersales/:serialNumber/qrcode`, `POST /api/aftersales/:serialNumber/submit`
- **Aftersales** (仅管理员): `POST /api/aftersales/:serialNumber/reassign`, `POST /api/aftersales/:serialNumber/force-close`, `DELETE /api/aftersales/:serialNumber`
- **Users** (技术员+管理员): `GET /api/users/assignable`
- **Users** (仅管理员): `POST /api/users`, `GET /api/users`, `PATCH /api/users/:id`, `POST /api/users/:id/reset-password`, `DELETE /api/users/:id`
- **Employees** (仅管理员): `POST /api/employees`, `GET /api/employees`, `PATCH /api/employees/:id`, `POST /api/employees/:id/reset-password`, `DELETE /api/employees/:id`
### Roles and permissions
- Roles are limited to `admin`, `technician`, and `employee`.
- `admin` has full backend access.
- `technician` only has aftersales/work-order module access.
- `employee` has no backend login access and does not require a password.
- Creating an employee through `/api/employees` creates employee master data and automatically generates one employee serial bound by `employeeId`.
- `admin` / `technician` creation requires an initial password; `employee` creation must not require one.
### Import Organization
Standard imports followed by third-party imports, then project imports (sorted alphabetically):
+1 -1
View File
@@ -62,7 +62,7 @@ deps:
# 生成 Swagger 文档
swagger:
@echo "生成 Swagger 文档..."
@swag init -g main.go
@go run github.com/swaggo/swag/cmd/swag@v1.16.6 init -g main.go
@echo "Swagger 文档已生成在 docs/ 目录"
# 初始化数据库
+27 -25
View File
@@ -35,7 +35,7 @@ backend-go/
│ ├── employees_controller.go # 员工赋码接口
│ ├── helper.go # 控制器通用辅助函数
│ ├── serials_controller.go # 序列号管理接口
│ └── users_controller.go # 用户管理接口(仅管理员)
│ └── users_controller.go # 员工主档管理接口(仅管理员)
├── database/ # 数据库连接和操作
│ └── database.go # 数据库初始化、连接池配置
├── docs/ # Swagger API 文档(自动生成)
@@ -58,8 +58,8 @@ backend-go/
│ ├── serials_service.go # 序列号业务逻辑
│ ├── aftersales_service_test.go # 售后工单单元测试
│ ├── services_test.go # 认证/序列号/员工/企业单元测试
│ ├── users_service.go # 用户管理业务逻辑
│ └── users_service_test.go # 用户管理单元测试
│ ├── users_service.go # 员工主档/后台账号业务逻辑
│ └── users_service_test.go # 员工主档/后台账号单元测试
├── tests/ # 集成测试
│ └── main_test.go # 端到端测试
├── data/ # 数据目录(SQLite 数据库存储位置)
@@ -197,7 +197,7 @@ curl -X GET http://localhost:3000/api/health
```bash
curl -X POST http://localhost:3000/api/auth/login \
-H "Content-Type: application/json" \
-d '{"username":"admin","password":"password123"}'
-d '{"username":"admin","password":"Beifan@2026"}'
```
首次启动且用户表为空时,系统会自动创建默认管理员账号(请在生产环境立即修改密码):
@@ -227,11 +227,11 @@ http://localhost:3000/swagger/index.html
如果修改了代码中的 API 注解,需要重新生成 Swagger 文档:
```bash
# 确保已安装 swag 工具
go install github.com/swaggo/swag/cmd/swag@latest
# 生成文档
swag init -g main.go
make swagger
# 或直接执行
go run github.com/swaggo/swag/cmd/swag@v1.16.6 init -g main.go
# 文档将生成在 docs/ 目录下
```
@@ -254,9 +254,9 @@ swag init -g main.go
| ---- | ----------------------------------- | ---------------- | -------- | ------ |
| POST | `/api/serials/generate` | 生成序列号 | 是 | 管理员 |
| POST | `/api/serials/generate-with-prefix` | 带前缀生成序列号 | 是 | 管理员 |
| POST | `/api/serials/:serialNumber/qrcode` | 生成序列号二维码 | 是 | 任何 |
| POST | `/api/serials/:serialNumber/qrcode` | 生成序列号二维码 | 是 | 管理员 |
| GET | `/api/serials/:serialNumber/query` | 查询序列号信息 | 否 | 任何 |
| GET | `/api/serials` | 获取序列号列表 | 是 | 任何 |
| GET | `/api/serials` | 获取序列号列表 | 是 | 管理员 |
| PATCH | `/api/serials/:serialNumber` | 更新序列号信息 | 是 | 管理员 |
| PUT | `/api/serials/:serialNumber` | 更新序列号信息 | 是 | 管理员 |
| POST | `/api/serials/:serialNumber/revoke` | 吊销序列号 | 是 | 管理员 |
@@ -280,9 +280,9 @@ swag init -g main.go
| 方法 | 路径 | 描述 | 需要认证 | 角色 |
| ---- | -------------------------------------- | ------------------ | -------- | ------ |
| POST | `/api/employee-serials/generate` | 生成员工序列号 | 是 | 管理员 |
| GET | `/api/employee-serials` | 获取员工序列号列表 | 是 | 任何 |
| GET | `/api/employee-serials` | 获取员工序列号列表 | 是 | 管理员 |
| GET | `/api/employee-serials/:serialNumber/query` | 查询员工序列号信息 | 否 | 任何 |
| POST | `/api/employee-serials/:serialNumber/qrcode` | 生成员工二维码 | 是 | 任何 |
| POST | `/api/employee-serials/:serialNumber/qrcode` | 生成员工二维码 | 是 | 管理员 |
| PATCH | `/api/employee-serials/:serialNumber` | 更新员工序列号信息 | 是 | 管理员 |
| PUT | `/api/employee-serials/:serialNumber` | 更新员工序列号信息 | 是 | 管理员 |
| POST | `/api/employee-serials/:serialNumber/revoke` | 吊销员工序列号 | 是 | 管理员 |
@@ -295,7 +295,8 @@ swag init -g main.go
**员工序列号特点**:
- 无有效期限制(与企业赋码不同)
- 包含部门(department)和员工姓名(employeeName)信息
- 创建员工主档时会自动生成 1 个员工码,并通过 `employeeId` 绑定员工
- 包含岗位(position)和员工姓名(employeeName)信息
- 序列号格式: `EMP26xxxxxx`(EMP + 年份后两位 + 6位随机字符)
### 售后工单
@@ -325,25 +326,26 @@ swag init -g main.go
- 工单状态机: `created``pending_confirmation``closed` / `rejected`,被退回后可重新提交
- 公开查询不返回手机号(脱敏)
### 用户管理(仅管理员)
### 员工管理(仅管理员)
| 方法 | 路径 | 描述 | 需要认证 | 角色 |
| ------ | ----------------------------------- | -------------------------- | -------- | ------------- |
| GET | `/api/users/assignable` | 可分配用户列表(用于售后) | 是 | 管理员/技术员 |
| POST | `/api/users` | 创建用户 | 是 | 管理员 |
| GET | `/api/users` | 用户列表(分页+筛选) | 是 | 管理员 |
| PATCH | `/api/users/:id` | 更新用户姓名/邮箱/角色 | 是 | 管理员 |
| POST | `/api/users/:id/reset-password` | 重置用户密码 | 是 | 管理员 |
| DELETE | `/api/users/:id` | 删除用户 | 是 | 管理员 |
| POST | `/api/employees` | 创建员工并自动生成员工码 | 是 | 管理员 |
| GET | `/api/employees` | 员工列表(分页+筛选) | 是 | 管理员 |
| PATCH | `/api/employees/:id` | 更新员工姓名/电话/工号/岗位/角色 | 是 | 管理员 |
| POST | `/api/employees/:id/reset-password` | 重置后台账号密码 | 是 | 管理员 |
| DELETE | `/api/employees/:id` | 删除员工 | 是 | 管理员 |
**用户角色**:
- `admin`完整权限,包括用户管理、强制关闭工单、工单分配(重新分配技术员)、删除工单
- `technician`可创建/编辑自己负责的售后工单,可使用 `assignable` 查询同事
- `user`:保留角色(暂未实际启用)
**员工角色**:
- `admin`管理员,拥有全部后台权限,包括企业管理、员工管理、工单分配、强制关闭和删除工单
- `technician`技术员,仅拥有工单模块权限,可创建/处理工单,可使用 `assignable` 查询可分配同事
- `employee`:员工,无后台登录权限,不需要密码,仅用于员工主档和员工码查询
**保护规则**:
- 不能删除自己;不能将自己的 admin 角色降级;不能删除最后一个 admin
- 默认创建用户密码 bcrypt 加密存储
- 创建 `admin` / `technician` 必须设置初始密码,密码 bcrypt 加密存储
- 创建 `employee` 不要求密码,且不能登录后台
## 测试
@@ -376,7 +378,7 @@ go tool cover -html=coverage.out
- 员工赋码测试(生成、查询、更新、吊销、二维码生成)
- 企业统计测试(统计概览)
- 售后工单测试(YYMMDDNN 序号生成、状态机、客户确认手机号校验、强制关闭)
- 用户管理测试(重复用户名、自降级保护、最后管理员保护、密码重置)
- 员工管理测试(创建员工自动生成员工码、重复工号、自降级保护、最后管理员保护、密码重置)
- **tests/**: 集成测试(健康检查、登录流程)
## 代码检查
+9 -6
View File
@@ -58,12 +58,15 @@ func (c *AuthController) Login(ctx *gin.Context) {
"message": "登录成功",
"accessToken": token,
"user": models.UserDTO{
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Role: user.Role,
CreatedAt: user.CreatedAt,
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Phone: user.Phone,
EmployeeNo: user.EmployeeNo,
Position: user.Position,
Role: user.Role,
CreatedAt: user.CreatedAt,
},
})
}
+26 -26
View File
@@ -10,7 +10,7 @@ import (
"git.beifan.cn/trace-system/backend-go/services"
)
// UsersController 用户管理控制器
// UsersController 员工管理控制器
type UsersController struct {
usersService services.UsersService
}
@@ -22,17 +22,17 @@ func NewUsersController() *UsersController {
}
}
// Create 创建用户(管理员)
// @Summary 创建用户
// @Tags 用户管理
// Create 创建员工(管理员)
// @Summary 创建员工
// @Tags 员工管理
// @Accept json
// @Produce json
// @Security BearerAuth
// @Param data body models.CreateUserDTO true "用户数据"
// @Param data body models.CreateUserDTO true "员工数据"
// @Success 200 {object} models.DataResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Router /users [post]
// @Router /employees [post]
func (c *UsersController) Create(ctx *gin.Context) {
var dto models.CreateUserDTO
if !BindJSON(ctx, &dto) {
@@ -45,14 +45,14 @@ func (c *UsersController) Create(ctx *gin.Context) {
return
}
SuccessResponse(ctx, "用户创建成功", gin.H{
"user": user,
SuccessResponse(ctx, "员工创建成功", gin.H{
"employee": user,
})
}
// FindAll 用户列表
// @Summary 用户列表
// @Tags 用户管理
// FindAll 员工列表
// @Summary 员工列表
// @Tags 员工管理
// @Produce json
// @Security BearerAuth
// @Param page query int false "页码"
@@ -61,7 +61,7 @@ func (c *UsersController) Create(ctx *gin.Context) {
// @Param search query string false "搜索"
// @Success 200 {object} models.PaginationResponse
// @Failure 401 {object} models.ErrorResponse
// @Router /users [get]
// @Router /employees [get]
func (c *UsersController) FindAll(ctx *gin.Context) {
page, _ := strconv.Atoi(ctx.DefaultQuery("page", "1"))
limit, _ := strconv.Atoi(ctx.DefaultQuery("limit", "20"))
@@ -74,7 +74,7 @@ func (c *UsersController) FindAll(ctx *gin.Context) {
return
}
SuccessResponse(ctx, "获取用户列表成功", gin.H{
SuccessResponse(ctx, "获取员工列表成功", gin.H{
"data": users,
"pagination": gin.H{
"page": page,
@@ -106,9 +106,9 @@ func (c *UsersController) FindAssignable(ctx *gin.Context) {
})
}
// Update 更新用户信息
// @Summary 更新用户
// @Tags 用户管理
// Update 更新员工信息
// @Summary 更新员工
// @Tags 员工管理
// @Accept json
// @Produce json
// @Security BearerAuth
@@ -117,7 +117,7 @@ func (c *UsersController) FindAssignable(ctx *gin.Context) {
// @Success 200 {object} models.DataResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Router /users/{id} [patch]
// @Router /employees/{id} [patch]
func (c *UsersController) Update(ctx *gin.Context) {
userModel, ok := GetCurrentUser(ctx)
if !ok {
@@ -141,14 +141,14 @@ func (c *UsersController) Update(ctx *gin.Context) {
return
}
SuccessResponse(ctx, "用户更新成功", gin.H{
"user": user,
SuccessResponse(ctx, "员工更新成功", gin.H{
"employee": user,
})
}
// ResetPassword 重置用户密码
// @Summary 重置密码
// @Tags 用户管理
// @Tags 员工管理
// @Accept json
// @Produce json
// @Security BearerAuth
@@ -157,7 +157,7 @@ func (c *UsersController) Update(ctx *gin.Context) {
// @Success 200 {object} models.BaseResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Router /users/{id}/reset-password [post]
// @Router /employees/{id}/reset-password [post]
func (c *UsersController) ResetPassword(ctx *gin.Context) {
id, err := strconv.ParseUint(ctx.Param("id"), 10, 32)
if err != nil {
@@ -178,16 +178,16 @@ func (c *UsersController) ResetPassword(ctx *gin.Context) {
SuccessResponse(ctx, "密码重置成功")
}
// Delete 删除用户
// @Summary 删除用户
// @Tags 用户管理
// Delete 删除员工
// @Summary 删除员工
// @Tags 员工管理
// @Produce json
// @Security BearerAuth
// @Param id path int true "用户 ID"
// @Success 200 {object} models.BaseResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Router /users/{id} [delete]
// @Router /employees/{id} [delete]
func (c *UsersController) Delete(ctx *gin.Context) {
userModel, ok := GetCurrentUser(ctx)
if !ok {
@@ -205,5 +205,5 @@ func (c *UsersController) Delete(ctx *gin.Context) {
return
}
SuccessResponse(ctx, "用户删除成功")
SuccessResponse(ctx, "员工删除成功")
}
+8 -5
View File
@@ -148,11 +148,14 @@ func seedAdminUser() {
}
admin := models.User{
Username: adminUsername,
Password: string(hashedPassword),
Name: adminName,
Email: adminEmail,
Role: "admin",
Username: adminUsername,
Password: string(hashedPassword),
Name: adminName,
Email: adminEmail,
Phone: "00000000000",
EmployeeNo: adminUsername,
Position: "管理员",
Role: "admin",
}
if err := DB.Create(&admin).Error; err != nil {
+1880 -1
View File
File diff suppressed because it is too large Load Diff
+1880 -1
View File
File diff suppressed because it is too large Load Diff
+1209 -1
View File
File diff suppressed because it is too large Load Diff
+53 -37
View File
@@ -8,16 +8,20 @@ import (
// User 模型
type User struct {
ID uint `gorm:"primaryKey" json:"id"`
Username string `gorm:"uniqueIndex;size:255" json:"username"`
Password string `gorm:"size:255" json:"-"`
Name string `gorm:"size:255" json:"name"`
Email string `gorm:"size:255" json:"email"`
Role string `gorm:"size:50;default:'user'" json:"role"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Serials []Serial `gorm:"foreignKey:CreatedBy" json:"-"`
ID uint `gorm:"primaryKey" json:"id"`
Username string `gorm:"uniqueIndex;size:255" json:"username"`
Password string `gorm:"size:255" json:"-"`
Name string `gorm:"size:255" json:"name"`
Email string `gorm:"size:255" json:"email"`
Phone string `gorm:"size:50" json:"phone"`
EmployeeNo string `gorm:"uniqueIndex;size:100" json:"employeeNo"`
Position string `gorm:"size:255" json:"position"`
Role string `gorm:"size:50;default:'employee'" json:"role"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
Serials []Serial `gorm:"foreignKey:CreatedBy" json:"-"`
EmployeeSerials []EmployeeSerial `gorm:"foreignKey:EmployeeID" json:"employeeSerials,omitempty"`
}
// Company 模型
@@ -48,12 +52,16 @@ type Serial struct {
// UserDTO 数据传输对象
type UserDTO struct {
ID uint `json:"id"`
Username string `json:"username"`
Name string `json:"name"`
Email string `json:"email"`
Role string `json:"role"`
CreatedAt time.Time `json:"createdAt"`
ID uint `json:"id"`
Username string `json:"username"`
Name string `json:"name"`
Email string `json:"email"`
Phone string `json:"phone"`
EmployeeNo string `json:"employeeNo"`
Position string `json:"position"`
Role string `json:"role"`
CreatedAt time.Time `json:"createdAt"`
EmployeeSerials []EmployeeSerial `json:"employeeSerials,omitempty"`
}
// LoginDTO 登录请求数据
@@ -76,18 +84,24 @@ type UpdateProfileDTO struct {
// CreateUserDTO 管理员创建用户请求
type CreateUserDTO struct {
Username string `json:"username" validate:"required,min=3,max=50"`
Password string `json:"password" validate:"required,min=6"`
Name string `json:"name" validate:"required"`
Email string `json:"email" validate:"omitempty,email"`
Role string `json:"role" validate:"required,oneof=admin technician employee user"`
Username string `json:"username,omitempty"`
Password string `json:"password,omitempty"`
Name string `json:"name" validate:"required"`
Email string `json:"email" validate:"omitempty,email"`
Phone string `json:"phone" validate:"required"`
EmployeeNo string `json:"employeeNo" validate:"required"`
Position string `json:"position" validate:"required"`
Role string `json:"role" validate:"required,oneof=admin technician employee"`
}
// UpdateUserDTO 管理员更新用户信息请求
type UpdateUserDTO struct {
Name string `json:"name,omitempty"`
Email string `json:"email,omitempty" validate:"omitempty,email"`
Role string `json:"role,omitempty" validate:"omitempty,oneof=admin technician employee user"`
Name string `json:"name,omitempty"`
Email string `json:"email,omitempty" validate:"omitempty,email"`
Phone string `json:"phone,omitempty"`
EmployeeNo string `json:"employeeNo,omitempty"`
Position string `json:"position,omitempty"`
Role string `json:"role,omitempty" validate:"omitempty,oneof=admin technician employee"`
}
// AdminResetPasswordDTO 管理员重置用户密码
@@ -187,19 +201,19 @@ type CompanyUpdateRequest struct {
// CompanyStatsOverviewDTO 企业统计概览
type CompanyStatsOverviewDTO struct {
TotalCompanies int64 `json:"totalCompanies"`
ActiveCompanies int64 `json:"activeCompanies"`
InactiveCompanies int64 `json:"inactiveCompanies"`
TotalSerials int64 `json:"totalSerials"`
ActiveSerials int64 `json:"activeSerials"`
RevokedSerials int64 `json:"revokedSerials"`
TotalEmployeeSerials int64 `json:"totalEmployeeSerials"`
ActiveEmployeeSerials int64 `json:"activeEmployeeSerials"`
RevokedEmployeeSerials int64 `json:"revokedEmployeeSerials"`
TotalAftersales int64 `json:"totalAftersales"`
PendingConfirmation int64 `json:"pendingConfirmation"`
ClosedAftersales int64 `json:"closedAftersales"`
RejectedAftersales int64 `json:"rejectedAftersales"`
TotalCompanies int64 `json:"totalCompanies"`
ActiveCompanies int64 `json:"activeCompanies"`
InactiveCompanies int64 `json:"inactiveCompanies"`
TotalSerials int64 `json:"totalSerials"`
ActiveSerials int64 `json:"activeSerials"`
RevokedSerials int64 `json:"revokedSerials"`
TotalEmployeeSerials int64 `json:"totalEmployeeSerials"`
ActiveEmployeeSerials int64 `json:"activeEmployeeSerials"`
RevokedEmployeeSerials int64 `json:"revokedEmployeeSerials"`
TotalAftersales int64 `json:"totalAftersales"`
PendingConfirmation int64 `json:"pendingConfirmation"`
ClosedAftersales int64 `json:"closedAftersales"`
RejectedAftersales int64 `json:"rejectedAftersales"`
}
// EmployeeSerial 员工序列号模型
@@ -209,12 +223,14 @@ type EmployeeSerial struct {
CompanyName string `gorm:"index;size:255" json:"companyName"`
Position string `gorm:"size:255" json:"position"`
EmployeeName string `gorm:"size:255" json:"employeeName"`
EmployeeID *uint `gorm:"index" json:"employeeId,omitempty"`
IsActive bool `gorm:"default:true" json:"isActive"`
CreatedBy *uint `json:"createdBy"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
User *User `gorm:"foreignKey:CreatedBy" json:"user,omitempty"`
Employee *User `gorm:"foreignKey:EmployeeID" json:"employee,omitempty"`
Company *Company `gorm:"foreignKey:CompanyName;references:CompanyName" json:"company,omitempty"`
}
+13 -9
View File
@@ -38,9 +38,9 @@ func SetupAPIRoutes(r *gin.RouterGroup) {
serialsRoutes.PATCH("/:serialNumber", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.Update)
serialsRoutes.POST("/generate", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.Generate)
serialsRoutes.POST("/generate-with-prefix", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.GenerateWithPrefix)
serialsRoutes.POST("/:serialNumber/qrcode", middleware.JWTAuthMiddleware(), serialsController.GenerateQRCode)
serialsRoutes.POST("/:serialNumber/qrcode", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.GenerateQRCode)
serialsRoutes.GET("/:serialNumber/query", serialsController.Query)
serialsRoutes.GET("/", middleware.JWTAuthMiddleware(), serialsController.FindAll)
serialsRoutes.GET("/", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.FindAll)
serialsRoutes.PUT("/:serialNumber", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.Update)
serialsRoutes.POST("/:serialNumber/revoke", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), serialsController.Revoke)
}
@@ -66,9 +66,9 @@ func SetupAPIRoutes(r *gin.RouterGroup) {
{
employeeSerialsRoutes.PATCH("/:serialNumber", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.Update)
employeeSerialsRoutes.POST("/generate", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.Generate)
employeeSerialsRoutes.POST("/:serialNumber/qrcode", middleware.JWTAuthMiddleware(), employeeSerialsController.GenerateQRCode)
employeeSerialsRoutes.POST("/:serialNumber/qrcode", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.GenerateQRCode)
employeeSerialsRoutes.GET("/:serialNumber/query", employeeSerialsController.Query)
employeeSerialsRoutes.GET("/", middleware.JWTAuthMiddleware(), employeeSerialsController.FindAll)
employeeSerialsRoutes.GET("/", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.FindAll)
employeeSerialsRoutes.PUT("/:serialNumber", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.Update)
employeeSerialsRoutes.POST("/:serialNumber/revoke", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.Revoke)
employeeSerialsRoutes.DELETE("/:serialNumber", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), employeeSerialsController.Delete)
@@ -76,14 +76,18 @@ func SetupAPIRoutes(r *gin.RouterGroup) {
// 用户管理路由(仅管理员)
usersController := controllers.NewUsersController()
employeesRoutes := r.Group("/employees")
{
employeesRoutes.POST("", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.Create)
employeesRoutes.GET("", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.FindAll)
employeesRoutes.PATCH("/:id", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.Update)
employeesRoutes.POST("/:id/reset-password", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.ResetPassword)
employeesRoutes.DELETE("/:id", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.Delete)
}
usersRoutes := r.Group("/users")
{
usersRoutes.GET("/assignable", middleware.JWTAuthMiddleware(), middleware.TechnicianMiddleware(), usersController.FindAssignable)
usersRoutes.POST("", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.Create)
usersRoutes.GET("", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.FindAll)
usersRoutes.PATCH("/:id", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.Update)
usersRoutes.POST("/:id/reset-password", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.ResetPassword)
usersRoutes.DELETE("/:id", middleware.JWTAuthMiddleware(), middleware.AdminMiddleware(), usersController.Delete)
}
// 售后工单路由
+24 -12
View File
@@ -23,6 +23,12 @@ func (s *AuthService) ValidateUser(username string, password string) (*models.Us
if result.Error != nil {
return nil, errors.New("用户名或密码不正确")
}
if user.Role != "admin" && user.Role != "technician" {
return nil, errors.New("用户名或密码不正确")
}
if user.Password == "" {
return nil, errors.New("用户名或密码不正确")
}
err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password))
if err != nil {
@@ -56,12 +62,15 @@ func (s *AuthService) GetProfile(userId uint) (*models.UserDTO, error) {
}
return &models.UserDTO{
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Role: user.Role,
CreatedAt: user.CreatedAt,
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Phone: user.Phone,
EmployeeNo: user.EmployeeNo,
Position: user.Position,
Role: user.Role,
CreatedAt: user.CreatedAt,
}, nil
}
@@ -109,11 +118,14 @@ func (s *AuthService) UpdateProfile(userId uint, name string, email string) (*mo
}
return &models.UserDTO{
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Role: user.Role,
CreatedAt: user.CreatedAt,
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Phone: user.Phone,
EmployeeNo: user.EmployeeNo,
Position: user.Position,
Role: user.Role,
CreatedAt: user.CreatedAt,
}, nil
}
+52 -4
View File
@@ -16,11 +16,14 @@ import (
"git.beifan.cn/trace-system/backend-go/database"
"git.beifan.cn/trace-system/backend-go/models"
"gorm.io/gorm"
)
// EmployeeSerialsService 员工序列号服务
type EmployeeSerialsService struct{}
const defaultEmployeeCompanyName = "内部员工"
// Generate 生成员工序列号
func (s *EmployeeSerialsService) Generate(
companyName string,
@@ -29,18 +32,62 @@ func (s *EmployeeSerialsService) Generate(
quantity int,
userId uint,
serialPrefix string,
) ([]models.EmployeeSerial, error) {
return s.generateWithDB(database.DB, companyName, position, employeeName, quantity, userId, nil, serialPrefix)
}
// GenerateForEmployee 为员工主档生成一个员工码
func (s *EmployeeSerialsService) GenerateForEmployee(
tx *gorm.DB,
employee models.User,
creatorID uint,
serialPrefix string,
) (*models.EmployeeSerial, error) {
db := tx
if db == nil {
db = database.DB
}
serials, err := s.generateWithDB(
db,
defaultEmployeeCompanyName,
employee.Position,
employee.Name,
1,
creatorID,
&employee.ID,
serialPrefix,
)
if err != nil {
return nil, err
}
if len(serials) == 0 {
return nil, errors.New("员工码生成失败")
}
return &serials[0], nil
}
func (s *EmployeeSerialsService) generateWithDB(
db *gorm.DB,
companyName string,
position string,
employeeName string,
quantity int,
userId uint,
employeeID *uint,
serialPrefix string,
) ([]models.EmployeeSerial, error) {
var serials []models.EmployeeSerial
// 检查公司是否存在,不存在则创建
var company models.Company
result := database.DB.Where("company_name = ?", companyName).First(&company)
result := db.Where("company_name = ?", companyName).First(&company)
if result.Error != nil {
company = models.Company{
CompanyName: companyName,
IsActive: true,
}
result = database.DB.Create(&company)
result = db.Create(&company)
if result.Error != nil {
return nil, fmt.Errorf("创建公司失败: %w", result.Error)
}
@@ -68,7 +115,7 @@ func (s *EmployeeSerialsService) Generate(
}
var existingSerial models.EmployeeSerial
checkResult := database.DB.Where("serial_number = ?", serialNumber).First(&existingSerial)
checkResult := db.Where("serial_number = ?", serialNumber).First(&existingSerial)
if checkResult.Error != nil {
serialNumbers[serialNumber] = true
i++
@@ -81,6 +128,7 @@ func (s *EmployeeSerialsService) Generate(
CompanyName: companyName,
Position: position,
EmployeeName: employeeName,
EmployeeID: employeeID,
CreatedBy: &userId,
IsActive: true,
}
@@ -88,7 +136,7 @@ func (s *EmployeeSerialsService) Generate(
}
// 保存到数据库
result = database.DB.Create(&serials)
result = db.Create(&serials)
if result.Error != nil {
return nil, fmt.Errorf("保存员工序列号失败: %w", result.Error)
}
+26 -7
View File
@@ -53,7 +53,7 @@ func TestAuthService_ValidateUser_Success(t *testing.T) {
Password: string(password),
Name: "测试用户",
Email: "test@example.com",
Role: "user",
Role: "technician",
}
database.DB.Create(&user)
@@ -75,7 +75,7 @@ func TestAuthService_ValidateUser_WrongPassword(t *testing.T) {
Password: string(password),
Name: "测试用户2",
Email: "test2@example.com",
Role: "user",
Role: "technician",
}
database.DB.Create(&user)
@@ -87,6 +87,25 @@ func TestAuthService_ValidateUser_WrongPassword(t *testing.T) {
database.DB.Unscoped().Delete(&user)
}
func TestAuthService_ValidateUser_EmployeeCannotLogin(t *testing.T) {
password, _ := bcrypt.GenerateFromPassword([]byte("password123"), bcrypt.DefaultCost)
user := models.User{
Username: "employee_no_login",
Password: string(password),
Name: "普通员工",
Role: "employee",
}
database.DB.Create(&user)
authService := AuthService{}
_, err := authService.ValidateUser("employee_no_login", "password123")
assert.Error(t, err)
assert.Contains(t, err.Error(), "用户名或密码不正确")
database.DB.Unscoped().Delete(&user)
}
func TestAuthService_ValidateUser_UserNotFound(t *testing.T) {
authService := AuthService{}
_, err := authService.ValidateUser("nonexistent", "password")
@@ -98,7 +117,7 @@ func TestAuthService_GenerateToken_Success(t *testing.T) {
user := &models.User{
ID: 1,
Username: "testuser",
Role: "user",
Role: "technician",
}
authService := AuthService{}
@@ -117,7 +136,7 @@ func TestAuthService_GetProfile_Success(t *testing.T) {
Password: string(password),
Name: "测试用户3",
Email: "test3@example.com",
Role: "user",
Role: "technician",
}
database.DB.Create(&user)
@@ -140,7 +159,7 @@ func TestAuthService_ChangePassword_Success(t *testing.T) {
Password: string(password),
Name: "测试用户4",
Email: "test4@example.com",
Role: "user",
Role: "technician",
}
database.DB.Create(&user)
@@ -165,7 +184,7 @@ func TestAuthService_ChangePassword_WrongCurrentPassword(t *testing.T) {
Password: string(password),
Name: "测试用户5",
Email: "test5@example.com",
Role: "user",
Role: "technician",
}
database.DB.Create(&user)
@@ -185,7 +204,7 @@ func TestAuthService_UpdateProfile_Success(t *testing.T) {
Password: string(password),
Name: "测试用户6",
Email: "test6@example.com",
Role: "user",
Role: "technician",
}
database.DB.Create(&user)
+111 -19
View File
@@ -17,18 +17,57 @@ type UsersService struct{}
func toUserDTO(user models.User) models.UserDTO {
return models.UserDTO{
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Role: user.Role,
CreatedAt: user.CreatedAt,
ID: user.ID,
Username: user.Username,
Name: user.Name,
Email: user.Email,
Phone: user.Phone,
EmployeeNo: user.EmployeeNo,
Position: user.Position,
Role: user.Role,
CreatedAt: user.CreatedAt,
EmployeeSerials: user.EmployeeSerials,
}
}
func hasBackendAccess(role string) bool {
return role == "admin" || role == "technician"
}
func isValidEmployeeRole(role string) bool {
return role == "admin" || role == "technician" || role == "employee"
}
// Create 创建用户
func (s *UsersService) Create(dto models.CreateUserDTO) (*models.UserDTO, error) {
name := strings.TrimSpace(dto.Name)
phone := strings.TrimSpace(dto.Phone)
employeeNo := strings.TrimSpace(dto.EmployeeNo)
position := strings.TrimSpace(dto.Position)
role := strings.TrimSpace(dto.Role)
username := strings.TrimSpace(dto.Username)
if username == "" {
username = employeeNo
}
if name == "" {
return nil, errors.New("姓名不能为空")
}
if phone == "" {
return nil, errors.New("电话不能为空")
}
if employeeNo == "" {
return nil, errors.New("工号不能为空")
}
if position == "" {
return nil, errors.New("岗位不能为空")
}
if !isValidEmployeeRole(role) {
return nil, errors.New("角色不正确")
}
if hasBackendAccess(role) && len(dto.Password) < 6 {
return nil, errors.New("管理员和技术员必须设置至少 6 位初始密码")
}
var existing models.User
err := database.DB.Where("username = ?", username).First(&existing).Error
@@ -39,21 +78,48 @@ func (s *UsersService) Create(dto models.CreateUserDTO) (*models.UserDTO, error)
return nil, fmt.Errorf("查询用户失败: %w", err)
}
hashed, err := bcrypt.GenerateFromPassword([]byte(dto.Password), bcrypt.DefaultCost)
if err != nil {
return nil, fmt.Errorf("密码加密失败: %w", err)
err = database.DB.Where("employee_no = ?", employeeNo).First(&existing).Error
if err == nil {
return nil, errors.New("工号已存在")
}
if !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, fmt.Errorf("查询员工工号失败: %w", err)
}
hashed := ""
if hasBackendAccess(role) {
hashBytes, err := bcrypt.GenerateFromPassword([]byte(dto.Password), bcrypt.DefaultCost)
if err != nil {
return nil, fmt.Errorf("密码加密失败: %w", err)
}
hashed = string(hashBytes)
}
user := models.User{
Username: username,
Password: string(hashed),
Name: dto.Name,
Email: dto.Email,
Role: dto.Role,
Username: username,
Password: hashed,
Name: name,
Email: dto.Email,
Phone: phone,
EmployeeNo: employeeNo,
Position: position,
Role: role,
}
if err := database.DB.Create(&user).Error; err != nil {
return nil, fmt.Errorf("创建用户失败: %w", err)
err = database.DB.Transaction(func(tx *gorm.DB) error {
if err := tx.Create(&user).Error; err != nil {
return fmt.Errorf("创建员工失败: %w", err)
}
serialService := EmployeeSerialsService{}
if _, err := serialService.GenerateForEmployee(tx, user, user.ID, ""); err != nil {
return err
}
return tx.Preload("EmployeeSerials").First(&user, user.ID).Error
})
if err != nil {
return nil, err
}
dtoOut := toUserDTO(user)
@@ -65,13 +131,14 @@ func (s *UsersService) FindAll(page int, limit int, role string, search string)
var users []models.User
var total int64
db := database.DB.Model(&models.User{})
db := database.DB.Model(&models.User{}).Preload("EmployeeSerials")
if role != "" {
db = db.Where("role = ?", role)
}
if search != "" {
pattern := "%" + search + "%"
db = db.Where("username LIKE ? OR name LIKE ? OR email LIKE ?", pattern, pattern, pattern)
db = db.Where("username LIKE ? OR name LIKE ? OR email LIKE ? OR phone LIKE ? OR employee_no LIKE ? OR position LIKE ?",
pattern, pattern, pattern, pattern, pattern, pattern)
}
if err := db.Count(&total).Error; err != nil {
@@ -114,12 +181,34 @@ func (s *UsersService) Update(userId uint, dto models.UpdateUserDTO, currentUser
}
if dto.Name != "" {
user.Name = dto.Name
user.Name = strings.TrimSpace(dto.Name)
}
if dto.Email != "" {
user.Email = dto.Email
}
if dto.Phone != "" {
user.Phone = strings.TrimSpace(dto.Phone)
}
if dto.EmployeeNo != "" {
employeeNo := strings.TrimSpace(dto.EmployeeNo)
var existing models.User
err := database.DB.Where("employee_no = ? AND id <> ?", employeeNo, userId).First(&existing).Error
if err == nil {
return nil, errors.New("工号已存在")
}
if !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, fmt.Errorf("查询员工工号失败: %w", err)
}
user.EmployeeNo = employeeNo
user.Username = employeeNo
}
if dto.Position != "" {
user.Position = strings.TrimSpace(dto.Position)
}
if dto.Role != "" {
if !isValidEmployeeRole(dto.Role) {
return nil, errors.New("角色不正确")
}
// 防止管理员把自己降级
if user.ID == currentUserId && user.Role == "admin" && dto.Role != "admin" {
return nil, errors.New("不能修改自己的管理员角色")
@@ -141,6 +230,9 @@ func (s *UsersService) ResetPassword(userId uint, newPassword string) error {
if err := database.DB.First(&user, userId).Error; err != nil {
return errors.New("用户不存在")
}
if !hasBackendAccess(user.Role) {
return errors.New("员工无后台登录权限,不能设置密码")
}
hashed, err := bcrypt.GenerateFromPassword([]byte(newPassword), bcrypt.DefaultCost)
if err != nil {
+55 -11
View File
@@ -13,18 +13,59 @@ import (
func TestUsersService_Create_Success(t *testing.T) {
svc := UsersService{}
dto, err := svc.Create(models.CreateUserDTO{
Username: "users_create_ok",
Password: "password123",
Name: "新技术员",
Email: "new@example.com",
Role: "technician",
Password: "password123",
Name: "新技术员",
Email: "new@example.com",
Phone: "13800000001",
EmployeeNo: "users_create_ok",
Position: "技术员",
Role: "technician",
})
assert.NoError(t, err)
assert.NotNil(t, dto)
assert.Equal(t, "users_create_ok", dto.Username)
assert.Equal(t, "technician", dto.Role)
assert.Equal(t, "13800000001", dto.Phone)
assert.Equal(t, "users_create_ok", dto.EmployeeNo)
assert.Len(t, dto.EmployeeSerials, 1)
database.DB.Unscoped().Where("username = ?", "users_create_ok").Delete(&models.User{})
database.DB.Unscoped().Where("employee_name = ?", "新技术员").Delete(&models.EmployeeSerial{})
}
func TestUsersService_Create_EmployeeWithoutPasswordGeneratesSerial(t *testing.T) {
svc := UsersService{}
dto, err := svc.Create(models.CreateUserDTO{
Name: "普通员工",
Phone: "13800000002",
EmployeeNo: "employee_no_pwd",
Position: "生产员工",
Role: "employee",
})
assert.NoError(t, err)
assert.NotNil(t, dto)
assert.Equal(t, "employee", dto.Role)
assert.Len(t, dto.EmployeeSerials, 1)
var user models.User
assert.NoError(t, database.DB.Where("employee_no = ?", "employee_no_pwd").First(&user).Error)
assert.Empty(t, user.Password)
database.DB.Unscoped().Where("employee_id = ?", user.ID).Delete(&models.EmployeeSerial{})
database.DB.Unscoped().Delete(&user)
}
func TestUsersService_Create_BackendRoleRequiresPassword(t *testing.T) {
svc := UsersService{}
_, err := svc.Create(models.CreateUserDTO{
Name: "无密码技术员",
Phone: "13800000003",
EmployeeNo: "tech_no_pwd",
Position: "技术员",
Role: "technician",
})
assert.Error(t, err)
assert.Contains(t, err.Error(), "必须设置")
}
func TestUsersService_Create_DuplicateUsername(t *testing.T) {
@@ -39,10 +80,13 @@ func TestUsersService_Create_DuplicateUsername(t *testing.T) {
svc := UsersService{}
_, err := svc.Create(models.CreateUserDTO{
Username: "users_create_dup",
Password: "password123",
Name: "duplicate",
Role: "technician",
Username: "users_create_dup",
Password: "password123",
Name: "duplicate",
Phone: "13800000004",
EmployeeNo: "users_create_dup_2",
Position: "技术员",
Role: "technician",
})
assert.Error(t, err)
assert.Contains(t, err.Error(), "用户名已存在")
@@ -164,7 +208,7 @@ func TestUsersService_Delete_BlocksLastAdmin(t *testing.T) {
func TestUsersService_FindAssignable_ReturnsAdminAndTechnician(t *testing.T) {
a := models.User{Username: "assignable_admin", Password: "x", Name: "A", Role: "admin"}
tech := models.User{Username: "assignable_tech", Password: "x", Name: "T", Role: "technician"}
plain := models.User{Username: "assignable_user", Password: "x", Name: "U", Role: "user"}
plain := models.User{Username: "assignable_user", Password: "x", Name: "U", Role: "employee"}
database.DB.Create(&a)
database.DB.Create(&tech)
database.DB.Create(&plain)
@@ -189,7 +233,7 @@ func TestUsersService_FindAssignable_ReturnsAdminAndTechnician(t *testing.T) {
func TestUsersService_FindAll_FilterByRole(t *testing.T) {
tech1 := models.User{Username: "findall_tech1", Password: "x", Name: "T1", Role: "technician"}
tech2 := models.User{Username: "findall_tech2", Password: "x", Name: "T2", Role: "technician"}
user1 := models.User{Username: "findall_user1", Password: "x", Name: "U1", Role: "user"}
user1 := models.User{Username: "findall_user1", Password: "x", Name: "U1", Role: "employee"}
database.DB.Create(&tech1)
database.DB.Create(&tech2)
database.DB.Create(&user1)
+4 -4
View File
@@ -73,11 +73,11 @@ func createTestUsers() {
Role: "admin",
},
{
Username: "user1",
Username: "employee1",
Password: string(testPassword),
Name: "普通用户",
Email: "user1@example.com",
Role: "user",
Name: "普通员工",
Email: "employee1@example.com",
Role: "employee",
},
}