MockConsumerExecStream.h

Go to the documentation of this file.

/*
// $Id: //open/dev/fennel/exec/MockConsumerExecStream.h#14 $
// 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) 2005-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_MockConsumerExecStream_Included
#define Fennel_MockConsumerExecStream_Included

#include "fennel/exec/SingleInputExecStream.h"
#include "fennel/tuple/TupleData.h"
#include "fennel/tuple/TuplePrinter.h"

using std::vector;
using std::string;
using std::ostream;

FENNEL_BEGIN_NAMESPACE

struct FENNEL_EXEC_EXPORT MockConsumerExecStreamParams
    : public SingleInputExecStreamParams
{
    bool saveData;
    ostream* echoData;

    MockConsumerExecStreamParams() : saveData(true), echoData(0)
    {
    }
};

class FENNEL_EXEC_EXPORT MockConsumerExecStream
    : public SingleInputExecStream
{
protected:
    bool saveData;
    ostream* echoData;
    vector<string> rowStrings;
private:
    long rowCount;
    TupleData inputTuple;
    TuplePrinter tuplePrinter;
    bool recvEOS;

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

    long getRowCount() const
    {
        return rowCount;
    }

    const vector<string>& getRowVector() {
        return const_cast<const vector<string>& >(rowStrings);
    }

    bool getRecvEOS() const
    {
        return recvEOS;
    }
};

FENNEL_END_NAMESPACE

#endif

// End MockConsumerExecStream.h

---