+++ /dev/null
-/**\r
-* Copyright (C) 2016 ZTE, Inc. and others. All rights reserved. (ZTE)\r
-*\r
-* Licensed under the Apache License, Version 2.0 (the "License");\r
-* you may not use this file except in compliance with the License.\r
-* You may obtain a copy of the License at\r
-*\r
-* http://www.apache.org/licenses/LICENSE-2.0\r
-*\r
-* Unless required by applicable law or agreed to in writing, software\r
-* distributed under the License is distributed on an "AS IS" BASIS,\r
-* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
-* See the License for the specific language governing permissions and\r
-* limitations under the License.\r
-*/\r
-package org.openo.msb.wrapper;\r
-\r
-import java.util.HashMap;\r
-import java.util.Map;\r
-import java.util.Set;\r
-\r
-import org.apache.commons.lang3.StringUtils;\r
-import org.openo.msb.api.CustomRouteInfo;\r
-import org.openo.msb.api.RouteServer;\r
-import org.openo.msb.api.exception.ExtendedInternalServerErrorException;\r
-import org.openo.msb.api.exception.ExtendedNotFoundException;\r
-import org.openo.msb.api.exception.ExtendedNotSupportedException;\r
-import org.openo.msb.wrapper.util.JedisUtil;\r
-import org.openo.msb.wrapper.util.RegExpTestUtil;\r
-import org.openo.msb.wrapper.util.RouteUtil;\r
-import org.slf4j.Logger;\r
-import org.slf4j.LoggerFactory;\r
-\r
-import redis.clients.jedis.Jedis;\r
-\r
-public class CustomRouteServiceWrapper {\r
-\r
-\r
- private static final Logger LOGGER = LoggerFactory.getLogger(CustomRouteServiceWrapper.class);\r
-\r
- private static CustomRouteServiceWrapper instance = new CustomRouteServiceWrapper();\r
-\r
- private CustomRouteServiceWrapper() {}\r
-\r
- public static CustomRouteServiceWrapper getInstance() {\r
- return instance;\r
- }\r
-\r
-\r
- /**\r
- * @Title: getAllCustomRouteService\r
- * @Description: TODO(获取全部内容服务列表)\r
- * @param: @return\r
- * @return: CustomRouteInfo[]\r
- */\r
- public CustomRouteInfo[] getAllCustomRouteInstances() {\r
-\r
-\r
- Jedis jedis = null;\r
- CustomRouteInfo[] customRouteList = null;\r
- try {\r
- jedis = JedisUtil.borrowJedisInstance();\r
- if (jedis == null) {\r
- throw new ExtendedInternalServerErrorException(\r
- "fetch from jedis pool failed,null object!");\r
- }\r
-\r
- // 获取全部服务列表\r
- String routekey =\r
- RouteUtil.getPrefixedKey("", RouteUtil.CUSTOMROUTE, "*",\r
- RouteUtil.ROUTE_PATH_INFO);\r
- Set<String> routeSet = jedis.keys(routekey);\r
- customRouteList = new CustomRouteInfo[routeSet.size()];\r
-\r
- int i = 0;\r
- for (String routePath : routeSet) {\r
- String[] routePathArray = routePath.split(":");\r
- CustomRouteInfo customRoute = getCustomRouteInstance(routePathArray[3], jedis);\r
- customRouteList[i] = customRoute;\r
- i++;\r
- }\r
-\r
-\r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new ExtendedInternalServerErrorException("call redis throw exception:"\r
- + e.getMessage());\r
-\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
-\r
- return customRouteList;\r
- }\r
-\r
-\r
-\r
- /**\r
- * @Title: getCustomRouteInstance\r
- * @Description: TODO(通过服务名获取单个内容服务对象信息)\r
- * @param: @param serviceName\r
- * @param: @return\r
- * @return: CustomRouteInfo\r
- */\r
- public CustomRouteInfo getCustomRouteInstance(String serviceName) {\r
-\r
- if (StringUtils.isBlank(serviceName)) {\r
- throw new ExtendedNotSupportedException("serviceName can't be empty");\r
- }\r
-\r
- CustomRouteInfo customRouteInfo;\r
-\r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.borrowJedisInstance();\r
- if (jedis == null) {\r
- throw new ExtendedInternalServerErrorException(\r
- "fetch from jedis pool failed,null object!");\r
- }\r
-\r
- customRouteInfo = getCustomRouteInstance(serviceName, jedis);\r
-\r
-\r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new ExtendedInternalServerErrorException("call redis throw exception:"\r
- + e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
-\r
- if (null == customRouteInfo) {\r
- String errInfo = "customRouteInfo not found: serviceName-" + serviceName;\r
- LOGGER.warn(errInfo);\r
- throw new ExtendedNotFoundException(errInfo);\r
-\r
- }\r
-\r
- return customRouteInfo;\r
-\r
- }\r
-\r
- public CustomRouteInfo getCustomRouteInstance(String serviceName, Jedis jedis) throws Exception {\r
-\r
-\r
- CustomRouteInfo customRouteInfo = null;\r
-\r
-\r
- // 获取info信息\r
- String routekey =\r
- RouteUtil.getPrefixedKey("", RouteUtil.CUSTOMROUTE, serviceName,\r
- RouteUtil.ROUTE_PATH_INFO);\r
- Map<String, String> infomap = jedis.hgetAll(routekey);\r
- if (!infomap.isEmpty()) {\r
- customRouteInfo = new CustomRouteInfo();\r
- customRouteInfo.setServiceName(serviceName);\r
- customRouteInfo.setUrl(infomap.get("url"));\r
- customRouteInfo.setControl(infomap.get("control"));\r
- customRouteInfo.setStatus(infomap.get("status"));\r
- customRouteInfo.setVisualRange(infomap.get("visualRange"));\r
- customRouteInfo.setUseOwnUpstream(infomap.get("useOwnUpstream"));\r
-\r
-\r
- // 获取负载均衡信息\r
- String serviceLBkey =\r
- RouteUtil.getPrefixedKey("", RouteUtil.CUSTOMROUTE, serviceName,\r
- RouteUtil.ROUTE_PATH_LOADBALANCE);\r
- Set<String> serviceLBset = jedis.keys(serviceLBkey + ":*");\r
- int serverNum = serviceLBset.size();\r
- RouteServer[] CustomRouteServerList = new RouteServer[serverNum];\r
- int i = 0;\r
- for (String serviceInfo : serviceLBset) {\r
- Map<String, String> serviceLBmap = jedis.hgetAll(serviceInfo);\r
- RouteServer server = new RouteServer();\r
- server.setIp(serviceLBmap.get("ip"));\r
- server.setPort(serviceLBmap.get("port"));\r
- server.setWeight(Integer.parseInt(serviceLBmap.get("weight")));\r
- CustomRouteServerList[i] = server;\r
- i++;\r
- }\r
-\r
- customRouteInfo.setServers(CustomRouteServerList);\r
- }\r
-\r
-\r
- return customRouteInfo;\r
- }\r
-\r
- /**\r
- * @Title: updateCustomRouteInstance\r
- * @Description: TODO(更新单个服务信息)\r
- * @param: @param serviceName\r
- * @param: @param CustomRouteInfo\r
- * @param: @return\r
- * @return: CustomRouteInfo\r
- */\r
- public synchronized CustomRouteInfo updateCustomRouteInstance(String serviceName,\r
- CustomRouteInfo customRouteInfo, String serverPort) {\r
- if (StringUtils.isBlank(serviceName)) {\r
- throw new ExtendedNotSupportedException("serviceName can't be empty");\r
- }\r
-\r
- try {\r
-\r
- if (serviceName.equals(customRouteInfo.getServiceName())) {\r
- // 删除已存在负载均衡服务器信息\r
- deleteCustomRoute(serviceName, RouteUtil.ROUTE_PATH_LOADBALANCE + "*", serverPort);\r
- } else {\r
- // 如果已修改服务名,先删除此服务全部已有信息\r
- deleteCustomRoute(serviceName, "*", serverPort);\r
- }\r
-\r
-\r
- saveCustomRouteInstance(customRouteInfo, serverPort);\r
-\r
-\r
-\r
- } catch (ExtendedNotSupportedException e) {\r
- throw e;\r
- } catch (Exception e) {\r
- LOGGER.error("updateCustomRoute throw exception", e);\r
- throw new ExtendedInternalServerErrorException("update CustomRoute throw exception"\r
- + e.getMessage());\r
-\r
- }\r
-\r
- return customRouteInfo;\r
-\r
- }\r
-\r
- /**\r
- * @Title updateCustomRouteStatus\r
- * @Description TODO(更新单个服务状态)\r
- * @param serviceName\r
- * @param status\r
- * @return\r
- * @return RouteResult\r
- */\r
- public synchronized CustomRouteInfo updateCustomRouteStatus(String serviceName, String status) {\r
-\r
- if (StringUtils.isBlank(serviceName)) {\r
- throw new ExtendedNotSupportedException("serviceName can't be empty");\r
- }\r
-\r
- if (!RouteUtil.contain(RouteUtil.statusRangeMatches, status)) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo Status FAIL:status is wrong,value range:("\r
- + RouteUtil.show(RouteUtil.statusRangeMatches) + ")");\r
- }\r
-\r
- CustomRouteInfo new_customRouteInfo = getCustomRouteInstance(serviceName);\r
-\r
-\r
-\r
- // 准备info信息\r
- String serviceInfokey =\r
- RouteUtil.getPrefixedKey("", RouteUtil.CUSTOMROUTE, serviceName,\r
- RouteUtil.ROUTE_PATH_INFO);\r
- Map<String, String> serviceInfoMap = new HashMap<String, String>();\r
- serviceInfoMap.put("status", status);\r
-\r
-\r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.borrowJedisInstance();\r
- if (jedis == null) {\r
- throw new ExtendedInternalServerErrorException(\r
- "fetch from jedis pool failed,null object!");\r
- }\r
- // 保存info信息\r
- jedis.hmset(serviceInfokey, serviceInfoMap);\r
- new_customRouteInfo.setStatus(status);\r
-\r
- } catch (Exception e) {\r
-\r
- LOGGER.error("update CustomRoute status throw exception", e);\r
- throw new ExtendedInternalServerErrorException(\r
- "update CustomRoute status throw exception" + e.getMessage());\r
-\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
-\r
- return new_customRouteInfo;\r
- }\r
-\r
- /**\r
- * @Title: saveCustomRouteInstance\r
- * @Description: TODO(存储单个服务信息)\r
- * @param: @param CustomRouteInfo\r
- * @param: @return\r
- * @return: CustomRouteInfo\r
- */\r
- public synchronized CustomRouteInfo saveCustomRouteInstance(CustomRouteInfo customRouteInfo,\r
- String serverPort) {\r
-\r
- if (StringUtils.isBlank(customRouteInfo.getServiceName())\r
- || customRouteInfo.getServers().length == 0) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo FAIL: Some required fields are empty");\r
- }\r
-\r
- \r
- if (!RegExpTestUtil.urlRegExpTest(customRouteInfo.getServiceName())) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo FAIL: ServiceName is not a valid format(ServiceName must be begin with /)");\r
- \r
- }\r
- \r
- if (StringUtils.isNotBlank(customRouteInfo.getUrl())){\r
- if (!RegExpTestUtil.urlRegExpTest(customRouteInfo.getUrl())) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo FAIL:url is not a valid format(url must be begin with /)");\r
- \r
- }\r
- }\r
-\r
- if (!RouteUtil.contain(RouteUtil.visualRangeRange, customRouteInfo.getVisualRange())) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo FAIL:VisualRange is wrong,value range:("\r
- + RouteUtil.show(RouteUtil.visualRangeMatches) + ")");\r
- }\r
-\r
- if (!RouteUtil.contain(RouteUtil.controlRangeMatches, customRouteInfo.getControl())) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo FAIL:control is wrong,value range:("\r
- + RouteUtil.show(RouteUtil.controlRangeMatches) + ")");\r
- }\r
-\r
- if (!RouteUtil.contain(RouteUtil.statusRangeMatches, customRouteInfo.getStatus())) {\r
- throw new ExtendedNotSupportedException(\r
- "save CustomRouteInfo FAIL:status is wrong,value range:("\r
- + RouteUtil.show(RouteUtil.statusRangeMatches) + ")");\r
- }\r
-\r
- if (!RouteUtil.contain(RouteUtil.useOwnUpstreamRangeMatches, customRouteInfo.getUseOwnUpstream())) {\r
- throw new ExtendedNotSupportedException(\r
- "save apiRouteInfo FAIL:useOwnUpstream is wrong,value range:("\r
- + RouteUtil.show(RouteUtil.useOwnUpstreamRangeMatches) + ")");\r
- }\r
-\r
- // 检查服务实例格式\r
- RouteServer[] serverList = customRouteInfo.getServers();\r
- for (int i = 0; i < serverList.length; i++) {\r
- RouteServer server = serverList[i];\r
- if (!RegExpTestUtil.ipRegExpTest(server.getIp())) {\r
- throw new ExtendedNotSupportedException("save CustomRouteInfo FAIL:IP("\r
- + server.getIp() + ")is not a valid ip address");\r
- }\r
-\r
- if (!RegExpTestUtil.portRegExpTest(server.getPort())) {\r
- throw new ExtendedNotSupportedException("save CustomRouteInfo FAIL:Port("\r
- + server.getPort() + ")is not a valid Port address");\r
- }\r
- }\r
-\r
-\r
- // 准备info信息\r
- String serviceInfokey =\r
- RouteUtil.getPrefixedKey(serverPort, RouteUtil.CUSTOMROUTE,\r
- customRouteInfo.getServiceName().trim(), RouteUtil.ROUTE_PATH_INFO);\r
- Map<String, String> serviceInfoMap = new HashMap<String, String>();\r
- serviceInfoMap.put("url", "/".equals(customRouteInfo.getUrl().trim())\r
- ? ""\r
- : customRouteInfo.getUrl().trim());\r
- serviceInfoMap.put("control", customRouteInfo.getControl());\r
- serviceInfoMap.put("status", customRouteInfo.getStatus());\r
- serviceInfoMap.put("visualRange", customRouteInfo.getVisualRange());\r
- serviceInfoMap.put("useOwnUpstream", customRouteInfo.getUseOwnUpstream());\r
-\r
-\r
-\r
- // 准备负载均衡信息\r
- String serviceLBkey =\r
- RouteUtil.getPrefixedKey(serverPort, RouteUtil.CUSTOMROUTE,\r
- customRouteInfo.getServiceName(), RouteUtil.ROUTE_PATH_LOADBALANCE);\r
-\r
-\r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.borrowJedisInstance();\r
- if (jedis == null) {\r
- throw new ExtendedInternalServerErrorException(\r
- "fetch from jedis pool failed,null object!");\r
- }\r
- // 保存info信息\r
- jedis.hmset(serviceInfokey, serviceInfoMap);\r
-\r
- // 保存负载均衡信息\r
-\r
- for (int i = 0; i < serverList.length; i++) {\r
- Map<String, String> servermap = new HashMap<String, String>();\r
- RouteServer server = serverList[i];\r
-\r
- servermap.put("ip", server.getIp());\r
- servermap.put("port", server.getPort());\r
- servermap.put("weight", Integer.toString(server.getWeight()));\r
-\r
- jedis.hmset(serviceLBkey + ":server" + (i + 1), servermap);\r
- }\r
-\r
-\r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new ExtendedInternalServerErrorException("call redis throw exception:"\r
- + e.getMessage());\r
-\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);;\r
- }\r
-\r
- return customRouteInfo;\r
- }\r
-\r
-\r
-\r
- /**\r
- * @Title: deleteCustomRoute\r
- * @Description: TODO(删除单个服务信息)\r
- * @param: @param type\r
- * @param: @param serviceName\r
- * @param: @param delKey\r
- * @param: @return\r
- * @return: void\r
- */\r
- public synchronized void deleteCustomRoute(String serviceName, String delKey, String serverPort) {\r
-\r
- if (StringUtils.isBlank(serviceName)) {\r
- throw new ExtendedNotSupportedException("serviceName can't be empty");\r
- }\r
-\r
- Jedis jedis = null;\r
-\r
- try {\r
- jedis = JedisUtil.borrowJedisInstance();\r
- if (jedis == null) {\r
- throw new ExtendedInternalServerErrorException(\r
- "fetch from jedis pool failed,null object!");\r
- }\r
-\r
- // 获取info信息\r
- String routekey =\r
- RouteUtil\r
- .getPrefixedKey(serverPort, RouteUtil.CUSTOMROUTE, serviceName, delKey);\r
- Set<String> infoSet = jedis.keys(routekey);\r
-\r
- if (infoSet.isEmpty()) {\r
- throw new ExtendedNotFoundException("delete CustomRoute FAIL:serviceName-"\r
- + serviceName + " not fond ");\r
- }\r
-\r
-\r
- String[] paths = new String[infoSet.size()];\r
-\r
- // Set-->数组\r
- infoSet.toArray(paths);\r
-\r
- jedis.del(paths);\r
-\r
- } catch (ExtendedNotFoundException e) {\r
- throw e;\r
- } catch (Exception e) {\r
-\r
- LOGGER.error("delete CustomRoute throw exception", e);\r
- throw new ExtendedInternalServerErrorException("delete CustomRoute throw exception:"\r
- + e.getMessage());\r
-\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
-\r
-\r
- }\r
-}\r