com.github.hermannpencole.nifi.config.service.AccessService Maven / Gradle / Ivy

Go to download
package com.github.hermannpencole.nifi.config.service;

import com.github.hermannpencole.nifi.swagger.ApiClient;
import com.github.hermannpencole.nifi.swagger.ApiException;
import com.github.hermannpencole.nifi.swagger.Configuration;
import com.github.hermannpencole.nifi.swagger.client.AccessApi;

import javax.inject.Inject;
import javax.inject.Singleton;

/**
 * Created by SFRJ2737 on 2017-05-28.
 *
 * @author hermann pencolé
 */
@Singleton
public class AccessService {

    @Inject
    private AccessApi apiInstance;

    /**
     * add token on http client. The token is ask to nifi.
     *
     * @param accessFromTicket accessFromTicket
     * @param username username
     * @param password password
     * @throws ApiException when communication probem
     */
    public void addTokenOnConfiguration(boolean accessFromTicket, String username, String password) throws ApiException {
        ApiClient client = Configuration.getDefaultApiClient();
        if (accessFromTicket) {
            String token = apiInstance.createAccessTokenFromTicket();
            client.setAccessToken(token);
        } else if (username != null) {
            String token = apiInstance.createAccessToken(username, password);
            client.setAccessToken(token);
        }
        Configuration.setDefaultApiClient(client);
    }

    /**
     * Configure the default http client
     *
     * @param basePath set the basePath
     * @param verifySsl set theverifySsl
     * @param debugging set the debugging
     * @param connectionTimeout set the  connectionTimeout
     * @param readTimeout set the readTimeout
     * @param writeTimeout set the writeTimeout
     * @throws ApiException when problem
     */
    public void setConfiguration(String basePath, boolean verifySsl, boolean debugging,
                                        int connectionTimeout, int readTimeout, int writeTimeout) throws ApiException {
        ApiClient client = Configuration.getDefaultApiClient()
        //ApiClient client = new ApiClient()
                .setBasePath(basePath)
                .setVerifyingSsl(verifySsl)
                .setConnectTimeout(connectionTimeout)
                .setReadTimeout(readTimeout)
                .setWriteTimeout(writeTimeout)
                .setDebugging(debugging);
        Configuration.setDefaultApiClient(client);
    }
}