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

Introduction

Here is the source code for org.talend.dataprep.api.service.command.dataset.CreateDataSet.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.talend.dataprep.command.Defaults.*;

import java.io.InputStream;
import java.net.URISyntaxException;

import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.entity.InputStreamEntity;
import org.springframework.context.annotation.Scope;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;
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 used to create a dataset. Basically pass through all data to the DataSet low level API.
 */
@Component
@Scope("prototype")
public class CreateDataSet extends GenericCommand<String> {

    /**
     * Default constructor.
     *
     * @param name name of the dataset.
     * @param contentType content-type of the dataset.
     * @param dataSetContent Dataset content or import parameters in json for remote datasets.
     */
    private CreateDataSet(String name, String tag, String contentType, InputStream dataSetContent) {
        super(GenericCommand.DATASET_GROUP);
        execute(() -> onExecute(name, tag, contentType, dataSetContent));
        onError(e -> {
            if (e instanceof TDPException) {
                return passthrough().apply(e);
            }
            return new TDPException(APIErrorCodes.UNABLE_TO_CREATE_DATASET, e);
        });
        on(HttpStatus.NO_CONTENT, HttpStatus.ACCEPTED).then(emptyString());
        on(HttpStatus.OK).then(asString());
    }

    private HttpRequestBase onExecute(String name, String tag, String contentType, InputStream dataSetContent) {
        try {
            URIBuilder uriBuilder = new URIBuilder(datasetServiceUrl + "/datasets");
            uriBuilder.addParameter("name", name);
            uriBuilder.addParameter("tag", tag);
            final HttpPost post = new HttpPost(uriBuilder.build());
            post.addHeader("Content-Type", contentType); //$NON-NLS-1$
            post.setEntity(new InputStreamEntity(dataSetContent));
            return post;
        } catch (URISyntaxException e) {
            throw new TDPException(CommonErrorCodes.UNEXPECTED_EXCEPTION, e);
        }
    }
}