使用Redis的Bitmap实现签到功能

一给  金牌会员 | 2024-12-8 16:01:01 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题
楼主

主题 994|帖子 994|积分 2982

马上注册，结交更多好友，享用更多功能，让你轻松玩转社区。

您需要 登录 才可以下载或查看，没有账号？立即注册

x
1.基础签到实现

  1. 1.1代码如下
复制代码
  1. @Service
  2. @Slf4j
  3. public class SignInService {
  5.     @Autowired
  6.     private StringRedisTemplate redisTemplate;
  7.    
  8.     private static final String SIGN_KEY_PREFIX = "sign:";
  9.    
  10.     /**
  11.      * 用户签到
  12.      * @param userId 用户ID
  13.      * @param date 签到日期
  14.      */
  15.     public boolean signIn(Long userId, LocalDate date) {
  16.         try {
  17.             String key = buildSignKey(userId, date);
  18.             // 获取当月的第几天
  19.             int dayOfMonth = date.getDayOfMonth();
  20.             
  21.             return Boolean.TRUE.equals(redisTemplate.opsForValue()
  22.                 .setBit(key, dayOfMonth - 1, true));
  23.         } catch (Exception e) {
  24.             log.error("签到失败", e);
  25.             throw new RuntimeException("签到失败", e);
  26.         }
  27.     }
  28.    
  29.     /**
  30.      * 检查用户是否签到
  31.      */
  32.     public boolean isSignedIn(Long userId, LocalDate date) {
  33.         try {
  34.             String key = buildSignKey(userId, date);
  35.             int dayOfMonth = date.getDayOfMonth();
  36.             
  37.             return Boolean.TRUE.equals(redisTemplate.opsForValue()
  38.                 .getBit(key, dayOfMonth - 1));
  39.         } catch (Exception e) {
  40.             log.error("检查签到状态失败", e);
  41.             throw new RuntimeException("检查签到状态失败", e);
  42.         }
  43.     }
  44.    
  45.     /**
  46.      * 获取用户当月签到次数
  47.      */
  48.     public long getMonthSignCount(Long userId, LocalDate date) {
  49.         try {
  50.             String key = buildSignKey(userId, date);
  51.             return redisTemplate.execute((RedisCallback<Long>) con ->
  52.                 con.bitCount(key.getBytes()));
  53.         } catch (Exception e) {
  54.             log.error("获取签到次数失败", e);
  55.             throw new RuntimeException("获取签到次数失败", e);
  56.         }
  57.     }
  58.    
  59.     private String buildSignKey(Long userId, LocalDate date) {
  60.         return String.format("%s%d:%s",
  61.             SIGN_KEY_PREFIX, userId, date.format(DateTimeFormatter.ofPattern("yyyyMM")));
  62.     }
  63. }
复制代码
2. 进阶功能实现

  1. 2.1代码如下
