com.github.housepower.data.ColumnWriterBuffer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of clickhouse-native-jdbc Show documentation
Show all versions of clickhouse-native-jdbc Show documentation
ClickHouse Native Protocol JDBC implementation
/*
* Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
*
* 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 com.github.housepower.data;
import com.github.housepower.buffer.ByteArrayWriter;
import com.github.housepower.serde.BinarySerializer;
import com.github.housepower.settings.ClickHouseDefines;
import static java.lang.Math.min;
import java.io.IOException;
import java.nio.Buffer;
import java.nio.ByteBuffer;
public class ColumnWriterBuffer {
private final ByteArrayWriter columnWriter;
public BinarySerializer column;
public ColumnWriterBuffer() {
this.columnWriter = new ByteArrayWriter(ClickHouseDefines.COLUMN_BUFFER_BYTES);
this.column = new BinarySerializer(columnWriter, false);
}
@SuppressWarnings("RedundantCast")
public void writeTo(BinarySerializer serializer) throws IOException {
// add a temp buffer to reduce memory requirement in case of large remaining data in buffer
byte[] writeBuffer = new byte[4*1024];
for (ByteBuffer buffer : columnWriter.getBufferList()) {
// upcast is necessary, see detail at:
// https://bitbucket.org/ijabz/jaudiotagger/issues/313/java-8-javalangnosuchmethoderror
((Buffer) buffer).flip();
while (buffer.hasRemaining()) {
int remaining = buffer.remaining();
int thisLength = min(remaining, writeBuffer.length);
buffer.get(writeBuffer, 0, thisLength);
serializer.writeBytes(writeBuffer, 0, thisLength);
}
}
}
public void reset() {
columnWriter.reset();
}
}