src/test/java/org/onap/dmaap/mr/test/clients/SimpleExamplePublisher.java

   1 /*******************************************************************************
   2  *  ============LICENSE_START=======================================================
   3  *  org.onap.dmaap
   4  *  ================================================================================
   5  *  Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  *  ================================================================================
   7  *  Modifications Copyright © 2021 Orange.
   8  *  ================================================================================
   9  *  Licensed under the Apache License, Version 2.0 (the "License");
  10  *  you may not use this file except in compliance with the License.
  11  *  You may obtain a copy of the License at
  12  *        http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  *  Unless required by applicable law or agreed to in writing, software
  15  *  distributed under the License is distributed on an "AS IS" BASIS,
  16  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  *  See the License for the specific language governing permissions and
  18  *  limitations under the License.
  19  *  ============LICENSE_END=========================================================
  20  *
  21  *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
  22  *
  23  *******************************************************************************/
  24
  25 package org.onap.dmaap.mr.test.clients;
  26
  27 import org.json.JSONObject;
  28 import org.onap.dmaap.mr.client.MRBatchingPublisher;
  29 import org.onap.dmaap.mr.client.MRClientFactory;
  30 import org.onap.dmaap.mr.client.MRPublisher.Message;
  31
  32 import java.io.File;
  33 import java.io.FileReader;
  34 import java.io.FileWriter;
  35 import java.io.IOException;
  36 import java.util.List;
  37 import java.util.Properties;
  38 import java.util.concurrent.TimeUnit;
  39
  40 /**
  41  * An example of how to use the Java publisher.
  42  *
  43  * @author author
  44  */
  45 public class SimpleExamplePublisher {
  46     static FileWriter routeWriter = null;
  47     static Properties props = null;
  48     static FileReader routeReader = null;
  49
  50     public void publishMessage(String producerFilePath) throws IOException, InterruptedException, Exception {
  51
  52         // create our publisher
  53         final MRBatchingPublisher pub = MRClientFactory.createBatchingPublisher(producerFilePath);
  54         // publish some messages
  55         final JSONObject msg1 = new JSONObject();
  56         msg1.put("Name", "Sprint");
  57
  58         pub.send("First cambria messge");
  59         pub.send("MyPartitionKey", msg1.toString());
  60
  61         final JSONObject msg2 = new JSONObject();
  62
  63
  64         // ...
  65
  66         // close the publisher to make sure everything's sent before exiting. The batching
  67         // publisher interface allows the app to get the set of unsent messages. It could
  68         // write them to disk, for example, to try to send them later.
  69         final List<Message> stuck = pub.close(20, TimeUnit.SECONDS);
  70         if (stuck.isEmpty()) {
  71             System.err.println(stuck.size() + " messages unsent");
  72         } else {
  73             System.out.println("Clean exit; all messages sent.");
  74         }
  75     }
  76
  77     public static void main(String[] args) throws InterruptedException, Exception {
  78
  79         String routeFilePath = "/src/main/resources/dme2/preferredRoute.txt";
  80
  81         SimpleExamplePublisher publisher = new SimpleExamplePublisher();
  82
  83
  84         File fo = new File(routeFilePath);
  85         if (!fo.exists()) {
  86             routeWriter = new FileWriter(new File(routeFilePath));
  87         }
  88         routeReader = new FileReader(new File(routeFilePath));
  89         props = new Properties();
  90         publisher.publishMessage("/src/main/resources/dme2/producer.properties");
  91     }
  92
  93 }
  94