user
73 lines
2.3 KiB
C++
73 lines
2.3 KiB
C++
#ifndef BASE_GENERATOR_H
|
|
#define BASE_GENERATOR_H
|
|
|
|
#include "generator_interface.h"
|
|
#include "range_limiter.h"
|
|
#include <QDateTime>
|
|
#include <QRandomGenerator>
|
|
|
|
class BaseGenerator : public GeneratorInterface, public virtual RangeLimiter {
|
|
private:
|
|
QString entityName_;
|
|
int lastPercentage_;
|
|
mutable int fieldWidth_;
|
|
qsizetype createdRecords_;
|
|
qsizetype maximumRecords_;
|
|
|
|
public:
|
|
ShouldLogSuccess defaultShouldLog(int percent, int seconds) const noexcept;
|
|
ShouldLogSuccess shouldLogAllSuccesses() const noexcept;
|
|
ShouldLogSuccess shouldLogEveryNPercent(int percent) const noexcept;
|
|
ShouldLogSuccess shouldLogEveryNSeconds(
|
|
int seconds, const ShouldLogSuccess &performOtherwise) const noexcept;
|
|
ShouldLogSuccess shouldLogFirstAndLastPercent() const noexcept;
|
|
|
|
public:
|
|
// Virtual methods
|
|
void generateAll(const ShouldLogSuccess &shouldLogSuccess) final;
|
|
|
|
void generateByFillRatio(double fillRatio,
|
|
const ShouldLogSuccess &shouldLogSuccess) final;
|
|
|
|
void generateByRecordCount(qint64 totalRecordsToCreate,
|
|
const ShouldLogSuccess &shouldLogSuccess) final;
|
|
|
|
protected:
|
|
inline BaseGenerator(const QString &entityName) noexcept;
|
|
|
|
// General methods
|
|
inline QString formatDateTime(const QDateTime &dateTime_) const noexcept;
|
|
inline qsizetype createdRecords() const noexcept;
|
|
|
|
private: // create* methods depends on clearStaleData and createRecord
|
|
QString createFailureMessage(const QString &errorMessage) const override;
|
|
QString createSuccessMessage() const override;
|
|
QString createTemplateMessage() const override;
|
|
|
|
// Non-virtual method
|
|
void generateImplementation(std::function<bool()> shouldContinue,
|
|
std::function<bool()> shouldLogSuccess);
|
|
|
|
void updatePercentage(); // Depends on createRecord
|
|
|
|
protected: // Non-const methods
|
|
virtual RepositoryResult createRecord() = 0;
|
|
virtual bool prepareData() = 0;
|
|
virtual void clearStaleData() = 0;
|
|
};
|
|
|
|
BaseGenerator::BaseGenerator(const QString &entityName) noexcept
|
|
: createdRecords_(0), entityName_(entityName), fieldWidth_(0),
|
|
lastPercentage_(-1) {}
|
|
|
|
inline qsizetype BaseGenerator::createdRecords() const noexcept {
|
|
return createdRecords_;
|
|
}
|
|
|
|
QString
|
|
BaseGenerator::formatDateTime(const QDateTime &dateTime_) const noexcept {
|
|
return dateTime_.toString("yyyy-MM-dd'T'HH:mm:ss.000");
|
|
}
|
|
|
|
#endif
|