GenericQueryBuilder.h

Go to the documentation of this file.

#pragma once
#include <memory>
#include <string>
#include <unordered_map>
 
#include "interfaces/AggregatorQuery.h"
#include "interfaces/ConditionQuery.h"
#include "interfaces/DataGeneratorInterface.h"
#include "interfaces/Validators.h"
 
namespace GenericQuery {
 
using ColumnStore::Column;
using ColumnStore::ConditionQuery;
using ColumnStore::DataRecord;
using ColumnStore::DataRecordMetadata;
using ColumnStore::DataSource;
using ColumnStore::DataType;
using ColumnStore::DataValue;
using ColumnStore::Metadata;
using ColumnStore::Query;
using ColumnStore::RecordValidator;
 
class GenericDataGenerator;
 
class GenericQueryBuilder {
    struct Join {
        std::string foreignKey;
        std::string primaryKey;
        std::vector<std::string> columns;
    };
 
    std::unordered_map<std::string, DataSource> data_sources;
 
    std::string baseSource;
 
    std::vector<std::string> baseColumns;
 
    std::vector<std::pair<std::string, Join>> joins;
 
    std::unordered_map<std::string, std::string> columnMap;
 
    Query query = Query(new ConditionQuery);
 
    std::vector<ColumnStore::AggregatorQuery> aggregations;
    std::vector<std::string> groupBys;
 
    friend class GenericDataGenerator;
    friend class GenericDataAggregator;
    friend class GenericGroupByAggregator;
 
   public:
    void registerDataSource(std::string name, DataSource source);
 
    void setBaseSource(std::string name, std::vector<std::string> columns);
 
    void join(std::string name, std::vector<std::string> columns,
              std::string foreignKey, std::string primaryKey);
 
    void where(Query query);
 
    Metadata generateMetadata();
 
    RecordValidator generateRecordValidator();
 
    DataSource build();
 
    void aggregate(ColumnStore::AggregatorQuery query) {
        aggregations.push_back(query);
    }
 
    void groupBy(std::string name) { groupBys.push_back(name); }
};
};  // namespace GenericQuery