src/main/java/org/onap/msb/sdk/httpclient/lb/RoundRobinLBStrategy.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.lb;
  15
  16 import java.util.concurrent.atomic.AtomicLong;
  17
  18 import org.onap.msb.sdk.httpclient.ServiceHttpEndPointObject;
  19 import org.onap.msb.sdk.httpclient.exception.RetrofitServiceRuntimeException;
  20
  21 /**
  22  * @author hu.rui
  23  *
  24  */
  25 public class RoundRobinLBStrategy implements ILoadBalanceStrategy {
  26
  27   private AtomicLong invokeCount = new AtomicLong(0);
  28
  29   /*
  30    * (non-Javadoc)
  31    *
  32    * @see
  33    * com.zte.ums.zenap.httpclient.retrofit.lb.ILoadBalanceStrategy#chooseEndPointObject(com.zte.ums.
  34    * zenap.httpclient.retrofit.lb.LoadBalanceContext)
  35    */
  36   @Override
  37   public ServiceHttpEndPointObject chooseEndPointObject(LoadBalanceContext lbCtx) {
  38
  39     long invokecount = invokeCount.getAndIncrement();
  40
  41     if (lbCtx.getEndPoints().size() == 0) {
  42       throw new RetrofitServiceRuntimeException("target endPoints is empty");
  43     }
  44
  45     int mod = (int) (invokecount % lbCtx.getEndPoints().size());
  46
  47     return lbCtx.getEndPoints().get(mod);
  48   }
  49
  50 }