1

zj

2025-10-13 8f886e9bb2f9204cf0b923979558482249aa8aef

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package com.ruoyi.web.controller.system;
 
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.system.domain.InvitationBlacklist;
import com.ruoyi.system.domain.IpBlacklist;
import com.ruoyi.system.service.impl.InvitationBlacklistServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
 
import java.util.Date;
import java.util.List;
 
@RestController
@RequestMapping("/system/Invitation")
public class InvitationBlacklistController extends BaseController {
 
    @Autowired
    InvitationBlacklistServiceImpl service;
 
    /**
     * 邀请码黑名单列表
     */
//    @PreAuthorize("@ss.hasPermi('im:group:list')")
    @GetMapping("/list")
    public TableDataInfo list(@RequestParam(value = "keywords", required = false) String keywords)
    {
        // 创建查询条件包装器
        LambdaQueryWrapper<InvitationBlacklist> queryWrapper = new LambdaQueryWrapper<>();
 
        // 只有当 keyword 不为空时才添加 OR 条件
        if (ObjectUtil.isNotEmpty(keywords)) {
            queryWrapper.and(wrapper -> wrapper.eq(InvitationBlacklist::getInvitationCode,keywords)
            );
        }
 
        // 默认按创建时间倒序
        queryWrapper.orderByDesc(InvitationBlacklist::getCreateTime);
        startPage();
        List<InvitationBlacklist> list = service.list(queryWrapper);
        return getDataTable(list);
    }
 
 
    /**
     * 新增邀请码黑名单
     */
    @PostMapping("/addCode")
    public AjaxResult addIp(@RequestParam("code") String code) {
 
        long count = service.count(new LambdaQueryWrapper<>(InvitationBlacklist.class).eq(InvitationBlacklist::getInvitationCode, code));
        if(count > 0){
            return AjaxResult.error("邀请码已存在！");
        }
        InvitationBlacklist invitationBlacklist = new InvitationBlacklist();
        invitationBlacklist.setInvitationCode(code);
        invitationBlacklist.setCreateTime(new Date());
        service.save(invitationBlacklist);
        return AjaxResult.success("添加成功");
    }
 
    /**
     * 删除邀请码黑名单
     */
    @PostMapping("/deleteCode")
    public AjaxResult deleteCode(@RequestParam("id") Integer id) {
 
        InvitationBlacklist invitationBlacklist  = service.getById(id);
        if(ObjectUtil.isEmpty(invitationBlacklist)){
            return AjaxResult.error("邀请码不存在！");
        }
        service.removeById(id);
        return AjaxResult.success("删除成功");
    }
 
 
}