org.plos.crepo.dao.ContentRepoBaseDao.java Source code

Introduction

Here is the source code for org.plos.crepo.dao.ContentRepoBaseDao.java

Source

/*
 * Copyright 2017 Public Library of Science
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

package org.plos.crepo.dao;

import com.google.common.base.Preconditions;
import org.apache.commons.io.IOUtils;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpRequestBase;
import org.plos.crepo.config.ContentRepoAccessConfig;
import org.plos.crepo.exceptions.ContentRepoException;
import org.plos.crepo.exceptions.ErrorType;
import org.plos.crepo.exceptions.NotFoundException;
import org.plos.crepo.util.HttpResponseUtil;
import org.slf4j.Logger;

import java.io.IOException;

public abstract class ContentRepoBaseDao {

    private final ContentRepoAccessConfig accessConfig;

    protected ContentRepoBaseDao(ContentRepoAccessConfig accessConfig) {
        this.accessConfig = Preconditions.checkNotNull(accessConfig);
    }

    protected String getRepoServer() {
        return accessConfig.getRepoServer();
    }

    protected CloseableHttpResponse executeRequest(HttpRequestBase request, ErrorType errorType) {
        CloseableHttpResponse response;
        try {
            response = accessConfig.open(request);
        } catch (IOException e) {
            getLog().error("Error handling the response, uri: " + request.getURI().toString() + " repoMessage: ",
                    e);
            throw new ContentRepoException.ContentRepoExceptionBuilder(errorType).baseException(e)
                    .url(request.getURI().toString()).build();
        }

        try {
            final int statusCode = response.getStatusLine().getStatusCode();

            if (statusCode == HttpStatus.SC_NOT_FOUND) {
                throw new NotFoundException(HttpResponseUtil.getErrorMessage(response));
            } else if (statusCode != HttpStatus.SC_OK && statusCode != HttpStatus.SC_CREATED) {
                final String cause = HttpResponseUtil.getErrorMessage(response);

                getLog().error("uri: " + request.getURI().toString() + " repoMessage: " + cause);
                throw new ContentRepoException.ContentRepoExceptionBuilder(errorType)
                        .url(request.getURI().toString()).repoMessage(cause).build();
            }

            return response;

        } catch (RuntimeException e) {
            IOUtils.closeQuietly(response);
            request.releaseConnection();
            throw e;
        }
    }

    public abstract Logger getLog();

}