DMAAP-MR - Merge MR repos

[dmaap/messagerouter/messageservice.git] / src / main / java / org / onap / dmaap / dmf / mr / resources / CambriaEventSet.java

diff --git a/src/main/java/org/onap/dmaap/dmf/mr/resources/CambriaEventSet.java b/src/main/java/org/onap/dmaap/dmf/mr/resources/CambriaEventSet.java
new file mode 100644 (file)
index 0000000..cd55d2f
--- /dev/null
+++ b/src/main/java/org/onap/dmaap/dmf/mr/resources/CambriaEventSet.java
@@ -0,0 +1,113 @@
+/*******************************************************************************
+ *  ============LICENSE_START=======================================================
+ *  org.onap.dmaap
+ *  ================================================================================
+ *  Copyright © 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=========================================================
+ *  
+ *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ *  
+ *******************************************************************************/
+package org.onap.dmaap.dmf.mr.resources;
+
+import com.att.nsa.apiServer.streams.ChunkedInputStream;
+import com.att.nsa.drumlin.service.standards.HttpStatusCodes;
+import org.onap.dmaap.dmf.mr.CambriaApiException;
+import org.onap.dmaap.dmf.mr.backends.Publisher.message;
+import org.onap.dmaap.dmf.mr.resources.streamReaders.CambriaJsonStreamReader;
+import org.onap.dmaap.dmf.mr.resources.streamReaders.CambriaRawStreamReader;
+import org.onap.dmaap.dmf.mr.resources.streamReaders.CambriaStreamReader;
+import org.onap.dmaap.dmf.mr.resources.streamReaders.CambriaTextStreamReader;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.zip.GZIPInputStream;
+
+/**
+ * An inbound event set.
+ * 
+ * @author peter
+ */
+public class CambriaEventSet {
+       private final reader fReader;
+
+       /**
+        * constructor initialization
+        * 
+        * @param mediaType
+        * @param originalStream
+        * @param chunked
+        * @param defPartition
+        * @throws CambriaApiException
+        */
+       public CambriaEventSet(String mediaType, InputStream originalStream,
+                       boolean chunked, String defPartition) throws CambriaApiException {
+               InputStream is = originalStream;
+               if (chunked) {
+                       is = new ChunkedInputStream(originalStream);
+               }
+
+               if (("application/json").equals(mediaType)) {
+                       if (chunked) {
+                               throw new CambriaApiException(
+                                               HttpServletResponse.SC_BAD_REQUEST,
+                                               "The JSON stream reader doesn't support chunking.");
+                       }
+                       fReader = new CambriaJsonStreamReader(is, defPartition);
+               } else if (("application/cambria").equals(mediaType)) {
+                       fReader = new CambriaStreamReader(is);
+               } else if (("application/cambria-zip").equals(mediaType)) {
+                       try {
+                               is = new GZIPInputStream(is);
+                       } catch (IOException e) {
+                               throw new CambriaApiException(HttpStatusCodes.k400_badRequest,
+                                               "Couldn't read compressed format: " + e);
+                       }
+                       fReader = new CambriaStreamReader(is);
+               } else if (("text/plain").equals(mediaType)) {
+                       fReader = new CambriaTextStreamReader(is, defPartition);
+               } else {
+                       fReader = new CambriaRawStreamReader(is, defPartition);
+               }
+       }
+
+       /**
+        * Get the next message from this event set. Returns null when the end of
+        * stream is reached. Will block until a message arrives (or the stream is
+        * closed/broken).
+        * 
+        * @return a message, or null
+        * @throws IOException
+        * @throws CambriaApiException
+        */
+       public message next() throws IOException, CambriaApiException {
+               return fReader.next();
+       }
+
+       /**
+        * 
+        * @author anowarul.islam
+        *
+        */
+       public interface reader {
+               /**
+                * 
+                * @return
+                * @throws IOException
+                * @throws CambriaApiException
+                */
+               message next() throws IOException, CambriaApiException;
+       }
+}