1 /*
2  * Copyright 2013-2019 the original author or authors.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      https://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.springframework.cloud.aws.messaging.listener.support;
18
19 import org.springframework.cloud.aws.messaging.listener.Visibility;
20 import org.springframework.core.MethodParameter;
21 import org.springframework.messaging.Message;
22 import org.springframework.messaging.handler.invocation.HandlerMethodArgumentResolver;
23 import org.springframework.util.ClassUtils;
24
25 /**
26  * @author Szymon Dembek
27  * @since 1.3
28  */
29 public class VisibilityHandlerMethodArgumentResolver
30         implements HandlerMethodArgumentResolver {
31
32     private final String visibilityHeaderName;
33
34     public VisibilityHandlerMethodArgumentResolver(String visibilityHeaderName) {
35         this.visibilityHeaderName = visibilityHeaderName;
36     }
37
38     @Override
39     public boolean supportsParameter(MethodParameter parameter) {
40         return ClassUtils.isAssignable(Visibility.class, parameter.getParameterType());
41     }
42
43     @Override
44     public Object resolveArgument(MethodParameter parameter, Message<?> message)
45             throws Exception {
46         if (!message.getHeaders().containsKey(this.visibilityHeaderName)
47                 || message.getHeaders().get(this.visibilityHeaderName) == null) {
48             throw new IllegalArgumentException(
49                     "No visibility object found for message header: '"
50                             + this.visibilityHeaderName + "'");
51         }
52         return message.getHeaders().get(this.visibilityHeaderName);
53     }
54
55 }
56