From 4a3f6e4546a213f593cd2386dfc3cb60295764ae Mon Sep 17 00:00:00 2001 From: Rossen Stoyanchev Date: Wed, 31 May 2017 22:43:25 -0400 Subject: [PATCH] Polish --- .../user/DefaultUserDestinationResolver.java | 28 +++++++++++-------- 1 file changed, 17 insertions(+), 11 deletions(-) diff --git a/spring-messaging/src/main/java/org/springframework/messaging/simp/user/DefaultUserDestinationResolver.java b/spring-messaging/src/main/java/org/springframework/messaging/simp/user/DefaultUserDestinationResolver.java index 4e15a242fae..9f0769af9c7 100644 --- a/spring-messaging/src/main/java/org/springframework/messaging/simp/user/DefaultUserDestinationResolver.java +++ b/spring-messaging/src/main/java/org/springframework/messaging/simp/user/DefaultUserDestinationResolver.java @@ -1,5 +1,5 @@ /* - * Copyright 2002-2016 the original author or authors. + * Copyright 2002-2017 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -130,7 +130,8 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { Set targetSet = new HashSet<>(); for (String sessionId : parseResult.getSessionIds()) { String actualDestination = parseResult.getActualDestination(); - String targetDestination = getTargetDestination(sourceDestination, actualDestination, sessionId, user); + String targetDestination = getTargetDestination( + sourceDestination, actualDestination, sessionId, user); if (targetDestination != null) { targetSet.add(targetDestination); } @@ -150,7 +151,7 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { switch (messageType) { case SUBSCRIBE: case UNSUBSCRIBE: - return parseSubscriptionMessage(message, headers, sourceDestination); + return parseSubscriptionMessage(message, sourceDestination); case MESSAGE: return parseMessage(headers, sourceDestination); default: @@ -158,7 +159,8 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { } } - private ParseResult parseSubscriptionMessage(Message message, MessageHeaders headers, String sourceDestination) { + private ParseResult parseSubscriptionMessage(Message message, String sourceDestination) { + MessageHeaders headers = message.getHeaders(); String sessionId = SimpMessageHeaderAccessor.getSessionId(headers); if (sessionId == null) { logger.error("No session id. Ignoring " + message); @@ -172,7 +174,8 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { Principal principal = SimpMessageHeaderAccessor.getUser(headers); String user = (principal != null ? principal.getName() : null); Set sessionIds = Collections.singleton(sessionId); - return new ParseResult(sourceDestination, actualDestination, sourceDestination, sessionIds, user); + return new ParseResult(sourceDestination, actualDestination, sourceDestination, + sessionIds, user); } private ParseResult parseMessage(MessageHeaders headers, String sourceDestination) { @@ -195,7 +198,8 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { if (!this.keepLeadingSlash) { actualDestination = actualDestination.substring(1); } - return new ParseResult(sourceDestination, actualDestination, subscribeDestination, sessionIds, userName); + return new ParseResult(sourceDestination, actualDestination, subscribeDestination, + sessionIds, userName); } private Set getSessionIdsByUser(String userName, String sessionId) { @@ -251,6 +255,8 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { */ private static class ParseResult { + private final String sourceDestination; + private final String actualDestination; private final String subscribeDestination; @@ -259,10 +265,10 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { private final String user; - private final String sourceDestination; public ParseResult(String sourceDest, String actualDest, String subscribeDest, Set sessionIds, String user) { + this.sourceDestination = sourceDest; this.actualDestination = actualDest; this.subscribeDestination = subscribeDest; @@ -271,6 +277,10 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { } + public String getSourceDestination() { + return this.sourceDestination; + } + public String getActualDestination() { return this.actualDestination; } @@ -286,10 +296,6 @@ public class DefaultUserDestinationResolver implements UserDestinationResolver { public String getUser() { return this.user; } - - public String getSourceDestination() { - return this.sourceDestination; - } } }