Tenlion/wg-basic
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加redis文件处理

Browse Source
This commit is contained in:
wanggeng 2022-08-02 17:58:25 +08:00
parent 56b317684d
commit a3a5b4014d
2 changed files with 80 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
redis-cache/src/main/java/ink/wgink/redis/cache/manager/app/RedisAppTokenService.java vendored
View File

@ -11,7 +11,10 @@ import org.springframework.context.annotation.Primary;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import java.util.*;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;
/**
@ -83,6 +86,6 @@ public class RedisAppTokenService implements IAppTokenService {
}
}
redisTemplate.delete(clearTokenKeys);
LOG.trace("本次共清理超时Token:{}个", clearTokenKeys.size());
LOG.trace(">>>>>> count of timeout token: {}", clearTokenKeys.size());
}
}

75
redis-cache/src/main/java/ink/wgink/redis/cache/manager/files/RedisFilesShowCodeService.java vendored Normal file
View File

@ -0,0 +1,75 @@
package ink.wgink.redis.cache.manager.files;
import ink.wgink.exceptions.ParamsException;
import ink.wgink.interfaces.manager.IFilesShowCodeService;
import ink.wgink.pojo.bos.files.FilesShowCode;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Primary;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import java.util.HashSet;
import java.util.Set;
/**
* @ClassName: RedisFilesShowCodeServiceImpl
* @Description: 文件展示码
* @Author: wanggeng
* @Date: 2022/8/1 17:42
* @Version: 1.0
*/
@Primary
@Component
public class RedisFilesShowCodeService implements IFilesShowCodeService {
private static final Logger LOG = LoggerFactory.getLogger(RedisFilesShowCodeService.class);
@Autowired
private RedisTemplate redisTemplate;
@Override
public void addShowCode(String code, String fileId, String fileName) {
redisTemplate.opsForValue().set(FILE_SHOW_CODE_KEY + fileId, new FilesShowCode(code, fileId, fileName));
}
@Override
public void clearTimeoutShowCode() {
Set<String> keySet = redisTemplate.keys(FILE_SHOW_CODE_KEY);
long currentTimeMillis = System.currentTimeMillis();
Set<String> clearKeys = new HashSet<>(16);
long clearTimeoutShowCodeCount = 0;
for (String key : keySet) {
FilesShowCode showCode = getShowCode(key);
if (currentTimeMillis - showCode.getLatestUpdateTime() > SHOW_CODE_TIMEOUT_MILLIS) {
clearKeys.add(FILE_SHOW_CODE_KEY + showCode.getFileId());
}
}
redisTemplate.delete(clearKeys);
LOG.debug(">>>>>> count of clear timeout show code: {}", clearTimeoutShowCodeCount);
}
@Override
public FilesShowCode getShowCode(String fileId, String code) {
FilesShowCode showCode = getShowCode(fileId);
if (showCode == null) {
throw new ParamsException("查看码不存在");
}
if (!StringUtils.equals(showCode.getCode(), code)) {
throw new ParamsException("查看码不匹配");
}
return showCode;
}
@Override
public synchronized FilesShowCode getShowCode(String fileId) {
FilesShowCode showCode = (FilesShowCode) redisTemplate.opsForValue().get(FILE_SHOW_CODE_KEY + fileId);
if (showCode != null) {
showCode.setLatestUpdateTime(System.currentTimeMillis());
redisTemplate.opsForValue().set(FILE_SHOW_CODE_KEY + fileId, showCode);
}
return showCode;
}
}