package com.ruoyi.approve.utils;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.ruoyi.approve.mapper.ApproveProcessMapper;
|
import com.ruoyi.approve.pojo.ApproveProcess;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.Resource;
|
import java.text.SimpleDateFormat;
|
import java.time.LocalDate;
|
import java.time.format.DateTimeFormatter;
|
import java.util.Calendar;
|
import java.util.Date;
|
import java.util.concurrent.TimeUnit;
|
|
|
//基于redis的一个每日计数器
|
@Component
|
public class DailyRedisCounter {
|
private static final String KEY_PREFIX = "daily_counter:";
|
private static final DateTimeFormatter DATE_FORMAT = DateTimeFormatter.ofPattern("yyyyMMdd");
|
private final StringRedisTemplate redisTemplate;
|
|
public DailyRedisCounter(StringRedisTemplate redisTemplate) {
|
this.redisTemplate = redisTemplate;
|
}
|
|
/**查缓存
|
* 获取指定计数器在今日的数值，并自增1
|
* @param counterName 计数器名称（例如：login_count、order_count）
|
* @return 今日自增后的计数值
|
*/
|
public long incrementAndGet(String counterName) {
|
String key = getKey(counterName);
|
long count = redisTemplate.opsForValue().increment(key, 1);
|
|
// 仅在第一次设置时设置过期时间（避免重复设置）
|
if (count == 1) {
|
long secondsUntilMidnight = calculateSecondsUntilMidnight();
|
redisTemplate.expire(key, secondsUntilMidnight, TimeUnit.SECONDS);
|
}
|
|
return count;
|
}
|
|
@Autowired
|
private ApproveProcessMapper approveProcessMapper;
|
|
/**
|
* 获取当前时间的 开始日期 ，结束日期
|
* @return
|
*/
|
public static StartAndEndDateDto getDateTime(){
|
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
|
Date date = new Date();
|
Calendar cal = Calendar.getInstance();
|
cal.setTime(date);
|
cal.add(Calendar.DATE,1);
|
String startDateTime = simpleDateFormat.format(date);
|
String endDateTime = simpleDateFormat.format(cal.getTime());
|
StartAndEndDateDto startAndEndDateDto = new StartAndEndDateDto();
|
startAndEndDateDto.setStartDate(startDateTime);
|
startAndEndDateDto.setEndDate(endDateTime);
|
return startAndEndDateDto;
|
}
|
|
/**查数据库
|
* 获取指定计数器在今日的数值，并自增1
|
* @return 今日自增后的计数值
|
*/
|
public long incrementAndGetByDb() {
|
String approveId = redisTemplate.opsForValue().get("approveNum");
|
if(approveId == null){
|
StartAndEndDateDto dateTime = getDateTime();
|
LambdaQueryWrapper<ApproveProcess> approveProcessLambdaQueryWrapper = new LambdaQueryWrapper<>();
|
approveProcessLambdaQueryWrapper
|
.eq(ApproveProcess::getApproveDelete,0)
|
.gt(ApproveProcess::getCreateTime,dateTime.getStartDate())
|
.lt(ApproveProcess::getCreateTime,dateTime.getEndDate());
|
Long aLong = approveProcessMapper.selectCount(approveProcessLambdaQueryWrapper);
|
if(aLong == null){
|
redisTemplate.opsForValue().set("approveNum","1",1L, TimeUnit.HOURS);
|
return 1;
|
}else{
|
aLong += 1;
|
redisTemplate.opsForValue().set("approveNum",aLong.toString(),1L, TimeUnit.HOURS);
|
return aLong;
|
}
|
}else{
|
Long num = Long.parseLong(approveId) + 1;
|
redisTemplate.opsForValue().set("approveNum",num.toString(),1L, TimeUnit.HOURS);
|
return Long.parseLong(approveId);
|
}
|
|
}
|
|
/**
|
* 获取指定计数器在今日的当前数值
|
* @param counterName 计数器名称
|
* @return 今日当前计数值，若不存在则返回0
|
*/
|
public long getCurrentCount(String counterName) {
|
String key = getKey(counterName);
|
String value = redisTemplate.opsForValue().get(key);
|
return value != null ? Long.parseLong(value) : 0;
|
}
|
|
/**
|
* 计算距离次日凌晨的秒数
|
*/
|
private long calculateSecondsUntilMidnight() {
|
LocalDate tomorrow = LocalDate.now().plusDays(1);
|
LocalDate midnight = tomorrow.atStartOfDay().toLocalDate();
|
return java.time.Duration.between(
|
LocalDate.now().atTime(23, 59, 59),
|
midnight.atTime(0, 0, 0)
|
).getSeconds() + 1;
|
}
|
|
/**
|
* 生成Redis键
|
*/
|
private String getKey(String counterName) {
|
String today = LocalDate.now().format(DATE_FORMAT);
|
return KEY_PREFIX + counterName + ":" + today;
|
}
|
}
|
