UnrealEngineUWP/Engine/Source/Programs/Horde/Samples/RemoteWorkerCpp/ComputeChannel.h

// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "ComputeBuffer.h"

class FComputeSocket;

//
// Allows bi-directional communication between two nodes using compute buffers
//
class FComputeChannel
{
public:
	// Reader for the channel
	FComputeBufferReader Reader;

	// Writer for the channel
	FComputeBufferWriter Writer;

	FComputeChannel();
	FComputeChannel(FComputeBufferReader InReader, FComputeBufferWriter InWriter);
	~FComputeChannel();

	// Tests whether the channel is valid
	bool IsValid() const;

	// Sends bytes to the remote. 
	size_t Send(const void* Data, size_t Size, int TimeoutMs = -1);

	// Reads as many bytes as are available from the socket.
	size_t Recv(void* Data, size_t Size, int TimeoutMs = -1);

	// Indicate to the remote that no more data will be sent.
	void MarkComplete();
};