如何使用Java处理庞大的单行文件

时间:2016-06-03 18:15:54

标签: java io large-files

我需要读取一个巨大的文件(15 + GB)并执行一些小的修改(添加一些换行符,以便不同的解析器可以实际使用它)。您可能认为通常有这样做的答案:

但我的整个档案都在一行。

到目前为止,我的一般方法非常基础:

char[] buffer = new char[X];
BufferedReader reader = new BufferedReader(new ReaderUTF8(new FileInputStream(new File("myFileName"))), X);
char[] bufferOut = new char[X+a little];
int bytesRead = -1;
int i = 0;
int offset = 0;
long totalBytesRead = 0;
int countToPrint = 0;
while((bytesRead = reader.read(buffer)) >= 0){
    for(i = 0; i < bytesRead; i++){
        if(buffer[i] == '}'){
            bufferOut[i+offset] = '}';
            offset++;
            bufferOut[i+offset] = '\n';
        }
        else{
            bufferOut[i+offset] = buffer[i];
        }
    }
    writer.write(bufferOut, 0, bytesRead+offset);
    offset = 0;
    totalBytesRead += bytesRead;
    countToPrint += 1;
    if(countToPrint == 10){
        countToPrint = 0;
        System.out.println("Read "+((double)totalBytesRead / originalFileSize * 100)+" percent.");
    }
}
writer.flush();

经过一些实验,我发现大于一百万的X值给出了最佳速度 - 看起来我每10分钟得到2%左右,而X值约为60,000 15小时内只有60%。剖析显示我在read()方法中花费了96%以上的时间,所以这绝对是我的瓶颈。在撰写本文时,我的800万X版本在2小时40分钟后完成了32%的文件,以防您想知道它是如何长期执行的。

有没有更好的方法来处理这么大的单行文件?有,是否有更快的方式来读取这种类型的文件,这使我可以相对简单地插入换行符?

我知道不同的语言或程序可能会优雅地处理这个问题,但我将其限制在Java的角度。

1 个答案:

答案 0 :(得分:10)

你正在使它变得更加复杂。通过使用标准类已经提供的缓冲,你应该每秒至少几MB的麻烦,没有任何麻烦。

这个简单的测试程序在不到2分钟的时间内在我的电脑上处理1GB(包括创建测试文件):

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.Writer;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Random;

public class TestFileProcessing {

    public static void main(String[] argv) {

        try {
            long time = System.currentTimeMillis();
            File from = new File("C:\\Test\\Input.txt");
            createTestFile(from, StandardCharsets.UTF_8, 1_000_000_000);
            System.out.println("Created file in: " + (System.currentTimeMillis() - time) + "ms");

            time = System.currentTimeMillis();
            File to = new File("C:\\Test\\Output.txt");
            doIt(from, to, StandardCharsets.UTF_8);
            System.out.println("Converted file in: " + (System.currentTimeMillis() - time) + "ms");
        } catch (IOException e) {
            throw new RuntimeException(e.getMessage(), e);
        }
    }

    public static void createTestFile(File file, Charset encoding, long size) throws IOException {
        Random r = new Random(12345);
        try (OutputStream fout = new FileOutputStream(file);
                BufferedOutputStream bout = new BufferedOutputStream(fout);
                Writer writer = new OutputStreamWriter(bout, encoding)) {
            for (long i=0; i<size; ++i) {
                int c = r.nextInt(26);
                if (c == 0)
                    writer.write('}');
                else
                    writer.write('a' + c);
            }
        }
    }

    public static void doIt(File from, File to, Charset encoding) throws IOException {
        try (InputStream fin = new FileInputStream(from);
                BufferedInputStream bin = new BufferedInputStream(fin);
                Reader reader = new InputStreamReader(bin, encoding);
                OutputStream fout = new FileOutputStream(to);
                BufferedOutputStream bout = new BufferedOutputStream(fout);
                Writer writer = new OutputStreamWriter(bout, encoding)) {
            int c;
            while ((c = reader.read()) >= 0) {
                if (c == '}')
                    writer.write('\n');
                writer.write(c);
            }
        }
    }

}

如您所见,没有使用复杂的逻辑或过多的缓冲区大小。使用的是简单地缓冲最接近硬件的流,FileInput / OutputStream。