springMVC源码分析--@SessionAttribute用法及原理解析SessionAttributesHandler和SessionAttributeStore

本文详细解析了Spring MVC框架中@SessionAttribute注解的工作原理及其应用场景,包括如何利用该注解实现跨请求的数据传递,并介绍了其内部实现机制。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

@SessionAttribute作用于处理器类上,用于在多个请求之间传递参数,类似于Session的Attribute,但不完全一样,一般来说@SessionAttribute设置的参数只用于暂时的传递,而不是长期的保存,长期保存的数据还是要放到Session中。

通过@SessionAttribute注解设置的参数有3类用法:

(1)在视图中通过request.getAttribute或session.getAttribute获取

(2)在后面请求返回的视图中通过session.getAttribute或者从model中获取

(3)自动将参数设置到后面请求所对应处理器的Model类型参数或者有@ModelAttribute注释的参数里面。

将一个参数设置到SessionAttribute中需要满足两个条件:

(1)在@SessionAttribute注解中设置了参数的名字或者类型

(2)在处理器中将参数设置到了model中。

@SessionAttribute用户后可以调用SessionStatus.setComplete来清除,这个方法只是清除SessionAttribute里的参数,而不会应用Session中的参数。

示例如下:注解@SessionAttribute中设置book、description和types={Double},这样值会被放到@SessionAttribute中,但Redirect跳转时就可以重新获得这些数据了,接下来操作sessionStatus.setComplete(),则会清除掉所有的数据,这样再次跳转时就无法获取数据了。

  1. @Controller
  2. @RequestMapping("/book")
  3. @SessionAttributes(value ={"book","description"},types={Double.class})
  4. public class RedirectController {
  5. @RequestMapping("/index")
  6. public String index(Model model){
  7. model.addAttribute("book", "金刚经");
  8. model.addAttribute("description","不擦擦擦擦擦擦擦车");
  9. model.addAttribute("price", new Double("1000.00"));
  10. //跳转之前将数据保存到book、description和price中,因为注解@SessionAttribute中有这几个参数
  11. return "redirect:get.action";
  12. }
  13. @RequestMapping("/get")
  14. public String get(@ModelAttribute ("book") String book,ModelMap model,
  15. SessionStatus sessionStatus){
  16. //可以获得book、description和price的参数
  17. System.out.println(model.get("book")+";"+model.get("description")+";"+model.get("price"));
  18. sessionStatus.setComplete();
  19. return "redirect:complete.action";
  20. }
  21. @RequestMapping("/complete")
  22. public String complete(ModelMap modelMap){
  23. //已经被清除,无法获取book的值
  24. System.out.println(modelMap.get("book"));
  25. modelMap.addAttribute("book", "妹纸");
  26. return "sessionAttribute";
  27. }
  28. }
接下来我们分析一下@SessionAttribute的实现机制

第一步:我们首先要获取注解@SessionAttribute的值的情况,在RequestMappingHandlerAdapter中的getModelFactory中处理。

  1. protected ModelAndView invokeHandlerMethod(HttpServletRequest request,
  2. HttpServletResponse response, HandlerMethod handlerMethod) throws Exception {
  3. //这里面对注解的@SessionAttribute的处理器做处理
  4. ModelFactory modelFactory = getModelFactory(handlerMethod, binderFactory);
  5. ........
  6. //会对@SessionAttribute操作的值进行处理
  7. modelFactory.initModel(webRequest, mavContainer, invocableMethod);
  8. ........
  9. return getModelAndView(mavContainer, modelFactory, webRequest);
  10. }
在getModelFactory中会创建@SessionAttribute的处理器SessionAttributeHandler

  1. private ModelFactory getModelFactory(HandlerMethod handlerMethod, WebDataBinderFactory binderFactory) {
  2. //创建SessionAttribute处理器
  3. SessionAttributesHandler sessionAttrHandler = getSessionAttributesHandler(handlerMethod);
  4. return new ModelFactory(attrMethods, binderFactory, sessionAttrHandler);
  5. }

