ISetProjector.h

Go to the documentation of this file.

//
// Created by Nico Schäfer on 11/14/17.
//
 
#ifndef JODA_ISetPROJECTOR_H
#define JODA_ISetPROJECTOR_H
#include <rapidjson/pointer.h>
#include <vector>
#include "joda/document/RapidJsonDocument.h"
 
namespace joda::query {
class ISetProjector {
 public:
  ISetProjector(const std::string &to) : ptr(to.c_str()), ptr_str(to){};
 
  virtual ~ISetProjector() = default;
 
  virtual void project(const RapidJsonDocument &json,
                       std::vector<std::unique_ptr<RJDocument>> &newDocs,
                       bool view = false) = 0;
 
  virtual std::string getToPointer() const final { return ptr_str; }
 
  RJPointer getRawToPointer() const { return ptr; }
 
  virtual std::string getType() = 0;
 
  virtual std::string toString() { return "'" + ptr_str + "':"; }
 
  virtual std::vector<std::string> getAttributes() const = 0;
 
 protected:
  RJPointer ptr;
  std::string ptr_str;
 
  unsigned long multiplicate(std::vector<std::unique_ptr<RJDocument>> &newDocs,
                             int times) {
    auto origDocs = newDocs.size();
    newDocs.reserve(origDocs * times);
    // Duplicate existing documents |VAL|-1 times
    for (auto i = 0; i < times - 1; ++i) {
      for (size_t j = 0; j < origDocs; ++j) {
        assert(j < newDocs.size() && j >= 0 && "Stay in range of array");
        auto doc = std::make_unique<RJDocument>(&newDocs[j]->GetAllocator());
        doc->CopyFrom(*newDocs[j], doc->GetAllocator());
        newDocs.push_back(std::move(doc));
      }
    }
    return origDocs;
  }
 
  void fillArrayRangeWithVal(std::vector<std::unique_ptr<RJDocument>> &newDocs,
                             unsigned long origDocs, int i, RJValue &val) {
    for (unsigned long j = origDocs * i; j < origDocs * (i + 1); ++j) {
      assert(j < newDocs.size() && j >= 0 && "Stay in range of array");
      RJValue newval;
      newval.CopyFrom(val, newDocs.front()->GetAllocator());
      ptr.Set(*newDocs[j], newval);
    }
  }
};
}  // namespace joda::query
#endif  // JODA_ISetPROJECTOR_H