implement document partial streaming

hmcts · May 15, 2024 · b3c16ae · b3c16ae
1 parent 252542c
commit b3c16ae
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/src/main/java/uk/gov/hmcts/reform/ccd/documentam/ApplicationParams.java b/src/main/java/uk/gov/hmcts/reform/ccd/documentam/ApplicationParams.java
@@ -28,8 +28,8 @@ public class ApplicationParams {
     @Value("${moving.case.types}")
     private List<String> movingCaseTypes;
 
-    @Value("${filtered.request.headers}")
-    private List<String> filteredRequestHeaders;
+    @Value("#{'${request.forwarded_headers.from_client}'.split(',')}")
+    private List<String> clientRequestHeadersToForward;
 
     public List<String> getMovingCaseTypes() {
         return Optional.ofNullable(movingCaseTypes).orElse(emptyList());

diff --git a/src/main/java/uk/gov/hmcts/reform/ccd/documentam/client/dmstore/DocumentStoreClient.java b/src/main/java/uk/gov/hmcts/reform/ccd/documentam/client/dmstore/DocumentStoreClient.java
@@ -139,7 +139,7 @@ private HttpGet prepareHttpRequest(UUID documentId, final Map<String, String> re
     }
 
     private void setRequestHeaders(final HttpGet httpGet, Map<String, String> requestHeaders) {
-        Set<String> filteredHeaders = new HashSet<>(applicationParams.getFilteredRequestHeaders());
+        Set<String> filteredHeaders = new HashSet<>(applicationParams.getClientRequestHeadersToForward());
 
         // map client request headers
         if (!filteredHeaders.isEmpty()) {
@@ -164,7 +164,7 @@ private void handleStreamResponse(HttpStatus statusCode,
                                       HttpServletResponse httpResponseOut,
                                       UUID documentId) {
         switch (statusCode) {
-            case OK -> {
+            case OK, PARTIAL_CONTENT -> {
                 httpResponseOut.setStatus(statusCode.value());
                 mapResponseHeaders(httpClientResponse, httpResponseOut);
                 try {

diff --git a/src/main/resources/application.yaml b/src/main/resources/application.yaml
@@ -128,6 +128,6 @@ retry:
   maxAttempts: ${RETRY_MAX_ATTEMPTS:3}
   maxDelay: ${RETRY_MAX_DELAY:1000}
 
-filtered:
-  request:
-    headers: ${FILTERED_REQUEST_HEADERS:RANGE}
+request:
+  forwarded_headers:
+    from_client: ${CLIENT_REQUEST_HEADERS_TO_FORWARD:RANGE}