js通用導出csv




/*
 * 參數:
 *    headers:表頭
 *             格式如:
 *             [
 *               {
 *                  column:"userid", //列字段
 *                  title: "姓名", //表頭標題
 *                  formatter:(value) =>{return ... }   //格式化函數  
 *                },
 *                ...
 *              ]
 *    
 *    rows:行
 *          格式如:[{userid:101,gender:0 },{userid:102,gender:1}]
 *    filename:導出保存的文件名
 */
export default function exportCsv(headers, rows, filename) {
    if (Array.isArray(headers) && headers.length > 0) { //表頭信息不能為空
        if (!filename || typeof filename != "string") {
            fileName = "export.csv"
        }
        let blob = getCsvBlob(headers, rows);
        if (navigator.msSaveOrOpenBlob) {
            navigator.msSaveOrOpenBlob(blob, filename);
        } else {
            let url = URL.createObjectURL(blob);
            let downloadLink = document.createElement('a');
            downloadLink.href = url;
            downloadLink.download = filename;
            document.body.appendChild(downloadLink);
            downloadLink.click();
            document.body.removeChild(downloadLink);
            URL.revokeObjectURL(url);
        }
    }
}

function getCsvBlob(headers, rows) {
    const BOM = '\uFEFF';
    let columnDelimiter = ','; //默認列分隔符','
    let rowDelimiter = '\r\n'; //默認行分隔符 '\r\n'
    let csv = headers.reduce((previous, header) => {
        return (previous ? previous + columnDelimiter : '') + (header.title || header.column);
    }, '');
    if (Array.isArray(rows) && rows.length > 0) {
        let columns = headers.map(header => header.column);
        csv = rows.reduce((previous, row) => {
            let rowCsv = columns.reduce((pre, column) => {
                if (row.hasOwnProperty(column)) {
                    let cell = row[column];
                    if (cell != null) {
                        let header = headers.find(item => item.column == column);
                        if (header.formatter != null && typeof (header.formatter) == "function") {
                            cell = header.formatter(cell);
                        }
                        if (cell != null) {
                            cell = cell.toString().replace(new RegExp(rowDelimiter, 'g'), ' '); // 若數據中本來就含行分隔符,則用' '替換
                            cell = new RegExp(columnDelimiter).test(cell) ? `"${cell}"` : cell; //若數據中本來就含列分隔符,則用""包起來
                            return pre ? pre + columnDelimiter + cell : pre + cell;
                        }
                    }
                    return pre ? pre + columnDelimiter : pre + " ";//reduce初始值為'',故第一次迭代時不會在行首加列分隔符。后面的遇到值為空或不存在的列要填充含空格的空白" ",則pre返回true,會加列分隔符
                }
                else {
                    return pre ? pre + columnDelimiter : pre + " ";//即使不存在該列也要填充空白,避免數據和表頭錯位不對應
                }
            }, '');
            return previous + rowDelimiter + rowCsv;
        }, csv);
    }
    let blob = new Blob([BOM + csv], { type: 'text/csv;charset=utf-8;' });
    return blob;
}



免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM