foundationdb/fdbserver/NetworkTest.h

/*
 * NetworkTest.h
 *
 * This source file is part of the FoundationDB open source project
 *
 * Copyright 2013-2022 Apple Inc. and the FoundationDB project authors
 *
 * 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
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef FDBSERVER_NETWORKTEST_H
#define FDBSERVER_NETWORKTEST_H
#pragma once

#include "fdbclient/FDBTypes.h"
#include "fdbrpc/fdbrpc.h"
#include "flow/FileIdentifier.h"

struct NetworkTestInterface {
	RequestStream<struct NetworkTestRequest> test;
	RequestStream<struct NetworkTestStreamingRequest> testStream;
	NetworkTestInterface() {}
	NetworkTestInterface(NetworkAddress remote);
	NetworkTestInterface(INetwork* local);
};

struct NetworkTestReply {
	constexpr static FileIdentifier file_identifier = 14465374;
	Value value;
	NetworkTestReply() {}
	NetworkTestReply(Value value) : value(value) {}
	template <class Ar>
	void serialize(Ar& ar) {
		serializer(ar, value);
	}
};

struct NetworkTestRequest {
	constexpr static FileIdentifier file_identifier = 4146513;
	Key key;
	uint32_t replySize;
	ReplyPromise<struct NetworkTestReply> reply;
	NetworkTestRequest() {}
	NetworkTestRequest(Key key, uint32_t replySize) : key(key), replySize(replySize) {}
	template <class Ar>
	void serialize(Ar& ar) {
		serializer(ar, key, replySize, reply);
	}
};

struct NetworkTestStreamingReply : ReplyPromiseStreamReply {
	constexpr static FileIdentifier file_identifier = 3726830;

	int index = 0;
	NetworkTestStreamingReply() = default;
	explicit NetworkTestStreamingReply(int index) : index(index) {}
	size_t expectedSize() const { return 4e6; /*sizeof(*this);*/ }

	template <class Ar>
	void serialize(Ar& ar) {
		serializer(ar, ReplyPromiseStreamReply::acknowledgeToken, ReplyPromiseStreamReply::sequence, index);
	}
};

struct NetworkTestStreamingRequest {
	constexpr static FileIdentifier file_identifier = 2794452;
	ReplyPromiseStream<struct NetworkTestStreamingReply> reply;
	template <class Ar>
	void serialize(Ar& ar) {
		serializer(ar, reply);
	}
};

Future<Void> networkTestServer();

Future<Void> networkTestClient(std::string const& testServers);

#endif