io.edurt.datacap.plugin.jdbc.dremio.DremioConnection Maven / Gradle / Ivy

Go to download
package io.edurt.datacap.plugin.jdbc.dremio;

import io.edurt.datacap.spi.connection.JdbcConfigure;
import io.edurt.datacap.spi.connection.JdbcConnection;
import io.edurt.datacap.spi.model.Response;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class DremioConnection
        extends JdbcConnection
{
    public DremioConnection(JdbcConfigure jdbcConfigure, Response response)
    {
        super(jdbcConfigure, response);
    }

    @Override
    protected String formatJdbcUrl()
    {
        JdbcConfigure jdbcConfigure = (JdbcConfigure) this.getConfigure();
        StringBuffer buffer = new StringBuffer();
        buffer.append("jdbc:");
        buffer.append(jdbcConfigure.getJdbcType());
        buffer.append("=");
        buffer.append(jdbcConfigure.getHost());
        buffer.append(":");
        buffer.append(jdbcConfigure.getPort());
        if (jdbcConfigure.getDatabase().isPresent()) {
            buffer.append(";");
            buffer.append(jdbcConfigure.getDatabase().get());
        }
        if (jdbcConfigure.getSsl().isPresent()) {
            if (jdbcConfigure.getSsl().get()) {
                buffer.append(String.format("?ssl=%s", jdbcConfigure.getSsl().get()));
            }
        }
        if (jdbcConfigure.getEnv().isPresent()) {
            Map env = jdbcConfigure.getEnv().get();
            List flatEnv = env.entrySet()
                    .stream()
                    .map(value -> String.format("%s=%s", value.getKey(), value.getValue()))
                    .collect(Collectors.toList());
            if (!jdbcConfigure.getSsl().isPresent()) {
                buffer.append("?");
            }
            else {
                buffer.append("&");
            }
            buffer.append(String.join("&", flatEnv));
        }
        return buffer.toString();
    }
}