com.firefly.mvc.web.support.ControllerMetaInfo Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly Show documentation

There is a newer version: 5.0.0-dev6

package com.firefly.mvc.web.support;

import com.firefly.annotation.MultipartSettings;
import com.firefly.annotation.RequestMapping;

import javax.servlet.MultipartConfigElement;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

public class ControllerMetaInfo extends HandlerMetaInfo {

    private final Set allowHttpMethod;
    private final MultipartConfigElement multipartConfigElement;

    public ControllerMetaInfo(Object object, Method method) {
        super(object, method);
        allowHttpMethod = new HashSet(Arrays.asList(method.getAnnotation(RequestMapping.class).method()));
        MultipartSettings s = method.getAnnotation(MultipartSettings.class);
        if (s != null) {
            multipartConfigElement = new MultipartConfigElement(s.location(), s.maxFileSize(), s.maxRequestSize(), s.fileSizeThreshold());
        } else {
            multipartConfigElement = null;
        }
    }

    public MultipartConfigElement getMultipartConfigElement() {
        return multipartConfigElement;
    }

    public boolean allowMethod(String method) {
        return allowHttpMethod.contains(method);
    }

    public String getAllowMethod() {
        StringBuilder s = new StringBuilder();
        for (String m : allowHttpMethod) {
            s.append(m).append(',');
        }
        s.deleteCharAt(s.length() - 1);
        return s.toString();
    }

    @Override
    public String toString() {
        return "ControllerMetaInfo [allowHttpMethod=" + allowHttpMethod + "]";
    }

}