/*
 * LogMetrics.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/SystemMonitor.h"
#include "fdbclient/NativeAPI.h"
#include "fdbserver/TesterInterface.h"
#include "fdbserver/workloads/workloads.h"
#include "fdbrpc/simulator.h"
#include "fdbserver/MasterInterface.h"
#include "fdbclient/SystemData.h"
#include "fdbserver/WorkerInterface.h"
#include "fdbserver/QuietDatabase.h"
#include "fdbserver/ServerDBInfo.h"
#include "flow/actorcompiler.h"  // This must be the last #include.

struct LogMetricsWorkload : TestWorkload {
	std::string dataFolder;
	double logAt, logDuration, logsPerSecond;

	LogMetricsWorkload( WorkloadContext const& wcx )
		: TestWorkload(wcx)
	{
		logAt = getOption( options, LiteralStringRef("logAt"), 0.0 );
		logDuration = getOption( options, LiteralStringRef("logDuration"), 30.0 );
		logsPerSecond = getOption( options, LiteralStringRef("logsPerSecond"), 20 );
		dataFolder = getOption( options, LiteralStringRef("dataFolder"), LiteralStringRef("") ).toString();
	}

	virtual std::string description() { return "LogMetricsWorkload"; }
	virtual Future<Void> setup( Database const& cx ) { return Void(); }
	virtual Future<Void> start( Database const& cx ) {
		if(clientId)
			return Void();
		return _start( cx, this );
	}

	ACTOR Future<Void> setSystemRate( LogMetricsWorkload *self, Database cx, uint32_t rate ) {
		// set worker interval and ss interval
		state BinaryWriter br(Unversioned());
		vector<std::pair<WorkerInterface, ProcessClass>> workers = wait( getWorkers( self->dbInfo ) );
		//vector<Future<Void>> replies;
		TraceEvent("RateChangeTrigger");
		SetMetricsLogRateRequest req(rate);
		for(int i = 0; i < workers.size(); i++) {
			workers[i].first.setMetricsRate.send( req );
		}
		//wait( waitForAll( replies ) );

		br << rate;
		loop {
			state Transaction tr(cx);
			try {
				Version v = wait( tr.getReadVersion() );
				tr.set(fastLoggingEnabled, br.toStringRef());
				tr.makeSelfConflicting();
				wait( tr.commit() );
				break;
			} catch(Error& e) {
				wait( tr.onError(e) );
			}
		}

		return Void();
	}

	ACTOR Future<Void> _start( Database cx, LogMetricsWorkload *self ) {
		wait( delay( self->logAt ) );

		wait( self->setSystemRate( self, cx, self->logsPerSecond ) );
		wait( timeout( recurring( &systemMonitor, 1.0 / self->logsPerSecond ), self->logDuration, Void() ) );

		// We're done, set everything back
		wait( self->setSystemRate( self, cx, 1.0 ) );

		return Void();
	}

	virtual Future<bool> check( Database const& cx ) { return true; }
	virtual void getMetrics( vector<PerfMetric>& m ) {
	}
};

WorkloadFactory<LogMetricsWorkload> LogMetricsWorkloadFactory("LogMetrics");