springboot+redis实现token机制

╰半夏微凉° 2022-12-17 05:25 127阅读 0赞

项目结构

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ZlbmdxaW5nNTU3OA_size_16_color_FFFFFF_t_70][]

pom.xml导入依赖

<dependencies>
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter-web</artifactId>
    		</dependency>
     
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter-test</artifactId>
    			<scope>test</scope>
    		</dependency>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-redis</artifactId>
                <version>1.4.5.RELEASE</version>
            </dependency>
            <dependency>
                <groupId>com.alibaba</groupId>
                <artifactId>fastjson</artifactId>
                <version>1.2.31</version>
            </dependency>
            <dependency>
                <groupId>nl.bitwalker</groupId>
                <artifactId>UserAgentUtils</artifactId>
                <version>1.2.4</version>
            </dependency>
            <dependency>
                <groupId>commons-codec</groupId>
                <artifactId>commons-codec</artifactId>
                <version>1.6</version>
            </dependency>
    	</dependencies>

application.properties

##指定使用redis数据库索引(默认为0)
    spring.redis.database=0
    ##指定Redis服务器地址
    spring.redis.host=192.168.126.149
    ##指定Redis端口号
    spring.redis.port=6379
    ##指定Redis密码
    spring.redis.password=123456

User.java

package com.fengqing.aapredis.bean;
     
    public class User implements Serializable{
        private Integer id;
     
        private String username;
     
        private String password;
     
        public User(Integer id, String username, String password) {
            this.id = id;
            this.username = username;
            this.password = password;
        }
     
        public User() {
        }
     
        public Integer getId() {
            return id;
        }
     
        public void setId(Integer id) {
            this.id = id;
        }
     
        public String getUsername() {
            return username;
        }
     
        public void setUsername(String username) {
            this.username = username;
        }
     
        public String getPassword() {
            return password;
        }
     
        public void setPassword(String password) {
            this.password = password;
        }
    }

Dto.java

package com.fengqing.aapredis.bean;
     
    public class Dto {
     
        private String token;
     
        private Long tokenCreatedDate;
     
        private Long tokenExpiryDate;
     
        private String isLogin;
     
        public String getToken() {
            return token;
        }
     
        public void setToken(String token) {
            this.token = token;
        }
     
        public Long getTokenCreatedDate() {
            return tokenCreatedDate;
        }
     
        public void setTokenCreatedDate(Long tokenCreatedDate) {
            this.tokenCreatedDate = tokenCreatedDate;
        }
     
        public Long getTokenExpiryDate() {
            return tokenExpiryDate;
        }
     
        public void setTokenExpiryDate(Long tokenExpiryDate) {
            this.tokenExpiryDate = tokenExpiryDate;
        }
     
        public String getIsLogin() {
            return isLogin;
        }
     
        public void setIsLogin(String isLogin) {
            this.isLogin = isLogin;
        }
    }

RedisConfig.java

package com.fengqing.aapredis.config;
     
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.data.redis.connection.RedisConnectionFactory;
    import org.springframework.data.redis.core.RedisTemplate;
     
    @Configuration
    public class RedisConfig {
     
        @Bean
        public RedisTemplate<String, String> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
            RedisTemplate<String, String> redisTemplate = new RedisTemplate<>();
            redisTemplate.setConnectionFactory(redisConnectionFactory);
            return redisTemplate;
        }
    }

RedisUtil.java

package com.fengqing.aapredis.util;
     
    import org.springframework.data.redis.core.RedisTemplate;
    import org.springframework.data.redis.core.ValueOperations;
    import org.springframework.stereotype.Component;
     
    import javax.annotation.Resource;
     
    @Component
    public class RedisUtil {
     
        @Resource
        private RedisTemplate<String, String> redisTemplate;
     
        public void set(String key, String value) {
            ValueOperations<String, String> valueOperations = redisTemplate.opsForValue();
            valueOperations.set(key, value);
        }
     
        public void setex(String key, String value, int seconds) {
            ValueOperations<String, String> valueOperations = redisTemplate.opsForValue();
            valueOperations.set(key, value, seconds);
        }
    }

TokenService.java

package com.fengqing.aapredis.service;
     
    import com.alibaba.fastjson.JSONObject;
    import com.fengqing.aapredis.bean.User;
    import com.fengqing.aapredis.util.RedisUtil;
    import nl.bitwalker.useragentutils.UserAgent;
    import org.apache.commons.codec.digest.DigestUtils;
    import org.springframework.stereotype.Service;
     
    import javax.annotation.Resource;
    import java.text.SimpleDateFormat;
    import java.util.Date;
    import java.util.Random;
     
    @Service("tokenService")
    public class TokenService {
     
        @Resource
        private RedisUtil redisUtil;
     
        //生成token(格式为token:设备-加密的用户名-时间-六位随机数)
        public String generateToken(String userAgentStr, String username) {
            StringBuilder token = new StringBuilder("token:");
            //设备
            UserAgent userAgent = UserAgent.parseUserAgentString(userAgentStr);
            if (userAgent.getOperatingSystem().isMobileDevice()) {
                token.append("MOBILE-");
            } else {
                token.append("PC-");
            }
            //加密的用户名
            token.append(DigestUtils.md5Hex(username) + "-");
            //时间
            token.append(new SimpleDateFormat("yyyyMMddHHmmssSSS").format(new Date()) + "-");
            //六位随机字符串
            token.append(new Random().nextInt(999999 - 111111 + 1) + 111111 );
            System.out.println("token-->" + token.toString());
            return token.toString();
        }
     
        //把token存到redis中
        public void save(String token, User user) {
            if (token.startsWith("token:PC")) {
                redisUtil.setex(token, JSONObject.toJSONString(user), 2*60*60);
            } else {
                redisUtil.set(token, JSONObject.toJSONString(user));
            }
        }
     
    }

UserService.java

package com.fengqing.aapredis.service;
     
    import com.fengqing.aapredis.bean.User;
    import org.springframework.stereotype.Service;
     
    @Service("userService")
    public class UserService {
        public User login(String username, String password) {
            if ("tom".equals(username) && "123".equals(password)){
                return new User(1, "tom", "123");
            } else {
                return null;
            }
        }
    }

UserController.java

package com.fengqing.aapredis.controller;
     
    import com.alibaba.fastjson.JSONObject;
    import com.fengqing.aapredis.bean.Dto;
    import com.fengqing.aapredis.bean.User;
    import com.fengqing.aapredis.service.TokenService;
    import com.fengqing.aapredis.service.UserService;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Controller;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.ResponseBody;
     
    import javax.servlet.http.HttpServletRequest;
     
    @Controller
    @RequestMapping("/user")
    public class UserController {
     
        @Autowired
        private UserService userService;
     
        @Autowired
        private TokenService tokenService;
     
        @RequestMapping("/login")
        @ResponseBody
        public String login(String username, String password, HttpServletRequest request) {
            Dto dto = new Dto();
            User user = this.userService.login(username, password);
            if (user != null) {
                String userAgent = request.getHeader("user-agent");
                String token = this.tokenService.generateToken(userAgent, username);
                this.tokenService.save(token, user);
     
                dto.setIsLogin("true");
                dto.setToken(token);
                dto.setTokenCreatedDate(System.currentTimeMillis());
                dto.setTokenExpiryDate(System.currentTimeMillis() + 2*60*60*1000);
            } else {
                dto.setIsLogin("false");
            }
            return JSONObject.toJSONString(dto);
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ZlbmdxaW5nNTU3OA_size_16_color_FFFFFF_t_70]: /images/20221123/43380acfed99412b89a6dbf7358f8675.png