Files
email-tracker/external/duckdb/third_party/tpce-tool/include/main/CE.h
2025-10-24 19:21:19 -05:00

209 lines
10 KiB
C++

/*
* 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
* - Doug Johnson
*/
/******************************************************************************
* Description: This class provides Customer Emulator functionality.
* It controls the mix of customer-initiated transactions
* and generates all required inputs for each of these
* transactions. These inputs are then made available to
* a sponsor provided callback interface to the SUT (see
* CESUTInterface.h). In addition, a set of constants used
* to uniquely identify each transaction type is exposed.
*
* The constructor for this class is overloaded. The first
* form of the constructor accepts 7 inputs, the last one
* of which is optional (i.e. it has a default value).
* - pSUT: a pointer to an instance of a sponsor provided
* subclassing of the CCESUTInterface class.
* - pLogger: a pointer to an
*instance of CEGenLogger or a sponsor provided subclassing of the CBaseLogger
*class.
* - inputFiles: a reference to an instance of the
* CInputFiles class containing all input files loaded
* into memory.
* - iCustomerCount: the total number of customers to
* emulate. C_IDs will be generated in the range of
* 1 to iCustomerCount.
* - iScaleFactor: the number of customers per tpsE. This
* should match the scale factor used at load time.
* - iDaysOfInitialTrades: the number of days of initial
* trades that was populated during load time.
* - RandomSeed: seed to be used for the RNG.
* - pTxnTunables: (optional). a pointer to a tuning
* structure used to configure the behavior of the CE
* class.
*
* The second form of the constructor accepts all of the
* same inputs as the first form. In addition, however,
* it accepts 3 additional inputs between iCustomerCount
* and iScaleFacto that facilitate partitioning by C_ID
* during runtime.
* - iMyStartingCustomerId: the starting customer ID for
* this instance's partition of C_IDs. This ID must be
* the first ID in a load unit.
* - iMyCustomerCount: the number of customers in this
* instance's partition. The number of customers specified
* must be an integral multiple of the load unit size.
* - iPartitionPercent: the percentage of C_IDs
* that should be generated within this instance's
* partition. 100-iParititionPercent percent of the C_IDs
* will be generated across the full range of C_IDs.
*
* Paritioning Example.
* Based on a load unit size of 1000, assume the following
* valid inputs.
* - iCustomerCount = 5000
* - iMyStartingCustomerID = 2001
* - iMyCustomerCount = 2000
* - iPartitionPercent = 40
* These setting will configure the CE to generated C_IDs
* as follows.
* - 40% of the time in the range [2001 - 4000]
* - 60% of the time in the range [1-5000]
*
* The CE provides the following entry point.
*
* - DoTxn: this entry point will select the next
* transaction type based on the mix settings, generate
* all required inputs for the selected transaction type,
* and provide those inputs to sponsor code at the
* appropriate callback interface.
*
******************************************************************************/
#ifndef CE_H
#define CE_H
#include "utilities/EGenUtilities_stdafx.h"
#include "CETxnInputGenerator.h"
#include "CETxnMixGenerator.h"
#include "CESUTInterface.h"
#include "BaseLogger.h"
#include "DriverParamSettings.h"
#include "input/DataFileManager.h"
namespace TPCE {
class CCE {
private:
CDriverGlobalSettings m_DriverGlobalSettings;
CDriverCESettings m_DriverCESettings;
CDriverCEPartitionSettings m_DriverCEPartitionSettings;
TDriverCETxnSettings m_DriverCETxnSettings;
CCESUTInterface *m_pSUT;
CBaseLogger *m_pLogger;
CCETxnMixGenerator m_TxnMixGenerator;
CCETxnInputGenerator m_TxnInputGenerator;
TBrokerVolumeTxnInput m_BrokerVolumeTxnInput;
TCustomerPositionTxnInput m_CustomerPositionTxnInput;
TMarketWatchTxnInput m_MarketWatchTxnInput;
TSecurityDetailTxnInput m_SecurityDetailTxnInput;
TTradeLookupTxnInput m_TradeLookupTxnInput;
TTradeOrderTxnInput m_TradeOrderTxnInput;
TTradeStatusTxnInput m_TradeStatusTxnInput;
TTradeUpdateTxnInput m_TradeUpdateTxnInput;
// Initialization that is common for all constructors.
void Initialize(PDriverCETxnSettings pTxnParamSettings);
// Automatically generate unique RNG seeds
void AutoSetRNGSeeds(UINT32 UniqueId);
/*
* Zero transaction input buffer.
*
* PARAMETERS:
* IN iTxnType - what transaction to zero the buffer for.
*
* RETURNS:
* none.
*/
void ZeroInputBuffer(int iTxnType);
public:
static const INT32 INVALID_TRANSACTION_TYPE = CCETxnMixGenerator::INVALID_TRANSACTION_TYPE;
static const INT32 SECURITY_DETAIL = CCETxnMixGenerator::SECURITY_DETAIL;
static const INT32 BROKER_VOLUME = CCETxnMixGenerator::BROKER_VOLUME;
static const INT32 CUSTOMER_POSITION = CCETxnMixGenerator::CUSTOMER_POSITION;
static const INT32 MARKET_WATCH = CCETxnMixGenerator::MARKET_WATCH;
static const INT32 TRADE_STATUS = CCETxnMixGenerator::TRADE_STATUS;
static const INT32 TRADE_LOOKUP = CCETxnMixGenerator::TRADE_LOOKUP;
static const INT32 TRADE_ORDER = CCETxnMixGenerator::TRADE_ORDER;
static const INT32 TRADE_UPDATE = CCETxnMixGenerator::TRADE_UPDATE;
// Trade-Result and Market-Feed are included for completness.
static const INT32 MARKET_FEED = CCETxnMixGenerator::MARKET_FEED;
static const INT32 TRADE_RESULT = CCETxnMixGenerator::TRADE_RESULT;
// Constructor - no partitioning by C_ID, automatic RNG seed generation
// (requires unique input)
CCE(CCESUTInterface *pSUT, CBaseLogger *pLogger, const DataFileManager &dfm, TIdent iConfiguredCustomerCount,
TIdent iActiveCustomerCount, INT32 iScaleFactor, INT32 iDaysOfInitialTrades, UINT32 UniqueId,
const PDriverCETxnSettings pParameterSettings = NULL);
// Constructor - no partitioning by C_ID, RNG seeds provided
CCE(CCESUTInterface *pSUT, CBaseLogger *pLogger, const DataFileManager &dfm, TIdent iConfiguredCustomerCount,
TIdent iActiveCustomerCount, INT32 iScaleFactor, INT32 iDaysOfInitialTrades, UINT32 UniqueId,
RNGSEED TxnMixRNGSeed, RNGSEED TxnInputRNGSeed, const PDriverCETxnSettings pParameterSettings = NULL);
// Constructor - partitioning by C_ID, automatic RNG seed generation
// (requires unique input)
CCE(CCESUTInterface *pSUT, CBaseLogger *pLogger, const DataFileManager &dfm, TIdent iConfiguredCustomerCount,
TIdent iActiveCustomerCount, TIdent iMyStartingCustomerId, TIdent iMyCustomerCount, INT32 iPartitionPercent,
INT32 iScaleFactor, INT32 iDaysOfInitialTrades, UINT32 UniqueId,
const PDriverCETxnSettings pParameterSettings = NULL);
// Constructor - partitioning by C_ID, RNG seeds provided
CCE(CCESUTInterface *pSUT, CBaseLogger *pLogger, const DataFileManager &dfm, TIdent iConfiguredCustomerCount,
TIdent iActiveCustomerCount, TIdent iMyStartingCustomerId, TIdent iMyCustomerCount, INT32 iPartitionPercent,
INT32 iScaleFactor, INT32 iDaysOfInitialTrades, UINT32 UniqueId, RNGSEED TxnMixRNGSeed, RNGSEED TxnInputRNGSeed,
const PDriverCETxnSettings pParameterSettings = NULL);
~CCE(void);
RNGSEED GetTxnInputGeneratorRNGSeed(void);
RNGSEED GetTxnMixGeneratorRNGSeed(void);
void SetTxnTunables(const PDriverCETxnSettings pTxnParamSettings);
void DoTxn(void);
};
} // namespace TPCE
#endif // CE_H