Java探索之——内存映射文件

内存映射文件是java NIO模块里非常有用的一个功能，它通过FileChannel中的map方法将文件片段映射到内存中，并且可以直接将对内存的操作反应到文件中，非常适合对大文件的操作，并且效率相比传统的字节IO流效率高出不少。

以下是我实现的一个简单的文件编辑器。

/**
 * 文件编辑器
 * @author rui.chen
 *
 */
public class MyFileEditor {
    private FileChannel channel;
    private RandomAccessFile file;
    
    public MyFileEditor(String path){
        try {
            File f = new File(path);
            if(!f.exists() || !f.isFile()){
                f.createNewFile();
            }
            file = new RandomAccessFile(f, "rw");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    public void open(){
        channel = file.getChannel();
    }
    
    /**
     * 在文件末尾添加内容
     */
    public void append(String content){
        try {
            override(content, (int) file.length());
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 在position位置写入content，并覆盖原有内容
     */
    public void override(String content, int position){
        try {
            MappedByteBuffer buffer = channel.map(MapMode.READ_WRITE, position, content.length());
            buffer.position(0);
            buffer.put(content.getBytes());
            buffer.flip();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 在position位置插入内容，不覆盖原有内容
     */
    public void insert(String content, int position){
        try {
            int length = (int) (file.length()-position+content.length());
            int fileLen = (int) file.length();
            MappedByteBuffer buffer = channel.map(MapMode.READ_WRITE, 
                    position, length);
            buffer.position(0);
            byte[] dst = new byte[length];
            byte[] contentBytes = content.getBytes();
            for(int i=0; i<contentBytes.length; i++){
                dst[i] = contentBytes[i];
            }
            byte[] old = new byte[fileLen-position];
            buffer.get(old, 0, fileLen-position);
            buffer.position(0);
            for(int i=0; i<old.length; i++){
                dst[i+contentBytes.length] = old[i];
            }
            buffer.put(dst);
            buffer.flip();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
    public void close(){
        if(channel.isOpen()){
            try {
                channel.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

通过下面的测试程序进行测试，文件大小为516MB，运行时间为44毫秒。

public class NIOTest {
    public static void main(String[] args){
        MyFileEditor editor = new MyFileEditor("D:\\server.log.2013-05-02");
        editor.open();
        long start = System.currentTimeMillis();
        editor.append("content be appended");
        System.out.println("time cost: "+(System.currentTimeMillis()-start)+" milliseconds");
        editor.close();
    }
}