它能够正确处理由逗号分隔的数据,并且能够忽略双引号中的逗号。
读取csv内容
const getCSVData = (file: RcFile): Promise<string> => {
return new Promise((resolve, reject) => {
const csvReader = new FileReader();
csvReader.readAsText(file);
csvReader.onload = (e) => {
const data = e.target?.result;
resolve(data as string);
};
csvReader.onerror = (error) => reject(error);
});
};
解析csv内容
/**
* 将csv格式字符串转为对象数组
* @param csv csv字符串
* @param choosenField 转换得到的字段,默认为全部获取
* @returns 返回转换得到的结果
*/
export const csvToObjArr = (csv: string, choosenField?: string[]) => {
const contents = csv.split(/\r\n|\n/);
const headers = contents[0].split(',');
const res = contents.slice(1, -1).map((line, index) => {
const obj: any = {};
// 以逗号分隔,但是不包括双引号中的逗号
line.split(/,(?=(?:[^\"]*\"[^\"]*\")*[^\"]*$)/).forEach((item, i) => {
if (item !== '') {
if(choosenField){
if(choosenField.includes(headers[i])){
obj[headers[i]] = item;
}
}else{
obj[headers[i]] = item;
}
}
});
return obj;
});
return res;
};