package com.walker.connector.util;

import com.walker.connector.Address;
import com.walker.connector.db.DamengConnector;
import com.walker.connector.db.MySqlConnector;
import com.walker.connector.db.OracleConnector;
import com.walker.connector.db.PostgresConnector;
import com.walker.connector.support.DatabaseConnector;
import com.walker.db.DatabaseType;
import com.walker.infrastructure.utils.StringUtils;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.interceptor.TransactionProxyFactoryBean;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

/**
 * 连接器工具类
 * @author shikeying
 * @date 2015年12月17日
 *
 */
public abstract class ConnectorUtils {

	/**
	 * 创建带事务管理的，数据库连接器对象。</p>
	 * 事务规则如下：
	 * <pre>
	 * "query*", "PROPAGATION_SUPPORTS,readOnly"
	 * "exec*", "PROPAGATION_REQUIRED, -Exception"
	 * </pre>
	 * @param connector
	 * @return
	 */
	public static final DatabaseConnector acquireTransactionProxyConnector(DatabaseConnector connector){
		PlatformTransactionManager transactionManager = new DataSourceTransactionManager(connector.getDataSource());

		// 定义事务规则
		Properties transactionAttributes = new Properties();
		transactionAttributes.put("query*", "PROPAGATION_SUPPORTS,readOnly");
		transactionAttributes.put("exec*", "PROPAGATION_REQUIRED, -Exception");

		TransactionProxyFactoryBean transactionProxy = new TransactionProxyFactoryBean();
		transactionProxy.setTransactionManager(transactionManager);
		transactionProxy.setTransactionAttributes(transactionAttributes);
		transactionProxy.setProxyTargetClass(true);
		transactionProxy.setTarget(connector);
		transactionProxy.afterPropertiesSet();
		return (DatabaseConnector)transactionProxy.getObject();
	}

	public static final DatabaseConnector createMySQLConnector(DataSource ds){
		return doCreateMySqLConnector(null, false, ds);
	}

	/**
	 * 创建MySQL的连接器
	 * @param address
	 * @return
	 */
	public static final DatabaseConnector createMySQLConnector(Address address){
		return doCreateMySqLConnector(address, false, null);
	}

	/**
	 * 创建MySQL的管理连接器，没有库名
	 * @param address
	 * @return
	 */
	public static final DatabaseConnector createMySQLManageConnector(Address address){
		return doCreateMySqLConnector(address, true, null);
	}

	private static DatabaseConnector doCreateMySqLConnector(Address address
			, boolean manageMode, DataSource ds){
		MySqlConnector dbConnector;

		if(ds != null){
			// 如果存在数据源，直接注入创建Connector对象
			dbConnector = new MySqlConnector(ds);
			return dbConnector;
		}

//		// 2023-05-04，使用系统配置统一数据源，否则会出现：业务用druid，而元数据用Hikari情况
//		if(JdbcInspector.getInstance() != null){
//			dbConnector = new MySqlConnector((DataSource) JdbcInspector.getInstance().getPrimaryDataSourceMeta());
//			return dbConnector;
//		}

		dbConnector = new MySqlConnector();
		dbConnector.setUrl(address.getUrl());
		dbConnector.setPort(address.getPort());
		//设置数据库名字，后面拼接字符集
//		dbConnector.setServiceName("walkersoft-share-marks?characterEncoding=UTF-8");
		if(manageMode){
			dbConnector.setManageMode();
		} else {
			if(StringUtils.isEmpty(address.getServiceName())){
				throw new IllegalArgumentException("未设置连接数据库名称：address.service = null.");
			}
			dbConnector.setServiceName(address.getServiceName()+"?characterEncoding=UTF-8");
		}

		Map<String, String> parameters = new HashMap<String, String>(5);
		parameters.put(DatabaseConnector.OPTION_USER, address.getAuthentication());
		parameters.put(DatabaseConnector.OPTION_PASSWORD, address.getCertification());
		if(address.getMaxActive() > 0){
			parameters.put(DatabaseConnector.OPTION_MAX_ACTIVE, String.valueOf(address.getMaxActive()));
		} else {
			parameters.put(DatabaseConnector.OPTION_MAX_ACTIVE, "6");
		}
		if(address.getMaxIdle() > 0){
			parameters.put(DatabaseConnector.OPTION_MAX_IDLE, String.valueOf(address.getMaxIdle()));
		} else {
			parameters.put(DatabaseConnector.OPTION_MAX_IDLE, "6");
		}
		if(address.getInitSize() > 0){
			parameters.put(DatabaseConnector.OPTION_INIT_SIZE, String.valueOf(address.getInitSize()));
		} else {
			parameters.put(DatabaseConnector.OPTION_INIT_SIZE, "3");
		}

		dbConnector.setParameters(parameters);
		dbConnector.initialize();
		return dbConnector;
	}

