1 files changed, 51 insertions, 0 deletions
diff --git a/activiti-extension/src/main/java/org/onap/workflow/utils/MsbUtils.java b/activiti-extension/src/main/java/org/onap/workflow/utils/MsbUtils.java
new file mode 100644
index 0000000..4757237
--- /dev/null
+++ b/activiti-extension/src/main/java/org/onap/workflow/utils/MsbUtils.java
@@ -0,0 +1,51 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * 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.workflow.utils;
+
+import org.apache.commons.lang3.StringUtils;
+import org.jvnet.hk2.annotations.Service;
+import org.onap.msb.sdk.discovery.MSBService;
+import org.onap.msb.sdk.discovery.common.RouteException;
+import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo;
+import org.onap.workflow.activitiext.restservicetask.PropertyUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+@Service
+public class MsbUtils {
+
+	private static final Logger LOG = LoggerFactory.getLogger(MsbUtils.class);
+
+	public static String getServiceAddress(String serviceName, String version) throws RouteException {
+
+		String baseUrl = "";
+		String msbIp = System.getenv("OPENPALETTE_MSB_IP");
+		String msbPort = System.getenv("OPENPALETTE_MSB_PORT");
+		
+		if(StringUtils.isNotEmpty(msbIp) && StringUtils.isNotEmpty(msbPort))
+		{
+			baseUrl = "http://" + msbIp + ":" + msbPort;
+		}else{
+			baseUrl = PropertyUtil.getBasePath();
+		}
+		
+		MicroServiceFullInfo msbInfo = new MSBService().queryMicroServiceInfo(baseUrl, serviceName, version);
+		
+		LOG.info("publishUrl: {}", msbInfo.getUrl());
+
+		return msbInfo.getUrl();
+	}
+}