初始化2

2026-08-28 15:06:01 +08:00
parent b34325a16f
commit a9c3fbeb41
537 changed files with 176215 additions and 17 deletions
@@ -0,0 +1,261 @@
+package handler
+
+import (
+	"net/http"
+	"strconv"
+
+	"bj_power_mes/common/httpx"
+	"bj_power_mes/internal/logic"
+	"bj_power_mes/internal/svc"
+
+	"github.com/zeromicro/go-zero/rest/pathvar"
+)
+
+// ctxUserId 从 JWT 上下文取用户ID（go-zero 将自定义 claim 放入 context）
+func ctxUserId(r *http.Request) int {
+	switch v := r.Context().Value("userId").(type) {
+	case float64:
+		return int(v)
+	case int:
+		return v
+	case int64:
+		return int(v)
+	case string:
+		n, _ := strconv.Atoi(v)
+		return n
+	}
+	return 0
+}
+
+// ctxOperator 当前操作人：优先取 name，其次 username
+func ctxOperator(r *http.Request) string {
+	if v, ok := r.Context().Value("name").(string); ok && v != "" {
+		return v
+	}
+	if v, ok := r.Context().Value("username").(string); ok && v != "" {
+		return v
+	}
+	return "system"
+}
+
+// pathId 读取路由参数 :id
+func pathId(r *http.Request) int {
+	vars := pathvar.Vars(r)
+	if vars == nil {
+		return 0
+	}
+	n, _ := strconv.Atoi(vars["id"])
+	return n
+}
+
+func LoginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.LoginReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		data, err := logic.New(svcCtx).Login(r.Context(), req)
+		if err != nil {
+			httpx.Fail(w, 1001, err.Error())
+			return
+		}
+		httpx.Ok(w, data)
+	}
+}
+
+func RefreshHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.RefreshReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		data, err := logic.New(svcCtx).Refresh(r.Context(), req)
+		if err != nil {
+			httpx.Fail(w, 1002, err.Error())
+			return
+		}
+		httpx.Ok(w, data)
+	}
+}
+
+func UserInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		data, err := logic.New(svcCtx).UserInfo(r.Context(), ctxUserId(r))
+		if err != nil {
+			httpx.Fail(w, 1003, err.Error())
+			return
+		}
+		httpx.Ok(w, data)
+	}
+}
+
+func SeedHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if err := logic.New(svcCtx).Seed(r.Context()); err != nil {
+			httpx.Fail(w, 1004, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "初始化完成", nil)
+	}
+}
+
+// ---------- 用户管理 ----------
+
+func ListUsersHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		data, err := logic.New(svcCtx).ListUsers(r.Context())
+		if err != nil {
+			httpx.Fail(w, 1005, err.Error())
+			return
+		}
+		httpx.Ok(w, data)
+	}
+}
+
+func CreateUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.UserReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		if err := logic.New(svcCtx).CreateUser(r.Context(), req); err != nil {
+			httpx.Fail(w, 1006, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "创建成功", nil)
+	}
+}
+
+func UpdateUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.UserReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		if err := logic.New(svcCtx).UpdateUser(r.Context(), req); err != nil {
+			httpx.Fail(w, 1007, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "更新成功", nil)
+	}
+}
+
+func DeleteUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if err := logic.New(svcCtx).DeleteUser(r.Context(), pathId(r)); err != nil {
+			httpx.Fail(w, 1008, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "删除成功", nil)
+	}
+}
+
+// ---------- 角色管理 ----------
+
+func ListRolesHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		data, err := logic.New(svcCtx).ListRoles(r.Context())
+		if err != nil {
+			httpx.Fail(w, 1009, err.Error())
+			return
+		}
+		httpx.Ok(w, data)
+	}
+}
+
+func CreateRoleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.RoleReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		if err := logic.New(svcCtx).CreateRole(r.Context(), req); err != nil {
+			httpx.Fail(w, 1010, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "创建成功", nil)
+	}
+}
+
+func UpdateRoleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.RoleReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		if err := logic.New(svcCtx).UpdateRole(r.Context(), req); err != nil {
+			httpx.Fail(w, 1011, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "更新成功", nil)
+	}
+}
+
+func DeleteRoleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if err := logic.New(svcCtx).DeleteRole(r.Context(), pathId(r)); err != nil {
+			httpx.Fail(w, 1012, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "删除成功", nil)
+	}
+}
+
+// ---------- 权限管理 ----------
+
+func ListPermissionsHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		data, err := logic.New(svcCtx).ListPermissions(r.Context())
+		if err != nil {
+			httpx.Fail(w, 1013, err.Error())
+			return
+		}
+		httpx.Ok(w, data)
+	}
+}
+
+func CreatePermissionHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.PermissionReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		if err := logic.New(svcCtx).CreatePermission(r.Context(), req); err != nil {
+			httpx.Fail(w, 1014, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "创建成功", nil)
+	}
+}
+
+func UpdatePermissionHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req logic.PermissionReq
+		if err := httpx.ParseJSON(r, &req); err != nil {
+			httpx.BadRequest(w, "请求体解析失败")
+			return
+		}
+		if err := logic.New(svcCtx).UpdatePermission(r.Context(), req); err != nil {
+			httpx.Fail(w, 1015, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "更新成功", nil)
+	}
+}
+
+func DeletePermissionHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if err := logic.New(svcCtx).DeletePermission(r.Context(), pathId(r)); err != nil {
+			httpx.Fail(w, 1016, err.Error())
+			return
+		}
+		httpx.OkMessage(w, "删除成功", nil)
+	}
+}