org.tinygroup.dbf.DbfWriter Maven / Gradle / Ivy

Go to download
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.gnu.org/licenses/gpl.html
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.tinygroup.dbf;

import java.io.*;
import java.util.List;

/**
 * Created by wcg on 2014/7/7.
 */
public abstract class DbfWriter implements Writer {
	
    public static final int HEADER_END_CHAR = 13;
    private String encode = "GBK"; //默认编码
    private String filename;
    protected List fields;//字段信息
    protected ByteArrayOutputStream bodybuffer = new ByteArrayOutputStream();
    protected ByteArrayOutputStream headbuffer = new ByteArrayOutputStream();
    private int position = 0;
    
    public void setFilename(String filename) {
    	this.filename = filename;
    }
    
    public void setFields(List fields) {
    	this.fields = fields;
    }
    
    public String getEncode() {
    	return encode;
    }
    public void setEncode(String encode) {
    	this.encode = encode; 
    }
    
    public static Writer generate(String filename,String encode)   {
        DbfWriter writer = new FoxproDBaseWriter();
        writer.setFilename(filename);
    	return writer;
    }
    
    public void save() throws IOException {
    	writeHeaders();
    	FileOutputStream dbffos = new FileOutputStream(new File(filename));
    	byte[] bodybytearray = bodybuffer.toByteArray();
    	headbuffer.write(bodybytearray,0,bodybytearray.length);
    	headbuffer.writeTo(dbffos);
    	headbuffer.close();
    	bodybuffer.close();
    	dbffos.close();
    }
    
    
    public abstract void writeFields(List f) throws IOException;
    
    public abstract void writeRecord(String...args) throws IOException,NullPointerException;
    
    protected abstract void writeHeaders() throws IOException;
    
    protected void next() {
		position ++;
	}
    
    protected int  getPostion() {
		return position;
	}

}