签入版本
This commit is contained in:
小蚂蚁云
@@ -0,0 +1,70 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
)
|
||||
|
||||
// ============================== 跨域中间件 ==============================
|
||||
|
||||
// CORS 跨域处理
|
||||
//
|
||||
// 读取配置 cors.allowOrigin：为空或为 * 时使用框架内置默认策略；
|
||||
// 配置为具体来源时显式回显，并对预检请求直接返回 204
|
||||
func CORS(r *ghttp.Request) {
|
||||
origin := g.Cfg().MustGet(r.Context(), "cors.allowOrigin").String()
|
||||
if origin == "" || origin == "*" {
|
||||
r.Response.CORSDefault()
|
||||
} else {
|
||||
r.Response.Header().Set("Access-Control-Allow-Origin", origin)
|
||||
r.Response.Header().Set("Access-Control-Allow-Credentials", "true")
|
||||
r.Response.Header().Set("Access-Control-Allow-Headers", "Origin,Authorization,Content-Type,Accept,Token,X-Requested-With")
|
||||
r.Response.Header().Set("Access-Control-Allow-Methods", "GET,POST,PUT,DELETE,OPTIONS")
|
||||
if r.Method == "OPTIONS" {
|
||||
r.Response.WriteStatus(204)
|
||||
return
|
||||
}
|
||||
}
|
||||
r.Middleware.Next()
|
||||
}
|
||||
@@ -0,0 +1,78 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/utility/response"
|
||||
)
|
||||
|
||||
// ============================== 演示模式中间件 ==============================
|
||||
|
||||
// Demo 演示模式拦截
|
||||
//
|
||||
// 当配置 app.demo 为 true 时启用，规则如下：
|
||||
// - 读操作（GET/HEAD/OPTIONS）放行，演示站可正常浏览
|
||||
// - 写操作（POST/PUT/PATCH/DELETE）一律拒绝并提示"演示环境，禁止操作!"
|
||||
// - 公开白名单路径（登录/注册/验证码/令牌兑换）放行，保证演示站可登录
|
||||
func Demo(r *ghttp.Request) {
|
||||
if !g.Cfg().MustGet(r.Context(), "app.demo").Bool() {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
switch r.Method {
|
||||
case http.MethodGet, http.MethodHead, http.MethodOptions:
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
if IsPublicPath(r.URL.Path) {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
response.Write(r, response.Fail(gerror.NewCode(consts.CodeDemoMode)))
|
||||
}
|
||||
@@ -0,0 +1,158 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strings"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/utility/cache"
|
||||
jwtutil "easygoadmin/utility/jwt"
|
||||
"easygoadmin/utility/response"
|
||||
)
|
||||
|
||||
// ============================== 身份认证中间件 ==============================
|
||||
|
||||
// JWT 身份认证中间件
|
||||
//
|
||||
// 处理流程：
|
||||
// 1. 免登录路由直接放行
|
||||
// 2. 解析 Bearer Token，校验签名、过期与签发者
|
||||
// 3. 校验 Redis 黑名单，登出或刷新后的旧令牌失效
|
||||
// 4. 加载角色与权限（缓存 300 秒）并注入请求上下文
|
||||
func JWT(r *ghttp.Request) {
|
||||
if IsPublicPath(r.URL.Path) {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
token := parseBearer(r.Header.Get("Authorization"))
|
||||
if token == "" {
|
||||
token = r.Get("token").String() // 兼容个别 URL 直传场景
|
||||
}
|
||||
if token == "" {
|
||||
writeUnauthorized(r)
|
||||
return
|
||||
}
|
||||
claims, err := jwtutil.Parse(r.Context(), token)
|
||||
if err != nil {
|
||||
writeUnauthorized(r)
|
||||
return
|
||||
}
|
||||
// 黑名单（登出/轮换后的旧令牌）
|
||||
if cache.Exists(r.Context(), consts.CacheKeyTokenBlacklist(claims.ID)) {
|
||||
writeUnauthorized(r)
|
||||
return
|
||||
}
|
||||
|
||||
ctx := r.Context()
|
||||
ctx = consts.WithUserID(ctx, claims.UserID)
|
||||
ctx = consts.WithTenantID(ctx, claims.TenantID)
|
||||
ctx = consts.WithUsername(ctx, claims.Username)
|
||||
ctx = consts.WithRealname(ctx, claims.Realname)
|
||||
ctx = consts.WithTokenJTI(ctx, claims.ID)
|
||||
if claims.ExpiresAt != nil {
|
||||
ctx = consts.WithTokenExpire(ctx, claims.ExpiresAt.Time)
|
||||
}
|
||||
|
||||
// 身份扩展（角色/权限/数据范围）：缓存优先
|
||||
if idn := loadIdentity(r, claims.UserID); idn != nil {
|
||||
ctx = consts.WithRoles(ctx, idn.Roles)
|
||||
ctx = consts.WithPerms(ctx, idn.Perms)
|
||||
ctx = consts.WithDataScope(ctx, idn.DataScope)
|
||||
ctx = consts.WithDeptID(ctx, idn.DeptId)
|
||||
}
|
||||
r.SetCtx(ctx)
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// loadIdentity 读取用户身份
|
||||
//
|
||||
// 优先读取身份缓存，未命中时回源 IdentityLoader 并写入缓存
|
||||
func loadIdentity(r *ghttp.Request, userID int) *Identity {
|
||||
if userID <= 0 {
|
||||
return nil
|
||||
}
|
||||
ck := consts.CacheKeyUserIdentity(userID)
|
||||
if v, ok := cache.Get(r.Context(), ck); ok {
|
||||
idn := &Identity{}
|
||||
if err := json.Unmarshal([]byte(v), idn); err == nil {
|
||||
return idn
|
||||
}
|
||||
}
|
||||
if IdentityLoader == nil {
|
||||
return &Identity{}
|
||||
}
|
||||
idn, err := IdentityLoader(r.Context(), userID)
|
||||
if err != nil || idn == nil {
|
||||
glogWarn(r.Context(), "加载用户身份失败 userID=%d err=%v", userID, err)
|
||||
return &Identity{}
|
||||
}
|
||||
if b, err := json.Marshal(idn); err == nil {
|
||||
_ = cache.Set(r.Context(), ck, string(b), 300)
|
||||
}
|
||||
return idn
|
||||
}
|
||||
|
||||
// parseBearer 解析 Bearer 令牌
|
||||
//
|
||||
// 兼容带前缀与直接传值两种形式
|
||||
func parseBearer(h string) string {
|
||||
if h == "" {
|
||||
return ""
|
||||
}
|
||||
if strings.HasPrefix(h, "Bearer ") || strings.HasPrefix(h, "bearer ") {
|
||||
return strings.TrimSpace(h[7:])
|
||||
}
|
||||
return strings.TrimSpace(h)
|
||||
}
|
||||
|
||||
// writeUnauthorized 输出未认证响应
|
||||
//
|
||||
// 状态码为 401，供前端识别并跳转登录页
|
||||
func writeUnauthorized(r *ghttp.Request) {
|
||||
res := response.Fail(gerror.NewCode(consts.CodeUnauthorized))
|
||||
response.WriteWithCode(r, res, 401)
|
||||
}
|
||||
@@ -0,0 +1,56 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
)
|
||||
|
||||
// ============================== 日志工具 ==============================
|
||||
|
||||
// glogWarn 中间件统一告警输出
|
||||
// 以警告级别记录中间件运行时的异常或降级信息
|
||||
func glogWarn(ctx context.Context, format string, args ...any) {
|
||||
g.Log().Warningf(ctx, format, args...)
|
||||
}
|
||||
@@ -0,0 +1,209 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/internal/router/meta"
|
||||
"easygoadmin/utility/response"
|
||||
"easygoadmin/utility/ua"
|
||||
)
|
||||
|
||||
// ============================== 操作日志中间件 ==============================
|
||||
//
|
||||
// 请求结束后根据路由元数据（gmeta log/logType 标签）组装记录，
|
||||
// 经有界通道异步批量落库，不阻塞业务请求。
|
||||
//
|
||||
// 开关：operationLog.enable 为总开关；operationLog.logQuery 控制查询类是否记录。
|
||||
|
||||
const (
|
||||
logChanSize = 2048 // 日志通道容量
|
||||
logBatchSize = 100 // 单批落库条数
|
||||
logFlushEvery = time.Second // 定时刷新间隔
|
||||
logMaxLen = 8000 // param/result 截断长度
|
||||
)
|
||||
|
||||
var (
|
||||
logOnce sync.Once // 保证日志工作协程仅启动一次
|
||||
logChan = make(chan *OperationLogEntry, logChanSize) // 日志缓冲通道
|
||||
logDrop int64 // 累计丢弃条数
|
||||
logMu sync.Mutex // 丢弃计数互斥锁
|
||||
)
|
||||
|
||||
// OperationLog 操作日志采集
|
||||
//
|
||||
// 依据路由元数据判断是否记录，组装日志记录后投入异步通道
|
||||
func OperationLog(r *ghttp.Request) {
|
||||
if !cfgBool(r.Context(), "operationLog.enable", true) {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
start := time.Now()
|
||||
r.Middleware.Next()
|
||||
|
||||
rm, ok := r.Context().Value(meta.CtxKey).(meta.RouteMeta)
|
||||
if !ok || rm.Log == "" {
|
||||
return
|
||||
}
|
||||
logType := consts.LogTypeMap[rm.LogType]
|
||||
if logType == consts.LogTypeQuery && !cfgBool(r.Context(), "operationLog.logQuery", true) {
|
||||
return
|
||||
}
|
||||
|
||||
code, _ := response.GetCode(r)
|
||||
entry := &OperationLogEntry{
|
||||
Title: rm.Log,
|
||||
Type: logType,
|
||||
Source: 0,
|
||||
Method: rm.Method + " " + rm.Path,
|
||||
RequestMethod: r.Method,
|
||||
URL: r.URL.String(),
|
||||
Param: trunc(collectParam(r), logMaxLen),
|
||||
Result: trunc(collectResult(r), logMaxLen),
|
||||
IP: realClientIP(r),
|
||||
Location: requestLocation(r.Context(), realClientIP(r)),
|
||||
OS: ua.ParseOS(r.Header.Get("User-Agent")),
|
||||
Browser: ua.ParseBrowser(r.Header.Get("User-Agent")),
|
||||
ConsumeTime: time.Since(start).Milliseconds(),
|
||||
Status: 0,
|
||||
Error: "",
|
||||
CreateUser: consts.GetUsername(r.Context()),
|
||||
}
|
||||
if code != 0 {
|
||||
entry.Status = 1
|
||||
if body, ok := response.GetBody(r); ok && body.Msg != "" {
|
||||
entry.Error = trunc(body.Msg, 500)
|
||||
}
|
||||
}
|
||||
|
||||
logOnce.Do(startLogWorker)
|
||||
select {
|
||||
case logChan <- entry:
|
||||
default:
|
||||
logMu.Lock()
|
||||
logDrop++
|
||||
n := logDrop
|
||||
logMu.Unlock()
|
||||
if n%100 == 1 {
|
||||
g.Log().Warningf(r.Context(), "操作日志通道已满，累计丢弃 %d 条", n)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// collectParam 组装脱敏后的请求参数
|
||||
//
|
||||
// GET 与 DELETE 请求取 query 参数，写操作优先取请求体
|
||||
func collectParam(r *ghttp.Request) string {
|
||||
if r.Method == "GET" || r.Method == "DELETE" {
|
||||
m := r.GetMap()
|
||||
if len(m) == 0 {
|
||||
return ""
|
||||
}
|
||||
return trunc(gjsonEncode(m), logMaxLen)
|
||||
}
|
||||
body := r.GetBodyString()
|
||||
if body != "" {
|
||||
return maskSecret(body)
|
||||
}
|
||||
return trunc(gjsonEncode(r.GetMap()), logMaxLen)
|
||||
}
|
||||
|
||||
// collectResult 组装响应信封内容
|
||||
func collectResult(r *ghttp.Request) string {
|
||||
body, ok := response.GetBody(r)
|
||||
if !ok {
|
||||
return ""
|
||||
}
|
||||
return trunc(gjsonEncode(body), logMaxLen)
|
||||
}
|
||||
|
||||
// maskSecret 脱敏敏感字段
|
||||
//
|
||||
// 覆盖密码、盐与令牌等字段
|
||||
func maskSecret(s string) string {
|
||||
for _, k := range []string{"password", "Password", "oldPassword", "newPassword", "salt", "access_token", "refresh_token"} {
|
||||
s = maskJSONField(s, k)
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// startLogWorker 启动异步落库工作协程
|
||||
//
|
||||
// 按批量大小或定时触发刷新，将缓冲日志写入 LogSink
|
||||
func startLogWorker() {
|
||||
go func() {
|
||||
batch := make([]*OperationLogEntry, 0, logBatchSize)
|
||||
flush := func() {
|
||||
if len(batch) == 0 || LogSink == nil {
|
||||
batch = batch[:0]
|
||||
return
|
||||
}
|
||||
ctx := context.Background()
|
||||
for _, e := range batch {
|
||||
if err := LogSink(ctx, e); err != nil {
|
||||
g.Log().Errorf(ctx, "操作日志落库失败: %v", err)
|
||||
}
|
||||
}
|
||||
batch = batch[:0]
|
||||
}
|
||||
ticker := time.NewTicker(logFlushEvery)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
select {
|
||||
case e := <-logChan:
|
||||
batch = append(batch, e)
|
||||
if len(batch) >= logBatchSize {
|
||||
flush()
|
||||
}
|
||||
case <-ticker.C:
|
||||
flush()
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
@@ -0,0 +1,74 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import "strings"
|
||||
|
||||
// ============================== 免登录路由白名单 ==============================
|
||||
|
||||
// publicPaths 免登录路由白名单
|
||||
//
|
||||
// 精确匹配，公开路由一律不带版本前缀；
|
||||
// 须与各注册处保持一致：auth.go 的 gmeta 标注（登录、验证码、令牌兑换）
|
||||
// 以及 router.go 中的健康检查
|
||||
var publicPaths = map[string]bool{
|
||||
"/api/login": true,
|
||||
"/api/register": true,
|
||||
"/api/captcha": true,
|
||||
"/api/oauth2/token": true, // refresh_token 兑换
|
||||
"/api/health": true, // 健康检查
|
||||
}
|
||||
|
||||
// IsPublicPath 判断是否为免登录路由
|
||||
//
|
||||
// 供身份认证与租户中间件据此放行；
|
||||
// 同时放行 OpenAPI 与 Swagger 文档路径
|
||||
func IsPublicPath(path string) bool {
|
||||
if publicPaths[path] {
|
||||
return true
|
||||
}
|
||||
// OpenAPI / Swagger 文档
|
||||
if path == "/api.json" || strings.HasPrefix(path, "/swagger") {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
@@ -0,0 +1,143 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/utility/cache"
|
||||
"easygoadmin/utility/response"
|
||||
)
|
||||
|
||||
// ============================== API 限流中间件 ==============================
|
||||
//
|
||||
// 固定窗口计数：键按窗口起点对齐（ratelimit:{ip}:{path}:{窗口起点}），
|
||||
// 计数存 utility/cache（Redis 优先、内存降级），多实例部署时共享计数。
|
||||
//
|
||||
// 规则：
|
||||
// - 仅限 /api/ 前缀接口；文档（/api.json）、静态资源、健康检查不受限
|
||||
// - 认证入口（登录/验证码/令牌兑换）按 rateLimit.authLimit 单独更严阈值
|
||||
// - 计数存储异常时放行（限流不阻断业务）
|
||||
|
||||
// RateLimit API 限流
|
||||
func RateLimit(r *ghttp.Request) {
|
||||
if !cfgBool(r.Context(), "rateLimit.enable", true) {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
path := r.URL.Path
|
||||
if !isRateLimitedPath(path) {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
window := cfgInt(r.Context(), "rateLimit.windowSeconds", 60)
|
||||
limit := cfgInt(r.Context(), "rateLimit.limit", 120)
|
||||
if isAuthPath(path) {
|
||||
limit = cfgInt(r.Context(), "rateLimit.authLimit", 10)
|
||||
}
|
||||
if limit <= 0 || window <= 0 {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
|
||||
// 窗口起点对齐：同一窗口内共享计数键
|
||||
windowStart := time.Now().Unix() / int64(window) * int64(window)
|
||||
key := fmt.Sprintf("ratelimit:%s:%s:%d", realClientIP(r), path, windowStart)
|
||||
n, err := incrCounter(r, key, window)
|
||||
if err != nil {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
if n > int64(limit) {
|
||||
response.Write(r, response.Fail(gerror.NewCode(consts.CodeTooManyRequests)))
|
||||
return
|
||||
}
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// isRateLimitedPath 判断是否为参与限流的接口路径
|
||||
// /api.json 与 /swagger 不带 /api/ 前缀，自然排除；健康检查显式豁免
|
||||
func isRateLimitedPath(path string) bool {
|
||||
if path == "/api/health" {
|
||||
return false
|
||||
}
|
||||
return path == "/api" || strings.HasPrefix(path, "/api/")
|
||||
}
|
||||
|
||||
// isAuthPath 判断是否为认证入口（适用更严格的独立阈值）
|
||||
func isAuthPath(path string) bool {
|
||||
switch path {
|
||||
case "/api/login", "/api/captcha", "/api/oauth2/token":
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// incrCounter 固定窗口计数自增
|
||||
// 键已含窗口起点，TTL 仅用于回收存储，续期不影响窗口归属
|
||||
func incrCounter(r *ghttp.Request, key string, windowSeconds int) (int64, error) {
|
||||
ttl := windowSeconds + 5
|
||||
if raw, ok := cache.Get(r.Context(), key); ok {
|
||||
n, _ := strconv.ParseInt(raw, 10, 64)
|
||||
n++
|
||||
_ = cache.Set(r.Context(), key, strconv.FormatInt(n, 10), ttl)
|
||||
return n, nil
|
||||
}
|
||||
if cache.SetNX(r.Context(), key, "1", ttl) {
|
||||
return 1, nil
|
||||
}
|
||||
// 并发下 SetNX 未抢到：读现值再加一
|
||||
if raw, ok := cache.Get(r.Context(), key); ok {
|
||||
n, _ := strconv.ParseInt(raw, 10, 64)
|
||||
n++
|
||||
_ = cache.Set(r.Context(), key, strconv.FormatInt(n, 10), ttl)
|
||||
return n, nil
|
||||
}
|
||||
return 1, nil
|
||||
}
|
||||
@@ -0,0 +1,89 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"runtime/debug"
|
||||
"strings"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/utility/response"
|
||||
)
|
||||
|
||||
// ============================== 统一响应兜底中间件 ==============================
|
||||
|
||||
// Response 统一响应兜底
|
||||
//
|
||||
// 负责 panic 恢复并输出统一 JSON 错误（前端契约 {code,msg,data}），
|
||||
// 置于跨域之后、认证之前，保证任何环节的 panic 都不会泄漏堆栈到前端。
|
||||
// 同时将请求元信息（IP、UA、方法、URL）写入上下文，供登录日志与操作日志采集。
|
||||
func Response(r *ghttp.Request) {
|
||||
// SVG 静态文件加固：强制附件下载 + nosniff（防内嵌脚本同源执行）
|
||||
if strings.HasSuffix(strings.ToLower(r.URL.Path), ".svg") {
|
||||
r.Response.Header().Set("Content-Disposition", "attachment")
|
||||
r.Response.Header().Set("X-Content-Type-Options", "nosniff")
|
||||
}
|
||||
// 请求元信息（所有路由均需，登录日志依赖）
|
||||
r.SetCtx(consts.WithRequestMeta(r.Context(),
|
||||
realClientIP(r),
|
||||
r.Header.Get("User-Agent"),
|
||||
r.Method,
|
||||
r.URL.String(),
|
||||
))
|
||||
defer func() {
|
||||
if e := recover(); e != nil {
|
||||
g.Log().Errorf(r.Context(), "panic recovered: %v\n%s", e, debug.Stack())
|
||||
if _, written := response.GetCode(r); written {
|
||||
return // 已写出响应，仅记录
|
||||
}
|
||||
response.WriteWithCode(r,
|
||||
response.Fail(gerror.NewCode(consts.CodeServerError, fmt.Sprintf("服务器内部错误: %v", e))),
|
||||
500)
|
||||
}
|
||||
}()
|
||||
r.Middleware.Next()
|
||||
}
|
||||
@@ -0,0 +1,111 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/utility/cache"
|
||||
"easygoadmin/utility/response"
|
||||
)
|
||||
|
||||
// ============================== 租户校验中间件 ==============================
|
||||
|
||||
// Tenant 租户状态校验
|
||||
//
|
||||
// 位于身份认证中间件之后，处理规则如下：
|
||||
// - 超管（上下文租户ID=0）直接放行
|
||||
// - 租户级校验状态与过期时间，摘要缓存 60 秒
|
||||
func Tenant(r *ghttp.Request) {
|
||||
if IsPublicPath(r.URL.Path) {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
tenantID := consts.CtxTenantID(r.Context())
|
||||
if tenantID == 0 {
|
||||
r.Middleware.Next()
|
||||
return
|
||||
}
|
||||
|
||||
info := loadTenantInfo(r, tenantID)
|
||||
if info != nil {
|
||||
if info.Status != 1 {
|
||||
response.WriteWithCode(r, response.Fail(gerror.NewCode(consts.CodeTenantDisabled)), 1)
|
||||
return
|
||||
}
|
||||
if info.ExpireTime != nil && info.ExpireTime.Time.Before(time.Now()) {
|
||||
response.WriteWithCode(r, response.Fail(gerror.NewCode(consts.CodeTenantExpired)), 1)
|
||||
return
|
||||
}
|
||||
}
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// loadTenantInfo 加载租户摘要
|
||||
//
|
||||
// 优先读取缓存（60 秒），未命中时回源 TenantLoader 并写缓存
|
||||
func loadTenantInfo(r *ghttp.Request, tenantID int) *TenantInfo {
|
||||
ck := consts.CacheKeyTenantInfo(tenantID)
|
||||
if v, ok := cache.Get(r.Context(), ck); ok {
|
||||
info := &TenantInfo{}
|
||||
if err := json.Unmarshal([]byte(v), info); err == nil {
|
||||
return info
|
||||
}
|
||||
}
|
||||
if TenantLoader == nil {
|
||||
return nil
|
||||
}
|
||||
info, err := TenantLoader(r.Context(), tenantID)
|
||||
if err != nil || info == nil {
|
||||
glogWarn(r.Context(), "加载租户信息失败 tenantID=%d err=%v", tenantID, err)
|
||||
return nil
|
||||
}
|
||||
if b, err := json.Marshal(info); err == nil {
|
||||
_ = cache.Set(r.Context(), ck, string(b), 60)
|
||||
}
|
||||
return info
|
||||
}
|
||||
@@ -0,0 +1,168 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net"
|
||||
"regexp"
|
||||
"strings"
|
||||
"unicode/utf8"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
|
||||
"easygoadmin/utility/ip2region"
|
||||
)
|
||||
|
||||
// ============================== 中间件通用工具 ==============================
|
||||
|
||||
// cfgBool 读取布尔配置
|
||||
// 配置缺失时返回默认值
|
||||
func cfgBool(ctx context.Context, key string, def bool) bool {
|
||||
v := g.Cfg().MustGet(ctx, key)
|
||||
if v.IsNil() {
|
||||
return def
|
||||
}
|
||||
return v.Bool()
|
||||
}
|
||||
|
||||
// cfgInt 读取整数配置
|
||||
// 配置缺失时返回默认值（显式配置 0 表示业务含义上的关闭，原样返回）
|
||||
func cfgInt(ctx context.Context, key string, def int) int {
|
||||
v := g.Cfg().MustGet(ctx, key)
|
||||
if v.IsNil() {
|
||||
return def
|
||||
}
|
||||
return v.Int()
|
||||
}
|
||||
|
||||
// gjsonEncode JSON 序列化
|
||||
// 序列化失败时返回空串
|
||||
func gjsonEncode(v any) string {
|
||||
b, err := json.Marshal(v)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
// trunc 按字节截断且保证 UTF-8 完整
|
||||
//
|
||||
// 截断点回退到完整的字符边界，避免 MySQL 严格模式拒绝非法字节
|
||||
func trunc(s string, n int) string {
|
||||
if len(s) <= n {
|
||||
return s
|
||||
}
|
||||
for n > 0 && !utf8.RuneStart(s[n]) {
|
||||
n--
|
||||
}
|
||||
return s[:n] + "..."
|
||||
}
|
||||
|
||||
// maskJSONField 脱敏 JSON 文本中的指定字段
|
||||
// 将该字段的值替换为 ***
|
||||
var maskReCache = map[string]*regexp.Regexp{}
|
||||
|
||||
func maskJSONField(s, field string) string {
|
||||
re, ok := maskReCache[field]
|
||||
if !ok {
|
||||
re = regexp.MustCompile(`("` + regexp.QuoteMeta(field) + `"\s*:\s*)"?[^",}]*"?`)
|
||||
maskReCache[field] = re
|
||||
}
|
||||
return re.ReplaceAllString(s, `${1}***`)
|
||||
}
|
||||
|
||||
// realClientIP 获取客户端真实IP
|
||||
//
|
||||
// 读取配置 api.proxyIps，直连来源为可信代理时，
|
||||
// 依次取 X-Forwarded-For 首段或 X-Real-IP
|
||||
func realClientIP(r *ghttp.Request) string {
|
||||
remote := r.GetRemoteIp()
|
||||
if !isTrustedProxy(r.Context(), remote) {
|
||||
return remote
|
||||
}
|
||||
if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
|
||||
if ip := strings.TrimSpace(strings.Split(xff, ",")[0]); ip != "" {
|
||||
return ip
|
||||
}
|
||||
}
|
||||
if real := strings.TrimSpace(r.Header.Get("X-Real-IP")); real != "" {
|
||||
return real
|
||||
}
|
||||
return remote
|
||||
}
|
||||
|
||||
// isTrustedProxy 判断来源是否为可信代理
|
||||
// 支持精确 IP 与 CIDR 两种配置形式
|
||||
func isTrustedProxy(ctx context.Context, ip string) bool {
|
||||
v := g.Cfg().MustGet(ctx, "api.proxyIps")
|
||||
if v.IsNil() || v.String() == "" {
|
||||
return false
|
||||
}
|
||||
parsed := net.ParseIP(ip)
|
||||
if parsed == nil {
|
||||
return false
|
||||
}
|
||||
for _, item := range strings.Split(v.String(), ",") {
|
||||
item = strings.TrimSpace(item)
|
||||
if item == "" {
|
||||
continue
|
||||
}
|
||||
if item == ip {
|
||||
return true
|
||||
}
|
||||
if _, cidr, err := net.ParseCIDR(item); err == nil && cidr.Contains(parsed) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// requestLocation 按请求IP解析归属地
|
||||
//
|
||||
// 基于 ip2region 实现，解析失败时返回 "-"
|
||||
func requestLocation(ctx context.Context, ip string) string {
|
||||
_ = ctx
|
||||
return ip2region.Location(ip)
|
||||
}
|
||||
@@ -0,0 +1,128 @@
|
||||
// +======================================================================
|
||||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||||
// +----------------------------------------------------------------------
|
||||
// | 开源协议: Apache-2.0
|
||||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||||
// +----------------------------------------------------------------------
|
||||
// | 官方网站: https://www.easygoadmin.vip
|
||||
// +----------------------------------------------------------------------
|
||||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||||
// +======================================================================
|
||||
// | 版权和免责声明:
|
||||
// | 一、知识产权声明
|
||||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||||
// | 二、使用限制与禁止条款
|
||||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||||
// | 三、责任豁免
|
||||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||||
// | 四、违约与法律责任
|
||||
// | 如发现任何违反本声明的行为，本团队有权：
|
||||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||||
// | 五、许可范围说明
|
||||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||||
// | 成的一切经济损失及商誉损害。
|
||||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||||
// | 许可协议》附件为准。
|
||||
// +======================================================================
|
||||
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/os/gtime"
|
||||
)
|
||||
|
||||
// ============================================================================
|
||||
// 依赖接线：中间件不直接依赖业务层，由 cmd 在启动时注册加载器与落地器。
|
||||
// ============================================================================
|
||||
|
||||
// ============================== 依赖类型 ==============================
|
||||
|
||||
// Identity 用户身份扩展信息
|
||||
// 包含角色编码与权限节点
|
||||
type Identity struct {
|
||||
Roles []string `json:"roles"` // 角色编码集合
|
||||
Perms []string `json:"perms"` // 权限节点集合
|
||||
DataScope int `json:"dataScope"` // 数据范围（多角色取最宽松；0=旧缓存缺字段，按全部处理）
|
||||
DeptId int `json:"deptId"` // 所属部门ID（0=未分配）
|
||||
}
|
||||
|
||||
// TenantInfo 租户摘要
|
||||
// 仅保留状态与过期校验热路径所需的最小字段
|
||||
type TenantInfo struct {
|
||||
ID int `json:"id"` // 租户ID
|
||||
Status int `json:"status"` // 状态：1-正常 2-禁用
|
||||
ExpireTime *gtime.Time `json:"expireTime"` // 过期时间，nil 表示永不过期
|
||||
}
|
||||
|
||||
// OperationLogEntry 操作日志记录
|
||||
// 字段覆盖操作日志表全部列
|
||||
type OperationLogEntry struct {
|
||||
Title string `json:"title"` // 日志标题
|
||||
Type int `json:"type"` // 操作类型
|
||||
Source int `json:"source"` // 来源
|
||||
Method string `json:"method"` // 路由模板
|
||||
RequestMethod string `json:"requestMethod"` // HTTP 方法
|
||||
URL string `json:"url"` // 请求地址
|
||||
Param string `json:"param"` // 脱敏后的请求参数
|
||||
Result string `json:"result"` // 响应报文（截断）
|
||||
IP string `json:"ip"` // 客户端IP
|
||||
Location string `json:"location"` // 归属地
|
||||
OS string `json:"os"` // 操作系统
|
||||
Browser string `json:"browser"` // 浏览器
|
||||
ConsumeTime int64 `json:"consumeTime"` // 耗时（毫秒）
|
||||
Status int `json:"status"` // 状态：0-正常 1-异常
|
||||
Error string `json:"error"` // 错误信息
|
||||
CreateUser string `json:"createUser"` // 操作人
|
||||
}
|
||||
|
||||
// ============================== 依赖注册 ==============================
|
||||
|
||||
var (
|
||||
// IdentityLoader 加载用户角色与权限
|
||||
// 为 nil 时跳过加载，仅认证身份
|
||||
IdentityLoader func(ctx context.Context, userID int) (*Identity, error)
|
||||
|
||||
// TenantLoader 加载租户状态摘要
|
||||
// 为 nil 时跳过租户校验
|
||||
TenantLoader func(ctx context.Context, tenantID int) (*TenantInfo, error)
|
||||
|
||||
// LogSink 操作日志异步落库
|
||||
// 为 nil 时降级为调试日志
|
||||
LogSink func(ctx context.Context, entry *OperationLogEntry) error
|
||||
)
|
||||
|
||||
// SetIdentityLoader 注册身份加载器
|
||||
func SetIdentityLoader(f func(ctx context.Context, userID int) (*Identity, error)) {
|
||||
IdentityLoader = f
|
||||
}
|
||||
|
||||
// SetTenantLoader 注册租户加载器
|
||||
func SetTenantLoader(f func(ctx context.Context, tenantID int) (*TenantInfo, error)) {
|
||||
TenantLoader = f
|
||||
}
|
||||
|
||||
// SetLogSink 注册操作日志落地器
|
||||
func SetLogSink(f func(ctx context.Context, entry *OperationLogEntry) error) {
|
||||
LogSink = f
|
||||
}
|
||||
|
||||
// Now 返回当前时间
|
||||
// 供各中间件统一使用
|
||||
func Now() time.Time { return time.Now() }
|
||||
Reference in New Issue
Block a user