进行token设计,方法验证
1.实现接口幂等性
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.stereotype.Service;import java.util.Arrays;
import java.util.UUID;
import java.util.concurrent.TimeUnit;/*** Spring Boot 如何实现接口幂等性 ?* 前端重复提交表单* 很多时候会因网络波动没有及时对用户做出提交成功响应,* 致使用户认为没有成功提交,然后一直点提交按钮,这时就会发生重复提交表单请求。** 防重令牌 Token*/
@Slf4j
@Service
public class TokenUtilService {@Autowiredprivate StringRedisTemplate redisTemplate;/*** 存入 Redis 的 Token 键的前缀*/private static final String IDEMPOTENT_TOKEN_PREFIX = "idempotent_token:";/*** 创建 Token 存入 Redis,并返回该 Token** @param value 用于辅助验证的 value 值* @return 生成的 Token 串*/public String generateToken(String value) {// 实例化生成 ID 工具对象String token = UUID.randomUUID().toString();// 设置存入 Redis 的 KeyString key = IDEMPOTENT_TOKEN_PREFIX + token;// 存储 Token 到 Redis,且设置过期时间为5分钟redisTemplate.opsForValue().set(key, value, 5, TimeUnit.MINUTES);// 返回 Tokenreturn token;}public boolean validToken(String token, String value) {// 设置 Lua 脚本,其中 KEYS[1] 是 key,KEYS[2] 是 valueString script = "if redis.call('get', KEYS[1]) == KEYS[2] then return redis.call('del', KEYS[1]) else return 0 end";RedisScript<Long> redisScript = new DefaultRedisScript<>(script, Long.class);// 根据 Key 前缀拼接 KeyString key = IDEMPOTENT_TOKEN_PREFIX + token;// 执行 Lua 脚本Long result = redisTemplate.execute(redisScript, Arrays.asList(key, value));// 根据返回结果判断是否成功成功匹配并删除 Redis 键值对,若果结果不为空和0,则验证通过if (result != null && result != 0L) {log.info("验证 token={},key={},value={} 成功", token, key, value);return true;}log.info("验证 token={},key={},value={} 失败", token, key, value);return false;}}
2.在这里插入代码片测试demo
package com.yomahub.tlog.example.feign.test;import lombok.extern.slf4j.Slf4j;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.MediaType;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;@Slf4j
@SpringBootTest
@RunWith(SpringRunner.class)
public class IdempotenceTest {@Autowiredprivate WebApplicationContext webApplicationContext;@Testpublic void interfaceIdempotenceTest() throws Exception {// 初始化 MockMvcMockMvc mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();String token = mockMvc.perform(MockMvcRequestBuilders.get("/token").accept(MediaType.TEXT_HTML)).andReturn().getResponse().getContentAsString();log.info("获取的 Token 串:{}", token);// 循环调用 5 次进行测试for (int i = 1; i <= 5; i++) {
// log.info("第{}次调用测试接口", i);// 调用验证接口并打印结果String result = mockMvc.perform(MockMvcRequestBuilders.post("/test").header("token", token).accept(MediaType.TEXT_HTML)).andReturn().getResponse().getContentAsString();log.info("result:{}",result);}}}
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!
