Dryad/DryadVertex/VertexHost/vertex/include/FifoOutputChannel.h

43 lines
1.2 KiB
C++

/*
Copyright (c) Microsoft Corporation
All rights reserved.
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
THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, EITHER
EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED WARRANTIES OR CONDITIONS OF
TITLE, FITNESS FOR A PARTICULAR PURPOSE, MERCHANTABLITY OR NON-INFRINGEMENT.
See the Apache Version 2.0 License for specific language governing permissions and
limitations under the License.
*/
#pragma once
#include <outputchannel.h>
#include <fifochannel.h>
class FifoOutputChannel: public OutputChannel
{
public:
FifoOutputChannel(UInt32 portNum, DryadVertexProgram* vertex,
WorkQueue *workQueue, RChannelWriter* outputChannel,
TransformType tType);
void Stop();
void SetInitialSizeHint(UInt64 hint);
const char* GetURI();
private:
FifoChannel *m_fifoChannel;
RChannelReaderHolderRef m_fifoReader;
RChannelWriterHolderRef m_fifoWriter;
RChannelWriter* m_origWriter;
void MakeFifo(UInt32 fifoLength, WorkQueue* workQueue);
};