gerrit.onap Code Review - dmaap/messagerouter/dmaapclient.git/blob

   1 /*******************************************************************************
   2  *  ============LICENSE_START=======================================================
   3  *  org.onap.dmaap
   4  *  ================================================================================
   5  *  Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  *  ================================================================================
   7  *  Licensed under the Apache License, Version 2.0 (the "License");
   8  *  you may not use this file except in compliance with the License.
   9  *  You may obtain a copy of the License at
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *  ============LICENSE_END=========================================================
  18  *
  19  *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
  20  *
  21  *******************************************************************************/
  22
  23 package org.onap.dmaap.messagerouter.dmaapclient.nsa.mr.dme.client;
  24
  25
  26 import java.io.IOException;
  27 import java.util.List;
  28 import java.util.Map;
  29
  30 import java.util.concurrent.TimeUnit;
  31
  32 import javax.ws.rs.core.MultivaluedMap;
  33
  34 import org.json.JSONObject;
  35 import org.onap.dmaap.messagerouter.dmaapclient.nsa.mr.client.MRBatchingPublisher;
  36 import org.onap.dmaap.messagerouter.dmaapclient.nsa.mr.client.MRClientFactory;
  37 import org.onap.dmaap.messagerouter.dmaapclient.nsa.mr.client.MRPublisher.message;
  38
  39 /**
  40  * An example of how to use the Java publisher.
  41  *
  42  * @author author
  43  */
  44 public class SimpleExamplePublisher {
  45         static String content = null;
  46         static String messageSize = null;
  47         static String transport = null;
  48         static String messageCount = null;
  49
  50         public void publishMessage(String producerFilePath) throws IOException, InterruptedException, Exception {
  51
  52                 // create our publisher
  53
  54                 // publish some messages
  55
  56
  57                 StringBuilder sb = new StringBuilder();
  58                 final MRBatchingPublisher pub = MRClientFactory.createBatchingPublisher(producerFilePath);
  59
  60                 if (content.equalsIgnoreCase("text/plain")) {
  61                         for (int i = 0; i < Integer.parseInt(messageCount); i++) {
  62                                 for (int j = 0; j < Integer.parseInt(messageSize); j++) {
  63                                         sb.append("T");
  64                                 }
  65
  66                                 pub.send(sb.toString());
  67                         }
  68                 } else if (content.equalsIgnoreCase("application/cambria")) {
  69                         for (int i = 0; i < Integer.parseInt(messageCount); i++) {
  70                                 for (int j = 0; j < Integer.parseInt(messageSize); j++) {
  71                                         sb.append("C");
  72                                 }
  73
  74                                 pub.send("Key", sb.toString());
  75                         }
  76                 } else if (content.equalsIgnoreCase("application/json")) {
  77                         for (int i = 0; i < Integer.parseInt(messageCount); i++) {
  78
  79                                         final JSONObject msg12 = new JSONObject();
  80                                         msg12.put("Name", "DMaaP Reference Client to Test jason Message");
  81
  82                                         pub.send(msg12.toString());
  83
  84                         }
  85                 }
  86
  87                 // ...
  88
  89                 // close the publisher to make sure everything's sent before exiting.
  90                 // The batching
  91                 // publisher interface allows the app to get the set of unsent messages.
  92                 // It could
  93                 // write them to disk, for example, to try to send them later.
  94         /*      final List<message> stuck = pub.close(20, TimeUnit.SECONDS);
  95                 if (stuck.size() > 0) {
  96                         System.err.println(stuck.size() + " messages unsent");
  97                 } else {
  98                         System.out.println("Clean exit; all messages sent.");
  99                 }*/
 100
 101                 if (transport.equalsIgnoreCase("HTTP")) {
 102                         MultivaluedMap<String, Object> headersMap = MRClientFactory.HTTPHeadersMap;
 103                         for (String key : headersMap.keySet()) {
 104                                 System.out.println("Header Key " + key);
 105                                 System.out.println("Header Value " + headersMap.get(key));
 106                         }
 107                 } else {
 108                         Map<String, String> dme2headersMap = MRClientFactory.DME2HeadersMap;
 109                         for (String key : dme2headersMap.keySet()) {
 110                                 System.out.println("Header Key " + key);
 111                                 System.out.println("Header Value " + dme2headersMap.get(key));
 112                         }
 113                 }
 114
 115         }
 116
 117         public static void main(String[] args) throws InterruptedException, Exception {
 118
 119                 String producerFilePath = args[0];
 120                 content = args[1];
 121                 messageSize = args[2];
 122                 transport = args[3];
 123                 messageCount = args[4];
 124                 /*String producerFilePath = null;
 125                 content = null;
 126                 messageSize =null;
 127                 transport =null;
 128                 messageCount = null;*/
 129                 SimpleExamplePublisher publisher = new SimpleExamplePublisher();
 130
 131                 publisher.publishMessage("D:\\SG\\producer.properties");
 132         }
 133
 134 }