package com.poteviohealth.cgp.integrate.database;

import com.alibaba.druid.pool.DruidDataSource;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

/**
 * 获取数据库连接
 * @author QIN
 * @date 2019/7/22
 */
@Data
@ConfigurationProperties(prefix = "spring.datasource.druid.database")
@Component
public class DatabaseConfig {
    private String jdbcUrl;
    private String username;
    private String password;
    private String driverClassName;
    public static final String DATABASE_NAME="datasource";

    /**
     * 获取数据源
     * @Author QIN
     * @Date 2019/7/22 15:02
     * @Param
     * @return DataSource
    **/
    public DataSource createDataSource() {
        DruidDataSource result = new DruidDataSource();
        result.setDriverClassName(getDriverClassName());
        result.setUrl(getJdbcUrl());
        result.setUsername(getUsername());
        result.setPassword(getPassword());
        return result;
    }

    /**
     * 获取连接
     * @Author QIN
     * @Date 2019/7/22 15:02
     * @Param
     * @return Connection
     **/
    public Connection getConnection() throws SQLException {
        return this.createDataSource().getConnection();
    }
}