复制代码
  1. @Service
  2. @Slf4j
  3. public class AdvancedSignInService {
  5.     @Autowired
  6.     private StringRedisTemplate redisTemplate;
  7.    
  8.     /**
  9.      * 获取当月连续签到天数
  10.      */
  11.     public int getContinuousSignCount(Long userId, LocalDate date) {
  12.         try {
  13.             String key = buildSignKey(userId, date);
  14.             int dayOfMonth = date.getDayOfMonth();
  15.             int count = 0;
  16.             
  17.             // 从当天开始往前查找连续签到
  18.             for (int i = dayOfMonth - 1; i >= 0; i--) {
  19.                 if (Boolean.TRUE.equals(redisTemplate.opsForValue().getBit(key, i))) {
  20.                     count++;
  21.                 } else {
  22.                     break;
  23.                 }
  24.             }
  25.             return count;
  26.         } catch (Exception e) {
  27.             log.error("获取连续签到天数失败", e);
  28.             throw new RuntimeException("获取连续签到天数失败", e);
  29.         }
  30.     }
  31.    
  32.     /**
  33.      * 获取当月签到日历
  34.      */
  35.     public List<Boolean> getMonthSignList(Long userId, LocalDate date) {
  36.         try {
  37.             String key = buildSignKey(userId, date);
  38.             int dayCount = date.lengthOfMonth();
  39.             List<Boolean> result = new ArrayList<>(dayCount);
  40.             
  41.             for (int i = 0; i < dayCount; i++) {
  42.                 result.add(Boolean.TRUE.equals(
  43.                     redisTemplate.opsForValue().getBit(key, i)));
  44.             }
  45.             return result;
  46.         } catch (Exception e) {
  47.             log.error("获取签到日历失败", e);
  48.             throw new RuntimeException("获取签到日历失败", e);
  49.         }
  50.     }
  51.    
  52.     /**
  53.      * 补签
  54.      */
  55.     public boolean retroactiveSign(Long userId, LocalDate date) {
  56.         try {
  57.             // 检查是否可以补签
  58.             if (isSignedIn(userId, date) || date.isAfter(LocalDate.now())) {
  59.                 return false;
  60.             }
  61.             
  62.             String key = buildSignKey(userId, date);
  63.             int dayOfMonth = date.getDayOfMonth();
  64.             
  65.             return Boolean.TRUE.equals(redisTemplate.opsForValue()
  66.                 .setBit(key, dayOfMonth - 1, true));
  67.         } catch (Exception e) {
  68.             log.error("补签失败", e);
  69.             throw new RuntimeException("补签失败", e);
  70.         }
  71.     }
  72.    
  73.     /**
  74.      * 获取首次签到日期
  75.      */
  76.     public LocalDate getFirstSignDate(Long userId, LocalDate date) {
  77.         try {
  78.             String key = buildSignKey(userId, date);
  79.             int dayCount = date.lengthOfMonth();
  80.             
  81.             for (int i = 0; i < dayCount; i++) {
  82.                 if (Boolean.TRUE.equals(
  83.                     redisTemplate.opsForValue().getBit(key, i))) {
  84.                     return date.withDayOfMonth(i + 1);
  85.                 }
  86.             }
  87.             return null;
  88.         } catch (Exception e) {
  89.             log.error("获取首次签到日期失败", e);
  90.             throw new RuntimeException("获取首次签到日期失败", e);
  91.         }
  92.     }
  93. }
复制代码
3. 签到统计和嘉奖体系

  1. 3.1代码如下
复制代码
  1. @Service
  2. @Slf4j
  3. public class SignInRewardService {
  5.     @Autowired
  6.     private AdvancedSignInService signInService;
  7.    
  8.     /**
  9.      * 计算签到奖励
  10.      */
  11.     public SignInReward calculateReward(Long userId, LocalDate date) {
  12.         int continuousDays = signInService.getContinuousSignCount(userId, date);
  13.         long monthTotal = signInService.getMonthSignCount(userId, date);
  14.         
  15.         SignInReward reward = new SignInReward();
  16.         reward.setPoints(calculatePoints(continuousDays));
  17.         reward.setBonus(calculateBonus(monthTotal));
  18.         
  19.         return reward;
  20.     }
  21.    
  22.     /**
  23.      * 计算连续签到积分
  24.      */
  25.     private int calculatePoints(int continuousDays) {
  26.         if (continuousDays >= 30) return 100;
  27.         if (continuousDays >= 20) return 50;
  28.         if (continuousDays >= 10) return 30;
  29.         if (continuousDays >= 7) return 20;
  30.         if (continuousDays >= 3) return 10;
  31.         return 5;
  32.     }
  33.    
  34.     /**
  35.      * 计算月度签到奖励
  36.      */
  37.     private BigDecimal calculateBonus(long monthTotal) {
  38.         if (monthTotal >= 28) return new BigDecimal("100");
  39.         if (monthTotal >= 20) return new BigDecimal("50");
  40.         if (monthTotal >= 15) return new BigDecimal("30");
  41.         if (monthTotal >= 10) return new BigDecimal("20");
  42.         return BigDecimal.ZERO;
  43.     }
  44. }
  46. @Data
  47. public class SignInReward {
  48.     private int points;
  49.     private BigDecimal bonus;
  50. }
