diff options
author | 2017-03-08 17:22:05 +0800 | |
---|---|---|
committer | 2017-03-09 09:24:37 +0800 | |
commit | d55c83c257914112016a9205b6ed81b73116a42a (patch) | |
tree | b7d28555552e2f09f5199abc4efd1caf42fc815f /holmes-actions/src | |
parent | bbd681988347c0a055def849abbb2e8dccb72d45 (diff) |
Add msb register code
Issue-ID:HOLMES-51
Change-Id: I5f12140459a22d16aa7b51884a3d54dd2aac4c8e
Signed-off-by: youbowu <wu.youbo@zte.com.cn>
Diffstat (limited to 'holmes-actions/src')
7 files changed, 290 insertions, 0 deletions
diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/api/entity/ServiceNode.java b/holmes-actions/src/main/java/org/openo/holmes/common/api/entity/ServiceNode.java new file mode 100644 index 0000000..429bf3b --- /dev/null +++ b/holmes-actions/src/main/java/org/openo/holmes/common/api/entity/ServiceNode.java @@ -0,0 +1,31 @@ +/**
+ * 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.openo.holmes.common.api.entity;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class ServiceNode {
+
+ private String ip;
+ private String port;
+ private int ttl;
+
+}
\ No newline at end of file diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/api/entity/ServiceRegisterEntity.java b/holmes-actions/src/main/java/org/openo/holmes/common/api/entity/ServiceRegisterEntity.java new file mode 100644 index 0000000..e7b032c --- /dev/null +++ b/holmes-actions/src/main/java/org/openo/holmes/common/api/entity/ServiceRegisterEntity.java @@ -0,0 +1,46 @@ +/**
+ * 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.openo.holmes.common.api.entity;
+
+import java.util.ArrayList;
+import java.util.List;
+import lombok.Getter;
+import lombok.Setter;
+
+@Setter
+@Getter
+public class ServiceRegisterEntity {
+
+ private String serviceName;
+ private String version;
+ private String url;
+ private String protocol;
+ private String visualRange = "1";
+ private List<ServiceNode> nodes = new ArrayList<>();
+
+ public void setSingleNode(String ip, String port, int ttl) {
+ ServiceNode node = new ServiceNode();
+ if (ip != null && ip.length() > 0) {
+ node.setIp(ip);
+ } else {
+ node.setIp(null);
+ }
+ node.setPort(port);
+ node.setTtl(ttl);
+ nodes.add(node);
+ }
+}
\ No newline at end of file diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/config/MicroServiceConfig.java b/holmes-actions/src/main/java/org/openo/holmes/common/config/MicroServiceConfig.java new file mode 100644 index 0000000..ba55562 --- /dev/null +++ b/holmes-actions/src/main/java/org/openo/holmes/common/config/MicroServiceConfig.java @@ -0,0 +1,36 @@ +/**
+ * 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.openo.holmes.common.config;
+
+public class MicroServiceConfig {
+
+ private static String getProperty(String name) {
+ String value = System.getenv(name);
+ if (value == null) {
+ value = System.getProperty(name);
+ }
+ return value;
+ }
+
+ public static String getMsbServerAddr() {
+ return getProperty("MSB_ADDR");
+ }
+
+ public static String getServiceIp() {
+ return getProperty("SERVICE_IP");
+ }
+
+}
diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/constant/AlarmConst.java b/holmes-actions/src/main/java/org/openo/holmes/common/constant/AlarmConst.java index c2248f1..2ddf78f 100644 --- a/holmes-actions/src/main/java/org/openo/holmes/common/constant/AlarmConst.java +++ b/holmes-actions/src/main/java/org/openo/holmes/common/constant/AlarmConst.java @@ -32,4 +32,6 @@ public interface AlarmConst { String ZH_CN = "zh_CN";
String ADMIN = "admin";
+
+ int RESPONSE_STATUS_OK = 200;
}
diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/utils/MSBRegisterUtil.java b/holmes-actions/src/main/java/org/openo/holmes/common/utils/MSBRegisterUtil.java new file mode 100644 index 0000000..49dff92 --- /dev/null +++ b/holmes-actions/src/main/java/org/openo/holmes/common/utils/MSBRegisterUtil.java @@ -0,0 +1,65 @@ +/**
+ * 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.openo.holmes.common.utils;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import java.io.IOException;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.entity.ByteArrayEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClients;
+import org.jvnet.hk2.annotations.Service;
+import org.openo.holmes.common.api.entity.ServiceRegisterEntity;
+import org.openo.holmes.common.config.MicroServiceConfig;
+import org.openo.holmes.common.constant.AlarmConst;
+
+@Slf4j
+@Service
+public class MSBRegisterUtil {
+
+ public boolean register(ServiceRegisterEntity entity) throws IOException {
+ CloseableHttpClient httpClient = HttpClients.createDefault();
+ try {
+ ObjectMapper mapper = new ObjectMapper();
+ String content = mapper.writeValueAsString(entity);
+ HttpPost httpPost = new HttpPost("http://" + MicroServiceConfig.getMsbServerAddr()
+ + ":8086/openoapi/microservices/v1/services?createOrUpdate=false");
+ if (StringUtils.isNotEmpty(content)) {
+ httpPost.setEntity(new ByteArrayEntity(content.getBytes()));
+ }
+ HttpResponse response;
+ try {
+ response = httpClient.execute(httpPost);
+ } catch (Exception e) {
+ log.warn("Registering the service to the bus failure", e);
+ return false;
+ }
+ if (response.getStatusLine().getStatusCode() == AlarmConst.RESPONSE_STATUS_OK) {
+ log.info("Registration successful service to the bus :" + response.getEntity());
+ return true;
+ } else {
+ log.warn("Registering the service to the bus failure");
+ return false;
+ }
+ } finally {
+ httpClient.close();
+ }
+ }
+}
\ No newline at end of file diff --git a/holmes-actions/src/test/java/org/openo/holmes/common/api/entity/ServiceNodeTest.java b/holmes-actions/src/test/java/org/openo/holmes/common/api/entity/ServiceNodeTest.java new file mode 100644 index 0000000..383cb91 --- /dev/null +++ b/holmes-actions/src/test/java/org/openo/holmes/common/api/entity/ServiceNodeTest.java @@ -0,0 +1,48 @@ +/**
+ * 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.openo.holmes.common.api.entity;
+
+import static org.hamcrest.core.IsEqual.equalTo;
+import static org.junit.Assert.assertThat;
+
+import org.junit.Test;
+
+public class ServiceNodeTest {
+
+ private ServiceNode serviceNode = new ServiceNode();
+
+ @Test
+ public void getterAndSetter4ip() {
+ String ip = "test";
+ serviceNode.setIp(ip);
+ assertThat(serviceNode.getIp(), equalTo(ip));
+ }
+
+ @Test
+ public void getterAndSetter4port() {
+ String port = "test";
+ serviceNode.setPort(port);
+ assertThat(serviceNode.getPort(), equalTo(port));
+ }
+
+ @Test
+ public void getterAndSetter4ttl() {
+ int ttl = 1;
+ serviceNode.setTtl(ttl);
+ assertThat(serviceNode.getTtl(), equalTo(ttl));
+ }
+}
\ No newline at end of file diff --git a/holmes-actions/src/test/java/org/openo/holmes/common/api/entity/ServiceRegisterEntityTest.java b/holmes-actions/src/test/java/org/openo/holmes/common/api/entity/ServiceRegisterEntityTest.java new file mode 100644 index 0000000..442549d --- /dev/null +++ b/holmes-actions/src/test/java/org/openo/holmes/common/api/entity/ServiceRegisterEntityTest.java @@ -0,0 +1,62 @@ +/**
+ * 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.openo.holmes.common.api.entity;
+
+import static org.hamcrest.core.IsEqual.equalTo;
+import static org.junit.Assert.assertThat;
+
+import org.junit.Test;
+
+public class ServiceRegisterEntityTest {
+
+ private ServiceRegisterEntity serviceRegisterEntity = new ServiceRegisterEntity();
+
+ @Test
+ public void getterAndSetter4protocol() {
+ String protocol = "test";
+ serviceRegisterEntity.setProtocol(protocol);
+ assertThat(serviceRegisterEntity.getProtocol(), equalTo(protocol));
+ }
+
+ @Test
+ public void getterAndSetter4serviceName() {
+ String serviceName = "test";
+ serviceRegisterEntity.setServiceName(serviceName);
+ assertThat(serviceRegisterEntity.getServiceName(), equalTo(serviceName));
+ }
+
+ @Test
+ public void getterAndSetter4url() {
+ String url = "test";
+ serviceRegisterEntity.setUrl(url);
+ assertThat(serviceRegisterEntity.getUrl(), equalTo(url));
+ }
+
+ @Test
+ public void getterAndSetter4version() {
+ String version = "test";
+ serviceRegisterEntity.setVersion(version);
+ assertThat(serviceRegisterEntity.getVersion(), equalTo(version));
+ }
+
+ @Test
+ public void getterAndSetter4visualRange() {
+ String visualRange = "test";
+ serviceRegisterEntity.setVisualRange(visualRange);
+ assertThat(serviceRegisterEntity.getVisualRange(), equalTo(visualRange));
+ }
+}
\ No newline at end of file |