com.stormpath.sdk.servlet.mvc.JacksonFieldValueResolver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of stormpath-sdk-servlet Show documentation

Servlet-specific additions allowing one to more easily deploy the Stormpath SDK in a servlet-container-based web application.

The newest version!

package com.stormpath.sdk.servlet.mvc;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.stormpath.sdk.lang.Strings;

import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.Collections;
import java.util.Map;

/**
 * 1.0.RC8
 */
public class JacksonFieldValueResolver implements RequestFieldValueResolver {

    public static final String MARSHALLED_OBJECT = JacksonFieldValueResolver.class.getName() + ".MARSHALLED_OBJECT";

    ObjectMapper objectMapper = new ObjectMapper();

    public ObjectMapper getObjectMapper() {
        return objectMapper;
    }

    public void setObjectMapper(ObjectMapper objectMapper) {
        this.objectMapper = objectMapper;
    }

    @Override
    public String getValue(HttpServletRequest request, String fieldName) {
        Map map = ensureBodyMap(request);
        Object value = map.get(fieldName);
        return value != null ? Strings.clean(value.toString()) : null;
    }

    @Override
    public Map getAllFields(HttpServletRequest request) {
        return ensureBodyMap(request);
    }

    @SuppressWarnings("unchecked")
    private Map ensureBodyMap(HttpServletRequest request) {
        Map map = (Map) request.getAttribute(MARSHALLED_OBJECT);
        if (map == null) {
            map = Collections.emptyMap();
            boolean bodyExists =
                    request.getContentLength() > 0 ||
                            //https://tools.ietf.org/html/rfc7230#section-3.3.2
                            //TODO not sure about this check contentLength should be enough need to check the RCF ref in the comment above
                            request.getHeader("Transfer-Encoding") != null;
            if (bodyExists) {
                map = readJsonBody(request);
            }
            request.setAttribute(MARSHALLED_OBJECT, map); //cache for repeated access;
        }
        return map;
    }

    @SuppressWarnings("unchecked")
    private Map readJsonBody(HttpServletRequest request) {
        try {
            return (Map) getObjectMapper().readValue(request.getInputStream(), Map.class);
        } catch (IOException e) {
            String msg = "Unable to read JSON value from request body: " + e.getMessage();
            throw new IllegalArgumentException(msg, e);
        }
    }
}