cn.icuter.jsql.data.LobOutputStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jsql-jdk1.6 Show documentation

JDBC framework for writing sql like java programing

The newest version!

package cn.icuter.jsql.data;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;

/**
 * @author edward
 * @since 2019-02-07
 */
public class LobOutputStream extends OutputStream {
    private JSQLClob clob;
    private JSQLBlob blob;
    private ByteArrayOutputStream out;

    LobOutputStream(JSQLBlob blob, long position) {
        this.blob = blob;
        init(blob.data, position);
    }

    LobOutputStream(JSQLClob clob, long position) {
        this.clob = clob;
        init(clob.data.toString().getBytes(StandardCharsets.UTF_8), position);
    }

    private void init(byte[] srcBytes, long position) {
        this.out = new ByteArrayOutputStream();
        for (int i = 0; i < position - 1; i++) {
            this.out.write(srcBytes[i]);
        }
    }

    @Override
    public void write(int b) throws IOException {
        out.write(b);
    }

    @Override
    public void flush() throws IOException {
        if (clob != null) {
            String replacing = out.toString("UTF-8");
            clob.data.replace(0, replacing.length(), replacing);
        }
        if (blob != null) {
            byte[] bytesInStream = out.toByteArray();
            if (bytesInStream.length >= blob.data.length) {
                blob.data = bytesInStream;
            } else {
                out.write(blob.data, bytesInStream.length, blob.data.length - bytesInStream.length);
                blob.data = out.toByteArray();
            }
        }
    }

    @Override
    public void close() throws IOException {
        flush();
    }
}