404 lines
9.6 KiB
Go
404 lines
9.6 KiB
Go
package controllers
|
||
|
||
import (
|
||
"encoding/json"
|
||
"server/models"
|
||
|
||
beego "github.com/beego/beego/v2/server/web"
|
||
)
|
||
|
||
// EmployeeController 员工控制器
|
||
type EmployeeController struct {
|
||
beego.Controller
|
||
}
|
||
|
||
// GetAllEmployees 获取所有员工（可选，用于管理员查看所有员工）
|
||
// @router /employees [get]
|
||
func (c *EmployeeController) GetAllEmployees() {
|
||
employees, err := models.GetAllEmployees()
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "获取员工列表失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
} else {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "获取员工列表成功",
|
||
"data": employees,
|
||
}
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// GetTenantEmployees 获取租户下的所有员工
|
||
// @router /employees/tenant/:tenantId [get]
|
||
func (c *EmployeeController) GetTenantEmployees() {
|
||
tenantId, err := c.GetInt(":tenantId")
|
||
if err != nil || tenantId <= 0 {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "租户ID无效",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
employees, err := models.GetTenantEmployees(tenantId)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "获取员工列表失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
// 格式化返回数据
|
||
employeeList := make([]map[string]interface{}, 0)
|
||
for _, emp := range employees {
|
||
employeeList = append(employeeList, map[string]interface{}{
|
||
"id": emp.Id,
|
||
"tenant_id": emp.TenantId,
|
||
"employee_no": emp.EmployeeNo,
|
||
"name": emp.Name,
|
||
"phone": emp.Phone,
|
||
"email": emp.Email,
|
||
"department_id": emp.DepartmentId,
|
||
"position_id": emp.PositionId,
|
||
"bank_name": emp.BankName,
|
||
"bank_account": emp.BankAccount,
|
||
"status": emp.Status,
|
||
"create_time": emp.CreateTime,
|
||
})
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "获取员工列表成功",
|
||
"data": employeeList,
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// GetEmployeeInfo 获取员工详情
|
||
// @router /employees/:id [get]
|
||
func (c *EmployeeController) GetEmployeeInfo() {
|
||
id, err := c.GetInt(":id")
|
||
if err != nil || id <= 0 {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工ID无效",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
employee, err := models.GetEmployeeById(id)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工不存在",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "获取员工信息成功",
|
||
"data": map[string]interface{}{
|
||
"id": employee.Id,
|
||
"tenant_id": employee.TenantId,
|
||
"employee_no": employee.EmployeeNo,
|
||
"name": employee.Name,
|
||
"phone": employee.Phone,
|
||
"email": employee.Email,
|
||
"department_id": employee.DepartmentId,
|
||
"position_id": employee.PositionId,
|
||
"bank_name": employee.BankName,
|
||
"bank_account": employee.BankAccount,
|
||
"status": employee.Status,
|
||
"create_time": employee.CreateTime,
|
||
},
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// AddEmployee 添加员工
|
||
// @router /employees [post]
|
||
func (c *EmployeeController) AddEmployee() {
|
||
var employeeData struct {
|
||
TenantId int `json:"tenant_id"`
|
||
EmployeeNo string `json:"employee_no"`
|
||
Name string `json:"name"`
|
||
Phone string `json:"phone"`
|
||
Email string `json:"email"`
|
||
DepartmentId int `json:"department_id"`
|
||
PositionId int `json:"position_id"`
|
||
BankName string `json:"bank_name"`
|
||
BankAccount string `json:"bank_account"`
|
||
Status int8 `json:"status"`
|
||
}
|
||
|
||
err := json.Unmarshal(c.Ctx.Input.RequestBody, &employeeData)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "请求参数错误: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
employee := &models.Employee{
|
||
TenantId: employeeData.TenantId,
|
||
EmployeeNo: employeeData.EmployeeNo,
|
||
Name: employeeData.Name,
|
||
Phone: employeeData.Phone,
|
||
Email: employeeData.Email,
|
||
DepartmentId: employeeData.DepartmentId,
|
||
PositionId: employeeData.PositionId,
|
||
BankName: employeeData.BankName,
|
||
BankAccount: employeeData.BankAccount,
|
||
Status: employeeData.Status,
|
||
}
|
||
|
||
// 如果没有指定租户ID，从当前登录用户获取（需要JWT中间件支持）
|
||
// 这里暂时使用传入的tenant_id，如果为0则使用默认值
|
||
if employee.TenantId == 0 {
|
||
// 可以从JWT token中获取租户ID，这里暂时设为1作为默认值
|
||
employee.TenantId = 1
|
||
}
|
||
|
||
// 默认密码
|
||
defaultPassword := "yunzer123"
|
||
id, err := models.AddEmployee(employee, defaultPassword)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "添加员工失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
employee.Id = int(id)
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "添加员工成功",
|
||
"data": employee,
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// UpdateEmployee 更新员工信息
|
||
// @router /employees/:id [put]
|
||
func (c *EmployeeController) UpdateEmployee() {
|
||
id, err := c.GetInt(":id")
|
||
if err != nil || id <= 0 {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工ID无效",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
var updateData struct {
|
||
EmployeeNo string `json:"employee_no"`
|
||
Name string `json:"name"`
|
||
Phone string `json:"phone"`
|
||
Email string `json:"email"`
|
||
DepartmentId int `json:"department_id"`
|
||
PositionId int `json:"position_id"`
|
||
BankName string `json:"bank_name"`
|
||
BankAccount string `json:"bank_account"`
|
||
Status int8 `json:"status"`
|
||
}
|
||
|
||
err = json.Unmarshal(c.Ctx.Input.RequestBody, &updateData)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "请求参数错误: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
employee, err := models.GetEmployeeById(id)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工不存在",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
// 更新字段
|
||
employee.EmployeeNo = updateData.EmployeeNo
|
||
employee.Name = updateData.Name
|
||
employee.Phone = updateData.Phone
|
||
employee.Email = updateData.Email
|
||
employee.DepartmentId = updateData.DepartmentId
|
||
employee.PositionId = updateData.PositionId
|
||
employee.BankName = updateData.BankName
|
||
employee.BankAccount = updateData.BankAccount
|
||
employee.Status = updateData.Status
|
||
|
||
if err := models.UpdateEmployee(employee); err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "更新员工信息失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "更新员工信息成功",
|
||
"data": employee,
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// DeleteEmployee 删除员工
|
||
// @router /employees/:id [delete]
|
||
func (c *EmployeeController) DeleteEmployee() {
|
||
id, err := c.GetInt(":id")
|
||
if err != nil || id <= 0 {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工ID无效",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
if err := models.DeleteEmployee(id); err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "删除员工失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "删除员工成功",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// ResetEmployeePassword 重置员工密码
|
||
// @router /employees/:id/reset-password [post]
|
||
func (c *EmployeeController) ResetEmployeePassword() {
|
||
id, err := c.GetInt(":id")
|
||
if err != nil || id <= 0 {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工ID无效",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
// 默认密码
|
||
defaultPassword := "yunzer123"
|
||
|
||
if err := models.ResetEmployeePassword(id, defaultPassword); err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "重置密码失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "密码重置成功，新密码为: " + defaultPassword,
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|
||
// ChangeEmployeePassword 修改员工密码
|
||
// @router /employees/:id/change-password [post]
|
||
func (c *EmployeeController) ChangeEmployeePassword() {
|
||
id, err := c.GetInt(":id")
|
||
if err != nil || id <= 0 {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "员工ID无效",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
var passwordData struct {
|
||
OldPassword string `json:"old_password"`
|
||
NewPassword string `json:"new_password"`
|
||
}
|
||
|
||
err = json.Unmarshal(c.Ctx.Input.RequestBody, &passwordData)
|
||
if err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "请求参数错误: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
if passwordData.NewPassword == "" {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "新密码不能为空",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
if err := models.ChangeEmployeePassword(id, passwordData.OldPassword, passwordData.NewPassword); err != nil {
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 1,
|
||
"message": "修改密码失败: " + err.Error(),
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
return
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 0,
|
||
"message": "密码修改成功",
|
||
"data": nil,
|
||
}
|
||
c.ServeJSON()
|
||
}
|
||
|