/*-
 * ============LICENSE_START=======================================================
 * onap
 * ================================================================================
 * Copyright (C) 2016 - 2017 ONAP
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */

package org.onap.ccsdk.sli.core.dblib.jdbc;

import java.sql.SQLFeatureNotSupportedException;
import org.apache.tomcat.jdbc.pool.DataSource;
import org.apache.tomcat.jdbc.pool.PoolProperties;
import org.onap.ccsdk.sli.core.dblib.CachedDataSource;
import org.onap.ccsdk.sli.core.dblib.DBConfigException;
import org.onap.ccsdk.sli.core.dblib.config.BaseDBConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


public class JdbcDBCachedDataSource extends CachedDataSource {

    private String dbDriver;
    private String dbUserId;
    private String dbPasswd;
    private String dbUrl;

    private int minLimit;
    private int maxLimit;
    private int initialLimit;

    private static final String AS_CONF_ERROR = "AS_CONF_ERROR: ";

    private static final Logger LOGGER = LoggerFactory.getLogger(JdbcDBCachedDataSource.class);

    /**
     * @param jdbcElem
     */
    public JdbcDBCachedDataSource(BaseDBConfiguration jdbcElem) {
        super(jdbcElem);
    }

    @Override
    protected DataSource configure(BaseDBConfiguration xmlElem) throws DBConfigException {
        BaseDBConfiguration jdbcConfig = xmlElem;
        if (jdbcConfig.getConnTimeout() > 0) {
            this.connReqTimeout = jdbcConfig.getConnTimeout();
        }
        if (jdbcConfig.getRequestTimeout() > 0) {
            this.dataReqTimeout = jdbcConfig.getRequestTimeout();
        }

        // set connection pool name
        String dbConnectionName = jdbcConfig.getDbConnectionName();
        super.setDbConnectionName(dbConnectionName);
        // Configure the JDBC connection
        dbUserId = jdbcConfig.getDbUserId();
        if (dbUserId == null) {
            String errorMsg = "Invalid XML contents: JDBCConnection missing dbUserId attribute";
            LOGGER.error(AS_CONF_ERROR + errorMsg);
            throw new DBConfigException(errorMsg);
        }

        dbPasswd = jdbcConfig.getDbPasswd();
        if (dbPasswd == null) {
            String errorMsg = "Invalid XML contents: JDBCConnection missing dbPasswd attribute";
            LOGGER.error(AS_CONF_ERROR + errorMsg);
            throw new DBConfigException(errorMsg);
        }

        dbDriver = jdbcConfig.getDriverName();
        if (dbDriver == null) {
            String errorMsg = "Invalid XML contents: JDBCConnection missing dbDriver attribute";
            LOGGER.error(AS_CONF_ERROR + errorMsg);
            throw new DBConfigException(errorMsg);
        }

        minLimit = jdbcConfig.getDbMinLimit();
//        if (minLimit == null)
//        {
//            String errorMsg =  "Invalid XML contents: JDBC Connection missing minLimit attribute";
//            LOGGER.error(AS_CONF_ERROR + errorMsg);
//            throw new DBConfigException(errorMsg);
//        }
        maxLimit = jdbcConfig.getDbMaxLimit();
//        if (maxLimit == null)
//        {
//            String errorMsg =  "Invalid XML contents: JDBC Connection missing maxLimit attribute";
//            LOGGER.error(AS_CONF_ERROR + errorMsg);
//            throw new DBConfigException(errorMsg);
//        }
        initialLimit = jdbcConfig.getDbInitialLimit();
//        if (initialLimit == null)
//        {
//            String errorMsg =  "Invalid XML contents: JDBC Connection missing initialLimit attribute";
//            LOGGER.error(AS_CONF_ERROR + errorMsg);
//            throw new DBConfigException(errorMsg);
//        }

        dbUrl = jdbcConfig.getDbUrl();
        if (dbUrl == null) {
            String errorMsg = "Invalid XML contents: JDBCConnection missing dbUrl attribute";
            LOGGER.error(AS_CONF_ERROR + errorMsg);
            throw new DBConfigException(errorMsg);
        }

        try {

            PoolProperties p = new PoolProperties();
            p.setDriverClassName(dbDriver);
            p.setUrl(dbUrl);
            p.setUsername(dbUserId);
            p.setPassword(dbPasswd);
            p.setJmxEnabled(true);
            p.setTestWhileIdle(false);
            p.setTestOnBorrow(true);
            p.setValidationQuery("SELECT 1");
            p.setTestOnReturn(false);
            p.setValidationInterval(30000);
            p.setTimeBetweenEvictionRunsMillis(30000);
            p.setInitialSize(initialLimit);
            p.setMaxActive(maxLimit);
            p.setMaxIdle(maxLimit);
            p.setMaxWait(10000);
            p.setRemoveAbandonedTimeout(60);
            p.setMinEvictableIdleTimeMillis(30000);
            p.setMinIdle(minLimit);
            p.setLogAbandoned(true);
            p.setRemoveAbandoned(true);
            p.setJdbcInterceptors("org.apache.tomcat.jdbc.pool.interceptor.ConnectionState;"
                + "org.apache.tomcat.jdbc.pool.interceptor.StatementFinalizer");

            final DataSource dataSource = new DataSource(p);

            synchronized (this) {
                initialized = true;
            }
                LOGGER.info(String.format("JdbcDBCachedDataSource <%s> configured successfully. Using URL: %s",
                    dbConnectionName, dbUrl));
            return dataSource;
        } catch (Exception exc) {
            initialized = false;
            LOGGER.error(String.format("AS_CONF_ERROR: Failed to initialize MySQLCachedDataSource <%s>. Reason: %s",
                dbConnectionName, exc.getMessage()));
            return null;
        }
    }

    public final String getDbUrl() {
        return dbUrl;
    }

    public final String getDbUserId() {
        return dbUserId;
    }

    public final String getDbPasswd() {
        return dbPasswd;
    }

    public static JdbcDBCachedDataSource createInstance(BaseDBConfiguration config) /*throws Exception*/ {
        return new JdbcDBCachedDataSource(config);
    }

    public String toString() {
        return getDbConnectionName();
    }

    public java.util.logging.Logger getParentLogger()
        throws SQLFeatureNotSupportedException {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void cleanUp() {
        DataSource dataSource = (DataSource) ds;
        dataSource.getPool().purge();
        dataSource.close(true);
        super.cleanUp();
    }
}