src/main/java/org/onap/msb/sdk/httpclient/msb/MSBServiceWrapperClient.java

   1 /*******************************************************************************
   2  * Copyright 2017 ZTE, Inc. and others.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   5  * in compliance with the License. You may obtain a copy of the License at
   6  *
   7  * http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software distributed under the License
  10  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  11  * or implied. See the License for the specific language governing permissions and limitations under
  12  * the License.
  13  ******************************************************************************/
  14 package org.onap.msb.sdk.httpclient.msb;
  15
  16 import java.util.concurrent.Callable;
  17
  18 import org.onap.msb.sdk.discovery.MSBService;
  19 import org.onap.msb.sdk.discovery.common.RouteException;
  20 import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo;
  21 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
  22 import org.onap.msb.sdk.discovery.entity.NodeAddress;
  23 import org.onap.msb.sdk.discovery.entity.RouteResult;
  24 import org.slf4j.Logger;
  25 import org.slf4j.LoggerFactory;
  26
  27
  28 public class MSBServiceWrapperClient {
  29
  30   private static final Logger logger = LoggerFactory.getLogger(MSBServiceWrapperClient.class);
  31
  32   private String msbSvrAddress;
  33
  34   private MSBService msbService = new MSBService();
  35
  36   private String nameSpace;
  37
  38   public MSBServiceWrapperClient(String msbSvrIp, int msbSvrPort) {
  39     super();
  40     this.msbSvrAddress = msbSvrIp + ":" + msbSvrPort;
  41
  42     logger.info("msb service info:msbSvrAddress:{},nameSpace:{}", this.msbSvrAddress,
  43         this.nameSpace);
  44
  45   }
  46
  47   public MicroServiceFullInfo queryMicroServiceInfo(String serviceName, String version)
  48       throws RouteException {
  49
  50     return invokeMsb(() -> msbService.queryMicroServiceInfo(msbSvrAddress, serviceName, version));
  51
  52
  53   }
  54
  55   private <V> V invokeMsb(Callable<V> callable) throws RouteException {
  56     try {
  57       return callable.call();
  58     } catch (Exception e) {
  59       logger.error("msb service info:msbSvrAddress:" + this.msbSvrAddress, e);
  60       if (e instanceof RouteException) {
  61         throw (RouteException) e;
  62       } else {
  63         throw new RuntimeException(e);
  64       }
  65     }
  66
  67   }
  68
  69
  70   /**
  71    * 注册微服务-默认追加方式,在msb上注册服务信息，服务的信息都可以填充在MicroServiceInfo上，包括：租户的信息等
  72    *
  73    * @param microServiceInfo 微服务注册实体类
  74    * @throws RouteException
  75    * @return MicroServiceFullInfo
  76    */
  77   public MicroServiceFullInfo registerMicroServiceInfo(MicroServiceInfo msinfo)
  78       throws RouteException {
  79
  80     return invokeMsb(() -> msbService.registerMicroServiceInfo(msbSvrAddress, msinfo));
  81   }
  82
  83
  84
  85   /**
  86    * 注册微服务,在msb上注册服务信息，服务的信息都可以填充在MicroServiceInfo上，包括：租户的信息等
  87    *
  88    * @param microServiceInfo 微服务注册实体类
  89    * @param createOrUpdate true：新增或追加更新 ,false：重新添加
  90    * @throws RouteException
  91    * @return MicroServiceFullInfo
  92    */
  93   public MicroServiceFullInfo registerMicroServiceInfo(MicroServiceInfo msinfo,
  94       boolean createOrUpdate) throws RouteException {
  95
  96     return invokeMsb(
  97         () -> msbService.registerMicroServiceInfo(msbSvrAddress, msinfo, createOrUpdate));
  98   }
  99
 100
 101
 102   /**
 103    * @Title cancelMicroServiceInfo
 104    * @Description TODO(注销全部微服务)
 105    * @param serviceName 服务名[必填,若自定义服务名包含/，用*代替]
 106    * @param version 版本号[若无版本号，传空字符串]
 107    * @throws RouteException
 108    * @return RouteResult
 109    */
 110   public RouteResult cancelMicroServiceInfo(String serviceName, String version)
 111       throws RouteException {
 112     return invokeMsb(() -> msbService.cancelMicroServiceInfo(msbSvrAddress, serviceName, version));
 113   }
 114
 115
 116
 117   /**
 118    * 注销单个微服务
 119    *
 120    * @param serviceName 服务名[必填,若自定义服务名包含/，用*代替]
 121    * @param version 版本号[若无版本号，传空字符串]
 122    * @param ip
 123    * @param port
 124    * @throws RouteException
 125    * @return RouteResult
 126    */
 127
 128   public RouteResult cancelMicroServiceInfo(String serviceName, String version, String ip,
 129       String port) throws RouteException {
 130
 131     return invokeMsb(
 132         () -> msbService.cancelMicroServiceInfo(msbSvrAddress, serviceName, version, ip, port));
 133   }
 134
 135
 136   public String getMsbSvrAddress() {
 137     return msbSvrAddress;
 138   }
 139
 140
 141
 142   /**
 143    * 请求服务实例TTL健康检查，默认使用本租户的信息
 144    *
 145    * @param serviceName 服务名
 146    * @param version 版本号[若无版本号，传空字符串]
 147    * @param ip 实例IP
 148    * @param port 实例端口
 149    * @throws RouteException
 150    * @return CheckNode
 151    */
 152   public NodeAddress healthCheckbyTTL(String serviceName, String version, String ip, String port)
 153       throws RouteException {
 154
 155     return invokeMsb(
 156         () -> msbService.healthCheckbyTTL(msbSvrAddress, serviceName, version, ip, port));
 157
 158   }
 159
 160
 161
 162 }