/*
 * FileSystem.actor.cpp
 *
 * This source file is part of the FoundationDB open source project
 *
 * Copyright 2013-2018 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.
 */

#include "flow/actorcompiler.h"
#include "fdbrpc/ContinuousSample.h"
#include "fdbclient/NativeAPI.h"
#include "fdbserver/TesterInterface.h"
#include "workloads.h"

struct FileSystemWorkload : TestWorkload {
	int actorCount, writeActorCount, fileCount, pathMinChars, pathCharRange, serverCount, userIDCount;
	double testDuration, transactionsPerSecond, deletedFilesRatio;
	bool discardEdgeMeasurements, performingWrites, loggingQueries;
	std::string operationName;

	vector<Future<Void>> clients;
	PerfIntCounter queries, writes;
	ContinuousSample<double> latencies;
	ContinuousSample<double> writeLatencies;

	class FileSystemOp {
	public:
		virtual Future<Optional<Version>> run( FileSystemWorkload *self, Transaction* tr ) = 0;
		virtual const char *name() = 0;
		virtual ~FileSystemOp() {}
	};

	FileSystemWorkload(WorkloadContext const& wcx)
		: TestWorkload(wcx),
		latencies( 2500 ), writeLatencies( 1000 ), queries("Queries"), writes("Latency")
	{
		testDuration = getOption( options, LiteralStringRef("testDuration"), 10.0 );
		transactionsPerSecond = getOption( options, LiteralStringRef("transactionsPerSecond"), 5000.0 ) / clientCount;
		double allowedLatency =  getOption( options, LiteralStringRef("allowedLatency"), 0.250 );
		actorCount = transactionsPerSecond * allowedLatency;
		fileCount = getOption( options, LiteralStringRef("fileCount"), 100000 );
		pathMinChars = std::max( getOption( options, LiteralStringRef("pathMinChars"), 32 ), 8 );
		pathCharRange = std::max( getOption( options, LiteralStringRef("pathMaxChars"), 128 ), pathMinChars ) - pathMinChars;
		discardEdgeMeasurements = getOption( options, LiteralStringRef("discardEdgeMeasurements"), true );
		deletedFilesRatio = getOption( options, LiteralStringRef("deletedFilesRatio"), 0.01 );
		serverCount = getOption( options, LiteralStringRef("serverCount"), 32 );
		userIDCount = getOption( options, LiteralStringRef("userIDCount"), std::max( 100, fileCount / 3000 ) );
		operationName = getOption( options, LiteralStringRef("operationName"), LiteralStringRef("modificationQuery") ).toString();
		performingWrites = getOption( options, LiteralStringRef("performingWrites"), false );
		writeActorCount = getOption( options, LiteralStringRef("writeActorCount"), 4 );
		loggingQueries = getOption( options, LiteralStringRef("loggingQueries"), false );
	}

	virtual std::string description() { return "ReadWrite"; }

	virtual Future<Void> setup( Database const& cx ) {
		return nodeSetup( cx, this );
	}

	virtual Future<Void> start( Database const& cx ) {
		return _start( cx, this );
	}

	virtual Future<bool> check( Database const& cx ) {
		clients.clear();
		return true;
	}

	virtual void getMetrics( vector<PerfMetric>& m ) {
		double duration = testDuration * (discardEdgeMeasurements ? 0.75 : 1.0);
		m.push_back( PerfMetric( "Measured Duration", duration, true ) );
		m.push_back( PerfMetric( "Transactions/sec", queries.getValue() / duration, false ) );
		m.push_back( PerfMetric( "Writes/sec", writes.getValue() / duration, false ) );
		m.push_back( PerfMetric( "Mean Latency (ms)", 1000 * latencies.mean(), true ) );
		m.push_back( PerfMetric( "Median Latency (ms, averaged)", 1000 * latencies.median(), true ) );
		m.push_back( PerfMetric( "90% Latency (ms, averaged)", 1000 * latencies.percentile( 0.90 ), true ) );
		m.push_back( PerfMetric( "98% Latency (ms, averaged)", 1000 * latencies.percentile( 0.98 ), true ) );
		m.push_back( PerfMetric( "Median Write Latency (ms, averaged)", 1000 * writeLatencies.median(), true ) );
	}

	Key keyForFileID( uint64_t id ) { return StringRef( format( "/files/id/%016llx", id ) ); }

