package com.engine.recruit.conn;

import org.apache.commons.lang3.StringUtils;
import weaver.conn.RecordSet;
import weaver.formmode.recruit.modeexpand.util.RecruitModeUtil;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * @author:dxfeng
 * @createTime: 2023/10/19
 * @version: 1.0
 */
public class RecruitRecordSet {

    /**
     * 获取单个记录映射
     *
     * @param rs RecordSet
     * @return
     */
    public static Map<String, Object> getSingleRecordMap(RecordSet rs) {
        Map<String, Object> dataMap = new RecruitDataMap<>();
        if (rs.next()) {
            String[] columnNames = rs.getColumnName();
            for (String columnName : columnNames) {
                dataMap.put(columnName.toLowerCase(), RecruitModeUtil.parseBlankToNull(rs.getString(columnName)));
            }
        }
        return dataMap;
    }

    /**
     * 插入数据
     *
     * @param dataMap   数据集合
     * @param tableName 表名
     */
    public static void insertData(Map<String, Object> dataMap, String tableName) {
        List<String> fieldList = new ArrayList<>();
        List<Object> dataList = new ArrayList<>();
        List<String> paramList = new ArrayList<>();

        dataMap.forEach((key, value) -> {
            fieldList.add(key);
            dataList.add(value);
            paramList.add("?");
        });
        String insertSql = " insert into " + tableName + "(" + StringUtils.join(fieldList, ",") + ") values (" + StringUtils.join(paramList, ",") + ")";
        RecordSet rs = new RecordSet();
        rs.executeUpdate(insertSql, dataList);
    }

}