	public static final DatabaseConnector createOracleConnector(DataSource ds){
		return doCreateOracleConnector(null, false, ds);
	}
	public static final DatabaseConnector createOracleConnector(Address address){
		return doCreateOracleConnector(address, false, null);
	}
	public static final DatabaseConnector createOracleManageConnector(Address address){
		return doCreateOracleConnector(address, true, null);
	}

	/**
	 * 创建达梦数据库连接器对象，不是管理端，只能连特定数据库。
	 * @param address
	 * @return
	 * @date 2023-03-03
	 */
	public static final DatabaseConnector createDamengConnector(Address address){
		return doCreateDamengConnector(address, false, null);
	}
	public static final DatabaseConnector createDamengManageConnector(Address address){
		return doCreateDamengConnector(address, true, null);
	}

	private static DatabaseConnector doCreateDamengConnector(Address address, boolean manageMode, DataSource ds){
		DamengConnector dbConnector;
		if(ds != null){
			// 如果存在数据源，直接注入创建Connector对象
			dbConnector = new DamengConnector(ds);
			return dbConnector;
		}

//		// 2023-05-04，使用系统配置统一数据源，否则会出现：业务用druid，而元数据用Hikari情况
//		if(JdbcInspector.getInstance() != null){
//			dbConnector = new DamengConnector((DataSource) JdbcInspector.getInstance().getPrimaryDataSourceMeta());
//			return dbConnector;
//		}

		dbConnector = new DamengConnector();
		dbConnector.setUrl(address.getUrl());
		dbConnector.setPort(address.getPort());
		if(manageMode){
			dbConnector.setManageMode();
		} else {
			if(StringUtils.isEmpty(address.getServiceName())){
				throw new IllegalArgumentException("未设置连接数据库名称：address.service = null.");
			}
			dbConnector.setServiceName(address.getServiceName());
		}
		Map<String, String> parameters = new HashMap<String, String>(5);
		parameters.put(DatabaseConnector.OPTION_USER, address.getAuthentication());
		parameters.put(DatabaseConnector.OPTION_PASSWORD, address.getCertification());
		parameters.put(DatabaseConnector.OPTION_MAX_ACTIVE, "6");
		parameters.put(DatabaseConnector.OPTION_MAX_IDLE, "3");
		parameters.put(DatabaseConnector.OPTION_INIT_SIZE, "3");

		dbConnector.setParameters(parameters);
		dbConnector.initialize();
		return dbConnector;
	}

