warning_context.h

Go to the documentation of this file.

#pragma once
 
#include <functional>
#include <mutex>
#include <vector>
 
#include "api.h"
#include "types.h"
#include "client_config.h"
#include "copy_from_error.h"
 
namespace kuzu {
namespace common {
class ValueVector;
}
namespace storage {
class ColumnChunkData;
}
 
namespace processor {
 
class SerialCSVReader;
 
struct WarningInfo {
    uint64_t queryID;
    PopulatedCopyFromError warning;
 
    WarningInfo(PopulatedCopyFromError warning, uint64_t queryID)
        : queryID(queryID), warning(std::move(warning)) {}
};
 
using populate_func_t = std::function<PopulatedCopyFromError(CopyFromFileError, common::idx_t)>;
using get_file_idx_func_t = std::function<common::idx_t(const CopyFromFileError&)>;
 
class KUZU_API WarningContext {
public:
    explicit WarningContext(main::ClientConfig* clientConfig);
 
    void appendWarningMessages(const std::vector<CopyFromFileError>& messages);
 
    void populateWarnings(uint64_t queryID, populate_func_t populateFunc = {},
        get_file_idx_func_t getFileIdxFunc = {});
    void defaultPopulateAllWarnings(uint64_t queryID);
 
    const std::vector<WarningInfo>& getPopulatedWarnings() const;
    uint64_t getWarningCount(uint64_t queryID);
    void clearPopulatedWarnings();
 
    void setIgnoreErrorsForCurrentQuery(bool ignoreErrors);
    // NOTE: this function only works if the logical operator is COPY FROM
    // for other operators setIgnoreErrorsForCurrentQuery() is not called
    bool getIgnoreErrorsOption() const;
 
private:
    std::mutex mtx;
    main::ClientConfig* clientConfig;
    std::vector<CopyFromFileError> unpopulatedWarnings;
    std::vector<WarningInfo> populatedWarnings;
    uint64_t queryWarningCount;
    uint64_t numStoredWarnings;
    bool ignoreErrorsOption;
};
 
} // namespace processor
} // namespace kuzu