李林
2024-03-07 e4c2789c8778fd0cc827692b9ebf554411d05673
user-server/src/main/java/com/yuanchu/mom/controller/UserController.java
@@ -7,8 +7,10 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.yuanchu.mom.annotation.ValueAuth;
import com.yuanchu.mom.common.PrintChina;
import com.yuanchu.mom.dto.Custom;
import com.yuanchu.mom.dto.UserPageDto;
import com.yuanchu.mom.pojo.User;
import com.yuanchu.mom.service.CustomService;
import com.yuanchu.mom.service.PowerService;
import com.yuanchu.mom.service.UserService;
import com.yuanchu.mom.utils.JackSonUtil;
@@ -40,16 +42,28 @@
    private Jwt jwt;
    private CustomService customService;
    @ApiOperation(value = "登录")
    @PostMapping("/enter")
    @ValueAuth
    public Result login(String account, String password) {
        User user = userService.selectUserByPwd(account, password);
        if (user == null) return Result.fail("登录失败");
        String jwtToken = jwt.createJwt(user.getName(), user.getId(), 60);
        String jwtReToken = jwt.createJwt(user.getName(), user.getId(), 24 * 60);
        RedisUtil.set("" + user.getId(), jwtToken, 60);
        RedisUtil.set(user.getId() + "Re", jwtReToken, 24 * 60);
        if (user == null) return Result.fail("账号密码错误或者账号已被停用");
        String jwtToken;
        String jwtReToken;
        if(RedisUtil.hasKey("" + user.getId())){
            jwtToken = RedisUtil.get("" + user.getId()) + "";
        }else{
            jwtToken = jwt.createJwt(user.getName(), user.getId(), 24 * 60);
            RedisUtil.set("" + user.getId(), jwtToken, 24 * 60);
        }
        if(RedisUtil.hasKey(user.getId() + "Re")){
            jwtReToken = RedisUtil.get(user.getId() + "Re") + "";
        }else{
            jwtReToken = jwt.createJwt(user.getName(), user.getId(), 48 * 60);
            RedisUtil.set(user.getId() + "Re", jwtReToken, 48 * 60);
        }
        Map<String, Object> map = new HashMap<>();
        map.put("token", jwtToken);
        map.put("reToken", jwtReToken);
@@ -71,10 +85,10 @@
            } else if (redisToken.equals(reToken)) {
                int id = Integer.parseInt(maps.get("id").replaceAll("\"", ""));
                String name = maps.get("name").replaceAll("\"", "");
                String jwtToken = jwt.createJwt(name, id, 60);
                String jwtReToken = jwt.createJwt(name, id, 24 * 60);
                RedisUtil.set("" + id, jwtToken, 60);
                RedisUtil.set(id + "Re", jwtReToken, 24 * 60);
                String jwtToken = jwt.createJwt(name, id,  24 * 60);
                String jwtReToken = jwt.createJwt(name, id, 48 * 60);
                RedisUtil.set("" + id, jwtToken, 24 * 60);
                RedisUtil.set(id + "Re", jwtReToken, 48 * 60);
                Map<String, String> map1 = new HashMap<>();
                map1.put("token", jwtToken);
                map1.put("reToken", jwtReToken);
@@ -108,4 +122,31 @@
        return Result.success(userService.addUser(user));
    }
    @ApiOperation(value = "获取用户信息枚举")
    @GetMapping("/getUserMenu")
    @ValueAuth
    public Result<?> getUserMenu() {
        return Result.success(userService.getUserMenu());
    }
    @ApiOperation(value = "获取客户列表")
    @PostMapping("/selectCustomPageList")
    public Result selectCustomPageList(@RequestBody Map<String, Object> data) throws Exception {
        Page page = JackSonUtil.unmarshal(JackSonUtil.marshal(data.get("page")), Page.class);
        Custom custom = JackSonUtil.unmarshal(JackSonUtil.marshal(data.get("entity")), Custom.class);
        return Result.success(customService.selectCustomPageList(page, custom));
    }
    @ApiOperation(value = "删除客户信息")
    @PostMapping("/delCustomById")
    public Result<?> delCustomById(Integer id) {
        return Result.success(customService.delCustomById(id));
    }
    @ApiOperation(value = "新增客户信息")
    @PostMapping("/addCustom")
    public Result<?> addCustom(@RequestBody Custom custom) {
        return Result.success(customService.addCustom(custom));
    }
}