com.sap.cloud.mt.runtime.HanaConnectionProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of multi-tenant-runtime Show documentation

Spring Boot Enablement Parent

There is a newer version: 3.3.3

/******************************************************************************
 * © 2020 SAP SE or an SAP affiliate company. All rights reserved.            *
 ******************************************************************************/

package com.sap.cloud.mt.runtime;

import com.sap.cloud.mt.subscription.DataSourceInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;

public class HanaConnectionProvider implements ConnectionProvider {
    private static Logger logger = LoggerFactory.getLogger(HanaConnectionProvider.class);

    public HanaConnectionProvider() {
    }

    @Override
    public Connection getConnection(String tenantId, DataSourceAndInfo dataSourceAndInfo) throws SQLException {
        DataSourceInfo info = dataSourceAndInfo.getDataSourceInfo();
        Connection connection = dataSourceAndInfo.getDataSource().getConnection();//NOSONAR
        try (Statement statement = connection.createStatement()) {
            logger.debug("Reconnect connection for user {}", info.getUser());
            statement.execute("connect " + info.getUser() +
                    " password \"" + info.getPassword() + "\" ");
            logger.debug("Set schema to {}", info.getSchema());
            connection.setSchema(info.getSchema());
            if (!connection.getSchema().equals(info.getSchema())) {
                throw new SQLException("Could not set schema " + info.getSchema());
            }
        }
        return connection;
    }
}