SpringMVC 多版本接口/IP访问控制/ANT打包发布到LINUX

package com.api.controller; import java.lang.reflect.Method; import java.util.Arrays; import javax.servlet.http.HttpServletRequest; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.servlet.HandlerMapping; import com.api.controller.core.room.RoomController1_0; import com.api.controller.factory.ControllerAbstractFactoryImpl; import com.api.modules.Conf; import com.api.modules.State; import com.api.util.Log; import com.api.util.ResultValue; import com.api.util.annotations.RequestAlias; import com.api.util.annotations.RequestStrategyUtils; import com.api.util.exception.ApiVersionException; /** * 核心Controller * * @see 只有这个controller提供对外访问 * * @author Allen 2017年5月24日 * */ @RequestMapping(value = "/{domain}/{version}") @Controller public class CoreController { /** class Constant **/ final String ROOM = "room"; final String USER = "user"; final String[] DOMAINS = new String[] { ROOM, USER }; String methodName = null;// 调用类名 Object[] obParam = null;// 参数组 Object result = null;// method返回值 /** * 多版本代理指向Controller * * @throws Exception */ @ResponseBody @RequestMapping("/**") public Object exeucte(@PathVariable String domain, @PathVariable String version, String callback, HttpServletRequest req) { Object ob = req.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE); if (ob == null) // 参数为空 return ResultValue.execute(State.NPE, callback); String val = ob.toString(); val = val.substring(val.indexOf(version) + version.length() + 1); if (Arrays.asList(DOMAINS).indexOf(domain) == -1) return ResultValue.execute(State.DOMAINNOTFOUND, callback); else { try { Class<?> clazz = selectClass(version, domain); if (clazz == null) return ResultValue.execute(State.VERSIONERROR, callback); analysis(val, version); int flag = targetMethod(clazz, req); if (flag == -1) {// IP访问权限不足 return ResultValue.execute(State.IPPERMISSIONS, callback); } else if (flag == -2) {// 找不到的访问域 return ResultValue.execute(State.DOMAINNOTFOUND, callback); } else if (result == null) { // 返回值为null,controller业务参数判定有无 return ResultValue.execute(State.NPE, callback); } else { // success return ResultValue.execute(State.SUCCESS, result, callback); } } catch (SecurityException e) { return ResultValue.execute(State.NPE, callback); } catch (NoSuchMethodException | IllegalArgumentException e) { return ResultValue.execute(State.DOMAINNOTFOUND, callback); } catch (ApiVersionException e) { return ResultValue.execute(State.VERSIONERROR, callback); } catch (Exception e) { Log.printlog(RoomController1_0.class, e); return ResultValue.execute(State.EXCEPTION, callback); } } } /** * 获取class * * @param version * @param domain * @return * @throws Exception */ private Class<?> selectClass(String version, String domain) throws Exception { switch (domain) { case ROOM: return ControllerAbstractFactoryImpl.instance(version).iRoomController().Class(); case USER: return ControllerAbstractFactoryImpl.instance(version).iUserController().Class(); default: return null; } } /** * 执行目标函数 * * @param clazz * @param method * @return * @throws Exception */ private int targetMethod(Class<?> clazz, HttpServletRequest req) throws Exception { Method[] methods = clazz.getMethods(); // 找到目标method并配置参数类型class for (Method m : methods) { // 通过注解判断是否匹配访问domain if (m.isAnnotationPresent(RequestAlias.class) && m.getAnnotation(RequestAlias.class).value().equals(methodName)) { if (!ipControl(getRemoteHost(req), clazz, m)) { // ip不匹配访问权限 return -1; } result = clazz.getMethod(m.getName(), m.getParameterTypes()).invoke(clazz.newInstance(), obParam); return 0; } } return -2; } /** * 解析请求url * * @param str */ private void analysis(String str, String version) { Object[] params = str.split("/"); methodName = params[0].toString(); obParam = new StringBuilder(str).delete(0, methodName.length()).insert(0, version).toString().split("/"); } /** * 访问者ip获取 * * @param request * @return */ private String getRemoteHost(HttpServletRequest request) { /** * method constant */ String X_FORWARDED_FOR = "x-forwarded-for"; String PROXY_CLIENT_IP = "Proxy-Client-IP"; String WL_PROXY_CLIENT_IP = "WL-Proxy-Client-IP"; String LOOPBACK = "0:0:0:0:0:0:0:1";// 环回地址 String UNKNOWN = "unknown"; String ip = request.getHeader(X_FORWARDED_FOR); if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) { ip = request.getHeader(PROXY_CLIENT_IP); } if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) { ip = request.getHeader(WL_PROXY_CLIENT_IP); } if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) { ip = request.getRemoteAddr(); } return ip.equals(LOOPBACK) ? Conf.LOCAL_IP : ip; } /** * IP访问控制 * * @param request * @param response * @param handler * @return */ private boolean ipControl(String requestIp, Class<?> clazz, Method m) throws Exception { if (clazz != null && m != null) { boolean isClazz = clazz.isAnnotationPresent(RequestStrategyUtils.class); boolean isMethod = m.isAnnotationPresent(RequestStrategyUtils.class); RequestStrategyUtils rc = null; // 如果方法和类声明中同时存在这个注解，那么方法中的会覆盖类中的设定。 rc = isMethod ? m.getAnnotation(RequestStrategyUtils.class) : isClazz ? clazz.getAnnotation(RequestStrategyUtils.class) : null; if (rc == null) return false; String[] value = rc.ip(); // 包含则true return Arrays.asList(value).stream().anyMatch(s -> s.equals(Conf.All_IP) || ((s.equals(Conf.LOCAL_IP) || (s.equals(Conf.DEVELOP_IP))) && s.indexOf(requestIp) != -1)); } return false; } }