com.nntk.restplus.strategy.HttpRequestSelector Maven / Gradle / Ivy

Go to download
package com.nntk.restplus.strategy;

import com.nntk.restplus.annotation.GET;
import com.nntk.restplus.annotation.POST;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.lang.annotation.Annotation;
import java.util.HashMap;
import java.util.Map;

@Component
public class HttpRequestSelector {

    private Map, HttpRequestBaseHandler> requestTypeMap = new HashMap<>();


    public boolean isRequestType(Class type) {
        return requestTypeMap.containsKey(type);
    }

    @Resource
    private GetRequestHandler getRequestHandler;

    @Resource
    private PostRequestHandler postRequestHandler;

    @PostConstruct
    public void postConstruct() {

        requestTypeMap.put(GET.class, getRequestHandler);
        requestTypeMap.put(POST.class, postRequestHandler);

    }

    public HttpRequestBaseHandler select(Class type) {
        HttpRequestBaseHandler handler = requestTypeMap.get(type);
        handler.setRequestType(type);
        return handler;
    }

}