#include <Interpreters/AsynchronousInsertLog.h>

#include <DataTypes/DataTypeDate.h>
#include <DataTypes/DataTypeDateTime.h>
#include <DataTypes/DataTypeDateTime64.h>
#include <DataTypes/DataTypeLowCardinality.h>
#include <DataTypes/DataTypeString.h>
#include <DataTypes/DataTypesNumber.h>
#include <DataTypes/DataTypeEnum.h>
#include <Parsers/ASTInsertQuery.h>
#include <Parsers/queryToString.h>


namespace DB
{

NamesAndTypesList AsynchronousInsertLogElement::getNamesAndTypes()
{
    auto type_status = std::make_shared<DataTypeEnum8>(
        DataTypeEnum8::Values
        {
            {"Ok",           static_cast<Int8>(Status::Ok)},
            {"ParsingError", static_cast<Int8>(Status::ParsingError)},
            {"FlushError",   static_cast<Int8>(Status::FlushError)},
        });

    return
    {
        {"event_date", std::make_shared<DataTypeDate>()},
        {"event_time", std::make_shared<DataTypeDateTime>()},
        {"event_time_microseconds", std::make_shared<DataTypeDateTime64>(6)},

        {"query", std::make_shared<DataTypeString>()},
        {"database", std::make_shared<DataTypeLowCardinality>(std::make_shared<DataTypeString>())},
        {"table", std::make_shared<DataTypeLowCardinality>(std::make_shared<DataTypeString>())},
        {"format", std::make_shared<DataTypeLowCardinality>(std::make_shared<DataTypeString>())},
        {"query_id", std::make_shared<DataTypeString>()},
        {"bytes", std::make_shared<DataTypeUInt64>()},
        {"exception", std::make_shared<DataTypeString>()},
        {"status", type_status},

        {"flush_time", std::make_shared<DataTypeDateTime>()},
        {"flush_time_microseconds", std::make_shared<DataTypeDateTime64>(6)},
        {"flush_query_id", std::make_shared<DataTypeString>()},
    };
}

void AsynchronousInsertLogElement::appendToBlock(MutableColumns & columns) const
{
    size_t i = 0;

    auto event_date = DateLUT::instance().toDayNum(event_time).toUnderType();
    columns[i++]->insert(event_date);
    columns[i++]->insert(event_time);
    columns[i++]->insert(event_time_microseconds);

    const auto & insert_query = assert_cast<const ASTInsertQuery &>(*query);
    columns[i++]->insert(queryToString(insert_query));

    if (insert_query.table_id)
    {
        columns[i++]->insert(insert_query.table_id.getDatabaseName());
        columns[i++]->insert(insert_query.table_id.getTableName());
    }
    else
    {
        columns[i++]->insertDefault();
        columns[i++]->insertDefault();
    }

    columns[i++]->insert(insert_query.format);
    columns[i++]->insert(query_id);
    columns[i++]->insert(bytes);
    columns[i++]->insert(exception);
    columns[i++]->insert(status);

    columns[i++]->insert(flush_time);
    columns[i++]->insert(flush_time_microseconds);
    columns[i++]->insert(flush_query_id);
}

}