summaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/dmaap/messagerouter/msgrtr/nsa/cambria/resources/CambriaEventSet.java
blob: 9eb16913aa2eb9bcfbff2ebe8b7dcb38e471b983 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*******************************************************************************
 *  ============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.messagerouter.msgrtr.nsa.cambria.resources;

import java.io.IOException;
import java.io.InputStream;
import java.util.zip.GZIPInputStream;

import javax.servlet.http.HttpServletResponse;

import org.onap.dmaap.messagerouter.msgrtr.nsa.cambria.CambriaApiException;
import org.onap.dmaap.messagerouter.msgrtr.nsa.cambria.backends.Publisher.message;
import org.onap.dmaap.messagerouter.msgrtr.nsa.cambria.resources.streamReaders.CambriaJsonStreamReader;
import org.onap.dmaap.messagerouter.msgrtr.nsa.cambria.resources.streamReaders.CambriaRawStreamReader;
import org.onap.dmaap.messagerouter.msgrtr.nsa.cambria.resources.streamReaders.CambriaStreamReader;
import org.onap.dmaap.messagerouter.msgrtr.nsa.cambria.resources.streamReaders.CambriaTextStreamReader;

import com.att.nsa.apiServer.streams.ChunkedInputStream;
import com.att.nsa.drumlin.service.standards.HttpStatusCodes;

/**
 * An inbound event set.
 * 
 * @author author
 */
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 author
	 *
	 */
	public interface reader {
		/**
		 * 
		 * @return
		 * @throws IOException
		 * @throws CambriaApiException
		 */
		message next() throws IOException, CambriaApiException;
	}
}