diff options
Diffstat (limited to 'src/main/java/org/openecomp/dmaapbc/client/MrTopicConnection.java')
| -rw-r--r-- | src/main/java/org/openecomp/dmaapbc/client/MrTopicConnection.java | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/src/main/java/org/openecomp/dmaapbc/client/MrTopicConnection.java b/src/main/java/org/openecomp/dmaapbc/client/MrTopicConnection.java new file mode 100644 index 0000000..1ab2431 --- /dev/null +++ b/src/main/java/org/openecomp/dmaapbc/client/MrTopicConnection.java @@ -0,0 +1,168 @@ +/*-
+ * ============LICENSE_START======================================================= + * OpenECOMP - org.openecomp.dmaapbc + * ================================================================================ + * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved. + * ================================================================================ + * 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. + * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.dmaapbc.client;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.OutputStream;
+import java.net.ProtocolException;
+import java.net.URL;
+import javax.net.ssl.HttpsURLConnection;
+
+import org.apache.commons.codec.binary.Base64;
+import org.apache.log4j.Logger;
+import org.openecomp.dmaapbc.model.ApiError;
+import org.openecomp.dmaapbc.model.MR_Cluster;
+
+public class MrTopicConnection {
+ static final Logger logger = Logger.getLogger(MrTopicConnection.class);
+ private String topicURL;
+
+ private HttpsURLConnection uc;
+
+
+ private String mmProvCred;
+
+
+
+ public MrTopicConnection(String user, String pwd ) {
+ mmProvCred = new String( user + ":" + pwd );
+
+ }
+
+ public boolean makeTopicConnection( MR_Cluster cluster, String topic ) {
+ logger.info( "connect to cluster: " + cluster.getFqdn() + " for topic: " + topic );
+
+
+ topicURL = cluster.getTopicProtocol() + "://" + cluster.getFqdn() + ":" + cluster.getTopicPort() + "/events/" + topic ;
+
+ return makeConnection( topicURL );
+ }
+
+ private boolean makeConnection( String pURL ) {
+ logger.info( "makeConnection to " + pURL );
+
+ try {
+ URL u = new URL( pURL );
+ uc = (HttpsURLConnection) u.openConnection();
+ uc.setInstanceFollowRedirects(false);
+ logger.info( "successful connect to " + pURL );
+ return(true);
+ } catch (Exception e) {
+ logger.error("Unexpected error during openConnection of " + pURL );
+ e.printStackTrace();
+ return(false);
+ }
+
+ }
+
+ static String bodyToString( InputStream is ) {
+ StringBuilder sb = new StringBuilder();
+ BufferedReader br = new BufferedReader( new InputStreamReader(is));
+ String line;
+ try {
+ while ((line = br.readLine()) != null ) {
+ sb.append( line );
+ }
+ } catch (IOException ex ) {
+ logger.error( "IOexception:" + ex);
+ }
+
+ return sb.toString();
+ }
+
+ public ApiError doPostMessage( String postMessage ) {
+ ApiError response = new ApiError();
+ String auth = "Basic " + Base64.encodeBase64String(mmProvCred.getBytes());
+
+
+
+ try {
+ byte[] postData = postMessage.getBytes();
+ logger.info( "post fields=" + postMessage );
+ uc.setRequestProperty("Authorization", auth);
+ logger.info( "Authenticating with " + auth );
+ uc.setRequestMethod("POST");
+ uc.setRequestProperty("Content-Type", "application/json");
+ uc.setRequestProperty( "charset", "utf-8");
+ uc.setRequestProperty( "Content-Length", Integer.toString( postData.length ));
+ uc.setUseCaches(false);
+ uc.setDoOutput(true);
+ OutputStream os = null;
+
+
+ try {
+ uc.connect();
+ os = uc.getOutputStream();
+ os.write( postData );
+
+ } catch (ProtocolException pe) {
+ // Rcvd error instead of 100-Continue
+ try {
+ // work around glitch in Java 1.7.0.21 and likely others
+ // without this, Java will connect multiple times to the server to run the same request
+ uc.setDoOutput(false);
+ } catch (Exception e) {
+ }
+ }
+ response.setCode( uc.getResponseCode());
+ logger.info( "http response code:" + response.getCode());
+ response.setMessage( uc.getResponseMessage() );
+ logger.info( "response message=" + response.getMessage() );
+
+
+ if ( response.getMessage() == null) {
+ // work around for glitch in Java 1.7.0.21 and likely others
+ // When Expect: 100 is set and a non-100 response is received, the response message is not set but the response code is
+ String h0 = uc.getHeaderField(0);
+ if (h0 != null) {
+ int i = h0.indexOf(' ');
+ int j = h0.indexOf(' ', i + 1);
+ if (i != -1 && j != -1) {
+ response.setMessage( h0.substring(j + 1) );
+ }
+ }
+ }
+ if ( response.is2xx() ) {
+ response.setFields( bodyToString( uc.getInputStream() ) );
+ logger.info( "responseBody=" + response.getFields() );
+ return response;
+
+ }
+
+ } catch (Exception e) {
+ response.setCode(500);
+ response.setMessage( "Unable to read response");
+ logger.warn( response.getMessage() );
+ e.printStackTrace();
+ }
+ finally {
+ try {
+ uc.disconnect();
+ } catch ( Exception e ) {}
+ }
+ return response;
+
+ }
+
+}
|