update_info.h

Go to the documentation of this file.

#pragma once
 
#include <array>
 
#include "column_chunk_data.h"
#include "constants.h"
#include "types.h"
 
namespace kuzu {
namespace common {
class ValueVector;
} // namespace common
 
namespace transaction {
class Transaction;
} // namespace transaction
 
namespace storage {
 
class ColumnChunkData;
struct VectorUpdateInfo {
    common::transaction_t version;
    std::array<common::sel_t, common::DEFAULT_VECTOR_CAPACITY> rowsInVector;
    common::sel_t numRowsUpdated;
    // Older versions.
    std::unique_ptr<VectorUpdateInfo> prev;
    // Newer versions.
    VectorUpdateInfo* next;
 
    std::unique_ptr<ColumnChunkData> data;
 
    explicit VectorUpdateInfo(const common::transaction_t transactionID,
        common::LogicalType dataType)
        : version{transactionID}, rowsInVector{}, numRowsUpdated{0}, prev{nullptr}, next{nullptr} {
        data = ColumnChunkFactory::createColumnChunkData(std::move(dataType), false,
            common::DEFAULT_VECTOR_CAPACITY, ResidencyState::IN_MEMORY);
    }
 
    std::unique_ptr<VectorUpdateInfo> movePrev() { return std::move(prev); }
    void setPrev(std::unique_ptr<VectorUpdateInfo> prev) { this->prev = std::move(prev); }
    VectorUpdateInfo* getPrev() const { return prev.get(); }
    void setNext(VectorUpdateInfo* next) { this->next = next; }
    VectorUpdateInfo* getNext() const { return next; }
};
 
class UpdateInfo {
public:
    UpdateInfo() {}
 
    VectorUpdateInfo* update(const transaction::Transaction* transaction, common::idx_t vectorIdx,
        common::sel_t rowIdxInVector, const common::ValueVector& values);
 
    void setVectorInfo(common::idx_t vectorIdx, std::unique_ptr<VectorUpdateInfo> vectorInfo) {
        vectorsInfo[vectorIdx] = std::move(vectorInfo);
    }
    void clearVectorInfo(common::idx_t vectorIdx) { vectorsInfo[vectorIdx] = nullptr; }
 
    common::idx_t getNumVectors() const { return vectorsInfo.size(); }
    VectorUpdateInfo* getVectorInfo(const transaction::Transaction* transaction,
        common::idx_t idx) const;
 
    common::row_idx_t getNumUpdatedRows(const transaction::Transaction* transaction) const;
 
    bool hasUpdates(const transaction::Transaction* transaction, common::row_idx_t startRow,
        common::length_t numRows) const;
 
private:
    VectorUpdateInfo& getOrCreateVectorInfo(const transaction::Transaction* transaction,
        common::idx_t vectorIdx, common::sel_t rowIdxInVector, const common::LogicalType& dataType);
 
private:
    std::vector<std::unique_ptr<VectorUpdateInfo>> vectorsInfo;
};
 
} // namespace storage
} // namespace kuzu