MockProducerExecStream.h

Go to the documentation of this file.

/*
// $Id: //open/dev/fennel/exec/MockProducerExecStream.h#18 $
// Fennel is a library of data storage and processing components.
// Copyright (C) 2005-2009 The Eigenbase Project
// Copyright (C) 2005-2009 SQLstream, Inc.
// Copyright (C) 2005-2009 LucidEra, Inc.
// Portions Copyright (C) 2004-2009 John V. Sichi
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the Free
// Software Foundation; either version 2 of the License, or (at your option)
// any later version approved by The Eigenbase Project.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#ifndef Fennel_MockProducerExecStream_Included
#define Fennel_MockProducerExecStream_Included

#include "fennel/exec/SingleOutputExecStream.h"
#include "fennel/tuple/TupleData.h"
#include <string>
#include <iostream>

FENNEL_BEGIN_NAMESPACE

template <class T>
class ColumnGenerator
{
public:
    virtual ~ColumnGenerator()
    {
    }

    virtual T next() = 0;
};

typedef ColumnGenerator<int64_t> Int64ColumnGenerator;

typedef boost::shared_ptr<Int64ColumnGenerator> SharedInt64ColumnGenerator;

class FENNEL_EXEC_EXPORT MockProducerExecStreamGenerator
{
public:
    virtual ~MockProducerExecStreamGenerator();

    virtual int64_t generateValue(uint iRow, uint iCol) = 0;
};

typedef boost::shared_ptr<MockProducerExecStreamGenerator>
    SharedMockProducerExecStreamGenerator;

typedef boost::shared_ptr<MockProducerExecStreamGenerator>
    SharedMockProducerExecStreamGenerator;

struct FENNEL_EXEC_EXPORT MockProducerExecStreamParams
    : public SingleOutputExecStreamParams
{
    uint64_t nRows;

    SharedMockProducerExecStreamGenerator pGenerator;

    bool saveTuples;

    std::ostream* echoTuples;

    SharedInt64ColumnGenerator pBatchGenerator;

    MockProducerExecStreamParams()
        : nRows(0), saveTuples(false), echoTuples(0) {}
};

class FENNEL_EXEC_EXPORT MockProducerExecStream
    : public SingleOutputExecStream
{
    uint cbTuple;
    uint64_t nRowsMax;
    uint64_t nRowsProduced;
    TupleData outputData;
    SharedMockProducerExecStreamGenerator pGenerator;
    SharedInt64ColumnGenerator pBatchGenerator;
    bool saveTuples;
    std::ostream* echoTuples;
    std::vector<std::string> savedTuples;

public:
    MockProducerExecStream();

    // implement ExecStream
    virtual void prepare(MockProducerExecStreamParams const &params);
    virtual void open(bool restart);
    virtual ExecStreamResult execute(ExecStreamQuantum const &quantum);

    const std::vector<std::string>& getRowVector() {
        return const_cast<const std::vector<std::string>&>(savedTuples);
    }

    uint64_t getProducedRowCount();
};

FENNEL_END_NAMESPACE

#endif

// End MockProducerExecStream.h

---