io.questdb.cairo.wal.WalEventCursor Maven / Gradle / Ivy
/*******************************************************************************
* ___ _ ____ ____
* / _ \ _ _ ___ ___| |_| _ \| __ )
* | | | | | | |/ _ \/ __| __| | | | _ \
* | |_| | |_| | __/\__ \ |_| |_| | |_) |
* \__\_\\__,_|\___||___/\__|____/|____/
*
* Copyright (c) 2014-2019 Appsicle
* Copyright (c) 2019-2023 QuestDB
*
* 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 io.questdb.cairo.wal;
import io.questdb.cairo.CairoException;
import io.questdb.cairo.ColumnType;
import io.questdb.cairo.sql.BindVariableService;
import io.questdb.cairo.vm.Vm;
import io.questdb.cairo.vm.api.MemoryMR;
import io.questdb.griffin.SqlException;
import io.questdb.std.BinarySequence;
import io.questdb.std.str.StringSink;
import static io.questdb.cairo.wal.WalTxnType.*;
import static io.questdb.cairo.wal.WalUtils.WALE_HEADER_SIZE;
public class WalEventCursor {
public static final long END_OF_EVENTS = -1L;
private final DataInfo dataInfo = new DataInfo();
private final MemoryMR eventMem;
private final SqlInfo sqlInfo = new SqlInfo();
private long memSize;
private long nextOffset = Integer.BYTES;
private long offset = Integer.BYTES; // skip wal meta version
private long txn = END_OF_EVENTS;
private byte type = NONE;
public WalEventCursor(MemoryMR eventMem) {
this.eventMem = eventMem;
}
public void drain() {
long o = offset;
while (true) {
if (memSize >= o + Integer.BYTES) {
final int value = eventMem.getInt(o);
o += Integer.BYTES;
if (value == SymbolMapDiffImpl.END_OF_SYMBOL_ENTRIES) {
offset = o;
break;
}
final int strLength = eventMem.getStrLen(o);
final long storageLength = Vm.getStorageLength(strLength);
o += storageLength;
} else {
throw CairoException.critical(0).put("WAL event file is too small, size=").put(memSize)
.put(", required=").put(o + Integer.BYTES);
}
}
}
public DataInfo getDataInfo() {
if (type != DATA) {
throw CairoException.critical(CairoException.ILLEGAL_OPERATION).put("WAL event type is not DATA, type=").put(type);
}
return dataInfo;
}
public SqlInfo getSqlInfo() {
if (type != SQL) {
throw CairoException.critical(CairoException.ILLEGAL_OPERATION).put("WAL event type is not SQL, type=").put(type);
}
return sqlInfo;
}
public long getTxn() {
return txn;
}
public byte getType() {
return type;
}
public boolean hasNext() {
offset = nextOffset;
int length = readInt();
if (length < 1) {
// EOF
return false;
}
nextOffset = length + nextOffset;
if (memSize < nextOffset + Integer.BYTES) {
eventMem.extend(nextOffset + Integer.BYTES);
memSize = eventMem.size();
}
txn = readLong();
if (txn == END_OF_EVENTS) {
return false;
}
readRecord();
return true;
}
public void reset() {
memSize = eventMem.size();
nextOffset = WALE_HEADER_SIZE; // skip wal meta version
txn = END_OF_EVENTS;
type = WalTxnType.NONE;
}
private void checkMemSize(long requiredBytes) {
if (memSize < offset + requiredBytes) {
throw CairoException.critical(0).put("WAL event file is too small, size=").put(memSize)
.put(", required=").put(offset + requiredBytes);
}
}
private BinarySequence readBin() {
checkMemSize(Long.BYTES);
final long binLength = eventMem.getBinLen(offset);
checkMemSize(binLength);
final BinarySequence value = eventMem.getBin(offset);
offset += binLength + Long.BYTES;
return value;
}
private boolean readBool() {
checkMemSize(Byte.BYTES);
final boolean value = eventMem.getBool(offset);
offset += Byte.BYTES;
return value;
}
private byte readByte() {
checkMemSize(Byte.BYTES);
final byte value = eventMem.getByte(offset);
offset += Byte.BYTES;
return value;
}
private char readChar() {
checkMemSize(Character.BYTES);
final char value = eventMem.getChar(offset);
offset += Character.BYTES;
return value;
}
private double readDouble() {
checkMemSize(Double.BYTES);
final double value = eventMem.getDouble(offset);
offset += Double.BYTES;
return value;
}
private float readFloat() {
checkMemSize(Float.BYTES);
final float value = eventMem.getFloat(offset);
offset += Float.BYTES;
return value;
}
private int readInt() {
checkMemSize(Integer.BYTES);
final int value = eventMem.getInt(offset);
offset += Integer.BYTES;
return value;
}
private long readLong() {
checkMemSize(Long.BYTES);
final long value = eventMem.getLong(offset);
offset += Long.BYTES;
return value;
}
private void readRecord() {
type = readByte();
switch (type) {
case DATA:
dataInfo.read();
break;
case SQL:
sqlInfo.read();
break;
case TRUNCATE:
break;
default:
throw CairoException.critical(CairoException.METADATA_VALIDATION).put("Unsupported WAL event type: ").put(type);
}
}
private short readShort() {
checkMemSize(Short.BYTES);
final short value = eventMem.getShort(offset);
offset += Short.BYTES;
return value;
}
private CharSequence readStr() {
checkMemSize(Integer.BYTES);
final int strLength = eventMem.getStrLen(offset);
final long storageLength = strLength > 0 ? Vm.getStorageLength(strLength) : Integer.BYTES;
checkMemSize(storageLength);
final CharSequence value = strLength >= 0 ? eventMem.getStr(offset) : null;
offset += storageLength;
return value;
}
void openOffset(long offset) {
reset();
if (offset > 0) {
this.offset = offset;
int size = readInt();
this.nextOffset = offset + size;
this.txn = readLong();
this.memSize = eventMem.size();
this.readRecord();
} else {
reset();
}
}
SymbolMapDiff readNextSymbolMapDiff(SymbolMapDiffImpl symbolMapDiff) {
final int columnIndex = readInt();
if (columnIndex == SymbolMapDiffImpl.END_OF_SYMBOL_DIFFS) {
return null;
}
final boolean nullFlag = readBool();
final int cleanTableSymbolCount = readInt();
final int size = readInt();
symbolMapDiff.of(columnIndex, cleanTableSymbolCount, size, nullFlag);
return symbolMapDiff;
}
SymbolMapDiffImpl.Entry readNextSymbolMapDiffEntry(SymbolMapDiffImpl.Entry entry) {
final int key = readInt();
if (key == SymbolMapDiffImpl.END_OF_SYMBOL_ENTRIES) {
entry.clear();
return null;
}
final CharSequence symbol = readStr();
entry.of(key, symbol);
return entry;
}
public class DataInfo implements SymbolMapDiffCursor {
private final SymbolMapDiffImpl symbolMapDiff = new SymbolMapDiffImpl(WalEventCursor.this);
private long endRowID;
private long maxTimestamp;
private long minTimestamp;
private boolean outOfOrder;
private long startRowID;
public long getEndRowID() {
return endRowID;
}
public long getMaxTimestamp() {
return maxTimestamp;
}
public long getMinTimestamp() {
return minTimestamp;
}
public long getStartRowID() {
return startRowID;
}
public boolean isOutOfOrder() {
return outOfOrder;
}
public SymbolMapDiff nextSymbolMapDiff() {
return readNextSymbolMapDiff(symbolMapDiff);
}
private void read() {
startRowID = readLong();
endRowID = readLong();
minTimestamp = readLong();
maxTimestamp = readLong();
outOfOrder = readBool();
}
}
public class SqlInfo {
private final StringSink sql = new StringSink();
private int cmdType;
private long rndSeed0;
private long rndSeed1;
public int getCmdType() {
return cmdType;
}
public long getRndSeed0() {
return rndSeed0;
}
public long getRndSeed1() {
return rndSeed1;
}
public CharSequence getSql() {
return sql;
}
public void populateBindVariableService(BindVariableService bindVariableService) {
bindVariableService.clear();
try {
populateIndexedVariables(bindVariableService);
populateNamedVariables(bindVariableService);
} catch (SqlException e) {
throw CairoException.critical(0).put(e.getMessage());
}
}
private void populateIndexedVariables(BindVariableService bindVariableService) throws SqlException {
final int count = readInt();
for (int i = 0; i < count; i++) {
final int type = readInt();
switch (ColumnType.tagOf(type)) {
case ColumnType.BOOLEAN:
bindVariableService.setBoolean(i, readBool());
break;
case ColumnType.BYTE:
bindVariableService.setByte(i, readByte());
break;
case ColumnType.SHORT:
bindVariableService.setShort(i, readShort());
break;
case ColumnType.CHAR:
bindVariableService.setChar(i, readChar());
break;
case ColumnType.INT:
bindVariableService.setInt(i, readInt());
break;
case ColumnType.FLOAT:
bindVariableService.setFloat(i, readFloat());
break;
case ColumnType.LONG:
case ColumnType.DATE:
case ColumnType.TIMESTAMP:
bindVariableService.setLong(i, readLong());
break;
case ColumnType.DOUBLE:
bindVariableService.setDouble(i, readDouble());
break;
case ColumnType.STRING:
bindVariableService.setStr(i, readStr());
break;
case ColumnType.BINARY:
bindVariableService.setBin(i, readBin());
break;
case ColumnType.GEOBYTE:
bindVariableService.setGeoHash(i, readByte(), type);
break;
case ColumnType.GEOSHORT:
bindVariableService.setGeoHash(i, readShort(), type);
break;
case ColumnType.GEOINT:
bindVariableService.setGeoHash(i, readInt(), type);
break;
case ColumnType.GEOLONG:
bindVariableService.setGeoHash(i, readLong(), type);
break;
default:
throw new UnsupportedOperationException("unsupported column type: " + ColumnType.nameOf(type));
}
}
}
private void populateNamedVariables(BindVariableService bindVariableService) throws SqlException {
final int count = readInt();
for (int i = 0; i < count; i++) {
// garbage, string intern?
final CharSequence name = readStr().toString();
final int type = readInt();
switch (ColumnType.tagOf(type)) {
case ColumnType.BOOLEAN:
bindVariableService.setBoolean(name, readBool());
break;
case ColumnType.BYTE:
bindVariableService.setByte(name, readByte());
break;
case ColumnType.SHORT:
bindVariableService.setShort(name, readShort());
break;
case ColumnType.CHAR:
bindVariableService.setChar(name, readChar());
break;
case ColumnType.INT:
bindVariableService.setInt(name, readInt());
break;
case ColumnType.FLOAT:
bindVariableService.setFloat(name, readFloat());
break;
case ColumnType.LONG:
case ColumnType.DATE:
case ColumnType.TIMESTAMP:
bindVariableService.setLong(name, readLong());
break;
case ColumnType.DOUBLE:
bindVariableService.setDouble(name, readDouble());
break;
case ColumnType.STRING:
bindVariableService.setStr(name, readStr());
break;
case ColumnType.BINARY:
bindVariableService.setBin(name, readBin());
break;
case ColumnType.GEOBYTE:
bindVariableService.setGeoHash(name, readByte(), type);
break;
case ColumnType.GEOSHORT:
bindVariableService.setGeoHash(name, readShort(), type);
break;
case ColumnType.GEOINT:
bindVariableService.setGeoHash(name, readInt(), type);
break;
case ColumnType.GEOLONG:
bindVariableService.setGeoHash(name, readLong(), type);
break;
case ColumnType.UUID:
bindVariableService.setUuid(name, readLong(), readLong());
break;
default:
throw new UnsupportedOperationException("unsupported column type: " + ColumnType.nameOf(type));
}
}
}
private void read() {
cmdType = readInt();
sql.clear();
sql.put(readStr());
rndSeed0 = readLong();
rndSeed1 = readLong();
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy