package com.iplatform.base.excel;

import com.iplatform.base.di.JdbcExcelDataImportor;
import com.iplatform.base.util.DataImportUtils;
import com.walker.di.UpdateResult;
import com.walker.di.UpdateType;
import com.walker.infrastructure.utils.PhoneNumberUtils;
import com.walker.infrastructure.utils.StringUtils;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;

import java.io.InputStream;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

/**
 * 系统用户导入器实现。
 * @author 时克英
 * @date 2023-02-07
 */
public class UserDataImportor extends JdbcExcelDataImportor {

//    public UserDataImportor(File file){
//        this.setBatchEnabled();
//        this.setBatchSleepMills(200);
//        this.setUpdateType(UpdateType.Override);    // 设置更新策略为:覆盖记录
//        try {
//            this.setSource(new BufferedInputStream(new FileInputStream(file)));
//        } catch (FileNotFoundException e) {
//            throw new RuntimeException(e);
//        }
//    }
    public UserDataImportor(InputStream inputStream){
        this.setBatchEnabled();
        this.setBatchSleepMills(200);
        this.setUpdateType(UpdateType.Override);    // 设置更新策略为:覆盖记录
        this.setSource(inputStream);
    }

    @Override
    protected String acquireTableName() {
        return "s_user_core";
    }

    @Override
    protected boolean isCheckDataExist() {
        return true;
    }

    @Override
    protected UpdateResult checkDataExist(String tableName, List<Map<String, Object>> mapList) {
        // 1 - 根据导入原始数据查询是否在表中存在这些记录
        MapSqlParameterSource sqlParameterSource = new MapSqlParameterSource();
        List<String> paramUserName = DataImportUtils.acquireWhereInStringValues("user_name", mapList);
        sqlParameterSource.addValue("userName", paramUserName);
        List<Map<String, Object>> existList = this.getDataImportService().queryListObjectWhereIn(SQL_QUERY, sqlParameterSource);

        if(!StringUtils.isEmptyList(existList)){
            if(this.logger.isDebugEnabled()){
                for(Map<String, Object> existOne : existList){
                    logger.debug("存在一条数据 = " + existOne);
                }
            }
        }

        // 创建更新结果对象
        UpdateResult updateResult = new UpdateResult();

        // 2 - 设置准备更新的'条件字段集合'，可支持多个
        List<String> whereColumnNames = Arrays.asList("user_name");

        // 3 - 找到新写入和需要更新的数据集合
        Object[] result = DataImportUtils.calculateInsertAndUpdateList(mapList, existList, whereColumnNames);
        List<Map<String, Object>> insertList = (List<Map<String, Object>>)result[0];
        List<Map<String, Object>> updateList = (List<Map<String, Object>>)result[1];

        // 4 - 设置要写入集合
        updateResult.setInsertList(insertList);

        // 5 - 如果更新策略为覆盖，则设置要更新集合
        if(this.getUpdateType() == UpdateType.Override){
            // 如果更新数据策略为: 覆盖模式，需要批量更新数据，这里获得更新集合给平台。
            updateResult.setUpdateColumnNames(Arrays.asList("dept_id", "org_id", "nick_name", "user_type", "email", "phonenumber", "sex"));
            updateResult.setWhereColumnNames(whereColumnNames);
            updateResult.setUpdateList(updateList);
        }

        return updateResult;
    }

    @Override
    protected String validateData(Map<String, String> data) {
        if(data.get("user_name") == null){
            return "登录名不存在";
        }
        if(data.get("nick_name") == null){
            return "昵称不存在";
        }
        if(data.get("user_type") == null){
            return "用户类型不存在";
        }
        String phoneNumber = data.get("phonenumber");
        if(StringUtils.isNotEmpty(phoneNumber) && !PhoneNumberUtils.isCellPhoneNumber(phoneNumber)){
            return "手机号格式错误";
        }
        return null;
    }

    private static final String SQL_QUERY = "select * from s_user_core where user_name in (:userName)";
}