	void initializeFile( Transaction *tr, FileSystemWorkload *self, uint64_t id ) {
		Key key = self->keyForFileID( id );

		int pathLen = self->pathMinChars + g_random->randomInt(0, self->pathCharRange);
		std::string path = "";
		for(int i=0; i<pathLen; i+=4)
			path += format( format( "%%0%dx", std::min( pathLen - i, 4 ) ).c_str(), g_random->randomInt( 0, 0xFFFF ) );
		uint64_t userID = g_random->randomInt(0, self->userIDCount);
		int serverID = g_random->randomInt(0, self->serverCount);
		bool deleted = g_random->random01() < self->deletedFilesRatio;
		UID uid = g_random->randomUniqueID();
		double time = now();

		tr->set( key, path );
		std::string keyStr(key.toString());
		tr->set( keyStr + "/size", format( "%d", g_random->randomInt( 0, 2 << 30 ) ) );
		tr->set( keyStr + "/server", format( "%d", g_random->randomInt( 0, self->serverCount ) ) );
		tr->set( keyStr + "/deleted", deleted ? LiteralStringRef("1") : LiteralStringRef("0") );
		tr->set( keyStr + "/server", format( "%d", serverID ) );
		tr->set( keyStr + "/created", doubleToTestKey( time ) );
		tr->set( keyStr + "/lastupdated", doubleToTestKey( time ) );
		tr->set( keyStr + "/userid", format( "%016llx", userID ) );

		if( deleted )
			tr->set( format( "/files/server/%08x/deleted/%016llx", serverID, id ), doubleToTestKey( time ) );
		tr->set( format( "/files/user/%016llx/updated/%016llx/%016llx", userID, *(uint64_t*)&time, id ), path );
		tr->set( format( "/files/user/%016llx/path/", userID ) + path, format( "%016llx", id ) );
		// This index was not specified in the original test: it removes duplicated paths
		tr->set( "/files/path/" + path, format( "%016llx", id ) );
	}

	ACTOR Future<Void> setupRange( Database cx, FileSystemWorkload* self, int begin, int end ) {
		state Transaction tr(cx);
		while (true) {
			try {
				Optional<Value> f = wait( tr.get( self->keyForFileID( begin ) ) );
				if (f.present()) break;  // The transaction already completed!

				for(int n=begin; n<end; n++)
					self->initializeFile( &tr, self, n );
				Void _ = wait( tr.commit() );
				break;
			} catch (Error& e) {
				Void _ = wait( tr.onError(e) );
			}
		}
		return Void();
	}

	ACTOR Future<Void> nodeSetup( Database cx, FileSystemWorkload* self ) {
		state int i;
		state vector<int> order;
		state int nodesToSetUp = self->fileCount / self->clientCount + 1;
		state int startingNode = nodesToSetUp * self->clientId;
		state int batchCount = 5;
		for(int o=0; o<=nodesToSetUp / batchCount; o++) order.push_back(o*batchCount);
		g_random->randomShuffle(order);
		for(i=0; i<order.size(); ) {
			vector<Future<Void>> fs;
			for(int j=0; j<100 && i<order.size(); j++) {
				fs.push_back( self->setupRange(
						cx, self, startingNode + order[i], std::min(startingNode+order[i]+batchCount, nodesToSetUp * (self->clientId+1)) ) );
				i++;
			}
			Void _ = wait( waitForAll(fs) );
		}
		TraceEvent("FileSetupOK").detail("ClientIdx", self->clientId).detail("ClientCount", self->clientCount)
			.detail("StartingFile", startingNode).detail("FilesToSetUp", nodesToSetUp);
		return Void();
	}

	ACTOR Future<Void> _start( Database cx, FileSystemWorkload *self ) {
		state FileSystemOp *operation;
		if( self->operationName == "deletionQuery" )
			operation = new ServerDeletionCountQuery();
		else
			operation = new RecentModificationQuery();
		Void _ = wait( timeout( self->operationClient( cx, self, operation, 0.01 ), 1.0, Void() ) );
		self->queries.clear();
		self->writes.clear();

		if( self->performingWrites ) {
			for(int c=0; c<self->writeActorCount; c++) {
				self->clients.push_back( timeout( self->writeClient( cx, self ), self->testDuration, Void() ) );
			}
		}
		for(int c=0; c<self->actorCount; c++) {
			self->clients.push_back( timeout(
				self->operationClient( cx, self, operation, self->actorCount / self->transactionsPerSecond ),
				self->testDuration, Void() ) );
		}
		Void _ = wait( waitForAll( self->clients ) );

		Void _ = wait( delay( 0.01 ) );  // Make sure the deletion happens after actor cancellation
		delete operation;
		return Void();
	}

	bool shouldRecord( double clientBegin ) {
		double n = now();
		return !discardEdgeMeasurements ||
			(n > (clientBegin + testDuration * 0.125) && n < (clientBegin + testDuration * 0.875));
	}

	ACTOR Future<Void> operationClient(
		Database cx, FileSystemWorkload *self, FileSystemOp *operation, double delay )
	{
		state double clientBegin = now();
		state double lastTime = now();
		loop {
			Void _ = wait( poisson( &lastTime, delay ) );
			state double tstart = now();
			state Transaction tr(cx);
			loop {
				try {
					Optional<Version> ver = wait( operation->run( self, &tr ) );
					if( ver.present() )
						break;
				} catch (Error& e) {
					Void _ = wait( tr.onError(e) );
				}
			}
			if( self->shouldRecord( clientBegin ) ) {
				++self->queries;
				double latency = now() - tstart;
				self->latencies.addSample( latency );
			}
		}
	}

