email-tracker/external/duckdb/third_party/tpce-tool/include/main/TxnHarnessTradeLookup.h

/*
 * Legal Notice
 *
 * This document and associated source code (the "Work") is a part of a
 * benchmark specification maintained by the TPC.
 *
 * The TPC reserves all right, title, and interest to the Work as provided
 * under U.S. and international laws, including without limitation all patent
 * and trademark rights therein.
 *
 * No Warranty
 *
 * 1.1 TO THE MAXIMUM EXTENT PERMITTED BY APPLICABLE LAW, THE INFORMATION
 *     CONTAINED HEREIN IS PROVIDED "AS IS" AND WITH ALL FAULTS, AND THE
 *     AUTHORS AND DEVELOPERS OF THE WORK HEREBY DISCLAIM ALL OTHER
 *     WARRANTIES AND CONDITIONS, EITHER EXPRESS, IMPLIED OR STATUTORY,
 *     INCLUDING, BUT NOT LIMITED TO, ANY (IF ANY) IMPLIED WARRANTIES,
 *     DUTIES OR CONDITIONS OF MERCHANTABILITY, OF FITNESS FOR A PARTICULAR
 *     PURPOSE, OF ACCURACY OR COMPLETENESS OF RESPONSES, OF RESULTS, OF
 *     WORKMANLIKE EFFORT, OF LACK OF VIRUSES, AND OF LACK OF NEGLIGENCE.
 *     ALSO, THERE IS NO WARRANTY OR CONDITION OF TITLE, QUIET ENJOYMENT,
 *     QUIET POSSESSION, CORRESPONDENCE TO DESCRIPTION OR NON-INFRINGEMENT
 *     WITH REGARD TO THE WORK.
 * 1.2 IN NO EVENT WILL ANY AUTHOR OR DEVELOPER OF THE WORK BE LIABLE TO
 *     ANY OTHER PARTY FOR ANY DAMAGES, INCLUDING BUT NOT LIMITED TO THE
 *     COST OF PROCURING SUBSTITUTE GOODS OR SERVICES, LOST PROFITS, LOSS
 *     OF USE, LOSS OF DATA, OR ANY INCIDENTAL, CONSEQUENTIAL, DIRECT,
 *     INDIRECT, OR SPECIAL DAMAGES WHETHER UNDER CONTRACT, TORT, WARRANTY,
 *     OR OTHERWISE, ARISING IN ANY WAY OUT OF THIS OR ANY OTHER AGREEMENT
 *     RELATING TO THE WORK, WHETHER OR NOT SUCH AUTHOR OR DEVELOPER HAD
 *     ADVANCE NOTICE OF THE POSSIBILITY OF SUCH DAMAGES.
 *
 * Contributors
 * - Sergey Vasilevskiy
 */

#ifndef TXN_HARNESS_TRADE_LOOKUP_H
#define TXN_HARNESS_TRADE_LOOKUP_H

#include "TxnHarnessDBInterface.h"

namespace TPCE {

class CTradeLookup {
	CTradeLookupDBInterface *m_db;

public:
	CTradeLookup(CTradeLookupDBInterface *pDB) : m_db(pDB){};

	void DoTxn(PTradeLookupTxnInput pTxnInput, PTradeLookupTxnOutput pTxnOutput) {
		TXN_HARNESS_SET_STATUS_SUCCESS;

		switch (pTxnInput->frame_to_execute) {
		case 1: {
			// Initialize
			TTradeLookupFrame1Input Frame1Input;
			TTradeLookupFrame1Output Frame1Output;

			// Copy Frame 1 Input
			Frame1Input.max_trades = pTxnInput->max_trades;
			memcpy(Frame1Input.trade_id, pTxnInput->trade_id, sizeof(Frame1Input.trade_id));

			// Execute Frame 1
			m_db->DoTradeLookupFrame1(&Frame1Input, &Frame1Output);

			// Validate Frame 1 Output
			if (Frame1Output.num_found != pTxnInput->max_trades) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF1_ERROR1);
			}

			// Copy Frame 1 Output
			pTxnOutput->frame_executed = 1;
			for (int i = 0; i < Frame1Output.num_found && i < TradeLookupFrame1MaxRows; i++) {
				pTxnOutput->is_cash[i] = Frame1Output.trade_info[i].is_cash;
				pTxnOutput->is_market[i] = Frame1Output.trade_info[i].is_market;
			}
			pTxnOutput->num_found = Frame1Output.num_found;

			break;
		}

