643 lines
22 KiB
Python
643 lines
22 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
用户管理系统 - 代码结构验证脚本
|
|||
|
|
验证所有模块、接口、服务的完整性
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import List, Dict, Tuple
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
# 颜色输出
|
|||
|
|
class Colors:
|
|||
|
|
HEADER = '\033[95m'
|
|||
|
|
OKBLUE = '\033[94m'
|
|||
|
|
OKCYAN = '\033[96m'
|
|||
|
|
OKGREEN = '\033[92m'
|
|||
|
|
WARNING = '\033[93m'
|
|||
|
|
FAIL = '\033[91m'
|
|||
|
|
ENDC = '\033[0m'
|
|||
|
|
BOLD = '\033[1m'
|
|||
|
|
UNDERLINE = '\033[4m'
|
|||
|
|
|
|||
|
|
# 项目根目录
|
|||
|
|
PROJECT_ROOT = Path(__file__).parent.absolute()
|
|||
|
|
|
|||
|
|
# 预期的文件结构
|
|||
|
|
EXPECTED_STRUCTURE = {
|
|||
|
|
"cmd/server/main.go": "主程序入口",
|
|||
|
|
"go.mod": "Go模块依赖",
|
|||
|
|
"Makefile": "构建脚本",
|
|||
|
|
|
|||
|
|
# Domain层
|
|||
|
|
"internal/domain/user.go": "用户实体",
|
|||
|
|
"internal/domain/role.go": "角色实体",
|
|||
|
|
"internal/domain/permission.go": "权限实体",
|
|||
|
|
"internal/domain/device.go": "设备实体",
|
|||
|
|
"internal/domain/user_role.go": "用户角色关联",
|
|||
|
|
"internal/domain/role_permission.go": "角色权限关联",
|
|||
|
|
"internal/domain/login_log.go": "登录日志",
|
|||
|
|
"internal/domain/operation_log.go": "操作日志",
|
|||
|
|
|
|||
|
|
# Repository层
|
|||
|
|
"internal/repository/user.go": "用户数据访问",
|
|||
|
|
"internal/repository/role.go": "角色数据访问",
|
|||
|
|
"internal/repository/permission.go": "权限数据访问",
|
|||
|
|
"internal/repository/device.go": "设备数据访问",
|
|||
|
|
"internal/repository/user_role.go": "用户角色关联",
|
|||
|
|
"internal/repository/role_permission.go": "角色权限关联",
|
|||
|
|
|
|||
|
|
# Service层
|
|||
|
|
"internal/service/auth.go": "认证服务",
|
|||
|
|
"internal/service/user.go": "用户服务",
|
|||
|
|
"internal/service/role.go": "角色服务",
|
|||
|
|
"internal/service/permission.go": "权限服务",
|
|||
|
|
"internal/service/device.go": "设备服务",
|
|||
|
|
|
|||
|
|
# Handler层
|
|||
|
|
"internal/api/handler/auth.go": "认证处理器",
|
|||
|
|
"internal/api/handler/user.go": "用户处理器",
|
|||
|
|
"internal/api/handler/role.go": "角色处理器",
|
|||
|
|
"internal/api/handler/permission.go": "权限处理器",
|
|||
|
|
"internal/api/handler/device.go": "设备处理器",
|
|||
|
|
|
|||
|
|
# Middleware层
|
|||
|
|
"internal/api/middleware/auth.go": "认证中间件",
|
|||
|
|
"internal/api/middleware/ratelimit.go": "限流中间件",
|
|||
|
|
"internal/api/middleware/cors.go": "CORS中间件",
|
|||
|
|
"internal/api/middleware/error.go": "错误处理中间件",
|
|||
|
|
"internal/api/middleware/logger.go": "日志中间件",
|
|||
|
|
|
|||
|
|
# 路由
|
|||
|
|
"internal/api/router/router.go": "路由配置",
|
|||
|
|
|
|||
|
|
# 缓存
|
|||
|
|
"internal/cache/cache_manager.go": "缓存管理器",
|
|||
|
|
"internal/cache/l1.go": "L1本地缓存",
|
|||
|
|
"internal/cache/l2.go": "L2 Redis缓存",
|
|||
|
|
|
|||
|
|
# 监控
|
|||
|
|
"internal/monitoring/metrics.go": "Prometheus指标",
|
|||
|
|
"internal/monitoring/health.go": "健康检查",
|
|||
|
|
"internal/monitoring/middleware.go": "监控中间件",
|
|||
|
|
|
|||
|
|
# 认证
|
|||
|
|
"internal/auth/jwt.go": "JWT管理",
|
|||
|
|
"internal/auth/oauth.go": "OAuth",
|
|||
|
|
"internal/auth/password.go": "密码加密",
|
|||
|
|
|
|||
|
|
# 安全
|
|||
|
|
"internal/security/ratelimit.go": "限流",
|
|||
|
|
"internal/security/validator.go": "参数验证",
|
|||
|
|
"internal/security/encryption.go": "加密工具",
|
|||
|
|
|
|||
|
|
# 数据库
|
|||
|
|
"internal/database/db.go": "数据库初始化",
|
|||
|
|
|
|||
|
|
# 配置
|
|||
|
|
"internal/config/config.go": "配置管理",
|
|||
|
|
|
|||
|
|
# 响应
|
|||
|
|
"internal/response/response.go": "响应封装",
|
|||
|
|
|
|||
|
|
# 错误
|
|||
|
|
"pkg/errors/errors.go": "错误定义",
|
|||
|
|
"pkg/response/response.go": "响应定义",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 预期的API接口
|
|||
|
|
EXPECTED_API_ENDPOINTS = {
|
|||
|
|
"认证接口": [
|
|||
|
|
"POST /api/v1/auth/register",
|
|||
|
|
"POST /api/v1/auth/login",
|
|||
|
|
"POST /api/v1/auth/refresh",
|
|||
|
|
"POST /api/v1/auth/logout",
|
|||
|
|
"POST /api/v1/auth/login/code",
|
|||
|
|
],
|
|||
|
|
"用户接口": [
|
|||
|
|
"GET /api/v1/users",
|
|||
|
|
"GET /api/v1/users/me",
|
|||
|
|
"GET /api/v1/users/:id",
|
|||
|
|
"POST /api/v1/users",
|
|||
|
|
"PUT /api/v1/users/:id",
|
|||
|
|
"DELETE /api/v1/users/:id",
|
|||
|
|
"PUT /api/v1/users/:id/password",
|
|||
|
|
"PUT /api/v1/users/:id/status",
|
|||
|
|
"GET /api/v1/users/:id/roles",
|
|||
|
|
"PUT /api/v1/users/:id/roles",
|
|||
|
|
],
|
|||
|
|
"角色接口": [
|
|||
|
|
"GET /api/v1/roles",
|
|||
|
|
"GET /api/v1/roles/:id",
|
|||
|
|
"POST /api/v1/roles",
|
|||
|
|
"PUT /api/v1/roles/:id",
|
|||
|
|
"DELETE /api/v1/roles/:id",
|
|||
|
|
"PUT /api/v1/roles/:id/status",
|
|||
|
|
"GET /api/v1/roles/:id/permissions",
|
|||
|
|
"PUT /api/v1/roles/:id/permissions",
|
|||
|
|
],
|
|||
|
|
"权限接口": [
|
|||
|
|
"GET /api/v1/permissions",
|
|||
|
|
"GET /api/v1/permissions/tree",
|
|||
|
|
"GET /api/v1/permissions/:id",
|
|||
|
|
"POST /api/v1/permissions",
|
|||
|
|
"PUT /api/v1/permissions/:id",
|
|||
|
|
"DELETE /api/v1/permissions/:id",
|
|||
|
|
"PUT /api/v1/permissions/:id/status",
|
|||
|
|
],
|
|||
|
|
"设备接口": [
|
|||
|
|
"GET /api/v1/devices",
|
|||
|
|
"GET /api/v1/devices/:id",
|
|||
|
|
"POST /api/v1/devices",
|
|||
|
|
"PUT /api/v1/devices/:id",
|
|||
|
|
"DELETE /api/v1/devices/:id",
|
|||
|
|
"PUT /api/v1/devices/:id/status",
|
|||
|
|
"GET /api/v1/devices/users/:id",
|
|||
|
|
],
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def print_success(msg: str):
|
|||
|
|
print(f"{Colors.OKGREEN}✓{Colors.ENDC} {msg}")
|
|||
|
|
|
|||
|
|
def print_error(msg: str):
|
|||
|
|
print(f"{Colors.FAIL}✗{Colors.ENDC} {msg}")
|
|||
|
|
|
|||
|
|
def print_warning(msg: str):
|
|||
|
|
print(f"{Colors.WARNING}⚠{Colors.ENDC} {msg}")
|
|||
|
|
|
|||
|
|
def print_info(msg: str):
|
|||
|
|
print(f"{Colors.OKBLUE}ℹ{Colors.ENDC} {msg}")
|
|||
|
|
|
|||
|
|
def verify_file_structure() -> Tuple[int, int, List[str]]:
|
|||
|
|
"""验证文件结构"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}1. 文件结构验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
total = len(EXPECTED_STRUCTURE)
|
|||
|
|
exists = 0
|
|||
|
|
missing_files = []
|
|||
|
|
|
|||
|
|
for file_path, description in EXPECTED_STRUCTURE.items():
|
|||
|
|
full_path = PROJECT_ROOT / file_path
|
|||
|
|
if full_path.exists():
|
|||
|
|
exists += 1
|
|||
|
|
size = full_path.stat().st_size
|
|||
|
|
print_success(f"{file_path:<50} ({size} bytes) - {description}")
|
|||
|
|
else:
|
|||
|
|
missing_files.append(file_path)
|
|||
|
|
print_error(f"{file_path:<50} - {description}")
|
|||
|
|
|
|||
|
|
print(f"\n{Colors.OKCYAN}文件结构完成度: {exists}/{total} ({exists/total*100:.1f}%){Colors.ENDC}")
|
|||
|
|
|
|||
|
|
return total, exists, missing_files
|
|||
|
|
|
|||
|
|
def verify_service_methods() -> Dict[str, List[str]]:
|
|||
|
|
"""验证Service层方法"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}2. Service层方法验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
# Service文件及其预期方法
|
|||
|
|
service_methods = {
|
|||
|
|
"auth.go": [
|
|||
|
|
"Register",
|
|||
|
|
"Login",
|
|||
|
|
"LoginByCode",
|
|||
|
|
"SocialLogin",
|
|||
|
|
"RefreshToken",
|
|||
|
|
"Logout",
|
|||
|
|
"GetUserInfo",
|
|||
|
|
"SendVerificationCode",
|
|||
|
|
],
|
|||
|
|
"user.go": [
|
|||
|
|
"GetUser",
|
|||
|
|
"GetUserByID",
|
|||
|
|
"ListUsers",
|
|||
|
|
"CreateUser",
|
|||
|
|
"UpdateUser",
|
|||
|
|
"DeleteUser",
|
|||
|
|
"UpdateUserStatus",
|
|||
|
|
"UpdatePassword",
|
|||
|
|
"GetUserRoles",
|
|||
|
|
"AssignRoles",
|
|||
|
|
],
|
|||
|
|
"role.go": [
|
|||
|
|
"GetRole",
|
|||
|
|
"ListRoles",
|
|||
|
|
"CreateRole",
|
|||
|
|
"UpdateRole",
|
|||
|
|
"DeleteRole",
|
|||
|
|
"UpdateRoleStatus",
|
|||
|
|
"GetRolePermissions",
|
|||
|
|
"AssignPermissions",
|
|||
|
|
],
|
|||
|
|
"permission.go": [
|
|||
|
|
"GetPermission",
|
|||
|
|
"ListPermissions",
|
|||
|
|
"GetPermissionTree",
|
|||
|
|
"CreatePermission",
|
|||
|
|
"UpdatePermission",
|
|||
|
|
"DeletePermission",
|
|||
|
|
"UpdatePermissionStatus",
|
|||
|
|
"Search",
|
|||
|
|
],
|
|||
|
|
"device.go": [
|
|||
|
|
"GetDevice",
|
|||
|
|
"ListDevices",
|
|||
|
|
"CreateDevice",
|
|||
|
|
"UpdateDevice",
|
|||
|
|
"DeleteDevice",
|
|||
|
|
"UpdateDeviceStatus",
|
|||
|
|
"GetUserDevices",
|
|||
|
|
],
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
results = {}
|
|||
|
|
|
|||
|
|
for service_file, expected_methods in service_methods.items():
|
|||
|
|
print(f"\n{Colors.OKCYAN}📁 internal/service/{service_file}{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
file_path = PROJECT_ROOT / "internal/service" / service_file
|
|||
|
|
if not file_path.exists():
|
|||
|
|
print_error(f"文件不存在")
|
|||
|
|
results[service_file] = {"exists": False, "methods": []}
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
found_methods = []
|
|||
|
|
missing_methods = []
|
|||
|
|
|
|||
|
|
for method in expected_methods:
|
|||
|
|
if f"func.*{method}" in content:
|
|||
|
|
found_methods.append(method)
|
|||
|
|
print_success(f" ✓ {method}")
|
|||
|
|
else:
|
|||
|
|
missing_methods.append(method)
|
|||
|
|
print_error(f" ✗ {method}")
|
|||
|
|
|
|||
|
|
print(f" {Colors.OKCYAN}完成度: {len(found_methods)}/{len(expected_methods)}{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
results[service_file] = {
|
|||
|
|
"exists": True,
|
|||
|
|
"found": found_methods,
|
|||
|
|
"missing": missing_methods,
|
|||
|
|
"total": len(expected_methods),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return results
|
|||
|
|
|
|||
|
|
def verify_handler_methods() -> Dict[str, List[str]]:
|
|||
|
|
"""验证Handler层方法"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}3. Handler层方法验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
handler_methods = {
|
|||
|
|
"auth.go": [
|
|||
|
|
"Register",
|
|||
|
|
"Login",
|
|||
|
|
"LoginByCode",
|
|||
|
|
"SocialLogin",
|
|||
|
|
"RefreshToken",
|
|||
|
|
"Logout",
|
|||
|
|
"SendVerificationCode",
|
|||
|
|
],
|
|||
|
|
"user.go": [
|
|||
|
|
"GetUser",
|
|||
|
|
"GetCurrentUser",
|
|||
|
|
"ListUsers",
|
|||
|
|
"CreateUser",
|
|||
|
|
"UpdateUser",
|
|||
|
|
"DeleteUser",
|
|||
|
|
"UpdateUserStatus",
|
|||
|
|
"UpdatePassword",
|
|||
|
|
"GetUserRoles",
|
|||
|
|
"AssignRoles",
|
|||
|
|
],
|
|||
|
|
"role.go": [
|
|||
|
|
"CreateRole",
|
|||
|
|
"ListRoles",
|
|||
|
|
"GetRole",
|
|||
|
|
"UpdateRole",
|
|||
|
|
"DeleteRole",
|
|||
|
|
"UpdateRoleStatus",
|
|||
|
|
"GetRolePermissions",
|
|||
|
|
"AssignPermissions",
|
|||
|
|
],
|
|||
|
|
"permission.go": [
|
|||
|
|
"CreatePermission",
|
|||
|
|
"ListPermissions",
|
|||
|
|
"GetPermissionTree",
|
|||
|
|
"GetPermission",
|
|||
|
|
"UpdatePermission",
|
|||
|
|
"DeletePermission",
|
|||
|
|
"UpdatePermissionStatus",
|
|||
|
|
],
|
|||
|
|
"device.go": [
|
|||
|
|
"GetMyDevices",
|
|||
|
|
"CreateDevice",
|
|||
|
|
"GetDevice",
|
|||
|
|
"UpdateDevice",
|
|||
|
|
"DeleteDevice",
|
|||
|
|
"UpdateDeviceStatus",
|
|||
|
|
"GetUserDevices",
|
|||
|
|
],
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
results = {}
|
|||
|
|
|
|||
|
|
for handler_file, expected_methods in handler_methods.items():
|
|||
|
|
print(f"\n{Colors.OKCYAN}📁 internal/api/handler/{handler_file}{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
file_path = PROJECT_ROOT / "internal/api/handler" / handler_file
|
|||
|
|
if not file_path.exists():
|
|||
|
|
print_error(f"文件不存在")
|
|||
|
|
results[handler_file] = {"exists": False, "methods": []}
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
found_methods = []
|
|||
|
|
missing_methods = []
|
|||
|
|
|
|||
|
|
for method in expected_methods:
|
|||
|
|
if f"func.*{method}" in content:
|
|||
|
|
found_methods.append(method)
|
|||
|
|
print_success(f" ✓ {method}")
|
|||
|
|
else:
|
|||
|
|
missing_methods.append(method)
|
|||
|
|
print_error(f" ✗ {method}")
|
|||
|
|
|
|||
|
|
print(f" {Colors.OKCYAN}完成度: {len(found_methods)}/{len(expected_methods)}{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
results[handler_file] = {
|
|||
|
|
"exists": True,
|
|||
|
|
"found": found_methods,
|
|||
|
|
"missing": missing_methods,
|
|||
|
|
"total": len(expected_methods),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return results
|
|||
|
|
|
|||
|
|
def verify_middleware() -> None:
|
|||
|
|
"""验证中间件"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}4. 中间件验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
middlewares = [
|
|||
|
|
("auth.go", "认证中间件"),
|
|||
|
|
("ratelimit.go", "限流中间件"),
|
|||
|
|
("cors.go", "CORS中间件"),
|
|||
|
|
("logger.go", "日志中间件"),
|
|||
|
|
("error.go", "错误处理中间件"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
middleware_dir = PROJECT_ROOT / "internal/api/middleware"
|
|||
|
|
|
|||
|
|
for middleware_file, description in middlewares:
|
|||
|
|
file_path = middleware_dir / middleware_file
|
|||
|
|
if file_path.exists():
|
|||
|
|
print_success(f"{middleware_file:<20} - {description}")
|
|||
|
|
else:
|
|||
|
|
print_error(f"{middleware_file:<20} - {description}")
|
|||
|
|
|
|||
|
|
def verify_cache() -> None:
|
|||
|
|
"""验证缓存"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}5. 缓存模块验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
cache_files = [
|
|||
|
|
("cache_manager.go", "缓存管理器 - L1+L2多级缓存"),
|
|||
|
|
("l1.go", "L1本地缓存实现"),
|
|||
|
|
("l2.go", "L2 Redis缓存实现"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
cache_dir = PROJECT_ROOT / "internal/cache"
|
|||
|
|
|
|||
|
|
for cache_file, description in cache_files:
|
|||
|
|
file_path = cache_dir / cache_file
|
|||
|
|
if file_path.exists():
|
|||
|
|
print_success(f"{cache_file:<20} - {description}")
|
|||
|
|
else:
|
|||
|
|
print_error(f"{cache_file:<20} - {description}")
|
|||
|
|
|
|||
|
|
def verify_monitoring() -> None:
|
|||
|
|
"""验证监控"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}6. 监控模块验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
monitoring_files = [
|
|||
|
|
("metrics.go", "Prometheus指标采集"),
|
|||
|
|
("health.go", "健康检查"),
|
|||
|
|
("middleware.go", "监控中间件"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
monitoring_dir = PROJECT_ROOT / "internal/monitoring"
|
|||
|
|
|
|||
|
|
for monitoring_file, description in monitoring_files:
|
|||
|
|
file_path = monitoring_dir / monitoring_file
|
|||
|
|
if file_path.exists():
|
|||
|
|
print_success(f"{monitoring_file:<20} - {description}")
|
|||
|
|
else:
|
|||
|
|
print_error(f"{monitoring_file:<20} - {description}")
|
|||
|
|
|
|||
|
|
def verify_database() -> None:
|
|||
|
|
"""验证数据库"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}7. 数据库验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
# 检查迁移文件
|
|||
|
|
migration_files = [
|
|||
|
|
("migrations/sqlite/V1__init.sql", "SQLite数据库迁移文件"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for migration_file, description in migration_files:
|
|||
|
|
file_path = PROJECT_ROOT / migration_file
|
|||
|
|
if file_path.exists():
|
|||
|
|
print_success(f"{migration_file:<50} - {description}")
|
|||
|
|
|
|||
|
|
# 检查表定义
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
expected_tables = ["users", "roles", "permissions", "user_roles", "role_permissions", "devices", "login_logs", "operation_logs"]
|
|||
|
|
print(f"\n {Colors.OKCYAN}数据表检查:{Colors.ENDC}")
|
|||
|
|
for table in expected_tables:
|
|||
|
|
if table in content:
|
|||
|
|
print_success(f" ✓ {table}")
|
|||
|
|
else:
|
|||
|
|
print_error(f" ✗ {table}")
|
|||
|
|
else:
|
|||
|
|
print_error(f"{migration_file:<50} - {description}")
|
|||
|
|
|
|||
|
|
def verify_config() -> None:
|
|||
|
|
"""验证配置"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}8. 配置文件验证{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
config_files = [
|
|||
|
|
("configs/config.yaml", "应用配置文件"),
|
|||
|
|
("deployment/docker-compose.yml", "Docker Compose配置"),
|
|||
|
|
("deployment/alertmanager/alertmanager.yml", "AlertManager配置"),
|
|||
|
|
("deployment/alertmanager/alerts.yml", "告警规则"),
|
|||
|
|
("deployment/grafana/dashboards/user-management.json", "Grafana仪表板"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for config_file, description in config_files:
|
|||
|
|
file_path = PROJECT_ROOT / config_file
|
|||
|
|
if file_path.exists():
|
|||
|
|
print_success(f"{config_file:<50} - {description}")
|
|||
|
|
else:
|
|||
|
|
print_error(f"{config_file:<50} - {description}")
|
|||
|
|
|
|||
|
|
def count_lines_of_code() -> Dict[str, int]:
|
|||
|
|
"""统计代码行数"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}9. 代码统计{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
stats = {
|
|||
|
|
"total": 0,
|
|||
|
|
"go": 0,
|
|||
|
|
"sql": 0,
|
|||
|
|
"yaml": 0,
|
|||
|
|
"json": 0,
|
|||
|
|
"md": 0,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for file_path in PROJECT_ROOT.rglob("*"):
|
|||
|
|
if file_path.is_file():
|
|||
|
|
# 跳过特定目录
|
|||
|
|
if any(part in file_path.parts for part in [".git", "node_modules", "vendor", "__pycache__"]):
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
ext = file_path.suffix.lower()
|
|||
|
|
try:
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
lines = len(f.readlines())
|
|||
|
|
|
|||
|
|
stats["total"] += lines
|
|||
|
|
|
|||
|
|
if ext == ".go":
|
|||
|
|
stats["go"] += lines
|
|||
|
|
elif ext == ".sql":
|
|||
|
|
stats["sql"] += lines
|
|||
|
|
elif ext in [".yaml", ".yml"]:
|
|||
|
|
stats["yaml"] += lines
|
|||
|
|
elif ext == ".json":
|
|||
|
|
stats["json"] += lines
|
|||
|
|
elif ext == ".md":
|
|||
|
|
stats["md"] += lines
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
print(f"{Colors.OKCYAN}总代码行数: {stats['total']:,}{Colors.ENDC}")
|
|||
|
|
print(f" Go: {stats['go']:,} 行")
|
|||
|
|
print(f" SQL: {stats['sql']:,} 行")
|
|||
|
|
print(f" YAML: {stats['yaml']:,} 行")
|
|||
|
|
print(f" JSON: {stats['json']:,} 行")
|
|||
|
|
print(f" Markdown:{stats['md']:,} 行")
|
|||
|
|
|
|||
|
|
return stats
|
|||
|
|
|
|||
|
|
def generate_report() -> Dict:
|
|||
|
|
"""生成验证报告"""
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}10. 生成验证报告{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
# 运行所有验证
|
|||
|
|
total_files, exists_files, missing_files = verify_file_structure()
|
|||
|
|
service_results = verify_service_methods()
|
|||
|
|
handler_results = verify_handler_methods()
|
|||
|
|
verify_middleware()
|
|||
|
|
verify_cache()
|
|||
|
|
verify_monitoring()
|
|||
|
|
verify_database()
|
|||
|
|
verify_config()
|
|||
|
|
code_stats = count_lines_of_code()
|
|||
|
|
|
|||
|
|
# 计算完成度
|
|||
|
|
file_completion = exists_files / total_files * 100
|
|||
|
|
|
|||
|
|
service_total = sum(r["total"] for r in service_results.values() if r.get("exists"))
|
|||
|
|
service_found = sum(len(r.get("found", [])) for r in service_results.values() if r.get("exists"))
|
|||
|
|
service_completion = service_found / service_total * 100 if service_total > 0 else 0
|
|||
|
|
|
|||
|
|
handler_total = sum(r["total"] for r in handler_results.values() if r.get("exists"))
|
|||
|
|
handler_found = sum(len(r.get("found", [])) for r in handler_results.values() if r.get("exists"))
|
|||
|
|
handler_completion = handler_found / handler_total * 100 if handler_total > 0 else 0
|
|||
|
|
|
|||
|
|
overall_completion = (file_completion + service_completion + handler_completion) / 3
|
|||
|
|
|
|||
|
|
# 打印总结
|
|||
|
|
print(f"\n{Colors.HEADER}{'='*60}{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}📊 验证总结{Colors.ENDC}")
|
|||
|
|
print(f"{Colors.HEADER}{'='*60}{Colors.ENDC}\n")
|
|||
|
|
|
|||
|
|
print(f"{Colors.OKCYAN}文件结构:{Colors.ENDC} {file_completion:.1f}% ({exists_files}/{total_files})")
|
|||
|
|
print(f"{Colors.OKCYAN}Service层:{Colors.ENDC} {service_completion:.1f}% ({service_found}/{service_total})")
|
|||
|
|
print(f"{Colors.OKCYAN}Handler层:{Colors.ENDC} {handler_completion:.1f}% ({handler_found}/{handler_total})")
|
|||
|
|
print(f"{Colors.OKCYAN}代码总量:{Colors.ENDC} {code_stats['total']:,} 行")
|
|||
|
|
print(f"\n{Colors.BOLD}{Colors.OKGREEN}总体完成度: {overall_completion:.1f}%{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
if overall_completion >= 95:
|
|||
|
|
print(f"{Colors.OKGREEN}✓ 项目完整度优秀，可以进行验收！{Colors.ENDC}")
|
|||
|
|
elif overall_completion >= 80:
|
|||
|
|
print(f"{Colors.WARNING}⚠ 项目基本完成，但还有部分功能需要补充{Colors.ENDC}")
|
|||
|
|
else:
|
|||
|
|
print(f"{Colors.FAIL}✗ 项目完成度较低，需要继续完善{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"file_completion": file_completion,
|
|||
|
|
"service_completion": service_completion,
|
|||
|
|
"handler_completion": handler_completion,
|
|||
|
|
"overall_completion": overall_completion,
|
|||
|
|
"total_files": total_files,
|
|||
|
|
"exists_files": exists_files,
|
|||
|
|
"missing_files": missing_files,
|
|||
|
|
"service_results": service_results,
|
|||
|
|
"handler_results": handler_results,
|
|||
|
|
"code_stats": code_stats,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
print(f"{Colors.BOLD}{Colors.HEADER}")
|
|||
|
|
print("╔════════════════════════════════════════════════════════╗")
|
|||
|
|
print("║ 用户管理系统 - 代码结构验证工具 ║")
|
|||
|
|
print("║ User Management System ║")
|
|||
|
|
print("╚════════════════════════════════════════════════════════╝")
|
|||
|
|
print(f"{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
report = generate_report()
|
|||
|
|
|
|||
|
|
# 保存报告到JSON
|
|||
|
|
report_file = PROJECT_ROOT / "validation_report.json"
|
|||
|
|
with open(report_file, 'w', encoding='utf-8') as f:
|
|||
|
|
json.dump(report, f, indent=2, ensure_ascii=False)
|
|||
|
|
|
|||
|
|
print(f"\n{Colors.OKCYAN}✓ 验证报告已保存到: {report_file}{Colors.ENDC}")
|
|||
|
|
|
|||
|
|
# 退出码
|
|||
|
|
if report["overall_completion"] >= 95:
|
|||
|
|
sys.exit(0)
|
|||
|
|
else:
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n{Colors.FAIL}✗ 验证过程出错: {e}{Colors.ENDC}")
|
|||
|
|
sys.exit(1)
|