	private static DatabaseConnector doCreateOracleConnector(Address address
			, boolean manageMode, DataSource ds){
		OracleConnector dbConnector;
		if(ds != null){
			// 如果存在数据源，直接注入创建Connector对象
			dbConnector = new OracleConnector(ds);
			return dbConnector;
		}

//		// 2023-05-04，使用系统配置统一数据源，否则会出现：业务用druid，而元数据用Hikari情况
//		if(JdbcInspector.getInstance() != null){
//			dbConnector = new OracleConnector((DataSource) JdbcInspector.getInstance().getPrimaryDataSourceMeta());
//			return dbConnector;
//		}

		dbConnector = new OracleConnector();
		dbConnector.setUrl(address.getUrl());
		dbConnector.setPort(address.getPort());
		if(manageMode){
			dbConnector.setManageMode();
		} else {
			if(StringUtils.isEmpty(address.getServiceName())){
				throw new IllegalArgumentException("未设置连接数据库名称：address.service = null.");
			}
			dbConnector.setServiceName(address.getServiceName());
		}

		Map<String, String> parameters = new HashMap<String, String>(5);
		parameters.put(DatabaseConnector.OPTION_USER, address.getAuthentication());
		parameters.put(DatabaseConnector.OPTION_PASSWORD, address.getCertification());
		parameters.put(DatabaseConnector.OPTION_MAX_ACTIVE, "6");
		parameters.put(DatabaseConnector.OPTION_MAX_IDLE, "3");
		parameters.put(DatabaseConnector.OPTION_INIT_SIZE, "3");

		dbConnector.setParameters(parameters);
		dbConnector.initialize();
		return dbConnector;
	}

	public static final DatabaseConnector createPostgresConnector(Address address){
		return doCreatePostgresConnector(address, false);
	}

	public static final DatabaseConnector createPostgresManageConnector(Address address){
		return doCreatePostgresConnector(address, true);
	}

	private static DatabaseConnector doCreatePostgresConnector(Address address, boolean manageMode){
		PostgresConnector dbConnector = new PostgresConnector();

//		// 2023-05-04，使用系统配置统一数据源，否则会出现：业务用druid，而元数据用Hikari情况
//		if(JdbcInspector.getInstance() != null){
//			dbConnector = new PostgresConnector((DataSource) JdbcInspector.getInstance().getPrimaryDataSourceMeta());
//			return dbConnector;
//		}

		dbConnector.setUrl(address.getUrl());
		dbConnector.setPort(address.getPort());
		if(manageMode){
			dbConnector.setManageMode();
		} else {
			if(StringUtils.isEmpty(address.getServiceName())){
				throw new IllegalArgumentException("未设置连接数据库名称：address.service = null.");
			}
			dbConnector.setServiceName(address.getServiceName());
		}

		Map<String, String> parameters = new HashMap<String, String>(5);
		parameters.put(DatabaseConnector.OPTION_USER, address.getAuthentication());
		parameters.put(DatabaseConnector.OPTION_PASSWORD, address.getCertification());
		parameters.put(DatabaseConnector.OPTION_MAX_ACTIVE, "20");
//		parameters.put(DatabaseConnector.OPTION_MAX_IDLE, "3");
		parameters.put(DatabaseConnector.OPTION_INIT_SIZE, "3");

		dbConnector.setParameters(parameters);
		dbConnector.initialize();
		return dbConnector;
	}

	/**
	 * 返回一个数据库连接器对象
	 * @param addr 数据库连接地址
	 * @param manage 是否管理连接，管理连接没有数据库名
	 * @return
	 */
	public static final DatabaseConnector getDbConnectorByType(Address addr
			, boolean manage, DatabaseType databaseType){
		if(databaseType == DatabaseType.MYSQL){
			if(manage){
				return ConnectorUtils.createMySQLManageConnector(addr);
			} else {
				return ConnectorUtils.createMySQLConnector(addr);
			}
		} else if(databaseType == DatabaseType.POSTGRES){
			if(manage){
				return ConnectorUtils.createPostgresManageConnector(addr);
			} else {
				return ConnectorUtils.createPostgresConnector(addr);
			}
		} else if(databaseType == DatabaseType.ORACLE){
			if(manage){
				return ConnectorUtils.createOracleManageConnector(addr);
			} else {
				return ConnectorUtils.createOracleConnector(addr);
			}
		} else if (databaseType == DatabaseType.DAMENG) {
			if(manage){
				return ConnectorUtils.createDamengManageConnector(addr);
			} else {
				return ConnectorUtils.createDamengConnector(addr);
			}
		} else {
			throw new UnsupportedOperationException("未实现对其他数据库的支持：DatabaseConnector");
		}
	}

}