+++ /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
-\r
-package org.openo.msb.wrapper.util;\r
-\r
-import java.sql.Date;\r
-import java.util.ArrayList;\r
-import java.util.HashMap;\r
-import java.util.HashSet;\r
-import java.util.List;\r
-import java.util.Map;\r
-import java.util.Set;\r
-import java.util.regex.Matcher;\r
-import java.util.regex.Pattern;\r
-\r
-import org.apache.commons.lang3.StringUtils;\r
-import org.openo.msb.api.MicroServiceFullInfo;\r
-import org.openo.msb.api.MicroServiceInfo;\r
-import org.openo.msb.api.Node;\r
-import org.openo.msb.api.NodeInfo;\r
-import org.openo.msb.api.Service;\r
-import org.openo.msb.wrapper.serviceListener.IMicroServiceChangeListener;\r
-import org.slf4j.Logger;\r
-import org.slf4j.LoggerFactory;\r
-\r
-import redis.clients.jedis.Jedis;\r
-\r
-public class MicroServiceDB {\r
-\r
- private static final Logger LOGGER = LoggerFactory.getLogger(MicroServiceDB.class);\r
-\r
- private static MicroServiceDB instance = new MicroServiceDB();\r
-\r
- private List<IMicroServiceChangeListener> serviceListenerlist =\r
- new ArrayList<IMicroServiceChangeListener>();\r
-\r
- private MicroServiceDB() {}\r
-\r
- public static MicroServiceDB getInstance() {\r
- return instance;\r
- }\r
-\r
- \r
- public void addServiceChangeListener(IMicroServiceChangeListener listener) {\r
- synchronized (serviceListenerlist) {\r
- serviceListenerlist.add(listener);\r
- }\r
- }\r
-\r
- \r
- public void removeServiceChangeListener(IMicroServiceChangeListener listener) {\r
- synchronized (serviceListenerlist) {\r
- serviceListenerlist.remove(listener);\r
- }\r
- }\r
- \r
- \r
- public MicroServiceFullInfo[] getAllMicroServiceInstances() throws Exception {\r
- Jedis jedis = null;\r
- MicroServiceFullInfo[] microServiceList;\r
- try {\r
- jedis = JedisUtil.getJedis();\r
-\r
- String routekey =\r
- MicroServiceUtil.getPrefixedKey("","*", MicroServiceUtil.SUFFIX_PATH_INFO);\r
- Set<String> serviceSet = jedis.keys(routekey);\r
- microServiceList = new MicroServiceFullInfo[serviceSet.size()];\r
-\r
- Pattern redisKeyPattern = MicroServiceUtil.getRedisKeyPattern();\r
- int i = 0;\r
- for (String servicePath : serviceSet) {\r
- Matcher matcher = redisKeyPattern.matcher(servicePath);\r
- if (matcher.matches()) {\r
- microServiceList[i] = getMicroServiceByJedis(jedis, matcher.group("servicename"),matcher.group("version"), "");\r
- i++;\r
- }\r
- }\r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new Exception("call redis throw exception:"+e.getMessage()); \r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
- \r
- return microServiceList;\r
- }\r
-\r
- public void saveMicroServiceInfo2Redis(MicroServiceInfo microServiceInfo,String serverPort) throws Exception {\r
- // 1.1 set info\r
- String serviceInfokey =\r
- MicroServiceUtil.getServiceInfoKey(serverPort,microServiceInfo.getServiceName(),\r
- microServiceInfo.getVersion());\r
- Map<String, String> serviceInfoMap = new HashMap<String, String>();\r
- serviceInfoMap.put("url", microServiceInfo.getUrl());\r
- serviceInfoMap.put("protocol", microServiceInfo.getProtocol());\r
- serviceInfoMap.put("visualRange",microServiceInfo.getVisualRange());\r
- serviceInfoMap.put("lb_policy",microServiceInfo.getLb_policy());\r
- serviceInfoMap.put("status", "0");\r
- \r
- \r
-\r
- // 1.2 set lb info\r
- String serviceLBkey =\r
- MicroServiceUtil.getPrefixedKey(serverPort,microServiceInfo.getServiceName(),\r
- microServiceInfo.getVersion(), MicroServiceUtil.ROUTE_PATH_LOADBALANCE);\r
-\r
-\r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.getJedis();\r
- // 2.1 save info\r
- jedis.hmset(serviceInfokey, serviceInfoMap);\r
-\r
-\r
- for(Node node:microServiceInfo.getNodes()){\r
- \r
- String key=serviceLBkey+":"+node.getIp()+"-"+node.getPort();\r
- \r
- Map<String,String> nodeMap = new HashMap<String,String>();\r
- \r
- nodeMap.put("ip", node.getIp());\r
- nodeMap.put("port", node.getPort());\r
- nodeMap.put("ttl", Integer.toString(node.getTtl()));\r
- long expiration_time=System.currentTimeMillis()+node.getTtl()*1000;\r
- nodeMap.put("expiration", Long.toString(expiration_time));\r
- \r
- if(jedis.keys(key).isEmpty()){\r
- nodeMap.put("created_at", Long.toString(System.currentTimeMillis()));\r
- }\r
-// else{\r
-// Map<String,String> nodeLBmap = jedis.hgetAll(key); \r
-// nodeMap.put("created_at", nodeLBmap.get("created_at"));\r
-// }\r
- nodeMap.put("updated_at", Long.toString(System.currentTimeMillis()));\r
- \r
- jedis.hmset(key, nodeMap);\r
- }\r
- \r
-// jedis.sadd(serviceLBkey, nodeArray);\r
-\r
- } catch (Exception e) {\r
- LOGGER.error("save to redis throw exception", e);\r
- throw new Exception("save to redis throw exception:"+e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
-\r
-\r
-\r
- }\r
- \r
- public void updateMicroServiceStatus(String serviceName, String version,String status) throws Exception{\r
- \r
- \r
- String serviceInfokey = MicroServiceUtil.getServiceInfoKey("",serviceName, version);\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 Exception("fetch from jedis pool failed,null object!");\r
- }\r
- jedis.hmset(serviceInfokey, serviceInfoMap);\r
- }\r
- catch (Exception e) {\r
- LOGGER.error("update MicroService status throw exception", e);\r
- throw new Exception("update MicroService status throw exception:"+e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
- \r
- }\r
- \r
- \r
- public void updateMicroServiceNode2Redis(String serviceName, String version,String ip,String port,int ttl) throws Exception {\r
- String serviceLBkey =\r
- MicroServiceUtil.getPrefixedKey("",serviceName,version, MicroServiceUtil.ROUTE_PATH_LOADBALANCE);\r
-\r
-\r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.getJedis();\r
- \r
- \r
- String nodeKey=serviceLBkey+":"+ip+"-"+port; \r
- Map<String,String> nodeLBmap = jedis.hgetAll(nodeKey);\r
- \r
- if(nodeLBmap.isEmpty()){\r
- throw new NullPointerException(" MicroService Node not fond "); \r
- }\r
- \r
- \r
- nodeLBmap.put("ttl", Integer.toString(ttl));\r
- long expiration_time=System.currentTimeMillis()+ttl*1000;\r
- nodeLBmap.put("expiration", Long.toString(expiration_time));\r
- nodeLBmap.put("updated_at", Long.toString(System.currentTimeMillis()));\r
- \r
- jedis.hmset(nodeKey, nodeLBmap);\r
-\r
-\r
- } \r
- catch (NullPointerException e){\r
- throw e; \r
- }\r
- catch (Exception e) {\r
- LOGGER.error("update MicroService Node throw exception", e);\r
- throw new Exception("update MicroService Node throw exception:"+e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
- }\r
- \r
- \r
- public void noticeUpdateApiListener(String serviceName,String version,Service microServiceInfo,String serverPort) {\r
- if (isNeedNotify(microServiceInfo)) {\r
- for (IMicroServiceChangeListener serviceListener : serviceListenerlist) {\r
- serviceListener.onChange(serviceName,version, microServiceInfo,serverPort);\r
- }\r
- }\r
- \r
- }\r
- \r
- public void noticeUpdateStatusListener(Service microServiceInfo,String status) {\r
- \r
- for (IMicroServiceChangeListener serviceListener : serviceListenerlist) {\r
- serviceListener.onStatusChange(microServiceInfo.getServiceName(),microServiceInfo.getUrl(),\r
- microServiceInfo.getVersion(),microServiceInfo.getProtocol(),status);\r
- }\r
- }\r
- \r
- \r
-\r
- public void noticeApiListener(Service microServiceInfo, String type,String serverPort) {\r
- if (isNeedNotify(microServiceInfo)) {\r
-\r
- if ("ADD".equals(type)) {\r
- for (IMicroServiceChangeListener serviceListener : serviceListenerlist) {\r
- serviceListener.onSave(microServiceInfo,serverPort);\r
- }\r
- } else if ("DELETE".equals(type)) {\r
- for (IMicroServiceChangeListener serviceListener : serviceListenerlist) {\r
- serviceListener.onDelete(microServiceInfo.getServiceName(),microServiceInfo.getUrl(),\r
- microServiceInfo.getVersion(),microServiceInfo.getProtocol(),serverPort);\r
- }\r
- } \r
-\r
- }\r
- }\r
-\r
-\r
- public MicroServiceFullInfo getMicroServiceInstance(String serviceName, String version,String serverPort)\r
- throws Exception {\r
- if (null == version || "null".equals(version)) {\r
- version = "";\r
- }\r
-\r
- Jedis jedis = null;\r
- MicroServiceFullInfo microServiceInfo = null;\r
- \r
- try {\r
- jedis = JedisUtil.getJedis();\r
- \r
- microServiceInfo= getMicroServiceByJedis(jedis,serviceName,version, serverPort);\r
-\r
- \r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new Exception("call redis throw exception:"+e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
- \r
- return microServiceInfo;\r
- \r
- }\r
-\r
- private MicroServiceFullInfo getMicroServiceByJedis(Jedis jedis,String serviceName, String version,String serverPort){\r
- MicroServiceFullInfo microServiceInfo = null;\r
- String serviceInfoKey = MicroServiceUtil.getServiceInfoKey(serverPort,serviceName, version);\r
- Map<String, String> infomap = jedis.hgetAll(serviceInfoKey);\r
- if (!infomap.isEmpty()) {\r
- microServiceInfo = new MicroServiceFullInfo();\r
- microServiceInfo.setServiceName(serviceName);\r
- microServiceInfo.setVersion(version);\r
- microServiceInfo.setUrl(infomap.get("url"));\r
- microServiceInfo.setProtocol(infomap.get("protocol"));\r
- microServiceInfo.setVisualRange(infomap.get("visualRange"));\r
- microServiceInfo.setStatus(infomap.get("status"));\r
- microServiceInfo.setLb_policy(infomap.get("lb_policy"));\r
-\r
- String nodeLBkey =\r
- MicroServiceUtil.getPrefixedKey(serverPort,microServiceInfo.getServiceName(),\r
- microServiceInfo.getVersion(),\r
- MicroServiceUtil.ROUTE_PATH_LOADBALANCE);\r
- \r
- Set<String> nodeKeys=jedis.keys(nodeLBkey+":*");\r
- \r
- Set<NodeInfo> nodes=new HashSet<NodeInfo>();\r
- for(String nodeKey:nodeKeys){\r
- Map<String,String> nodeLBmap = jedis.hgetAll(nodeKey);\r
- NodeInfo nodeInfo=new NodeInfo();\r
- nodeInfo.setNodeId(serviceName+"_"+nodeLBmap.get("ip")+"_"+nodeLBmap.get("port"));\r
- nodeInfo.setIp(nodeLBmap.get("ip"));\r
- nodeInfo.setPort(nodeLBmap.get("port"));\r
- nodeInfo.setTtl(Integer.parseInt(nodeLBmap.get("ttl")));\r
- nodeInfo.setCreated_at(new Date(Long.parseLong(nodeLBmap.get("created_at"))));\r
- nodeInfo.setUpdated_at(new Date(Long.parseLong(nodeLBmap.get("updated_at"))));\r
- nodeInfo.setExpiration(new Date(Long.parseLong(nodeLBmap.get("expiration"))));\r
- \r
- nodes.add(nodeInfo);\r
- }\r
- \r
- microServiceInfo.setNodes(nodes);\r
- }\r
- \r
- \r
- \r
- \r
- return microServiceInfo;\r
- }\r
- \r
-\r
- public void deleteMicroService(String serviceName, String version,String serverPort) throws Exception {\r
- if (null == version || "null".equals(version)) {\r
- version = "";\r
- }\r
- \r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.getJedis();\r
- String routekey = MicroServiceUtil.getPrefixedKey(serverPort,serviceName, version, "*");\r
- Set<String> infoSet = jedis.keys(routekey);\r
-\r
- String[] paths = new String[infoSet.size()];\r
-\r
- infoSet.toArray(paths);\r
-\r
- jedis.del(paths);\r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new Exception("call redis throw exception:"+e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
- }\r
-\r
- public void deleteNode(String serviceName, String version, String ip,String port) throws Exception {\r
- if (null == version || "null".equals(version)) {\r
- version = "";\r
- }\r
- \r
- Jedis jedis = null;\r
- try {\r
- jedis = JedisUtil.getJedis();\r
- String serviceLBkey =\r
- MicroServiceUtil.getPrefixedKey("",serviceName, version,\r
- MicroServiceUtil.ROUTE_PATH_LOADBALANCE,ip+"-"+port);\r
- jedis.del(serviceLBkey);\r
- } catch (Exception e) {\r
- LOGGER.error("call redis throw exception", e);\r
- throw new Exception("call redis throw exception:"+e.getMessage());\r
- } finally {\r
- JedisUtil.returnJedisInstance(jedis);\r
- }\r
-\r
- }\r
-\r
-\r
- /**\r
- * Determine whether the service needs to send a notification \r
- * TODO: filter according to the agreement,\r
- * the only notice of agreement for REST \ UI interface MSB - REST \r
- * @param protocol\r
- * @return\r
- */\r
- private boolean isNeedNotifyByProtocol(String protocol) {\r
- return "UI".equalsIgnoreCase(protocol) ||("REST".equalsIgnoreCase(protocol));\r
- }\r
- \r
- /**\r
- * Determine whether the service needs to send a notification \r
- * TODO: according to the visual range filter conditions \r
- * @param visualRange\r
- * @return\r
- */\r
- private boolean isNeedNotifyByVisualRange(String visualRange) {\r
- String[] rangeArray=StringUtils.split(visualRange, "|");\r
- return RouteUtil.contain(RouteUtil.visualRangeMatches, rangeArray);\r
- }\r
-\r
- /**\r
- * According to the MicroServiceInfo entity information to judge whether need to send a notification\r
- * @param microServiceInfo\r
- * @return\r
- */\r
- private boolean isNeedNotify(Service microServiceInfo) {\r
- if (null != microServiceInfo) {\r
- return isNeedNotifyByProtocol(microServiceInfo.getProtocol()) &&\r
- isNeedNotifyByVisualRange(microServiceInfo.getVisualRange());\r
- } else {\r
- return false;\r
- }\r
- }\r
- \r
- \r
- \r
-}\r