SimplePhysicalNode.h

Go to the documentation of this file.

/*****************************************************************************
 *                                                                           *
 *  Copyright 2018 Rice University                                           *
 *                                                                           *
 *  Licensed under the Apache License, Version 2.0 (the "License");          *
 *  you may not use this file except in compliance with the License.         *
 *  You may obtain a copy of the License at                                  *
 *                                                                           *
 *      http://www.apache.org/licenses/LICENSE-2.0                           *
 *                                                                           *
 *  Unless required by applicable law or agreed to in writing, software      *
 *  distributed under the License is distributed on an "AS IS" BASIS,        *
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
 *  See the License for the specific language governing permissions and      *
 *  limitations under the License.                                           *
 *                                                                           *
 *****************************************************************************/
#ifndef PDB_SIMPLEPHYSICALNODE_H
#define PDB_SIMPLEPHYSICALNODE_H

#include "BroadcastJoinBuildHTJobStage.h"
#include "HashPartitionedJoinBuildHTJobStage.h"
#include "JobStageBuilders/TupleSetJobStageBuilder.h"
#include "AggregationJobStage.h"
#include "TupleSetJobStage.h"
#include "AbstractPhysicalNode.h"

namespace pdb {

class SimplePhysicalNode;
typedef std::shared_ptr<SimplePhysicalNode> SimplePhysicalNodePtr;

class SimplePhysicalNode : public AbstractPhysicalNode {

public:

  SimplePhysicalNode(string jobId,
                     AtomicComputationPtr node,
                     const Handle<ComputePlan> &computePlan,
                     LogicalPlanPtr logicalPlan,
                     ConfigurationPtr conf);

  PhysicalOptimizerResultPtr analyze(const StatisticsPtr &stats, int nextStageID) override;

  PhysicalOptimizerResultPtr analyze(TupleSetJobStageBuilderPtr &shared_ptr,
                                     SimplePhysicalNodePtr &prevNode,
                                     const StatisticsPtr &stats,
                                     int nextStageID);

  const AtomicComputationPtr &getNode() const;

  void addConsumer(const AbstractPhysicalNodePtr &consumer) override;

  bool hasConsumers() override;

  bool isConsuming(Handle<SetIdentifier> &set) override;

  double getCost(const StatisticsPtr &stats) override;

  SimplePhysicalNodePtr getSimpleNodeHandle();

protected:

  virtual PhysicalOptimizerResultPtr analyzeSingleConsumer(TupleSetJobStageBuilderPtr &tupleStageBuilder,
                                                           SimplePhysicalNodePtr &prevNode,
                                                           const StatisticsPtr &stats,
                                                           int nextStageID);

  virtual PhysicalOptimizerResultPtr analyzeOutput(TupleSetJobStageBuilderPtr &tupleStageBuilder,
                                                   SimplePhysicalNodePtr &prevNode,
                                                   const StatisticsPtr &stats,
                                                   int nextStageID);

  virtual PhysicalOptimizerResultPtr analyzeMultipleConsumers(TupleSetJobStageBuilderPtr &ptr,
                                                              SimplePhysicalNodePtr &prevNode,
                                                              const StatisticsPtr &stats,
                                                              int nextStageID);

  std::string getNodeIdentifier() override;

 protected:

  double getCost(Handle<SetIdentifier> source, const StatisticsPtr &stats);

  const Handle<SetIdentifier> &getSourceSetIdentifier() const;

  std::list<SimplePhysicalNodePtr> activeConsumers;

  Handle<SetIdentifier> sourceSetIdentifier;

  AtomicComputationPtr node;

};

}

#endif //PDB_SIMPLEPHYSICALNODE_H