Java tutorial
/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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 com.alibaba.jstorm.daemon.nimbus.metric.uploader; import backtype.storm.generated.TopologyMetric; import org.apache.http.HttpEntity; import org.apache.http.NameValuePair; import org.apache.http.client.entity.UrlEncodedFormEntity; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpGet; import org.apache.http.client.methods.HttpPost; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClientBuilder; import org.apache.http.message.BasicNameValuePair; import org.apache.http.util.EntityUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.net.URLEncoder; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; public class AlimonitorClient extends DefaultMetricUploader { private static final Logger LOG = LoggerFactory.getLogger(AlimonitorClient.class); // Send to localhost:15776 by default public static final String DEFAUT_ADDR = "127.0.0.1"; public static final String DEFAULT_PORT = "15776"; public static final int DEFAUTL_FLAG = 0; public static final String DEFAULT_ERROR_INFO = ""; private final String COLLECTION_FLAG = "collection_flag"; private final String ERROR_INFO = "error_info"; private final String MSG = "MSG"; private String port; private String requestIP; private String monitorName; private int collectionFlag; private String errorInfo; private boolean post; public AlimonitorClient() { } public AlimonitorClient(String requestIP, String port, boolean post) { this.requestIP = requestIP; this.port = port; this.post = post; this.monitorName = null; this.collectionFlag = 0; this.errorInfo = null; } public void setIpAddr(String ipAddr) { this.requestIP = ipAddr; } public void setPort(String port) { this.port = port; } public void setMonitorName(String monitorName) { this.monitorName = monitorName; } public void setCollectionFlag(int flag) { this.collectionFlag = flag; } public void setErrorInfo(String msg) { this.errorInfo = msg; } public void setPostFlag(boolean post) { this.post = post; } public String buildURL() { return "http://" + requestIP + ":" + port + "/passive"; } public String buildRqstAddr() { return "http://" + requestIP + ":" + port + "/passive?name=" + monitorName + "&msg="; } public Map buildAliMonitorMsg(int collection_flag, String error_message) { // Json format of the message sent to Alimonitor // { // "collection_flag":int, // "error_info":string, // "MSG": ojbect | array // } Map ret = new HashMap(); ret.put(COLLECTION_FLAG, collection_flag); ret.put(ERROR_INFO, error_message); ret.put(MSG, null); return ret; } private void addMsgData(Map jsonObj, Map<String, Object> map) { jsonObj.put(MSG, map); } private boolean sendRequest(int collection_flag, String error_message, Map<String, Object> msg) throws Exception { boolean ret = false; if (msg.size() == 0) return ret; Map jsonObj = buildAliMonitorMsg(collection_flag, error_message); addMsgData(jsonObj, msg); String jsonMsg = jsonObj.toString(); LOG.info(jsonMsg); if (post) { String url = buildURL(); ret = httpPost(url, jsonMsg); } else { String request = buildRqstAddr(); StringBuilder postAddr = new StringBuilder(); postAddr.append(request); postAddr.append(URLEncoder.encode(jsonMsg)); ret = httpGet(postAddr); } return ret; } private boolean httpGet(StringBuilder postAddr) { boolean ret = false; CloseableHttpClient httpClient = HttpClientBuilder.create().build(); CloseableHttpResponse response = null; try { HttpGet request = new HttpGet(postAddr.toString()); response = httpClient.execute(request); HttpEntity entity = response.getEntity(); if (entity != null) { LOG.info(EntityUtils.toString(entity)); } EntityUtils.consume(entity); ret = true; } catch (Exception e) { LOG.error("Exception when sending http request to ali monitor", e); } finally { try { if (response != null) response.close(); httpClient.close(); } catch (Exception e) { LOG.error("Exception when closing httpclient", e); } } return ret; } private boolean httpPost(String url, String msg) { boolean ret = false; CloseableHttpClient httpClient = HttpClientBuilder.create().build(); CloseableHttpResponse response = null; try { HttpPost request = new HttpPost(url); List<NameValuePair> nvps = new ArrayList<>(); nvps.add(new BasicNameValuePair("name", monitorName)); nvps.add(new BasicNameValuePair("msg", msg)); request.setEntity(new UrlEncodedFormEntity(nvps)); response = httpClient.execute(request); HttpEntity entity = response.getEntity(); if (entity != null) { LOG.info(EntityUtils.toString(entity)); } EntityUtils.consume(entity); ret = true; } catch (Exception e) { LOG.error("Exception when sending http request to ali monitor", e); } finally { try { if (response != null) response.close(); httpClient.close(); } catch (Exception e) { LOG.error("Exception when closing httpclient", e); } } return ret; } protected Map<String, Object> convertMap(String clusterName, String topologyId, TopologyMetric tpMetric) { return null; } @Override public boolean upload(String clusterName, String topologyId, TopologyMetric tpMetric, Map<String, Object> metricContext) { Map<String, Object> metricMap = convertMap(clusterName, topologyId, tpMetric); if (metricMap == null || metricMap.isEmpty()) { return false; } try { sendRequest(collectionFlag, errorInfo, metricMap); return true; } catch (Exception e) { LOG.error("Failed to upload metric to ali monitor", e); return false; } } }