LimitSwitchCtrl.java 2.61 KB
package com.ui.ctrl;

import com.ui.contants.HttpUriContants;
import com.ui.http.HttpRestClient;
import com.ui.model.BaseResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.servlet.ModelAndView;

import java.util.HashMap;
import java.util.Map;

/**
 * 限流调整
 *
 * @author bblu 2016-10-26
 */
@Controller
@RequestMapping("limitSwitch")
public class LimitSwitchCtrl {

    private Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    HttpRestClient httpRestClient;

    /**
     * 展示当前配置
     *
     * @return ModelAndView
     */
    @RequestMapping(value = "toLimitSwitch")
    public ModelAndView viewNginxConf() {
        BaseResponse<Map<String, Object>> response = httpRestClient.exchangeForget(HttpUriContants.VIEW_LIMIT_CONF, new ParameterizedTypeReference<BaseResponse<Map<String, Object>>>() {
        }, null);
        Map<String, Object> data = response.getData();

        ModelAndView mdv = new ModelAndView("switch/limitSwitch");
        mdv.addObject("open_limit_flow", data.get("open_limit_flow"));
        mdv.addObject("limit_config", data.get("limit_config"));
        mdv.addObject("limit_service_config", data.get("limit_service_config"));
        return mdv;
    }

    /**
     * 展示修改后配置文件内容
     *
     * @param openLimitFlow      限流开关true/false
     * @param limitConfig        api接口限流配置
     * @param limitServiceConfig service接口限流配置
     * @return BaseResponse
     */
    @RequestMapping(value = "viewToChangeLimitConf")
    @ResponseBody
    public BaseResponse viewToChangeLimitConf(String openLimitFlow, String limitConfig, String limitServiceConfig) {
        Map<String, Object> map = new HashMap<>();
        map.put("limitFlow", openLimitFlow);
        map.put("apiConfs", limitConfig);
        map.put("serviceConfs", limitServiceConfig);
        return httpRestClient.defaultPost(HttpUriContants.VIEW_TOCHANGE_LIMIT_CONF, map, BaseResponse.class);
    }

    /**
     * 修改配置,并重启nginx
     *
     * @return BaseResponse
     */
    @RequestMapping(value = "switchConf")
    @ResponseBody
    public BaseResponse switchConf() {
        Map<String, Object> map = new HashMap<>();
        return httpRestClient.defaultGet(HttpUriContants.SWITCH_LIMIT, BaseResponse.class, map);
    }

}