tcjlj/flvx
1
0
Fork 0
mirror of https://github.com/Sagit-chu/flvx.git synced 2026-09-29 07:56:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 批量设置用户最大连接数

Browse Source 
前端:
- 用户列表添加 checkbox 多选（表格和网格视图）
- 选中用户后显示「批量设置连接数」按钮
- 批量设置对话框，输入连接数后批量应用

后端:
- 新增 BatchUpdateUserMaxConn repo 方法
- 新增 /api/v1/user/batch-set-max-conn API
- 返回成功/失败计数
- 加入 admin-only 路由白名单
This commit is contained in:
sagitchu
2026-04-25 22:03:42 +08:00
parent 3f374df724
commit 28568244bb
6 changed files with 175 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
go-backend/internal/http/handler/handler.go
+1
View File
@@ -138,6 +138,7 @@ func (h *Handler) Register(mux *http.ServeMux) {
mux.HandleFunc("/api/v1/user/update", h.userUpdate)
mux.HandleFunc("/api/v1/user/delete", h.userDelete)
mux.HandleFunc("/api/v1/user/reset", h.userResetFlow)
mux.HandleFunc("/api/v1/user/batch-set-max-conn", h.userBatchSetMaxConn)
mux.HandleFunc("/api/v1/user/quota/reset", h.userQuotaReset)
mux.HandleFunc("/api/v1/user/groups", h.userGroups)
mux.HandleFunc("/api/v1/config/get", h.getConfigByName)
go-backend/internal/http/handler/mutations.go
+20
View File
@@ -214,6 +214,26 @@ func (h *Handler) userUpdate(w http.ResponseWriter, r *http.Request) {
response.WriteJSON(w, response.OKEmpty())
}
func (h *Handler) userBatchSetMaxConn(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
response.WriteJSON(w, response.ErrDefault("请求失败"))
return
}
var req struct {
UserIDs []int64 `json:"userIds"`
MaxConn int `json:"maxConn"`
}
if err := decodeJSON(r.Body, &req); err != nil || len(req.UserIDs) == 0 {
response.WriteJSON(w, response.ErrDefault("请求参数错误"))
return
}
s, f := h.repo.BatchUpdateUserMaxConn(req.UserIDs, req.MaxConn)
response.WriteJSON(w, response.OK(map[string]interface{}{
"success": s,
"failed": f,
}))
}
func (h *Handler) userGroups(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
response.WriteJSON(w, response.ErrDefault("请求失败"))
go-backend/internal/http/middleware/auth.go
+1 -1
View File
@@ -137,7 +137,7 @@ func requiresAdmin(path string) bool {
}
switch path {
case "/api/v1/user/create", "/api/v1/user/list", "/api/v1/user/update", "/api/v1/user/delete", "/api/v1/user/reset":
case "/api/v1/user/create", "/api/v1/user/list", "/api/v1/user/update", "/api/v1/user/delete", "/api/v1/user/reset", "/api/v1/user/batch-set-max-conn":
return true
case "/api/v1/config/update", "/api/v1/config/update-single":
return true
go-backend/internal/store/repo/repository_mutations.go
+17
View File
@@ -1320,6 +1320,23 @@ func (r *Repository) BatchUpdateForwardStatus(ids []int64, status int) (int, int
return s, f
}
func (r *Repository) BatchUpdateUserMaxConn(ids []int64, maxConn int) (int, int) {
if r == nil || r.db == nil {
return 0, len(ids)
}
s := 0
f := 0
now := time.Now().UnixMilli()
for _, id := range ids {
if err := r.db.Model(&model.User{}).Where("id = ?", id).Updates(map[string]interface{}{"max_conn": maxConn, "updated_time": now}).Error; err != nil {
f++
} else {
s++
}
}
return s, f
}
func (r *Repository) CreateTunnelTx(tx *gorm.DB, name string, trafficRatio float64, typeVal int, flow int64, now int64, status int, inIP interface{}, inx int, ipPreference string) (int64, error) {
inIPVal := nullStringFromInterface(inIP)
tunnel := model.Tunnel{
vite-frontend/src/api/index.ts
+2
View File
@@ -240,6 +240,8 @@ export const resetUserFlow = (data: { id: number; type: number }) =>
Network.post("/user/reset", data);
export const resetUserQuota = (data: UserQuotaResetPayload) =>
Network.post("/user/quota/reset", data);
export const batchSetUserMaxConn = (data: { userIds: number[]; maxConn: number }) =>
Network.post("/user/batch-set-max-conn", data);
export const getUserGroups = (id: number) =>
Network.post<number[]>("/user/groups", { id });
vite-frontend/src/pages/user.tsx
+134
View File
@@ -58,6 +58,7 @@ import {
getSpeedLimitList,
resetUserFlow,
resetUserQuota,
batchSetUserMaxConn,
getUserGroupList,
getUserGroups,
getMonitorPermissionList,
@@ -201,6 +202,10 @@ export default function UserPage() {
size: 10,
total: 0,
});
const [selectedUserIds, setSelectedUserIds] = useState<Set<number>>(new Set());
const [isBatchMaxConnModalOpen, setIsBatchMaxConnModalOpen] = useState(false);
const [batchMaxConnValue, setBatchMaxConnValue] = useState(0);
const [batchMaxConnLoading, setBatchMaxConnLoading] = useState(false);
const searchDebounceRef = useRef<ReturnType<typeof setTimeout> | null>(null);
// 用户表单相关状态
@@ -904,6 +909,60 @@ export default function UserPage() {
}
};
const handleBatchSetMaxConn = async () => {
if (selectedUserIds.size === 0) {
toast.error("请先选择用户");
return;
}
setBatchMaxConnLoading(true);
try {
const response = await batchSetUserMaxConn({
userIds: Array.from(selectedUserIds),
maxConn: batchMaxConnValue,
});
if (response.code === 0) {
const data = response.data as { success: number; failed: number } | null;
const { success = 0, failed = 0 } = data || {};
if (failed === 0) {
toast.success(`成功设置 ${success} 个用户的连接数限制`);
} else {
toast.success(`成功 ${success} 个，失败 ${failed} 个`);
}
setIsBatchMaxConnModalOpen(false);
setSelectedUserIds(new Set());
await loadUsers(searchKeyword);
} else {
toast.error(response.msg || "批量设置失败");
}
} catch {
toast.error("批量设置失败");
} finally {
setBatchMaxConnLoading(false);
}
};
const toggleSelectUser = (userId: number) => {
setSelectedUserIds((prev) => {
const next = new Set(prev);
if (next.has(userId)) {
next.delete(userId);
} else {
next.add(userId);
}
return next;
});
};
const toggleSelectAll = () => {
if (selectedUserIds.size === users.length) {
setSelectedUserIds(new Set());
} else {
setSelectedUserIds(new Set(users.map((u) => u.id)));
}
};
// 隧道流量重置相关函数
const handleResetTunnelFlow = (userTunnel: UserTunnel) => {
setTunnelToReset(userTunnel);
@@ -1046,6 +1105,19 @@ export default function UserPage() {
</div>
<div className="flex items-center gap-2">
{selectedUserIds.size > 0 && (
<Button
color="warning"
size="sm"
variant="flat"
onPress={() => {
setBatchMaxConnValue(0);
setIsBatchMaxConnModalOpen(true);
}}
>
批量设置连接数 ({selectedUserIds.size})
</Button>
)}
<Button
isIconOnly
size="sm"
@@ -1092,6 +1164,12 @@ export default function UserPage() {
}}
>
<TableHeader>
<TableColumn>
<Checkbox
isSelected={selectedUserIds.size === users.length && users.length > 0}
onValueChange={toggleSelectAll}
/>
</TableColumn>
<TableColumn>用户名</TableColumn>
<TableColumn>流量统计</TableColumn>
<TableColumn>配额限制</TableColumn>
@@ -1109,6 +1187,12 @@ export default function UserPage() {
return (
<TableRow key={user.id}>
<TableCell>
<Checkbox
isSelected={selectedUserIds.has(user.id)}
onValueChange={() => toggleSelectUser(user.id)}
/>
</TableCell>
<TableCell>
<div className="flex items-center gap-3">
<div className="flex items-center justify-center shrink-0 w-10 h-10 rounded-full bg-primary text-white font-bold text-sm relative">
@@ -1270,6 +1354,11 @@ export default function UserPage() {
<Card className="overflow-hidden h-full">
<CardHeader className="pb-2 md:pb-2">
<div className="flex justify-between items-start w-full">
<Checkbox
className="absolute top-2 right-2 z-10"
isSelected={selectedUserIds.has(user.id)}
onValueChange={() => toggleSelectUser(user.id)}
/>
<div className="flex items-center gap-3 flex-1 min-w-0">
<div className="flex items-center justify-center shrink-0 w-10 h-10 rounded-full bg-primary text-white font-bold text-sm relative">
{(user.name || user.user).slice(0, 2).toUpperCase()}
@@ -2524,6 +2613,51 @@ export default function UserPage() {
</ModalFooter>
</ModalContent>
</Modal>
{/* 批量设置最大连接数对话框 */}
<Modal
backdrop="blur"
classNames={{
base: "!w-[calc(100%-32px)] !mx-auto sm:!w-full rounded-2xl",
}}
isOpen={isBatchMaxConnModalOpen}
placement="center"
scrollBehavior="inside"
size="md"
onClose={() => setIsBatchMaxConnModalOpen(false)}
>
<ModalContent>
<ModalHeader>批量设置最大连接数</ModalHeader>
<ModalBody>
<p className="text-sm text-default-500">
已选择 <span className="font-medium text-foreground">{selectedUserIds.size}</span> 个用户
</p>
<Input
label="最大连接数"
placeholder="0 或空表示不限制"
type="number"
min="0"
value={batchMaxConnValue === 0 ? "" : String(batchMaxConnValue)}
onChange={(e) => {
const value = Math.max(Number(e.target.value) || 0, 0);
setBatchMaxConnValue(value);
}}
/>
</ModalBody>
<ModalFooter>
<Button variant="light" onPress={() => setIsBatchMaxConnModalOpen(false)}>
取消
</Button>
<Button
color="warning"
isLoading={batchMaxConnLoading}
onPress={handleBatchSetMaxConn}
>
确认设置
</Button>
</ModalFooter>
</ModalContent>
</Modal>
</AnimatedPage>
);
}