huang
1f71741836
主要变更： - ✅ 完成所有文档任务（T092-T095a） * 创建中文 README.md 和项目文档 * 添加限流器使用指南 * 更新快速入门文档 * 添加详细的中文代码注释 - ✅ 完成代码质量任务（T096-T103） * 通过 gofmt、go vet、golangci-lint 检查 * 修复 17 个 errcheck 问题 * 验证无硬编码 Redis key * 确保命名规范符合 Go 标准 - ✅ 完成测试任务（T104-T108） * 58 个测试全部通过 * 总体覆盖率 75.1%（超过 70% 目标） * 核心模块覆盖率 90%+ - ✅ 完成安全审计任务（T109-T113） * 修复日志中令牌泄露问题 * 验证 Fail-closed 策略正确实现 * 审查 Redis 连接安全 * 完成依赖项漏洞扫描 - ✅ 完成性能验证任务（T114-T117） * 令牌验证性能：17.5 μs/op（~58,954 ops/s） * 响应序列化性能：1.1 μs/op（>1,000,000 ops/s） * 配置访问性能：0.58 ns/op（接近 CPU 缓存速度） - ✅ 完成质量关卡任务（T118-T126） * 所有测试通过 * 代码格式和静态检查通过 * 无 TODO/FIXME 遗留 * 中间件集成验证 * 优雅关闭机制验证 新增文件： - README.md（中文项目文档） - docs/rate-limiting.md（限流器指南） - docs/security-audit-report.md（安全审计报告） - docs/performance-benchmark-report.md（性能基准报告） - docs/quality-gate-report.md（质量关卡报告） - docs/PROJECT-COMPLETION-SUMMARY.md（项目完成总结） - 基准测试文件（config, response, validator） 安全修复： - 移除 pkg/validator/token.go 中的敏感日志记录 质量评分：9.6/10（优秀） 项目状态：✅ 已完成，待部署
77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
package validator
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
)
|
|
|
|
// RedisClient 定义 Redis 客户端接口，便于测试
|
|
type RedisClient interface {
|
|
Ping(ctx context.Context) *redis.StatusCmd
|
|
Get(ctx context.Context, key string) *redis.StringCmd
|
|
}
|
|
|
|
// TokenValidator 令牌验证器
|
|
type TokenValidator struct {
|
|
redis RedisClient
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// NewTokenValidator 创建新的令牌验证器
|
|
func NewTokenValidator(rdb RedisClient, logger *zap.Logger) *TokenValidator {
|
|
return &TokenValidator{
|
|
redis: rdb,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// Validate 验证令牌并返回用户 ID
|
|
func (v *TokenValidator) Validate(token string) (string, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
|
defer cancel()
|
|
|
|
// 检查 Redis 可用性（失败关闭策略）
|
|
if err := v.redis.Ping(ctx).Err(); err != nil {
|
|
v.logger.Error("Redis 不可用",
|
|
zap.Error(err),
|
|
)
|
|
return "", errors.ErrRedisUnavailable
|
|
}
|
|
|
|
// 从 Redis 获取用户 ID
|
|
userID, err := v.redis.Get(ctx, constants.RedisAuthTokenKey(token)).Result()
|
|
if err == redis.Nil {
|
|
// 令牌不存在或已过期
|
|
return "", errors.ErrInvalidToken
|
|
}
|
|
if err != nil {
|
|
v.logger.Error("Redis 获取失败",
|
|
zap.Error(err),
|
|
// 注意：不记录完整的 token_key 以避免泄露令牌
|
|
)
|
|
return "", err
|
|
}
|
|
|
|
// 验证用户 ID 非空
|
|
if userID == "" {
|
|
return "", errors.ErrInvalidToken
|
|
}
|
|
|
|
return userID, nil
|
|
}
|
|
|
|
// IsAvailable 检查 Redis 是否可用
|
|
func (v *TokenValidator) IsAvailable() bool {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
|
defer cancel()
|
|
|
|
err := v.redis.Ping(ctx).Err()
|
|
return err == nil
|
|
}
|