getSessionAttributesHandler的操作就是获取或者初始化SessionAttributesHandler

  1. //已经获取过的SessionAttribute放到Map中,如果没有则需要初始化
  2. private SessionAttributesHandler getSessionAttributesHandler(HandlerMethod handlerMethod) {
  3. Class<?> handlerType = handlerMethod.getBeanType();
  4. SessionAttributesHandler sessionAttrHandler = this.sessionAttributesHandlerCache.get(handlerType);
  5. if (sessionAttrHandler == null) {
  6. synchronized (this.sessionAttributesHandlerCache) {
  7. sessionAttrHandler = this.sessionAttributesHandlerCache.get(handlerType);
  8. if (sessionAttrHandler == null) {
  9. //初始化sessionAttrHandler,并放到map中
  10. sessionAttrHandler = new SessionAttributesHandler(handlerType, sessionAttributeStore);
  11. this.sessionAttributesHandlerCache.put(handlerType, sessionAttrHandler);
  12. }
  13. }
  14. }
  15. return sessionAttrHandler;
  16. }

SessionAttributesHandler的构造函数中的操作如下,其实就是解析被@SessionAttribute注解的处理器,这样就完成了@SessionAttribute注解中设置的key的解析。

  1. public SessionAttributesHandler(Class<?> handlerType, SessionAttributeStore sessionAttributeStore) {
  2. Assert.notNull(sessionAttributeStore, "SessionAttributeStore may not be null.");
  3. this.sessionAttributeStore = sessionAttributeStore;
  4. //解析被@SessionAttribute注解的处理器
  5. SessionAttributes annotation = AnnotationUtils.findAnnotation(handlerType, SessionAttributes.class);
  6. if (annotation != null) {
  7. this.attributeNames.addAll(Arrays.asList(annotation.names()));
  8. this.attributeTypes.addAll(Arrays.asList(annotation.types()));
  9. }
  10. for (String attributeName : this.attributeNames) {
  11. this.knownAttributeNames.add(attributeName);
  12. }
  13. }

接下来我们看看springMVC对@SessionAttribute的处理操作,在ModelFactory.initModel会对@SessionAttribute的注解进行处理操作。

  1. protected ModelAndView invokeHandlerMethod(HttpServletRequest request,
  2. HttpServletResponse response, HandlerMethod handlerMethod) throws Exception {
  3. //这里面对注解的@SessionAttribute的处理器做处理
  4. ModelFactory modelFactory = getModelFactory(handlerMethod, binderFactory);
  5. ........
  6. //会对@SessionAttribute操作的值进行处理
  7. modelFactory.initModel(webRequest, mavContainer, invocableMethod);
  8. ........
  9. return getModelAndView(mavContainer, modelFactory, webRequest);
  10. }

initModel其实做了两步操作,一是:获取上一次请求保存在SessionAttributeHandler中值,给这次请求的值用,二是将这次请求的处理结果可能会对上次的@SessionAttribute中的值进行改变的值进行保存给下一次请求使用。

  1. public void initModel(NativeWebRequest request, ModelAndViewContainer mavContainer, HandlerMethod handlerMethod)
  2. throws Exception {
  3. //获取所有的@SessionAttribute注解设置的key中值
  4. Map<String, ?> sessionAttributes = this.sessionAttributesHandler.retrieveAttributes(request);
  5. //将获取的值传递给下一个请求使用
  6. mavContainer.mergeAttributes(sessionAttributes);
  7. invokeModelAttributeMethods(request, mavContainer);
  8. //请求访问完之后将修改的值重新放到@SessionAttributeStore设置的key中
  9. for (String name : findSessionAttributeArguments(handlerMethod)) {
  10. if (!mavContainer.containsAttribute(name)) {
  11. Object value = this.sessionAttributesHandler.retrieveAttribute(request, name);
  12. if (value == null) {
  13. throw new HttpSessionRequiredException("Expected session attribute '" + name + "'");
  14. }
  15. mavContainer.addAttribute(name, value);
  16. }
  17. }
  18. }
