diff options
Diffstat (limited to 'apiroute/apiroute-service/src/main/java/org/openo/msb/wrapper/ApiRouteServiceWrapper.java')
| -rw-r--r-- | apiroute/apiroute-service/src/main/java/org/openo/msb/wrapper/ApiRouteServiceWrapper.java | 653 |
1 files changed, 653 insertions, 0 deletions
diff --git a/apiroute/apiroute-service/src/main/java/org/openo/msb/wrapper/ApiRouteServiceWrapper.java b/apiroute/apiroute-service/src/main/java/org/openo/msb/wrapper/ApiRouteServiceWrapper.java new file mode 100644 index 0000000..6f0ed7d --- /dev/null +++ b/apiroute/apiroute-service/src/main/java/org/openo/msb/wrapper/ApiRouteServiceWrapper.java @@ -0,0 +1,653 @@ +/**
+* Copyright (C) 2016 ZTE, Inc. and others. All rights reserved. (ZTE)
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+package org.openo.msb.wrapper;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.net.URL;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Set;
+
+import org.apache.commons.lang3.StringUtils;
+import org.openo.msb.api.ApiRouteInfo;
+import org.openo.msb.api.DiscoverInfo;
+import org.openo.msb.api.RouteServer;
+import org.openo.msb.api.exception.ExtendedInternalServerErrorException;
+import org.openo.msb.api.exception.ExtendedNotFoundException;
+import org.openo.msb.api.exception.ExtendedNotSupportedException;
+import org.openo.msb.wrapper.util.JedisUtil;
+import org.openo.msb.wrapper.util.RegExpTestUtil;
+import org.openo.msb.wrapper.util.RouteUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import redis.clients.jedis.Jedis;
+
+
+
+/**
+ * @ClassName: ApiRouteServiceWrapper
+ * @Description: TODO(ApiRoute服务类)
+ * @author tanghua10186366
+ * @date 2015年9月25日 上午9:44:04
+ *
+ */
+public class ApiRouteServiceWrapper {
+
+ private static final Logger LOGGER = LoggerFactory.getLogger(ApiRouteServiceWrapper.class);
+
+
+ private static ApiRouteServiceWrapper instance = new ApiRouteServiceWrapper();
+
+ private ApiRouteServiceWrapper() {}
+
+ public static ApiRouteServiceWrapper getInstance() {
+ return instance;
+ }
+
+ /**
+ * @Title: getAllApiRouteInstances
+ * @Description: TODO(获取全部服务列表)
+ * @param: @return
+ * @return: ApiRouteInfoBean[]
+ */
+ public ApiRouteInfo[] getAllApiRouteInstances() {
+
+
+ Jedis jedis = null;
+ ApiRouteInfo[] apiRouteList = null;
+ try {
+ jedis = JedisUtil.borrowJedisInstance();
+ if (jedis == null) {
+ throw new ExtendedInternalServerErrorException(
+ "fetch from jedis pool failed,null object!");
+ }
+
+ // 获取全部服务列表
+ String routekey =
+ RouteUtil
+ .getPrefixedKey("", RouteUtil.APIROUTE, "*", RouteUtil.ROUTE_PATH_INFO);
+ Set<String> routeSet = jedis.keys(routekey);
+ apiRouteList = new ApiRouteInfo[routeSet.size()];
+
+ int i = 0;
+ for (String routePath : routeSet) {
+ String[] routePathArray = routePath.split(":");
+ ApiRouteInfo apiRoute =
+ getApiRouteInstance(routePathArray[3], routePathArray[4], jedis);
+ apiRouteList[i] = apiRoute;
+ i++;
+ }
+
+
+ } catch (Exception e) {
+ LOGGER.error("call redis throw exception", e);
+ throw new ExtendedInternalServerErrorException("call redis throw exception:"
+ + e.getMessage());
+
+ } finally {
+ JedisUtil.returnJedisInstance(jedis);
+ }
+
+ return apiRouteList;
+ }
+
+
+
+ public static boolean checkRedisConnect() {
+
+ Jedis jedis = null;
+ try {
+ jedis = JedisUtil.borrowJedisInstance();
+ if (jedis != null) {
+ return true;
+ }
+ } catch (Exception e) {
+ LOGGER.error("call redis throw exception", e);
+ } finally {
+ JedisUtil.returnJedisInstance(jedis);
+ }
+
+ return false;
+ }
+
+ /**
+ * @Title: getApiRouteInstance
+ * @Description: TODO(通过服务名+版本号获取单个服务对象信息)
+ * @param: @param serviceName
+ * @param: @param version
+ * @param: @return
+ * @return: ApiRouteInfo
+ */
+ public ApiRouteInfo getApiRouteInstance(String serviceName, String version) {
+
+ if (StringUtils.isBlank(serviceName)) {
+ throw new ExtendedNotSupportedException("serviceName can't be empty");
+ }
+
+ if (StringUtils.isNotBlank(version)) {
+ if (!RegExpTestUtil.versionRegExpTest(version)) {
+ throw new ExtendedNotSupportedException("version (" + version
+ + ") is not a valid format");
+ }
+ }
+
+
+ ApiRouteInfo apiRouteInfo = null;
+
+ Jedis jedis = null;
+ try {
+ jedis = JedisUtil.borrowJedisInstance();
+ if (jedis == null) {
+ throw new ExtendedInternalServerErrorException(
+ "fetch from jedis pool failed,null object!");
+ }
+
+ apiRouteInfo = getApiRouteInstance(serviceName, version, jedis);
+
+
+ } catch (Exception e) {
+ LOGGER.error("call redis throw exception", e);
+ throw new ExtendedInternalServerErrorException("call redis throw exception:"
+ + e.getMessage());
+
+ } finally {
+ JedisUtil.returnJedisInstance(jedis);
+ }
+
+ if (null == apiRouteInfo) {
+ String errInfo =
+ "ApiRouteInfo not found: serviceName-" + serviceName + " ,version-" + version;
+ LOGGER.warn(errInfo);
+ throw new ExtendedNotFoundException(errInfo);
+
+ }
+
+ return apiRouteInfo;
+
+ }
+
+ public ApiRouteInfo getApiRouteInstance(String serviceName, String version, Jedis jedis)
+ throws Exception {
+ if ("null".equals(version)) {
+ version = "";
+ }
+
+ ApiRouteInfo apiRouteInfo = null;
+
+
+ // 获取info信息
+ String routekey =
+ RouteUtil.getPrefixedKey("", RouteUtil.APIROUTE, serviceName, version,
+ RouteUtil.ROUTE_PATH_INFO);
+ Map<String, String> infomap = jedis.hgetAll(routekey);
+ if (!infomap.isEmpty()) {
+ apiRouteInfo = new ApiRouteInfo();
+ apiRouteInfo.setServiceName(serviceName);
+ apiRouteInfo.setVersion(version);
+ apiRouteInfo.setUrl(infomap.get("url"));
+ apiRouteInfo.setMetricsUrl(infomap.get("metricsUrl"));
+ apiRouteInfo.setApiJson(infomap.get("apijson"));
+ apiRouteInfo.setApiJsonType(infomap.get("apiJsonType"));
+ apiRouteInfo.setControl(infomap.get("control"));
+ apiRouteInfo.setStatus(infomap.get("status"));
+ apiRouteInfo.setVisualRange(infomap.get("visualRange"));
+ apiRouteInfo.setUseOwnUpstream(infomap.get("useOwnUpstream"));
+
+
+ // 获取负载均衡信息
+ String serviceLBkey =
+ RouteUtil.getPrefixedKey("", RouteUtil.APIROUTE, serviceName, version,
+ RouteUtil.ROUTE_PATH_LOADBALANCE);
+ Set<String> serviceLBset = jedis.keys(serviceLBkey + ":*");
+ int serverNum = serviceLBset.size();
+ RouteServer[] apiRouteServerList = new RouteServer[serverNum];
+ int i = 0;
+ for (String serviceInfo : serviceLBset) {
+ Map<String, String> serviceLBmap = jedis.hgetAll(serviceInfo);
+ RouteServer server = new RouteServer();
+ server.setIp(serviceLBmap.get("ip"));
+ server.setPort(serviceLBmap.get("port"));
+ server.setWeight(Integer.parseInt(serviceLBmap.get("weight")));
+ apiRouteServerList[i] = server;
+ i++;
+ }
+
+ apiRouteInfo.setServers(apiRouteServerList);
+
+ // 获取生命周期信息
+
+// ApiRouteLifeCycle lifeCycle = new ApiRouteLifeCycle();
+// String serviceLifekey =
+// RouteUtil.getPrefixedKey("", RouteUtil.APIROUTE, serviceName, version,
+// RouteUtil.APIROUTE_PATH_LIFE);
+// Map<String, String> serviceLifeMap = jedis.hgetAll(serviceLifekey);
+//
+// lifeCycle.setInstallPath(serviceLifeMap.get("path"));
+// lifeCycle.setStartScript(serviceLifeMap.get("start"));
+// lifeCycle.setStopScript(serviceLifeMap.get("stop"));
+//
+// apiRouteInfo.setLifeCycle(lifeCycle);
+ }
+
+
+ return apiRouteInfo;
+ }
+
+ /**
+ * @Title: updateApiRouteInstance
+ * @Description: TODO(更新单个服务信息)
+ * @param: @param serviceName
+ * @param: @param version
+ * @param: @param apiRouteInfo
+ * @param: @return
+ * @return: ApiRouteInfo
+ */
+ public synchronized ApiRouteInfo updateApiRouteInstance(String serviceName, String version,
+ ApiRouteInfo apiRouteInfo, String serverPort) {
+
+ if ("null".equals(version)) {
+ version = "";
+ }
+
+ if (StringUtils.isBlank(serviceName)) {
+ throw new ExtendedNotSupportedException("serviceName can't be empty");
+ }
+
+ if (StringUtils.isNotBlank(version)) {
+ if (!RegExpTestUtil.versionRegExpTest(version)) {
+ throw new ExtendedNotSupportedException("version (" + version
+ + ") is not a valid format");
+ }
+ }
+
+
+
+
+ try {
+
+
+ if (serviceName.equals(apiRouteInfo.getServiceName())
+ && version.equals(apiRouteInfo.getVersion())) {
+ // 删除已存在负载均衡服务器信息
+ deleteApiRoute(serviceName, version, RouteUtil.ROUTE_PATH_LOADBALANCE + "*",
+ serverPort);
+
+ } else {
+ // 如果已修改服务名或者版本号，先删除此服务全部已有信息
+ deleteApiRoute(serviceName, version, "*", serverPort);
+ }
+
+
+ saveApiRouteInstance(apiRouteInfo, serverPort);
+
+
+ } catch (ExtendedNotSupportedException e) {
+ throw e;
+ } catch (Exception e) {
+ LOGGER.error("update ApiRoute throw exception", e);
+ throw new ExtendedInternalServerErrorException("update apiRouteInfo throw exception"
+ + e.getMessage());
+
+ }
+
+ return apiRouteInfo;
+
+ }
+
+ /**
+ * @Title updateApiRouteStatus
+ * @Description TODO(更新单个服务状态)
+ * @param serviceName
+ * @param version
+ * @param status
+ * @return
+ * @return RouteResult
+ */
+ public synchronized ApiRouteInfo updateApiRouteStatus(String serviceName, String version,
+ String status) {
+
+ if ("null".equals(version)) {
+ version = "";
+ }
+
+ if (StringUtils.isBlank(serviceName)) {
+ throw new ExtendedNotSupportedException("serviceName can't be empty");
+ }
+
+ if (StringUtils.isNotBlank(version)) {
+ if (!RegExpTestUtil.versionRegExpTest(version)) {
+ throw new ExtendedNotSupportedException("version (" + version
+ + ") is not a valid format");
+ }
+ }
+
+ if (!RouteUtil.contain(RouteUtil.statusRangeMatches, status)) {
+ throw new ExtendedNotSupportedException(
+ "save ApiRouteInfo Status FAIL:status is wrong,value range:("
+ + RouteUtil.show(RouteUtil.statusRangeMatches) + ")");
+ }
+
+ ApiRouteInfo new_apiRouteInfo = getApiRouteInstance(serviceName, version);
+
+
+ // 准备info信息
+ String serviceInfokey =
+ RouteUtil.getPrefixedKey("", RouteUtil.APIROUTE, serviceName, version,
+ RouteUtil.ROUTE_PATH_INFO);
+ Map<String, String> serviceInfoMap = new HashMap<String, String>();
+ serviceInfoMap.put("status", status);
+
+
+ Jedis jedis = null;
+ try {
+ jedis = JedisUtil.borrowJedisInstance();
+ if (jedis == null) {
+ throw new Exception("fetch from jedis pool failed,null object!");
+ }
+ // 保存info信息
+ jedis.hmset(serviceInfokey, serviceInfoMap);
+ new_apiRouteInfo.setStatus(status);
+
+
+ } catch (Exception e) {
+ LOGGER.error("update ApiRoute status throw exception", e);
+ throw new ExtendedInternalServerErrorException("update ApiRoute status throw exception"
+ + e.getMessage());
+
+ } finally {
+ JedisUtil.returnJedisInstance(jedis);
+ }
+
+ return new_apiRouteInfo;
+ }
+
+
+ /**
+ * @Title: saveApiRouteInstance
+ * @Description: TODO(存储单个服务信息)
+ * @param: @param apiRouteInfo
+ * @param: @return
+ * @return: ApiRouteInfo
+ */
+ public synchronized ApiRouteInfo saveApiRouteInstance(ApiRouteInfo apiRouteInfo,
+ String serverPort) {
+
+
+
+ if (StringUtils.isBlank(apiRouteInfo.getServiceName())
+ || apiRouteInfo.getServers().length == 0) {
+ throw new ExtendedNotSupportedException(
+ "save apiRouteInfo FAIL: Some required fields are empty");
+ }
+
+ if (StringUtils.isNotBlank(apiRouteInfo.getVersion())) {
+ if (!RegExpTestUtil.versionRegExpTest(apiRouteInfo.getVersion())) {
+ throw new ExtendedNotSupportedException("version (" + apiRouteInfo.getVersion()
+ + ") is not a valid format");
+ }
+ }
+
+ if (StringUtils.isNotBlank(apiRouteInfo.getUrl())) {
+ if (!RegExpTestUtil.urlRegExpTest(apiRouteInfo.getUrl())) {
+ throw new ExtendedNotSupportedException(
+ "save apiRouteInfo FAIL:url is not a valid format(url must be begin with /)");
+
+ }
+ }
+
+ if (!RouteUtil.contain(RouteUtil.visualRangeRange, apiRouteInfo.getVisualRange())) {
+ throw new ExtendedNotSupportedException(
+ "save apiRouteInfo FAIL:VisualRange is wrong,value range:("
+ + RouteUtil.show(RouteUtil.visualRangeMatches) + ")");
+ }
+
+ if (!RouteUtil.contain(RouteUtil.controlRangeMatches, apiRouteInfo.getControl())) {
+ throw new ExtendedNotSupportedException(
+ "save apiRouteInfo FAIL:control is wrong,value range:("
+ + RouteUtil.show(RouteUtil.controlRangeMatches) + ")");
+ }
+
+ if (!RouteUtil.contain(RouteUtil.statusRangeMatches, apiRouteInfo.getStatus())) {
+ throw new ExtendedNotSupportedException(
+ "save apiRouteInfo FAIL:status is wrong,value range:("
+ + RouteUtil.show(RouteUtil.statusRangeMatches) + ")");
+ }
+
+ if (!RouteUtil.contain(RouteUtil.useOwnUpstreamRangeMatches, apiRouteInfo.getUseOwnUpstream())) {
+ throw new ExtendedNotSupportedException(
+ "save apiRouteInfo FAIL:useOwnUpstream is wrong,value range:("
+ + RouteUtil.show(RouteUtil.useOwnUpstreamRangeMatches) + ")");
+ }
+
+ // 检查服务实例格式
+ RouteServer[] serverList = apiRouteInfo.getServers();
+ for (int i = 0; i < serverList.length; i++) {
+ RouteServer server = serverList[i];
+ if (!RegExpTestUtil.ipRegExpTest(server.getIp())) {
+ throw new ExtendedNotSupportedException("save apiRouteInfo FAIL:IP("
+ + server.getIp() + ")is not a valid ip address");
+ }
+
+ if (!RegExpTestUtil.portRegExpTest(server.getPort())) {
+ throw new ExtendedNotSupportedException("save apiRouteInfo FAIL:Port("
+ + server.getPort() + ")is not a valid Port address");
+ }
+ }
+
+ // 准备info信息
+ String serviceInfokey =
+ RouteUtil.getPrefixedKey(serverPort, RouteUtil.APIROUTE,
+ apiRouteInfo.getServiceName().trim(), apiRouteInfo.getVersion().trim(),
+ RouteUtil.ROUTE_PATH_INFO);
+ Map<String, String> serviceInfoMap = new HashMap<String, String>();
+ serviceInfoMap.put("url", "/".equals(apiRouteInfo.getUrl().trim()) ? "" : apiRouteInfo
+ .getUrl().trim());
+ serviceInfoMap.put("apijson", apiRouteInfo.getApiJson());
+ serviceInfoMap.put("apiJsonType", apiRouteInfo.getApiJsonType());
+ serviceInfoMap.put("metricsUrl", apiRouteInfo.getMetricsUrl());
+ serviceInfoMap.put("control", apiRouteInfo.getControl());
+ serviceInfoMap.put("status", apiRouteInfo.getStatus());
+ serviceInfoMap.put("visualRange", apiRouteInfo.getVisualRange());
+ serviceInfoMap.put("useOwnUpstream", apiRouteInfo.getUseOwnUpstream());
+
+ // 准备负载均衡信息
+ String serviceLBkey =
+ RouteUtil.getPrefixedKey(serverPort, RouteUtil.APIROUTE,
+ apiRouteInfo.getServiceName(), apiRouteInfo.getVersion(),
+ RouteUtil.ROUTE_PATH_LOADBALANCE);
+
+
+ Jedis jedis = null;
+ try {
+ jedis = JedisUtil.borrowJedisInstance();
+ if (jedis == null) {
+ throw new ExtendedInternalServerErrorException(
+ "fetch from jedis pool failed,null object!");
+ }
+ // 保存info信息
+ jedis.hmset(serviceInfokey, serviceInfoMap);
+
+ // 保存负载均衡信息
+ for (int i = 0; i < serverList.length; i++) {
+ Map<String, String> servermap = new HashMap<String, String>();
+ RouteServer server = serverList[i];
+
+ servermap.put("ip", server.getIp());
+ servermap.put("port", server.getPort());
+ servermap.put("weight", Integer.toString(server.getWeight()));
+
+ jedis.hmset(serviceLBkey + ":server" + (i + 1), servermap);
+ }
+ // 保存生命周期信息
+
+// ApiRouteLifeCycle lifeCycle = apiRouteInfo.getLifeCycle();
+// if (lifeCycle != null) {
+// String serviceLifekey =
+// RouteUtil.getPrefixedKey(serverPort, RouteUtil.APIROUTE,
+// apiRouteInfo.getServiceName(), apiRouteInfo.getVersion(),
+// RouteUtil.APIROUTE_PATH_LIFE);
+// Map<String, String> serviceLifeMap = new HashMap<String, String>();
+// serviceLifeMap.put("path", lifeCycle.getInstallPath());
+// serviceLifeMap.put("start", lifeCycle.getStartScript());
+// serviceLifeMap.put("stop", lifeCycle.getStopScript());
+// jedis.hmset(serviceLifekey, serviceLifeMap);
+// }
+
+
+
+ } catch (Exception e) {
+ LOGGER.error("call redis throw exception", e);
+ throw new ExtendedInternalServerErrorException("call redis throw exception:"
+ + e.getMessage());
+
+ } finally {
+ JedisUtil.returnJedisInstance(jedis);
+ }
+
+ return apiRouteInfo;
+ }
+
+
+
+ /**
+ * @Title: deleteApiRoute
+ * @Description: TODO(删除单个服务信息)
+ * @param: @param type
+ * @param: @param serviceName
+ * @param: @param version
+ * @param: @param delKey
+ * @param: @return
+ * @return: void
+ */
+ public synchronized void deleteApiRoute(String serviceName, String version, String delKey,
+ String serverPort) {
+
+ if ("null".equals(version)) {
+ version = "";
+ }
+
+ if (StringUtils.isBlank(serviceName)) {
+ throw new ExtendedNotSupportedException("serviceName can't be empty");
+ }
+
+ if (StringUtils.isNotBlank(version)) {
+ if (!RegExpTestUtil.versionRegExpTest(version)) {
+ throw new ExtendedNotSupportedException("version (" + version
+ + ") is not a valid format");
+ }
+ }
+
+
+ Jedis jedis = null;
+ try {
+ jedis = JedisUtil.borrowJedisInstance();
+ if (jedis == null) {
+ throw new ExtendedInternalServerErrorException(
+ "fetch from jedis pool failed,null object!");
+ }
+
+ // 获取info信息
+ String routekey =
+ RouteUtil.getPrefixedKey(serverPort, RouteUtil.APIROUTE, serviceName, version,
+ delKey);
+ Set<String> infoSet = jedis.keys(routekey);
+
+ if (infoSet.isEmpty()) {
+ throw new ExtendedNotFoundException("delete ApiRoute FAIL:serviceName-"
+ + serviceName + ",version:" + version + " not fond ");
+ }
+
+ String[] paths = new String[infoSet.size()];
+
+ // Set-->数组
+ infoSet.toArray(paths);
+
+ jedis.del(paths);
+
+
+ } catch (ExtendedNotFoundException e) {
+ throw e;
+ } catch (Exception e) {
+ LOGGER.error("delete ApiRoute throw exception", e);
+ throw new ExtendedInternalServerErrorException("delete ApiRoute throw exception:"
+ + e.getMessage());
+ } finally {
+ JedisUtil.returnJedisInstance(jedis);
+ }
+
+
+ }
+
+ /**
+ * @Title: getAllApiDocs
+ * @Description: TODO(获取本地ext\initSwaggerJson目录的全部json文件目录)
+ * @param: @return
+ * @return: String[]
+ */
+ public String[] getAllApiDocs() {
+ URL apiDocsPath = ApiRouteServiceWrapper.class.getResource("/ext/initSwaggerJson");
+ if (apiDocsPath != null) {
+ String path = apiDocsPath.getPath();
+
+ try {
+ return readfile(path);
+ } catch (FileNotFoundException e) {
+ // TODO Auto-generated catch block
+ LOGGER.error("read ApiDocs Files throw FileNotFoundException", e);
+ throw new ExtendedInternalServerErrorException("read ApiDocs Files throw FileNotFoundException:"
+ + e.getMessage());
+ } catch (IOException e) {
+ // TODO Auto-generated catch block
+ LOGGER.error("read ApiDocs Files throw IOexception", e);
+ throw new ExtendedInternalServerErrorException("read ApiDocs Files throw IOexception:"
+ + e.getMessage());
+ }
+
+ }
+
+
+ return null;
+ }
+
+ /**
+ * 读取某个文件夹下的所有文件
+ */
+ public String[] readfile(String filepath) throws FileNotFoundException, IOException {
+ File file = new File(filepath);
+ if (file.isDirectory()) {
+ String[] filelist = file.list();
+ return filelist;
+ }
+ return null;
+ }
+
+ public String getApiGatewayPort() {
+ // return JedisUtil.serverIp+":"+JedisUtil.serverPort;
+ return System.getenv("APIGATEWAY_EXPOSE_PORT") == null ? String
+ .valueOf(JedisUtil.serverPort) : System.getenv("APIGATEWAY_EXPOSE_PORT");
+
+ }
+
+ public DiscoverInfo getServiceDiscoverInfo() {
+ return RouteUtil.discoverInfo;
+
+ }
+
+
+}
|