	static int testKeyToInt( const KeyRef& p ) {
		int x = 0;
		sscanf( p.toString().c_str(), "%d", &x );
		return x;
	}

	ACTOR Future<Void> writeClient(
		Database cx, FileSystemWorkload *self )
	{
		state double clientBegin = now();
		loop {
			state int fileID = g_random->randomInt(0, self->fileCount);
			state bool isDeleting = g_random->random01() < 0.25;
			state int size = isDeleting ? 0 : g_random->randomInt( 0, 2 << 30 );
			state std::string keyStr = self->keyForFileID( fileID ).toString();
			state double tstart = now();
			state Transaction tr(cx);
			loop {
				try {
					state double time = now();
					if( isDeleting ) {
						state Optional<Value> deleted = wait( tr.get( StringRef( keyStr + "/deleted") ) );
						ASSERT( deleted.present() );
						Optional<Value> serverStr = wait( tr.get( StringRef(keyStr + "/server") ) );
						ASSERT( serverStr.present() );
						int serverID = testKeyToInt( serverStr.get() );
						if( deleted.get().toString() == "1" ) {
							tr.set( keyStr + "/deleted", LiteralStringRef("0") );
							tr.clear( format( "/files/server/%08x/deleted/%016llx", serverID, fileID ) );
						} else {
							tr.set( keyStr + "/deleted", LiteralStringRef("1") );
							tr.set( format( "/files/server/%08x/deleted/%016llx", serverID, fileID ), doubleToTestKey( time ) );
						}
					} else {
						tr.set( keyStr + "/size", format( "%d", size ) );
					}
					tr.set( keyStr + "/lastupdated", doubleToTestKey( time ) );
					Void _ = wait( tr.commit() );
					break;
				} catch (Error& e) {
					Void _ = wait( tr.onError(e) );
				}
			}
			if( self->shouldRecord( clientBegin ) ) {
				++self->writes;
				self->writeLatencies.addSample( now() - tstart );
			}
		}
	}

	ACTOR Future<Optional<Version>> modificationQuery( FileSystemWorkload *self, Transaction *tr ) {
		state uint64_t userID = g_random->randomInt(0, self->userIDCount);
		state std::string base = format("/files/user/%016llx", userID);
		if( self->loggingQueries )
			TraceEvent("UserQuery").detail("UserID", userID).detail("PathBase", base);
		Key keyEnd(base + "/updated0");
		Standalone<RangeResultRef> val = wait( tr->getRange(
				firstGreaterOrEqual( keyEnd ) - 10, firstGreaterOrEqual( keyEnd ), 10 ) );
		Key keyBegin(base + "/updated/");
		for(int i=val.size() - 1; i >= 0; i--) {
			if( val[i].key.startsWith( keyBegin ) && self->loggingQueries ) {
				TraceEvent("UserQueryResults").detail("UserID", userID)
					.detail("PathBase", base).detail("LastModified", printable( val[i].key.substr(54) ));
				break;
			}
		}
		return Optional<Version>(Version(0));
	}

	ACTOR Future<Optional<Version>> deletionQuery( FileSystemWorkload *self, Transaction *tr ) {
		state uint64_t serverID = g_random->randomInt(0, self->serverCount);
		state std::string base = format("/files/server/%08x/deleted", serverID);
		if( self->loggingQueries )
			TraceEvent("DeletionQuery").detail("ServerID", serverID).detail("PathBase", base);
		state Key keyBegin(base + "/");
		state Key keyEnd(base + "0");
		state KeySelectorRef begin = firstGreaterThan( keyBegin );
		state KeySelectorRef end = firstGreaterOrEqual( keyEnd );
		state int transfered = 1000;
		state int transferSize = 1000;
		state uint64_t deletedFiles = 0;
		while( transfered == transferSize ) {
			Standalone<RangeResultRef> val = wait( tr->getRange( begin, end, transferSize ) );
			transfered = val.size();
			deletedFiles += transfered;
			begin = begin + transfered;
		}
		if( self->loggingQueries ) {
			TraceEvent("DeletionQueryResults").detail("serverID", serverID)
				.detail("PathBase", base).detail("DeletedFiles", deletedFiles);
		}
		return Optional<Version>(Version(0));
	}

	class RecentModificationQuery : public FileSystemOp {
		virtual Future<Optional<Version>> run( FileSystemWorkload *self, Transaction* tr ) {
			return self->modificationQuery( self, tr );
		}
		virtual const char* name() { return "RecentUserModifications"; }
	};

	class ServerDeletionCountQuery : public FileSystemOp {
		virtual Future<Optional<Version>> run( FileSystemWorkload *self, Transaction* tr ) {
			return self->deletionQuery( self, tr );
		}
		virtual const char* name() { return "ServerDeletions"; }
	};
};

WorkloadFactory<FileSystemWorkload> FileSystemWorkloadFactory("FileSystem");
