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

74 lines
2.1 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 "DrCommon.h"
#include <channelinterface.h>
#include <channelparser.h>
#include <channelmarshaler.h>
class DataBlockItem : public RChannelDataItem
{
public:
DataBlockItem(DrMemoryBuffer* buf);
DataBlockItem(Size_t size);
~DataBlockItem();
Size_t GetAllocatedSize();
Size_t GetAvailableSize();
void SetAvailableSize(Size_t size);
void * GetDataAddress();
DrMemoryBuffer * GetData();
private:
DrRef<DrMemoryBuffer> m_buf;
};
class DataBlockParser : public RChannelItemParser
{
public:
DataBlockParser() {};
DataBlockParser(DObjFactoryBase* factory);
RChannelItem* ParseNextItem(ChannelDataBufferList* bufferList,
Size_t startOffset,
Size_t* pOutLength);
RChannelItem* ParsePartialItem(ChannelDataBufferList* bufferList,
Size_t startOffset,
RChannelBufferMarker*
markerBuffer);
};
typedef StdParserFactory<DataBlockParser> DataBlockParserFactory;
class DataBlockMarshaler : public RChannelItemMarshaler
{
public:
DrError MarshalItem(ChannelMemoryBufferWriter* writer,
RChannelItem* item,
bool flush,
RChannelItemRef* pFailureItem);
};
typedef StdMarshalerFactory<DataBlockMarshaler> DataBlockMarshalerFactory;