main.java.com.cloudant.client.internal.views.ViewMultipleRequester Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cloudant-client Show documentation
Official Cloudant client for Java
There is a newer version: 2.20.1
/*
 * Copyright © 2015, 2020 IBM Corp. 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.
 */

package com.cloudant.client.internal.views;

import com.cloudant.client.api.views.ViewMultipleRequest;
import com.cloudant.client.api.views.ViewResponse;
import com.cloudant.client.org.lightcouch.CouchDbException;
import com.cloudant.http.Http;
import com.cloudant.http.HttpConnection;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URI;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class ViewMultipleRequester implements ViewMultipleRequest {

    private final List> requestParameters = new
            ArrayList>();

    public List> getViewResponses() throws IOException {
        //build the queries array of data to POST
        JsonArray queries = new JsonArray();
        ViewQueryParameters viewQueryParameters = null;
        for (ViewQueryParameters params : requestParameters) {
            if (viewQueryParameters == null) {
                viewQueryParameters = params;
            }
            queries.add(params.asJson());
        }
        JsonObject queryJson = new JsonObject();
        queryJson.add("queries", queries);
        String requestBody = queryJson.toString();
        //construct and execute the POST request
        URI multiRequestUri = viewQueryParameters.getViewURIBuilder().path("queries").build();
        JsonObject jsonResponse = null;
        try {
            jsonResponse = performMultiRequest(multiRequestUri, viewQueryParameters, requestBody);
        } catch (CouchDbException e) {
            if (e.getStatusCode() == HttpURLConnection.HTTP_INTERNAL_ERROR && "badmatch".equals(e.getError())) {
                // CouchDB versions prior to 2.2 don't support .../_view/{viewname}/queries
                // POST directly to view instead
                multiRequestUri = viewQueryParameters.getViewURIBuilder().build();
                jsonResponse = performMultiRequest(multiRequestUri, viewQueryParameters, requestBody);
            } else {
                throw e;
            }
        }
        //loop the returned array creating the ViewResponse objects
        List> responses = new ArrayList>();
        JsonArray jsonResponses = jsonResponse.getAsJsonArray("results");
        if (jsonResponses != null) {
            int index = 0;
            for (ViewQueryParameters params : requestParameters) {
                JsonObject response = jsonResponses.get(index).getAsJsonObject();
                responses.add(new ViewResponseImpl(params, response, null));
                index++;
            }
            return responses;
        } else {
            return Collections.emptyList();
        }
    }

    public void add(ViewQueryParameters viewQueryParameters) {
        requestParameters.add(viewQueryParameters);
    }

    private JsonObject performMultiRequest(URI multiRequestUri,
                                           ViewQueryParameters viewQueryParameters,
                                           String body) throws CouchDbException, IOException {
        //construct and execute the POST request
        HttpConnection post = Http.POST(multiRequestUri, "application/json");
        post.setRequestBody(body);
        return ViewRequester.executeRequestWithResponseAsJson(viewQueryParameters, post);
    }
}