CWriteToBinaryThread.cpp

#include "CWriteToBinaryThread.h"
#include <QAbstractItemModel>
#include <QDataStream>
#include <QDateTime>
#include <QDebug>
#include "DataModel.h"
#include <QFile>
#include <QMutexLocker>
#include "DataEntry.h"
#include <QDir>

CWriteToBinaryThread::CWriteToBinaryThread(QAbstractItemModel* model, QQueue<DataEntry>* dataQueue, QObject* parent) :
	QThread(parent),
	m_model(model),
	m_dataQueue(dataQueue)
{
	
}

void CWriteToBinaryThread::start(qint64 num)
{
	QMutexLocker locker(&m_mutex);

	m_stop = false;

	m_num = num;

	QThread::start();
}

CWriteToBinaryThread::~CWriteToBinaryThread()
{

}

void CWriteToBinaryThread::saveDataToFile(QFile& file)
{
	QMutexLocker locker(&m_mutex);

	QDataStream out(&file);

	qint64 nStartTimeStamp = m_dataQueue->at(0).getTime();

	//TODO 设置文件区分间隔的时间单位
	qint64 nEndTimeStamp = getEndOfTimerstamp(nStartTimeStamp, EDEVIDE_TIMEUNIT::HOUR);

	DataEntry dataEntry;

	while(nStartTimeStamp <= nEndTimeStamp)
	{
		if (m_dataQueue->isEmpty())
		{
			return;
		}

		dataEntry = m_dataQueue->dequeue();

		nStartTimeStamp = dataEntry.getTime();

		out << dataEntry;

		m_num--;
		//TODO 判断测试数据的间隔时间来决定需要自增多少,如下以毫秒为间隔
	}
}

void CWriteToBinaryThread::readDataFromFile(QFile& file)
{
	QDataStream in(&file);

	QList<DataEntry> entries;  // 用于存储读取的所有数据条目

	//TODO optimize this

	while (!in.atEnd())
	{
		DataEntry entry;
		in >> entry;

		// 将读取的条目添加到列表中
		entries.append(entry);
	}

	m_stop = true;

	emit signalFinishReadData(entries);
}

qint64 CWriteToBinaryThread::getEndOfTimerstamp(qint64 currentTimestamp, EDEVIDE_TIMEUNIT divideUnit)
{
	QDateTime dateTime = QDateTime::fromMSecsSinceEpoch(currentTimestamp);

	// 获取当前分钟的年、月、日、小时和分钟值
	int year = dateTime.date().year();
	int month = dateTime.date().month();
	int day = dateTime.date().day();
	int hour = dateTime.time().hour();
	int minute = dateTime.time().minute();
	int second = dateTime.time().second();

	QDateTime endOfTime;

	switch (divideUnit)
	{
	case EDEVIDE_TIMEUNIT::YEAR:
		endOfTime = QDateTime(QDate(year, 12, 31), QTime(23, 59, 59, 999));
		break;
	case EDEVIDE_TIMEUNIT::MONTH:
	{
		QDate lastDayOfMonth(year, month, QDate(year, month, 1).daysInMonth());
		endOfTime = QDateTime(lastDayOfMonth, QTime(23, 59, 59, 999));
	}
	break;
	case EDEVIDE_TIMEUNIT::DAY:
		endOfTime = QDateTime(QDate(year, month, day), QTime(23, 59, 59, 999));
		break;
	case EDEVIDE_TIMEUNIT::HOUR:
		endOfTime = QDateTime(QDate(year, month, day), QTime(hour, 59, 59, 999));
		break;
	case EDEVIDE_TIMEUNIT::MINUTE:
		endOfTime = QDateTime(QDate(year, month, day), QTime(hour, minute, 59, 999));
		break;
	case EDEVIDE_TIMEUNIT::SECOND:
		endOfTime = QDateTime(QDate(year, month, day), QTime(hour, minute, second, 999));
		break;
	default:
		qWarning() << "未知的时间单位";
		return -1;  // 返回一个无效的时间戳
	}

	return endOfTime.toMSecsSinceEpoch();
}

void CWriteToBinaryThread::run()
{
	while(m_num)
	{
		if (m_stop)
		{
			break;
		}
#if 1
		qint64 nTime = m_dataQueue->at(0).getTime();
		QDateTime dateTime = QDateTime::fromMSecsSinceEpoch(nTime);

		// 按年月日构建目录路径
		QString yearPath = "data/" + dateTime.toString("yyyy");
		QString monthPath = yearPath + "/" + dateTime.toString("MM");
		QString dayPath = monthPath + "/" + dateTime.toString("dd");

		QString hourPath = dayPath + "/" + dateTime.toString("hh");

		// 构建文件名格式为yyyy-MM-dd-hh
		//QString fileName = dateTime.toString("yyyy-MM-dd-hh-mm") + ".bin";
		QString fileName = dateTime.toString("yyyy-MM-dd-hh") + ".bin";
		QString fullPath = hourPath + "/" + fileName;

		QDir dir;
		if (!dir.exists(hourPath)) {
			if (!dir.mkpath(hourPath)) {
				qWarning() << "无法创建目录:" << hourPath;
				break;
			}
		}

		QFile file(fullPath);

		if (file.open(QIODevice::WriteOnly | QIODevice::Append)) 
		{
			// Save all data to the file
			saveDataToFile(file);
		}
		file.close();

		m_stop = true;
#endif

#if 0
		//QString fileName = "data/2024-09-25-08-12.bin";

		QFile file(m_strFilePath);

		if (file.open(QIODevice::ReadOnly))
		{
			readDataFromFile(file);
		}
		file.close();

		m_stop = true;
#endif

	}

}

void CWriteToBinaryThread::stop()
{
	QMutexLocker locker(&m_mutex);

	m_stop = true;
}

void CWriteToBinaryThread::start()
{
	QMutexLocker locker(&m_mutex);

	m_stop = false;

	QThread::start();
}

bool CWriteToBinaryThread::isRunning() const
{
	return !m_stop;
}

void CWriteToBinaryThread::setFilePath(const QString& filePath)
{
	m_strFilePath = filePath;
}