复制代码
4. Controller层实现

  1. 4.1.代码如下
复制代码
  1. @RestController
  2. @RequestMapping("/api/sign")
  3. @Slf4j
  4. public class SignInController {
  6.     @Autowired
  7.     private SignInService signInService;
  8.    
  9.     @Autowired
  10.     private AdvancedSignInService advancedSignInService;
  11.    
  12.     @Autowired
  13.     private SignInRewardService rewardService;
  14.    
  15.     @PostMapping("/in")
  16.     public ResponseEntity<?> signIn(@RequestParam Long userId) {
  17.         try {
  18.             LocalDate today = LocalDate.now();
  19.             boolean success = signInService.signIn(userId, today);
  20.             
  21.             if (success) {
  22.                 SignInReward reward = rewardService.calculateReward(userId, today);
  23.                 return ResponseEntity.ok(reward);
  24.             }
  25.             
  26.             return ResponseEntity.badRequest().body("签到失败");
  27.         } catch (Exception e) {
  28.             log.error("签到异常", e);
  29.             return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
  30.                 .body("系统错误");
  31.         }
  32.     }
  33.    
  34.     @GetMapping("/calendar")
  35.     public ResponseEntity<?> getSignCalendar(
  36.             @RequestParam Long userId,
  37.             @RequestParam(required = false) String monthStr) {
  38.         try {
  39.             LocalDate date = monthStr != null ?
  40.                 YearMonth.parse(monthStr).atDay(1) : LocalDate.now();
  41.                
  42.             List<Boolean> calendar = advancedSignInService.getMonthSignList(userId, date);
  43.             return ResponseEntity.ok(calendar);
  44.         } catch (Exception e) {
  45.             log.error("获取签到日历异常", e);
  46.             return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
  47.                 .body("系统错误");
  48.         }
  49.     }
  50.    
  51.     @PostMapping("/retroactive")
  52.     public ResponseEntity<?> retroactiveSign(
  53.             @RequestParam Long userId,
  54.             @RequestParam String dateStr) {
  55.         try {
  56.             LocalDate date = LocalDate.parse(dateStr);
  57.             boolean success = advancedSignInService.retroactiveSign(userId, date);
  58.             
  59.             if (success) {
  60.                 return ResponseEntity.ok().build();
  61.             }
  62.             
  63.             return ResponseEntity.badRequest().body("补签失败");
  64.         } catch (Exception e) {
  65.             log.error("补签异常", e);
  66.             return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
  67.                 .body("系统错误");
  68.         }
  69.     }
  70. }
复制代码
5. 性能优化发起

  1. 5.1 使用Pipeline批量操作：
复制代码
  1. public List<Boolean> getMonthSignListWithPipeline(Long userId, LocalDate date) {
  2.     String key = buildSignKey(userId, date);
  3.     int dayCount = date.lengthOfMonth();
  4.    
  5.     List<Boolean> result = new ArrayList<>(dayCount);
  6.     redisTemplate.executePipelined((RedisCallback<Object>) connection -> {
  7.         for (int i = 0; i < dayCount; i++) {
  8.             connection.getBit(key.getBytes(), i);
  9.         }
  10.         return null;
  11.     }).forEach(b -> result.add((Boolean) b));
  12.    
  13.     return result;
  14. }
复制代码
  1. 5.2.使用缓存减少Redis访问：
复制代码
  1. @Cacheable(value = "sign_calendar", key = "#userId + ':' + #date.format(T(java.time.format.DateTimeFormatter).ofPattern('yyyyMM'))")
  2. public List<Boolean> getMonthSignList(Long userId, LocalDate date) {
  3.     // 实现逻辑
  4. }
复制代码
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

倒序浏览
返回列表

快速回复

高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

一给

金牌会员
这个人很懒什么都没写!

楼主热帖

标签云

AI 运维 CIO 存储 服务器
微信订阅号
微信服务号
微信客服
小程序
H5
关于我们 商务合作 网站地图
©Copyright 2022-2024 杭州祥升科技有限公司 版权所有 浙ICP备20004199号
快速回复 返回顶部 返回列表