转载URL : http://blog.csdn.net/v123411739/article/details/50620289
前言
最近在开发的时候,接到了一个开发任务,要将百万行级别的txt数据插入到数据库中,由于内存方面的原因,因此不可能一次读取所有内容,后来在网上找到了解决方法,可以使用NIO技术来处理,于是找到了这篇文章http://www.sharejs.com/codes/java/1334,后来在试验过程中发现了一点小bug,由于是按字节读取,汉字又是2个字节,因此会出现汉字读取“一半”导致乱码的情况,于是花了几天时间将这个问题解决了。
例子
假设我们一次读取的字节是从下图的start到end,因为结尾是汉字,所以有几率出现上述的情况。
解决方法如下:将第9行这半行(第9行阴影的部分)跟上一次读取留下来的半行(第9行没阴影的部分)按顺序存放在字节数组,然后转成字符串;中间第10行到第17行正常转换成字符串;第18行这半行(第18行阴影的部分)留着跟下一次读取的第1行(第18行没阴影的部分)连接成一行,因为是先拼接成字节数组再转字符串,因此不会出现乱码的情况。
代码
[java] view plain copy- import java.io.File;
- import java.io.IOException;
- import java.io.RandomAccessFile;
- import java.nio.ByteBuffer;
- import java.nio.channels.FileChannel;
- import java.util.Date;
- public class TestNIO3 {
- public static void main(String args[]) throws Exception {
- int bufSize = 1000000;//一次读取的字节长度
- File fin = new File("D:\\test\\20151224_1375789.txt");//读取的文件
- File fout = new File("D:\\test\\111111111111111111.txt");//写出的文件
- Date startDate = new Date();
- FileChannel fcin = new RandomAccessFile(fin, "r").getChannel();
- ByteBuffer rBuffer = ByteBuffer.allocate(bufSize);
- FileChannel fcout = new RandomAccessFile(fout, "rws").getChannel();
- ByteBuffer wBuffer = ByteBuffer.allocateDirect(bufSize);
- readFileByLine(bufSize, fcin, rBuffer, fcout, wBuffer);
- Date endDate = new Date();
- System.out.print(startDate+"|"+endDate);//测试执行时间
- }
- public static void readFileByLine(int bufSize, FileChannel fcin,
- ByteBuffer rBuffer, FileChannel fcout, ByteBuffer wBuffer) {
- String enterStr = "\n";
- try {
- byte[] bs = new byte[bufSize];
- //temp:由于是按固定字节读取,在一次读取中,第一行和最后一行经常是不完整的行,因此定义此变量来存储上次的最后一行和这次的第一行的内容,
- //并将之连接成完成的一行,否则会出现汉字被拆分成2个字节,并被提前转换成字符串而乱码的问题,数组大小应大于文件中最长一行的字节数
- byte[] temp = new byte[500];
- while (fcin.read(rBuffer) != -1) {
- int rSize = rBuffer.position();
- rBuffer.rewind();
- rBuffer.get(bs);
- rBuffer.clear();
- //windows下ascii值13、10是换行和回车,unix下ascii值10是换行
- //从开头顺序遍历,找到第一个换行符
- int startNum=0;
- int length=0;
- for(int i=0;i<rSize;i++){
- if(bs[i]==10){//找到换行字符
- startNum=i;
- for(int k=0;k<500;k++){
- if(temp[k]==0){//temp已经存储了上一次读取的最后一行,因此遍历找到空字符位置,继续存储此次的第一行内容,连接成完成一行
- length=i+k;
- for(int j=0;j<=i;j++){
- temp[k+j]=bs[j];
- }
- break;
- }
- }
- break;
- }
- }
- //将拼凑出来的完整的一行转换成字符串
- String tempString1 = new String(temp, 0, length+1, "GBK");
- //清空temp数组
- for(int i=0;i<temp.length;i++){
- temp[i]=0;
- }
- //从末尾倒序遍历,找到第一个换行符
- int endNum=0;
- int k = 0;
- for(int i=rSize-1;i>=0;i--){
- if(bs[i]==10){
- endNum=i;//记录最后一个换行符的位置
- for(int j=i+1;j<rSize;j++){
- temp[k++]=bs[j];//将此次读取的最后一行的不完整字节存储在temp数组,用来跟下一次读取的第一行拼接成完成一行
- bs[j]=0;
- }
- break;
- }
- }
- //去掉第一行和最后一行不完整的,将中间所有完整的行转换成字符串
- String tempString2 = new String(bs, startNum+1, endNum-startNum, "GBK");
- //拼接两个字符串
- String tempString = tempString1 + tempString2;
- // System.out.print(tempString);
- int fromIndex = 0;
- int endIndex = 0;
- while ((endIndex = tempString.indexOf(enterStr, fromIndex)) != -1) {
- String line = tempString.substring(fromIndex, endIndex)+enterStr;//按行截取字符串
- System.out.print(line);
- //写入文件
- writeFileByLine(fcout, wBuffer, line);
- fromIndex = endIndex + 1;
- }
- }
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- /**
- * 写到文件上
- * @param fcout
- * @param wBuffer
- * @param line
- */
- @SuppressWarnings("static-access")
- public static void writeFileByLine(FileChannel fcout, ByteBuffer wBuffer,
- String line) {
- try {
- fcout.write(wBuffer.wrap(line.getBytes("UTF-8")), fcout.size());
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- }