org.talend.dataprep.api.service.command.dataset.CloneDataSet.java Source code

Introduction

Here is the source code for org.talend.dataprep.api.service.command.dataset.CloneDataSet.java

Source

//  ============================================================================
//
//  Copyright (C) 2006-2016 Talend Inc. - www.talend.com
//
//  This source code is available under agreement available at
//  https://github.com/Talend/data-prep/blob/master/LICENSE
//
//  You should have received a copy of the agreement
//  along with this program; if not, write to Talend SA
//  9 rue Pages 92150 Suresnes, France
//
//  ============================================================================

package org.talend.dataprep.api.service.command.dataset;

import static org.springframework.http.MediaType.APPLICATION_JSON_VALUE;
import static org.springframework.http.MediaType.TEXT_PLAIN_VALUE;

import java.io.IOException;
import java.net.URISyntaxException;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.client.utils.URIBuilder;
import org.springframework.context.annotation.Scope;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;
import org.talend.daikon.exception.ExceptionContext;
import org.talend.dataprep.api.service.command.common.HttpResponse;
import org.talend.dataprep.command.GenericCommand;
import org.talend.dataprep.exception.TDPException;
import org.talend.dataprep.exception.error.APIErrorCodes;
import org.talend.dataprep.exception.error.CommonErrorCodes;

/**
 * Command to clone a dataset.
 */
@Component
@Scope("request")
public class CloneDataSet extends GenericCommand<HttpResponse> {

    /**
     * Constructor.
     *
     * @param dataSetId the requested dataset id.
     * @param folderPath the folder to clone the dataset
     * @param cloneName the cloned name
     */
    public CloneDataSet(String dataSetId, String folderPath, String cloneName) {
        super(GenericCommand.DATASET_GROUP);
        execute(() -> {
            try {
                URIBuilder uriBuilder = new URIBuilder(datasetServiceUrl + "/datasets/clone/" + dataSetId);
                if (StringUtils.isNotEmpty(folderPath)) {
                    uriBuilder.addParameter("folderPath", folderPath);
                }
                if (StringUtils.isNotEmpty(cloneName)) {
                    uriBuilder.addParameter("cloneName", cloneName);
                }
                return new HttpPut(uriBuilder.build());
            } catch (URISyntaxException e) {
                throw new TDPException(CommonErrorCodes.UNEXPECTED_EXCEPTION, e);
            }
        });

        onError(e -> new TDPException(APIErrorCodes.UNABLE_TO_COPY_DATASET_CONTENT, e,
                ExceptionContext.build().put("id", dataSetId)));

        on(HttpStatus.OK, HttpStatus.BAD_REQUEST).then((httpRequestBase, httpResponse) -> {
            try {
                // we transfer status code and content type
                return new HttpResponse(httpResponse.getStatusLine().getStatusCode(), //
                        IOUtils.toString(httpResponse.getEntity().getContent()), //
                        httpResponse.getStatusLine().getStatusCode() == HttpStatus.BAD_REQUEST.value() ? //
                APPLICATION_JSON_VALUE : TEXT_PLAIN_VALUE);
            } catch (IOException e) {
                throw new TDPException(CommonErrorCodes.UNEXPECTED_EXCEPTION, e);
            } finally {
                httpRequestBase.releaseConnection();
            }
        });
    }

}