asterix-runtime/src/main/java/edu/uci/ics/asterix/runtime/evaluators/common/AccessibleByteArrayEval.java

package edu.uci.ics.asterix.runtime.evaluators.common;

import java.io.DataOutput;
import java.io.DataOutputStream;
import java.io.IOException;

import edu.uci.ics.hyracks.algebricks.common.exceptions.AlgebricksException;
import edu.uci.ics.hyracks.algebricks.runtime.base.IEvaluator;
import edu.uci.ics.hyracks.dataflow.common.comm.io.ByteArrayAccessibleOutputStream;
import edu.uci.ics.hyracks.dataflow.common.data.accessors.IFrameTupleReference;

public class AccessibleByteArrayEval implements IEvaluator {

    private DataOutput out;
    private ByteArrayAccessibleOutputStream baaos;
    private DataOutput dataOutput;

    public AccessibleByteArrayEval(DataOutput out) {
        this.out = out;
        this.baaos = new ByteArrayAccessibleOutputStream();
        this.dataOutput = new DataOutputStream(baaos);
    }

    public AccessibleByteArrayEval(DataOutput out, ByteArrayAccessibleOutputStream baaos) {
        this.out = out;
        this.baaos = baaos;
        this.dataOutput = new DataOutputStream(baaos);
    }

    public DataOutput getDataOutput() {
        return dataOutput;
    }

    @Override
    public void evaluate(IFrameTupleReference tuple) throws AlgebricksException {
        try {
            out.write(baaos.getByteArray(), 0, baaos.size());
        } catch (IOException e) {
            throw new AlgebricksException(e);
        }
    }

    public void setBaaos(ByteArrayAccessibleOutputStream baaos) {
        this.baaos = baaos;
    }

    public ByteArrayAccessibleOutputStream getBaaos() {
        return baaos;
    }

    public void reset() {
        baaos.reset();
    }
}