src/main/java/org/onap/msb/sdk/httpclient/handler/impl/LBBuilder.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 /**
  15  *
  16  */
  17 package org.onap.msb.sdk.httpclient.handler.impl;
  18
  19 import org.onap.msb.sdk.httpclient.annotaion.LoadBalance;
  20 import org.onap.msb.sdk.httpclient.annotaion.LoadBalance.LBSTYLE;
  21 import org.onap.msb.sdk.httpclient.exception.RetrofitServiceRuntimeException;
  22 import org.onap.msb.sdk.httpclient.handler.HandlerContextBuilder;
  23 import org.onap.msb.sdk.httpclient.handler.RetrofitServiceHandlerContext;
  24 import org.onap.msb.sdk.httpclient.lb.ILoadBalanceStrategy;
  25 import org.onap.msb.sdk.httpclient.lb.RoundRobinLBStrategy;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 /**
  30  * @author 10071214
  31  *
  32  */
  33 public class LBBuilder implements HandlerContextBuilder {
  34
  35
  36   private final static Logger logger = LoggerFactory.getLogger(LBBuilder.class);
  37
  38   /*
  39    * (non-Javadoc)
  40    *
  41    * @see
  42    * com.zte.ums.zenap.versey.rpc.retrofit2.service.impl.handler.HandlerContextBuilder#build(com.zte
  43    * .ums.zenap.versey.rpc.retrofit2.service.impl.handler.RetrofitServiceHandlerContext)
  44    */
  45   @Override
  46   public void build(RetrofitServiceHandlerContext ctx) throws RetrofitServiceRuntimeException {
  47     initlb(ctx);
  48
  49   }
  50
  51   private void initlb(RetrofitServiceHandlerContext ctx) throws RetrofitServiceRuntimeException {
  52     LoadBalance lb = ctx.getRetrofitSrvInterfaceClazz().getAnnotation(LoadBalance.class);
  53
  54     LBSTYLE lbStyle = null;
  55     ILoadBalanceStrategy lbStrategy;
  56     if (lb != null) {
  57       lbStyle = lb.lbStyle();
  58       try {
  59         lbStrategy = (ILoadBalanceStrategy) Class.forName(lb.lbClassName()).newInstance();
  60       } catch (Exception e) {
  61         logger.error("init lb strategy error", e);
  62         throw new RetrofitServiceRuntimeException("init lb strategy error", e);
  63
  64       }
  65     } else {
  66       lbStyle = LBSTYLE.CLIENT;
  67       lbStrategy = new RoundRobinLBStrategy();
  68
  69     }
  70
  71     ctx.setLbStyle(lbStyle);
  72     ctx.setLbStrategy(lbStrategy);
  73
  74   }
  75
  76 }