package com.artfess.base.manager.impl;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import javax.annotation.Resource;

import com.artfess.base.dao.CommonDao;
import com.artfess.base.enums.ResponseErrorEnums;
import com.artfess.base.exception.BaseException;
import com.artfess.base.exception.SystemException;
import com.artfess.base.interceptor.MasterSlaveAutoRoutingPlugin;
import com.artfess.base.manager.CommonManager;
import com.artfess.base.manager.QueryFilterHelper;
import com.artfess.base.query.PageBean;
import com.artfess.base.query.PageList;
import com.artfess.base.query.QueryFilter;
import com.artfess.base.util.AppUtil;
import com.artfess.base.util.BeanUtils;
import com.artfess.base.util.StringUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import org.springframework.dao.DuplicateKeyException;
import org.springframework.stereotype.Service;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;
import org.springframework.util.Assert;

import com.baomidou.mybatisplus.core.metadata.IPage;


/**
 * 通用管理器的默认实现
 * 
 * @company 阿特菲斯信息技术有限公司
 * @author heyifan
 * @email heyf@jee-soft.cn
 * @date 2018年4月25日
 */
@Service
public class DefaultCommonManager implements CommonManager, QueryFilterHelper<Map<String, Object>> {
	private final Logger logger = LoggerFactory.getLogger(getClass());
	@Resource
	CommonDao commonDao;

	@Override
	public int execute(String sql) {
		Assert.notNull(sql, "sql can not be empty.");
		return this.execute(sql, new HashMap<>());
	}

	/**
	 * 将参数化sql和值转换为map型
	 * <p>
	 * 该方法返回修改后的sql，传入的map会被清空后将objs按顺序放入map，sql转化后示例： select * from ... where a=?
	 * and b=? 转换为 select * from ... where a=#{key0} and b=#{key1}
	 * </p>
	 * 
	 * @param map
	 * @param sql
	 * @param objs
	 */
	private String convertToMap(Map<String, Object> map, String sql, Object... objs) {
		Assert.notNull(map, "map不能为空");
		Assert.isTrue(StringUtil.isNotEmpty(sql), "sql不能为空");
		Assert.isTrue(BeanUtils.isNotEmpty(objs), "objs不能为空");
		Pattern pattern = Pattern.compile("(\\?)");
		Matcher matcher = pattern.matcher(sql);
		StringBuffer sb = new StringBuffer();
		int count = 0;
		while (matcher.find()) {
			matcher.appendReplacement(sb, String.format("#{key%s}", count));
			count++;
		}
		matcher.appendTail(sb);

		if (count != objs.length) {
			throw new BaseException(String.format("sql：%s中参数化的数量和传递进来的参数：%s数量不匹配。", sql, objs));
		}
		map.clear();
		for (int i = 0; i < objs.length; i++) {
			map.put("key" + i, objs[i]);
		}
		return sb.toString();
	}

	@Override
	public int execute(String sql, Object... objs) {
		Map<String, Object> map = new HashMap<>();
		String newSql = convertToMap(map, sql, objs);
		return this.execute(newSql, map);
	}

	@Override
	public int execute(String sql, Map<String, Object> map) {
		map.put("sql", sql);
		// 当前处于切换到外部数据源状态时,使用新事务
		if (MasterSlaveAutoRoutingPlugin.getInExternalDatasource()) {
			PlatformTransactionManager transactionManager = (PlatformTransactionManager) AppUtil
					.getBean(TransactionManager.class);
			DefaultTransactionDefinition def = new DefaultTransactionDefinition();
			def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);
			// 事物隔离级别，开启新事务，这样会比较安全些。
			TransactionStatus status = transactionManager.getTransaction(def);
			// 获得事务状态
			try { // 逻辑代码，可以写上你的逻辑处理代码
				int update = commonDao.update(map);
				transactionManager.commit(status);
				return update;
			}catch (DuplicateKeyException e) {
				transactionManager.rollback(status);
				throw e;
			} catch (Exception e) {
				transactionManager.rollback(status);
				throw new BaseException(ResponseErrorEnums.DATABASE_ERROR, e);
			}
		} else {
			return commonDao.update(map);
		}

	}

	@Override
	public List<Map<String, Object>> query(String sql) {
		return this.query(sql, new HashMap<>());
	}

	@Override
	public List<Map<String, Object>> query(String sql, Object... objs) {
		Map<String, Object> map = new HashMap<>();
		String newSql = convertToMap(map, sql, objs);
		return this.query(newSql, map);
	}

	@Override
	public List<Map<String, Object>> query(String sql, Map<String, Object> map) {
		Assert.notNull(sql, "sql can not be empty.");
		Assert.notNull(map, "map can not be empty.");
		if (map.containsKey("sql")) {
			logger.warn(
					"can not define 'sql' in map when invoke CommonManager.query(sql, map), it will be override by sql string.");
		}
		map.put("sql", sql);

        // 当前处于切换到外部数据源状态时,使用新事务
        if (MasterSlaveAutoRoutingPlugin.getInExternalDatasource()) {
            PlatformTransactionManager transactionManager = (PlatformTransactionManager) AppUtil
                    .getBean(TransactionManager.class);
            DefaultTransactionDefinition def = new DefaultTransactionDefinition();
            def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);
            // 事物隔离级别，开启新事务，这样会比较安全些。
            TransactionStatus status = transactionManager.getTransaction(def);
            // 获得事务状态
            try { // 逻辑代码，可以写上你的逻辑处理代码
                List<Map<String, Object>> result = commonDao.query(map);
                transactionManager.commit(status);
                return result;
            } catch (Exception e) {
                transactionManager.rollback(status);
                throw new BaseException(ResponseErrorEnums.DATABASE_ERROR, e);
            }
        } else {
            return commonDao.query(map);
        }
	}

	@Override
	public PageList<Map<String, Object>> query(String sql, PageBean pageBean) {
		Assert.notNull(sql, "sql can not be empty.");
		IPage<Map<String, Object>> result = commonDao.queryByPage(convert2IPage(pageBean), sql);
		return new PageList<Map<String, Object>>(result);
	}

	@Override
	public PageList<Map<String, Object>> query(String sql, QueryFilter<?> queryFilter) throws SystemException {
		Assert.notNull(sql, "sql can not be empty.");
		Assert.notNull(queryFilter, "queryFilter can not be empty.");
		PageBean pageBean = queryFilter.getPageBean();
		IPage<Map<String, Object>> result = commonDao.queryByQueryFilter(convert2IPage(pageBean), sql,
				convert2Wrapper(queryFilter, null));
		return new PageList<Map<String, Object>>(result);
	}

	@Override
	public PageList<Map<String, Object>> queryByCustomSql(String sql, QueryFilter<?> queryFilter)
			throws SystemException {
		Assert.notNull(sql, "sql can not be empty.");
		Assert.notNull(queryFilter, "queryFilter can not be empty.");
		PageBean pageBean = queryFilter.getPageBean();
		IPage<Map<String, Object>> result = commonDao.queryByCustomSql(convert2IPage(pageBean), sql,
				convert2Wrapper(queryFilter, null));
		return new PageList<Map<String, Object>>(result);
	}

}
