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.support;
18
19 import org.springframework.cloud.aws.messaging.config.annotation.NotificationSubject;
20 import org.springframework.cloud.aws.messaging.support.converter.NotificationRequestConverter;
21 import org.springframework.core.MethodParameter;
22 import org.springframework.messaging.Message;
23 import org.springframework.messaging.converter.MessageConverter;
24 import org.springframework.messaging.converter.StringMessageConverter;
25 import org.springframework.messaging.handler.invocation.HandlerMethodArgumentResolver;
26 import org.springframework.util.ClassUtils;
27
28 /**
29  * @author Alain Sahli
30  */
31 public class NotificationSubjectArgumentResolver
32         implements HandlerMethodArgumentResolver {
33
34     private final MessageConverter converter;
35
36     public NotificationSubjectArgumentResolver() {
37         this.converter = new NotificationRequestConverter(new StringMessageConverter());
38     }
39
40     @Override
41     public boolean supportsParameter(MethodParameter parameter) {
42         return (parameter.hasParameterAnnotation(NotificationSubject.class)
43                 && ClassUtils.isAssignable(String.class, parameter.getParameterType()));
44     }
45
46     @Override
47     public Object resolveArgument(MethodParameter parameter, Message<?> message)
48             throws Exception {
49         return ((NotificationRequestConverter.NotificationRequest) this.converter
50                 .fromMessage(message, String.class)).getSubject();
51     }
52
53 }
54