68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
package middleware
|
||
|
||
import (
|
||
"net/http"
|
||
"time"
|
||
|
||
"bdrp-server/app/main/api/internal/service"
|
||
"bdrp-server/app/main/model"
|
||
jwtx "bdrp-server/common/jwt"
|
||
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
const (
|
||
HeaderMembershipExpired = "X-Membership-Expired"
|
||
)
|
||
|
||
// MembershipExpiredInterceptor 检测代理会员是否过期，过期则写入响应头
|
||
// 依赖 ctx 中的 claims（由 AuthInterceptorMiddleware 注入）和 svcCtx 中的 AgentModel
|
||
type MembershipExpiredInterceptor struct {
|
||
AgentModel model.AgentModel
|
||
}
|
||
|
||
func NewMembershipExpiredInterceptor(agentModel model.AgentModel) *MembershipExpiredInterceptor {
|
||
return &MembershipExpiredInterceptor{
|
||
AgentModel: agentModel,
|
||
}
|
||
}
|
||
|
||
func (m *MembershipExpiredInterceptor) Handle(next http.HandlerFunc) http.HandlerFunc {
|
||
return func(w http.ResponseWriter, r *http.Request) {
|
||
// 先执行业务逻辑
|
||
next(w, r)
|
||
|
||
// 业务完成后，尝试检测会员过期状态
|
||
claims, err := getClaimsFromRequest(r)
|
||
if err != nil || claims == nil {
|
||
return
|
||
}
|
||
|
||
// 只检查正式用户
|
||
if claims.UserType != model.UserTypeNormal {
|
||
return
|
||
}
|
||
|
||
agent, err := m.AgentModel.FindOneByUserId(r.Context(), claims.UserId)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
if service.IsMembershipExpired(agent, time.Now()) {
|
||
w.Header().Set(HeaderMembershipExpired, "true")
|
||
logx.Infof("检测到代理会员已过期，写入响应头，代理ID: %d, 用户ID: %d", agent.Id, claims.UserId)
|
||
}
|
||
}
|
||
}
|
||
|
||
func getClaimsFromRequest(r *http.Request) (*jwtx.JwtClaims, error) {
|
||
value := r.Context().Value(jwtx.ExtraKey)
|
||
if value == nil {
|
||
return nil, nil
|
||
}
|
||
if claims, ok := value.(*jwtx.JwtClaims); ok {
|
||
return claims, nil
|
||
}
|
||
return nil, nil
|
||
}
|