Reset only the selected forwarding rule's displayed upload/download usage without affecting user totals, tunnel quotas, historical statistics, nftables baselines, or running services. Closes #523
5.4 KiB
规则流量清零设计
背景
Issue #523 希望“规则”页面中每条隧道规则显示的流量使用量支持手动清零。
当前规则流量保存在 forward.in_flow 和 forward.out_flow。流量上报时,同一份增量还会累计到用户总流量、用户隧道流量和相关配额统计中。因此,本功能必须将“规则展示计数器清零”与“用户或隧道配额重置”严格区分。
目标
为单条规则提供手动流量清零能力:
- 将所选规则的上传流量和下载流量清零。
- 管理员可以清零任意规则。
- 普通用户只能清零自己的规则。
- 清零后,新产生的流量继续从零正常累计。
非目标
本功能不会:
- 修改用户总流量
user.in_flow或user.out_flow。 - 修改用户隧道流量
user_tunnel.in_flow或user_tunnel.out_flow。 - 修改每日或每月配额用量。
- 修改历史流量统计。
- 重置 nftables 节点计数器或其增量计算基线。
- 重启、暂停、恢复或重新部署规则服务。
- 增加批量流量清零功能。
后端设计
API
新增接口:
POST /api/v1/forward/reset-flow
请求体:
{
"id": 123
}
成功响应沿用统一 envelope:
{
"code": 0,
"msg": "success",
"data": null,
"ts": 0
}
具体 msg、data 和 ts 值继续由现有 response helper 生成。
参数与权限校验
Handler 执行以下步骤:
- 只接受
POST请求。 - 从 JSON 请求体读取正整数规则 ID。
- 调用现有
resolveForwardAccess:- 管理员角色可以访问任意存在的规则。
- 普通用户仅能访问
forward.user_id等于当前用户 ID 的规则。 - 对普通用户访问他人规则的情况,沿用现有逻辑返回“转发不存在”,避免暴露规则存在性。
- 调用 Repository 完成清零。
- 返回统一成功响应。
Repository
新增方法:
func (r *Repository) ResetForwardFlow(forwardID int64, now int64) error
该方法只更新指定 forward 记录:
in_flow = 0
out_flow = 0
updated_time = now
Repository 不直接操作 Handler 的身份信息,也不更新任何其他表。
并发与后续流量
清零使用单条 SQL UPDATE。agent 流量上报和 nftables 流量采集仍使用原有增量累加逻辑。清零不会重置采集基线,因此下一次采集只会把清零之后新计算出的增量加回规则计数,不会把清零前的累计值整体恢复。
若清零 SQL 与流量增量 SQL 同时执行,数据库按实际语句执行顺序决定最终值;每条更新本身保持原子性。本功能不引入暂停采集或跨节点同步流程。
前端设计
API 封装
在 vite-frontend/src/api/index.ts 新增:
export const resetForwardFlow = (id: number) =>
Network.post("/forward/reset-flow", { id });
入口
在规则页面所有单条规则操作入口中增加“流量清零”操作:
- 分组表格视图。
- 精简表格视图。
- 卡片视图。
按钮使用独立的清零/刷新语义图标和提示文本,不复用删除按钮样式。
当规则的 inFlow + outFlow 等于零时,按钮禁用,避免重复请求。
确认交互
点击按钮后打开确认弹窗,显示规则名称,并明确说明:
- 仅清零当前规则显示的上传和下载流量。
- 不影响用户总流量、用户隧道配额和历史统计。
- 操作不可撤销。
确认期间显示 loading 状态并阻止重复提交。
成功与失败
- 成功:关闭弹窗,显示成功 toast,并刷新规则列表。
- 失败:保留弹窗,显示后端错误信息或通用失败 toast。
- 刷新后,该规则上传和下载均显示为零;后续流量继续正常累计。
错误处理
- 非 POST 请求:返回现有通用请求失败响应。
- 请求体无法解析、ID 缺失或 ID 非正数:返回“请求参数错误”。
- 规则不存在或普通用户访问他人规则:返回“转发不存在”。
- Repository 更新失败:返回包含 Repository 错误信息的统一错误响应。
- 前端网络错误:显示“流量清零失败”。
测试策略
Repository 测试
验证:
- 指定规则的
in_flow、out_flow被清零。 - 指定规则的
updated_time被更新。 - 其他规则的流量不变。
- 用户总流量不变。
- 用户隧道流量不变。
- Repository 未初始化时返回错误。
Handler 测试
验证:
- 管理员能够清零任意存在的规则。
- 普通用户能够清零自己的规则。
- 普通用户不能清零他人的规则。
- 不存在的规则返回错误。
- 无效 ID 返回参数错误。
- 非 POST 请求返回请求失败。
- 成功请求不修改用户和用户隧道流量。
前端验证
项目没有配置前端测试框架,因此不新增前端单元测试。使用以下命令验证:
(cd vite-frontend && pnpm run build)
(cd vite-frontend && pnpm run lint)
后端使用:
(cd go-backend && go test ./...)
文件范围
预计修改:
go-backend/internal/http/handler/handler.gogo-backend/internal/http/handler/mutations.gogo-backend/internal/http/handler/*_test.gogo-backend/internal/store/repo/repository_mutations.gogo-backend/internal/store/repo/*_test.govite-frontend/src/api/index.tsvite-frontend/src/pages/forward.tsx
不需要数据库迁移或新增依赖。