1 files changed, 177 insertions, 0 deletions
diff --git a/rest-client/src/main/java/org/openo/baseservice/roa/util/ServiceUtil.java b/rest-client/src/main/java/org/openo/baseservice/roa/util/ServiceUtil.java
new file mode 100644
index 0000000..aca60c0
--- /dev/null
+++ b/rest-client/src/main/java/org/openo/baseservice/roa/util/ServiceUtil.java
@@ -0,0 +1,177 @@
+/*
+ * Copyright (c) 2016, Huawei Technologies Co., Ltd.
+ *
+ * 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.openo.baseservice.roa.util;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Enumeration;
+import java.util.Properties;
+
+/**
+ * Service connection configuration util.<br/>
+ * <p>
+ * Get host and port from the Client Configure the connection environment and service Configure
+ * profile
+ * </p>
+ * 
+ * @author
+ * @version SDNO 0.5 28-May-2016
+ */
+public class ServiceUtil {
+
+    private static final Logger LOG = LoggerFactory.getLogger(ServiceUtil.class);
+
+    private final Properties allConfigure = new Properties();
+
+    private final Properties serviceConfigure;
+
+    private String serviceStage;
+
+    private String serviceName;
+
+    /**
+     * Constructor<br/>
+     * <p>
+     * Load profile information.
+     * </p>
+     * 
+     * @since SDNO 0.5
+     * @param serviceName user-specified service name.
+     * @param url invoked service url.
+     */
+    public ServiceUtil(final String serviceName, final String url) {
+        final String fomattedUrl = formatUrl(url);
+        serviceConfigure = loadProperties("service-configure.properties");
+        if(null == serviceName || serviceName.isEmpty()) {
+            this.serviceName = getServiceNameWhitUrl(fomattedUrl);
+        } else {
+            this.serviceName = serviceName;
+        }
+        loadPropertyFile();
+    }
+
+    /**
+     * Get the service user-specified host.
+     * <br/>
+     * 
+     * @return host
+     * @since SDNO 0.5
+     */
+    public String getServiceHost() {
+        final String host = allConfigure.getProperty(serviceName + "." + serviceStage + ".host");
+        if(null == host) {
+            return "";
+        }
+        return host;
+    }
+
+    /**
+     * Get the service user-specified port.
+     * <br/>
+     * 
+     * @return port
+     * @since SDNO 0.5
+     */
+    public int getServicePort() {
+        final String portStr = allConfigure.getProperty(serviceName + "." + serviceStage + ".port");
+        if(null == portStr) {
+            return -1;
+        }
+        return Integer.parseInt(portStr);
+    }
+
+    private String getServiceNameWhitUrl(final String url) {
+        final Enumeration<?> keys = serviceConfigure.propertyNames();
+        while(keys.hasMoreElements()) {
+            final String key = (String)keys.nextElement();
+            if(key.endsWith("urls")) {
+                final String urls = serviceConfigure.getProperty(key);
+                for(String tempUrl : urls.split(",")) {
+                    tempUrl = formatUrl(tempUrl);
+                    if(url.startsWith(tempUrl)) {
+                        return key.split("\\.")[0];
+                    }
+                }
+
+            }
+        }
+        return "";
+    }
+
+    private static String formatUrl(final String url) {
+        String outUrl = url;
+        if(outUrl.contains("?")) {
+            outUrl = outUrl.split("\\?")[0];
+        }
+        outUrl = outUrl.replace("\\", "/");
+        outUrl = outUrl.replaceAll("[/]{2,}", "/");
+        outUrl = outUrl.endsWith("/") ? outUrl.substring(0, outUrl.length() - 1) : outUrl;
+        outUrl = outUrl.endsWith("/*") ? outUrl.substring(0, outUrl.length() - 2) : outUrl;
+        return outUrl;
+    }
+
+    /**
+     * Loads the client and service configuration files.
+     * <br/>
+     * 
+     * @since SDNO 0.5
+     */
+    private void loadPropertyFile() {
+        final Properties clientConfigure = loadProperties(serviceName + "-client-configure.properties");
+        allConfigure.putAll(clientConfigure);
+        allConfigure.putAll(serviceConfigure);
+        serviceStage = allConfigure.getProperty(serviceName + ".stage");
+    }
+
+    /**
+     * Loads the client and service configuration files.
+     * <br/>
+     * 
+     * @param classProperties: service profile file name.
+     * @return Service configuration.
+     * @since SDNO 0.5
+     */
+    private Properties loadProperties(final String classProperties) {
+        final Properties properties = new Properties();
+        InputStream inputStream = null;
+        ClassLoader classloader = null;
+        try {
+            classloader = this.getClass().getClassLoader();
+            if(classloader != null) {
+                inputStream = classloader.getResourceAsStream(classProperties);
+            }
+            if(inputStream != null) {
+                properties.load(inputStream);
+                inputStream.close();
+            }
+        } catch(final IOException e) {
+            LOG.error("load file error: ", e);
+        } finally {
+            if(inputStream != null) {
+                try {
+                    inputStream.close();
+                } catch(final IOException ee) {
+                    LOG.error("close inputStream  error: ", ee);
+                }
+                inputStream = null;
+            }
+        }
+        return properties;
+    }
+}