Skip to content

Commit

Permalink
Issue #3493 - backport BeanParam support from 3.x
Browse files Browse the repository at this point in the history
Co-Authored-By: Richard Obersheimer <richard.obersheimer@anderscore.com>
Signed-off-by: Denis Karabetsky <karabetsky@gmail.com>
  • Loading branch information
2 people authored and senivam committed Aug 15, 2023
1 parent 81d03fc commit 3035f37
Show file tree
Hide file tree
Showing 16 changed files with 1,055 additions and 108 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,219 @@
/*
* Copyright (c) 2012, 2023 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the
* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
* version 2 with the GNU Classpath Exception, which is available at
* https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
*/

package org.glassfish.jersey.client.proxy;

import java.beans.IntrospectionException;
import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import javax.ws.rs.BeanParam;
import javax.ws.rs.CookieParam;
import javax.ws.rs.FormParam;
import javax.ws.rs.HeaderParam;
import javax.ws.rs.MatrixParam;
import javax.ws.rs.PathParam;
import javax.ws.rs.QueryParam;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.Cookie;
import javax.ws.rs.core.Form;
import javax.ws.rs.core.MultivaluedHashMap;
import javax.ws.rs.core.MultivaluedMap;

/**
* Collector to retrieve parameters for setting up the HTTP request sent in the invoke method of WebResourceFactory
* The addParameter method takes a single annotated method parameter or annotated field or property of a BeanParam
* and adds the information to the web target, headers, cookie list or form.
*/
class RequestParameters {

private WebTarget newTarget;
private final MultivaluedHashMap<String, Object> headers;
private final LinkedList<Cookie> cookies;
private final Form form;

private static final List<Class<?>> PARAM_ANNOTATION_CLASSES = Arrays.asList(PathParam.class, QueryParam.class,
HeaderParam.class, CookieParam.class, MatrixParam.class, FormParam.class, BeanParam.class);

RequestParameters(final WebTarget newTarget, final MultivaluedMap<String, Object> headers,
final List<Cookie> cookies, final Form form) {

this.headers = new MultivaluedHashMap<String, Object>(headers);
this.cookies = new LinkedList<>(cookies);
this.form = new Form();
this.form.asMap().putAll(form.asMap());

this.newTarget = newTarget;
}

void addParameter(final Object value, final Map<Class<?>, Annotation> anns)
throws IntrospectionException, InvocationTargetException, IllegalAccessException {

Annotation ann;
if ((ann = anns.get(PathParam.class)) != null) {
newTarget = newTarget.resolveTemplate(((PathParam) ann).value(), value);
} else if ((ann = anns.get((QueryParam.class))) != null) {
if (value instanceof Collection) {
newTarget = newTarget.queryParam(((QueryParam) ann).value(), convert((Collection<?>) value));
} else {
newTarget = newTarget.queryParam(((QueryParam) ann).value(), value);
}
} else if ((ann = anns.get((HeaderParam.class))) != null) {
if (value instanceof Collection) {
headers.addAll(((HeaderParam) ann).value(), convert((Collection<?>) value));
} else {
headers.addAll(((HeaderParam) ann).value(), value);
}

} else if ((ann = anns.get((CookieParam.class))) != null) {
final String name = ((CookieParam) ann).value();
Cookie c;
if (value instanceof Collection) {
for (final Object v : ((Collection<?>) value)) {
if (!(v instanceof Cookie)) {
c = new Cookie(name, v.toString());
} else {
c = (Cookie) v;
if (!name.equals(((Cookie) v).getName())) {
// is this the right thing to do? or should I fail? or ignore the difference?
c = new Cookie(name, c.getValue(), c.getPath(), c.getDomain(), c.getVersion());
}
}
cookies.add(c);
}
} else {
if (!(value instanceof Cookie)) {
cookies.add(new Cookie(name, value.toString()));
} else {
c = (Cookie) value;
if (!name.equals(((Cookie) value).getName())) {
// is this the right thing to do? or should I fail? or ignore the difference?
cookies.add(new Cookie(name, c.getValue(), c.getPath(), c.getDomain(), c.getVersion()));
}
}
}
} else if ((ann = anns.get((MatrixParam.class))) != null) {
if (value instanceof Collection) {
newTarget = newTarget.matrixParam(((MatrixParam) ann).value(), convert((Collection<?>) value));
} else {
newTarget = newTarget.matrixParam(((MatrixParam) ann).value(), value);
}
} else if ((ann = anns.get((FormParam.class))) != null) {
if (value instanceof Collection) {
for (final Object v : ((Collection<?>) value)) {
form.param(((FormParam) ann).value(), v.toString());
}
} else {
form.param(((FormParam) ann).value(), value.toString());
}
} else if ((anns.get((BeanParam.class))) != null) {
if (value instanceof Collection) {
for (final Object v : ((Collection<?>) value)) {
addBeanParameter(v);
}
} else {
addBeanParameter(value);
}
}
}

private void addBeanParameter(final Object beanParam)
throws IllegalAccessException, IntrospectionException, InvocationTargetException {
Class<?> beanClass = beanParam.getClass();
List<Field> fields = new ArrayList<>();
getAllFields(fields, beanClass);

for (final Field field : fields) {
Object value = null;
final Map<Class<?>, Annotation> anns = new HashMap<>();

// get field annotations
for (final Annotation ann : field.getAnnotations()) {
anns.put(ann.annotationType(), ann);
}

if (hasAnyParamAnnotation(anns)) {
value = field.get(beanParam);
} else {
// get getter annotations if there are no field annotations
for (final PropertyDescriptor pd : Introspector.getBeanInfo(beanClass).getPropertyDescriptors()) {
if (pd.getName().equals(field.getName()) && pd.getReadMethod() != null) {
for (final Annotation ann : pd.getReadMethod().getAnnotations()) {
anns.put(ann.annotationType(), ann);
}
if (hasAnyParamAnnotation(anns)) {
value = pd.getReadMethod().invoke(beanParam);
}
}
}
}

if (value != null) {
addParameter(value, anns);
}
}
}

private List<Field> getAllFields(List<Field> fields, Class<?> type) {
fields.addAll(Arrays.asList(type.getDeclaredFields()));

if (type.getSuperclass() != null) {
getAllFields(fields, type.getSuperclass());
}

return fields;
}

private Object[] convert(final Collection<?> value) {
return value.toArray();
}

public static boolean hasAnyParamAnnotation(final Map<Class<?>, Annotation> anns) {
for (final Class<?> paramAnnotationClass : PARAM_ANNOTATION_CLASSES) {
if (anns.containsKey(paramAnnotationClass)) {
return true;
}
}
return false;
}

WebTarget getNewTarget() {
return newTarget;
}

MultivaluedHashMap<String, Object> getHeaders() {
return headers;
}

LinkedList<Cookie> getCookies() {
return cookies;
}

Form getForm() {
return form;
}

}
Loading

0 comments on commit 3035f37

Please sign in to comment.