解决了RedisServiceImpl可能的NullPointerException问题 · Pull Request !23 · guchengwuyue/yshop意象商城系统

Pull Request 描述

优化概述
本次提交优化了`RedisServiceImpl`的代码，主要解决了可能的`NullPointerException`问题，并提取了重复代码。通过这些改动，提高了代码的健壮性、可维护性和安全性。

改动清单
1. **空值检查**：
   - 在`DataType dataType = redisTemplate.type(keyString)`后增加空值检查。
   - 在`Object value = redisTemplate.opsForValue().get(keyString)`后增加空值检查。

2. **提取重复代码**：
   - 提取了`isFilteredKey`方法用于检查键是否需要过滤，减少代码冗余。

3. **增强输入验证**：
   - 在`delete`方法中增加对`key`的空值和空字符串检查。
   - 在`getCodeVal`方法中增加对`key`的空值和空字符串检查。
   - 在`saveCode`方法中增加对`key`和`val`的空值和空字符串检查。

#代码更正示例
```java
package co.yixiang.modules.monitor.service.impl;

import co.yixiang.modules.monitor.domain.vo.RedisVo;
import co.yixiang.modules.monitor.service.RedisService;
import co.yixiang.utils.PageUtil;
import co.yixiang.utils.ShopKeyUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Pageable;
import org.springframework.data.redis.connection.DataType;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;

@Service
public class RedisServiceImpl implements RedisService {

@Autowired
    private RedisTemplate<String, Object> redisTemplate;

@Value("${loginCode.expiration}")
    private Long expiration;

@Override
    public Page<RedisVo> findByKey(String key, Pageable pageable) {
        List<RedisVo> redisVos = new ArrayList<>();
        if (!"*".equals(key)) {
            key = "*" + key + "*";
        }
        for (Object s : redisTemplate.keys(key)) {
            String keyString = s.toString();
            // 过滤掉权限的缓存
            if (isFilteredKey(keyString)) {
                continue;
            }
            DataType dataType = redisTemplate.type(keyString);
            if (dataType == null || !"string".equals(dataType.code())) {
                continue;
            }
            Object value = redisTemplate.opsForValue().get(keyString);
            if (value == null) {
                continue;
            }
            RedisVo redisVo = new RedisVo(keyString, value.toString());
            redisVos.add(redisVo);
        }
        Page<RedisVo> page = new PageImpl<>(
                PageUtil.toPage(pageable.getPageNumber(), pageable.getPageSize(), redisVos),
                pageable,
                redisVos.size());
        return page;
    }

@Override
    public void delete(String key) {
        if (key != null && !key.trim().isEmpty()) {
            redisTemplate.delete(key);
        }
    }

@Override
    public void flushdb() {
        redisTemplate.getConnectionFactory().getConnection().flushDb();
    }

@Override
    public String getCodeVal(String key) {
        if (key == null || key.trim().isEmpty()) {
            return "";
        }
        try {
            Object value = redisTemplate.opsForValue().get(key);
            return value != null ? value.toString() : "";
        } catch (Exception e) {
            return "";
        }
    }

@Override
    public void saveCode(String key, Object val) {
        if (key != null && !key.trim().isEmpty() && val != null) {
            redisTemplate.opsForValue().set(key, val);
            redisTemplate.expire(key, expiration, TimeUnit.MINUTES);
        }
    }

private boolean isFilteredKey(String key) {
        return key.contains("role::loadPermissionByUser")
                || key.contains("user::loadUserByUsername")
                || key.contains("wechat")
                || key.contains("wxpay")
                || key.contains(ShopKeyUtils.getSiteUrl());
    }
}
```

测试情况
- 所有现有测试用例通过。
- 新增了针对空值检查的测试用例，确保没有`NullPointerException`抛出。
- 代码通过了项目的格式化检查。

请审核我的Pull Request，如果有任何问题或建议，我会及时做出修正。

guchengwuyue/yshop意象商城系统 .gitee-modal { width: 500px !important; }

搜索帮助

guchengwuyue/yshop意象商城系统