Initial commit for AAI-UI(sparky-backend)

[aai/sparky-be.git] / src / main / java / org / openecomp / sparky / dal / servlet / ResettableStreamHttpServletRequest.java

diff --git a/src/main/java/org/openecomp/sparky/dal/servlet/ResettableStreamHttpServletRequest.java b/src/main/java/org/openecomp/sparky/dal/servlet/ResettableStreamHttpServletRequest.java
new file mode 100644 (file)
index 0000000..d7b1e6b
--- /dev/null
+++ b/src/main/java/org/openecomp/sparky/dal/servlet/ResettableStreamHttpServletRequest.java
@@ -0,0 +1,129 @@
+/**
+ * ============LICENSE_START===================================================
+ * SPARKY (AAI UI service)
+ * ============================================================================
+ * Copyright © 2017 AT&T Intellectual Property.
+ * Copyright © 2017 Amdocs
+ * 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 and OpenECOMP are trademarks
+ * and service marks of AT&T Intellectual Property.
+ */
+
+package org.openecomp.sparky.dal.servlet;
+
+import com.google.common.primitives.Bytes;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Arrays;
+
+import javax.servlet.ReadListener;
+import javax.servlet.ServletInputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletRequestWrapper;
+
+/**
+ * The Class ResettableStreamHttpServletRequest.
+ */
+public class ResettableStreamHttpServletRequest extends HttpServletRequestWrapper {
+
+  private byte[] requestBody = new byte[0];
+  private boolean bufferFilled = false;
+
+  /**
+   * Constructs a request object wrapping the given request.
+   *
+   * @param request The request to wrap
+   * @throws IllegalArgumentException if the request is null
+   */
+  public ResettableStreamHttpServletRequest(HttpServletRequest request) {
+    super(request);
+  }
+
+  /**
+   * Get request body.
+   * 
+   * @return Bytes with the request body contents.
+   * @throws IOException In case stream reqding fails.
+   */
+  public byte[] getRequestBody() throws IOException {
+    if (bufferFilled) {
+      return Arrays.copyOf(requestBody, requestBody.length);
+    }
+
+    InputStream inputStream = super.getInputStream();
+
+    byte[] buffer = new byte[102400];
+
+    int bytesRead;
+    while ((bytesRead = inputStream.read(buffer)) != -1) {
+      requestBody = Bytes.concat(this.requestBody, Arrays.copyOfRange(buffer, 0, bytesRead));
+    }
+
+    bufferFilled = true;
+
+    return requestBody;
+  }
+
+  @Override
+  public ServletInputStream getInputStream() throws IOException {
+    return new CustomServletInputStream(getRequestBody());
+  }
+
+  /**
+   * The Class CustomServletInputStream.
+   */
+  private static class CustomServletInputStream extends ServletInputStream {
+
+    private ByteArrayInputStream buffer;
+
+    /**
+     * Instantiates a new custom servlet input stream.
+     *
+     * @param contents the contents
+     */
+    public CustomServletInputStream(byte[] contents) {
+      this.buffer = new ByteArrayInputStream(contents);
+    }
+
+    /* (non-Javadoc)
+     * @see java.io.InputStream#read()
+     */
+    @Override
+    public int read() throws IOException {
+      return buffer.read();
+    }
+
+    @Override
+    public boolean isFinished() {
+      return buffer.available() == 0;
+    }
+
+    @Override
+    public boolean isReady() {
+      return true;
+    }
+
+    @Override
+    public void setReadListener(ReadListener arg0) {
+      throw new RuntimeException("Not implemented");
+    }
+
+  }
+
+}