		case 2: {
			// Initialize
			TTradeLookupFrame2Input Frame2Input;
			TTradeLookupFrame2Output Frame2Output;

			// Copy Frame 2 Input
			Frame2Input.acct_id = pTxnInput->acct_id;
			Frame2Input.max_trades = pTxnInput->max_trades;
			Frame2Input.start_trade_dts = pTxnInput->start_trade_dts;
			Frame2Input.end_trade_dts = pTxnInput->end_trade_dts;

			// Execute Frame 2
			m_db->DoTradeLookupFrame2(&Frame2Input, &Frame2Output);

			// Validate Frame 2 Output
			if (Frame2Output.num_found < 0 || Frame2Output.num_found > Frame2Input.max_trades) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF2_ERROR1);
			}
			if (Frame2Output.num_found == 0) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF2_WARN1);
			}

			// Copy Frame 2 Output
			pTxnOutput->frame_executed = 2;
			for (int i = 0; i < Frame2Output.num_found && i < TradeLookupFrame2MaxRows; i++) {
				pTxnOutput->is_cash[i] = Frame2Output.trade_info[i].is_cash;
				pTxnOutput->trade_list[i] = Frame2Output.trade_info[i].trade_id;
			}
			pTxnOutput->num_found = Frame2Output.num_found;

			break;
		}

		case 3: {
			// Initialize
			TTradeLookupFrame3Input Frame3Input;
			TTradeLookupFrame3Output Frame3Output;

			// Copy Frame 3 Input
			Frame3Input.max_trades = pTxnInput->max_trades;
			strncpy(Frame3Input.symbol, pTxnInput->symbol, sizeof(Frame3Input.symbol));
			Frame3Input.start_trade_dts = pTxnInput->start_trade_dts;
			Frame3Input.end_trade_dts = pTxnInput->end_trade_dts;
			Frame3Input.max_acct_id = pTxnInput->max_acct_id;

			// Execute Frame 3
			m_db->DoTradeLookupFrame3(&Frame3Input, &Frame3Output);

			// Validate Frame 3 Output
			if (Frame3Output.num_found < 0 || Frame3Output.num_found > Frame3Input.max_trades) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF3_ERROR1);
			}
			if (Frame3Output.num_found == 0) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF3_WARN1);
			}

			// Copy Frame 3 Output
			pTxnOutput->frame_executed = 3;
			pTxnOutput->num_found = Frame3Output.num_found;
			for (int i = 0; i < Frame3Output.num_found && i < TradeLookupFrame3MaxRows; i++) {
				pTxnOutput->is_cash[i] = Frame3Output.trade_info[i].is_cash;
				pTxnOutput->trade_list[i] = Frame3Output.trade_info[i].trade_id;
			}

			break;
		}

		case 4: {
			// Initialize
			TTradeLookupFrame4Input Frame4Input;
			TTradeLookupFrame4Output Frame4Output;

			// Copy Frame 4 Input
			Frame4Input.acct_id = pTxnInput->acct_id;
			Frame4Input.trade_dts = pTxnInput->start_trade_dts;

			// Execute Frame 4
			m_db->DoTradeLookupFrame4(&Frame4Input, &Frame4Output);

			// Validate Frame 4 Output
			if (Frame4Output.num_trades_found == 0) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF4_WARN1);
			} else if (Frame4Output.num_trades_found != 1) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF4_ERROR1);
			} else if (Frame4Output.num_found < 1 || Frame4Output.num_found > 20) {
				TXN_HARNESS_PROPAGATE_STATUS(CBaseTxnErr::TLF4_ERROR2);
			}

			// Copy Frame 4 Output
			pTxnOutput->frame_executed = 4;
			pTxnOutput->num_found = Frame4Output.num_found;
			pTxnOutput->trade_list[0] = Frame4Output.trade_id;

			break;
		}

		default: {
			// should never get here.
			assert(false);
			break;
		}
		}
	}
};

} // namespace TPCE

#endif // TXN_HARNESS_TRADE_LOOKUP_H