初始化2

bj_power_mes/common/httpx/not_found.go

@@ -0,0 +1,49 @@
+package httpx
+
+import (
+	"net/http"
+	"os"
+	"path"
+	"strings"
+)
+
+const basename = "/"
+
+type NotFoundHandler struct {
+	fs         http.FileSystem
+	fileServer http.Handler
+}
+
+// NewNotFoundHandler 静态资源服务：命中即返回文件，否则回退到 index.html（前端 SPA 路由）
+func NewNotFoundHandler(fs http.FileSystem) NotFoundHandler {
+	return NotFoundHandler{
+		fs:         fs,
+		fileServer: http.FileServer(fs),
+	}
+}
+
+func (n NotFoundHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	if strings.HasPrefix(r.URL.Path, "/api/") {
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+	filePath := strings.TrimPrefix(path.Clean(r.URL.Path), basename)
+	if len(filePath) == 0 {
+		filePath = basename
+	}
+
+	file, err := n.fs.Open(filePath)
+	switch {
+	case err == nil:
+		n.fileServer.ServeHTTP(w, r)
+		_ = file.Close()
+		return
+	case os.IsNotExist(err):
+		r.URL.Path = "/" // vue app 虚拟路由统一回 index.html
+		n.fileServer.ServeHTTP(w, r)
+		return
+	default:
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+}

bj_power_mes/common/httpx/response.go

@@ -0,0 +1,58 @@
+package httpx
+
+import (
+	"encoding/json"
+	"net/http"
+)
+
+// Resp 统一返回结构
+type Resp struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Data    any    `json:"data,omitempty"`
+}
+
+func writeJSON(w http.ResponseWriter, status int, body any) {
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.WriteHeader(status)
+	_ = json.NewEncoder(w).Encode(body)
+}
+
+// Ok 成功
+func Ok(w http.ResponseWriter, data any) {
+	writeJSON(w, http.StatusOK, Resp{Code: 0, Message: "success", Data: data})
+}
+
+// OkCode 成功但自定义 message
+func OkMessage(w http.ResponseWriter, message string, data any) {
+	writeJSON(w, http.StatusOK, Resp{Code: 0, Message: message, Data: data})
+}
+
+// Fail 业务失败（HTTP 200，业务码非0）
+func Fail(w http.ResponseWriter, code int, message string) {
+	writeJSON(w, http.StatusOK, Resp{Code: code, Message: message})
+}
+
+// FailHTTP 直接返回 HTTP 错误
+func FailHTTP(w http.ResponseWriter, status int, message string) {
+	writeJSON(w, status, Resp{Code: status, Message: message})
+}
+
+// BadRequest 参数错误
+func BadRequest(w http.ResponseWriter, message string) {
+	writeJSON(w, http.StatusBadRequest, Resp{Code: 400, Message: message})
+}
+
+func InternalError(w http.ResponseWriter, message string) {
+	writeJSON(w, http.StatusInternalServerError, Resp{Code: 500, Message: message})
+}
+
+func Unauthorized(w http.ResponseWriter, message string) {
+	writeJSON(w, http.StatusUnauthorized, Resp{Code: 401, Message: message})
+}
+
+// ParseJSON 解析请求体
+func ParseJSON(r *http.Request, v any) error {
+	dec := json.NewDecoder(r.Body)
+	return dec.Decode(v)
+}