Redis实现投票功能

短命女 2023-10-11 23:14 70阅读 0赞

#### 一、背景介绍 ####

投票功能是一个非常常见的Web应用场景，SpringBoot作为当今流行的Web开发框架，为了提高开发效率和性能，通常需要整合一些第三方组件。Redis是一种高性能的键值对存储数据库，而Mybatis-plus则是Mybatis的扩展版本，提供了更强大和便捷的数据库操作方式。本文将介绍如何将Redis和Mybatis-plus整合到SpringBoot中，实现投票功能。

#### 二、开发环境 ####

*  JDK 1.8
 *  SpringBoot 2.5.0
 *  Redis 6.2.4
 *  Mybatis-plus 3.4.3
 *  IntelliJ IDEA

#### 三、技术实现 ####

##### 1. 配置Redis #####

在SpringBoot的配置文件application.yml中添加Redis的配置：

spring:
      # Redis相关配置
      redis:
        # Redis服务器IP地址
        host: localhost
        # Redis服务器端口号
        port: 6379
        # Redis服务器密码
        password: 
        # Redis连接池最大连接数
        jedis:
          pool:
            max-active: 8
        # Redis连接池最大等待时间（单位：毫秒）
        lettuce:
          pool:
            max-wait: -1ms
        timeout: 5000ms

##### 2. 配置Mybatis-plus #####

在SpringBoot的配置类中添加Mybatis-plus的配置：

@Configuration
    @MapperScan("com.example.mapper")
    public class MybatisPlusConfig {
        
    
        /**
         * Mybatis-plus分页插件配置
         */
        @Bean
        public PaginationInterceptor paginationInterceptor() {
        
            return new PaginationInterceptor();
        }
    
        /**
         * Mybatis-plus通用Mapper配置
         */
        @Bean
        public MapperScannerConfigurer mapperScannerConfigurer() {
        
            MapperScannerConfigurer scannerConfigurer = new MapperScannerConfigurer();
            scannerConfigurer.setSqlSessionFactoryBeanName("sqlSessionFactory");
            scannerConfigurer.setBasePackage("com.example.mapper");
            return scannerConfigurer;
        }
    }

##### 3. 实现投票功能 #####

首先创建一个投票的实体类Vote，包含投票项的id和投票数count：

@Data
    @AllArgsConstructor
    @NoArgsConstructor
    public class Vote implements Serializable {
        
        private Long id;
        private Integer count;
    }

然后创建投票的数据库表vote，包含两个字段id和count，id为主键：

CREATE TABLE `vote` (
      `id` bigint(20) NOT NULL,
      `count` int(11) DEFAULT NULL,
      PRIMARY KEY (`id`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

接着创建投票的Mapper接口VoteMapper和对应的XML文件VoteMapper.xml，定义增加投票数和查询投票数的方法：

public interface VoteMapper extends BaseMapper<Vote> {
        
        /**
         * 增加投票数
         * @param id 投票项id
         * @return
         */
        int increaseCount(@Param("id") Long id);
    
        /**
         * 查询投票数
         * @param id 投票项id
         * @return
         */
        int selectCount(@Param("id") Long id);
    }

<?xml version="1.0" encoding="UTF-8" ?>
    <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
      "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
    <mapper namespace="com.example.mapper.VoteMapper">
    
        
        <update id="increaseCount">
            update vote set count = count + 1
            where id = #{id}
        </update>
    
        
        <select id="selectCount" resultType="int">
            select count
            from vote
            where id = #{id}
        </select>
    
    </mapper>

接下来创建投票的Service类VoteService，其中增加投票数和查询投票数的方法使用了Redis缓存：

@Service
    public class VoteService {
        
    
        @Autowired
        private VoteMapper voteMapper;
    
        @Autowired
        private RedisTemplate<String, Object> redisTemplate;
    
        /**
         ** @param id 投票项id
         */
        public void increaseCount(Long id) {
        
            ValueOperations<String, Object> operations = redisTemplate.opsForValue();
            String key = "vote:" + id;
            // 先从缓存中获取投票数
            Integer count = (Integer) operations.get(key);
            // 如果缓存中没有，则从数据库中获取，并存入缓存
            if (count == null) {
        
                count = voteMapper.selectCount(id);
                if (count != null) {
        
                    operations.set(key, count);
                }
            }
            // 如果缓存中有，则增加投票数并更新缓存
            if (count != null) {
        
                operations.increment(key);
                voteMapper.increaseCount(id);
            }
        }
    
        /**
         * 查询投票数
         * @param id 投票项id
         * @return
         */
        public Integer selectCount(Long id) {
        
            ValueOperations<String, Object> operations = redisTemplate.opsForValue();
            String key = "vote:" + id;
            // 先从缓存中获取投票数
            Integer count = (Integer) operations.get(key);
            // 如果缓存中没有，则从数据库中获取，并存入缓存
            if (count == null) {
        
                count = voteMapper.selectCount(id);
                if (count != null) {
        
                    operations.set(key, count);
                }
            }
            return count;
        }
    }

最后创建投票的Controller类VoteController，提供增加投票数和查询投票数的接口：

@RestController
    public class VoteController {
        
    
        @Autowired
        private VoteService voteService;
    
        /**
         * 增加投票数接口
         * @param id 投票项id
         * @return
         */
        @PostMapping("/vote/increase")
        public String increaseCount(@RequestParam Long id) {
        
            voteService.increaseCount(id);
            return "success";
        }
    
        /**
         * 查询投票数接口
         * @param id 投票项id
         * @return
         */
        @GetMapping("/vote/select")
        public Integer selectCount(@RequestParam Long id) {
        
            Integer count = voteService.selectCount(id);
            return count == null ? 0 : count;
        }
    }

#### 四、测试运行 ####

启动SpringBoot应用后，在浏览器中访问http://localhost:8080/vote/select?id=1，可以查询id为1的投票项的投票数；再访问http://localhost:8080/vote/increase?id=1，可以对id为1的投票项进行投票。同时可以在Redis客户端中查看投票项的投票数是否正确。

#### 五、总结 ####

本文介绍了如何将Redis和Mybatis-plus整合到SpringBoot中，以实现投票功能。其中Redis缓存可以增加应用性能，Mybatis-plus可以简化数据库操作。代码已上传至Github：https://github.com/chatbot-ai/spring-boot-redis-mybatis-vote。