1 files changed, 76 insertions, 0 deletions
diff --git a/src/main/java/org/onap/msb/sdk/httpclient/handler/impl/LBBuilder.java b/src/main/java/org/onap/msb/sdk/httpclient/handler/impl/LBBuilder.java
new file mode 100644
index 0000000..98b2a32
--- /dev/null
+++ b/src/main/java/org/onap/msb/sdk/httpclient/handler/impl/LBBuilder.java
@@ -0,0 +1,76 @@
+/*******************************************************************************
+ * Copyright 2017 ZTE, Inc. and others.
+ * 
+ * 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.onap.msb.sdk.httpclient.handler.impl;
+
+import org.onap.msb.sdk.httpclient.annotaion.LoadBalance;
+import org.onap.msb.sdk.httpclient.annotaion.LoadBalance.LBSTYLE;
+import org.onap.msb.sdk.httpclient.exception.RetrofitServiceRuntimeException;
+import org.onap.msb.sdk.httpclient.handler.HandlerContextBuilder;
+import org.onap.msb.sdk.httpclient.handler.RetrofitServiceHandlerContext;
+import org.onap.msb.sdk.httpclient.lb.ILoadBalanceStrategy;
+import org.onap.msb.sdk.httpclient.lb.RoundRobinLBStrategy;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * @author 10071214
+ *
+ */
+public class LBBuilder implements HandlerContextBuilder {
+
+
+  private final static Logger logger = LoggerFactory.getLogger(LBBuilder.class);
+
+  /*
+   * (non-Javadoc)
+   * 
+   * @see
+   * com.zte.ums.zenap.versey.rpc.retrofit2.service.impl.handler.HandlerContextBuilder#build(com.zte
+   * .ums.zenap.versey.rpc.retrofit2.service.impl.handler.RetrofitServiceHandlerContext)
+   */
+  @Override
+  public void build(RetrofitServiceHandlerContext ctx) throws RetrofitServiceRuntimeException {
+    initlb(ctx);
+
+  }
+
+  private void initlb(RetrofitServiceHandlerContext ctx) throws RetrofitServiceRuntimeException {
+    LoadBalance lb = ctx.getRetrofitSrvInterfaceClazz().getAnnotation(LoadBalance.class);
+
+    LBSTYLE lbStyle = null;
+    ILoadBalanceStrategy lbStrategy;
+    if (lb != null) {
+      lbStyle = lb.lbStyle();
+      try {
+        lbStrategy = (ILoadBalanceStrategy) Class.forName(lb.lbClassName()).newInstance();
+      } catch (Exception e) {
+        logger.error("init lb strategy error", e);
+        throw new RetrofitServiceRuntimeException("init lb strategy error", e);
+
+      }
+    } else {
+      lbStyle = LBSTYLE.CLIENT;
+      lbStrategy = new RoundRobinLBStrategy();
+
+    }
+
+    ctx.setLbStyle(lbStyle);
+    ctx.setLbStrategy(lbStrategy);
+
+  }
+
+}