sessionAttributesHandler.retrieveAttributes的操作是将request中值,按照注解@SessionAttribute中key取值处理,然后保存到attribute中,作为这次请求的传递值使用。

  1. public Map<String, Object> retrieveAttributes(WebRequest request) {
  2. Map<String, Object> attributes = new HashMap<String, Object>();
  3. //获取注解@SessionAttribute中设置的key
  4. for (String name : this.knownAttributeNames) {
  5. //如果设置的key有值则把它保存到attribute中,给跳转之后的请求使用
  6. Object value = this.sessionAttributeStore.retrieveAttribute(request, name);
  7. if (value != null) {
  8. attributes.put(name, value);
  9. }
  10. }
  11. return attributes;
  12. }
这样就完成了将值保存在SessionAttributeHandler中,这样下一次请求过来时依然可以从SessionAttributeHandler中获取上次的结果,完成了类似Session的实现机制,但明显感觉和Session不一样,所有的请求其值是保存在一个同一个SessionAttributeHandler中。


SessionAttributeHandler其实维持了一个Map结构来存取数据,功能主要有解析@SessionAttribute注解,get和set相关值,源码如下:

  1. public class SessionAttributesHandler {
  2. private final Set<String> attributeNames = new HashSet<String>();
  3. private final Set<Class<?>> attributeTypes = new HashSet<Class<?>>();
  4. private final Set<String> knownAttributeNames =
  5. Collections.newSetFromMap(new ConcurrentHashMap<String, Boolean>(4));
  6. private final SessionAttributeStore sessionAttributeStore;
  7. //构造函数,解析@SessionAttribute注解,将其设置额key等信息保存
  8. public SessionAttributesHandler(Class<?> handlerType, SessionAttributeStore sessionAttributeStore) {
  9. Assert.notNull(sessionAttributeStore, "SessionAttributeStore may not be null.");
  10. this.sessionAttributeStore = sessionAttributeStore;
  11. SessionAttributes annotation = AnnotationUtils.findAnnotation(handlerType, SessionAttributes.class);
  12. if (annotation != null) {
  13. this.attributeNames.addAll(Arrays.asList(annotation.names()));
  14. this.attributeTypes.addAll(Arrays.asList(annotation.types()));
  15. }
  16. for (String attributeName : this.attributeNames) {
  17. this.knownAttributeNames.add(attributeName);
  18. }
  19. }
  20. public boolean hasSessionAttributes() {
  21. return ((this.attributeNames.size() > 0) || (this.attributeTypes.size() > 0));
  22. }
  23. //判断类型
  24. public boolean isHandlerSessionAttribute(String attributeName, Class<?> attributeType) {
  25. Assert.notNull(attributeName, "Attribute name must not be null");
  26. if (this.attributeNames.contains(attributeName) || this.attributeTypes.contains(attributeType)) {
  27. this.knownAttributeNames.add(attributeName);
  28. return true;
  29. }
  30. else {
  31. return false;
  32. }
  33. }
  34. //保存
  35. public void storeAttributes(WebRequest request, Map<String, ?> attributes) {
  36. for (String name : attributes.keySet()) {
  37. Object value = attributes.get(name);
  38. Class<?> attrType = (value != null) ? value.getClass() : null;
  39. if (isHandlerSessionAttribute(name, attrType)) {
  40. this.sessionAttributeStore.storeAttribute(request, name, value);
  41. }
  42. }
  43. }
  44. //获取
  45. public Map<String, Object> retrieveAttributes(WebRequest request) {
  46. Map<String, Object> attributes = new HashMap<String, Object>();
  47. for (String name : this.knownAttributeNames) {
  48. Object value = this.sessionAttributeStore.retrieveAttribute(request, name);
  49. if (value != null) {
  50. attributes.put(name, value);
  51. }
  52. }
  53. return attributes;
  54. }
  55. //清除所有内容
  56. public void cleanupAttributes(WebRequest request) {
  57. for (String attributeName : this.knownAttributeNames) {
  58. this.sessionAttributeStore.cleanupAttribute(request, attributeName);
  59. }
  60. }
  61. //获取所有值
  62. Object retrieveAttribute(WebRequest request, String attributeName) {
  63. return this.sessionAttributeStore.retrieveAttribute(request, attributeName);
  64. }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值