使用 C++23 协程实现第一个 co_await 同步风格调用接口--Qt计算文件哈希值

news2024/10/6 10:29:45

C++加入了协程 coroutine的特性，一直没有动手实现过。看了网上很多文章，已经了解了协程作为“可被中断和恢复的函数”的一系列特点。在学习过程中，我发现大多数网上的例子，要不就是在main()函数的控制台程序里演示yeild,await, resume的特性，要不就是讲述很多概念，很少有演示协程究竟如何把异步变成同步调用的。本次，我们就通过一个简单的计算文件哈希值的例子，来演示如何进行协程操作。

1. 原始的哈希值计算

假设存在一个最简单的哈希计算需求，要计算一个大文件的指纹。我们很容易实现一个演示算法：

void DlgCT::on_pushButton_normal_clicked(){
	QFile fp(filename);
	char buf[1024];
	unsigned long long hashfile = 0;
	if (fp.open(QIODevice::ReadOnly))
	{
		int rlen = fp.read(buf,1024);
		while (rlen>0)
		{
			for (int i=0;i<rlen;++i)
			{
				unsigned char c = hashfile>>56;
				hashfile <<=8;
				hashfile ^= (buf[i] ^ c );
			}
			rlen = fp.read(buf,1024);
			//假多线程，也可以看做是Qt的有栈协程，人为释放资源
			QCoreApplication::processEvents();
		}
	}
}

上面的代码，在文件比较大时，如果没有‘’QCoreApplication::processEvents();”显然会阻塞界面，导致按钮弹不起来，界面卡死。当然，可以通过适时调用QCoreApplication::processEvents();保持消息循环。这是一种假多线程，也可以看做是Qt的有栈协程，人为释放资源让给其他消息。

2. 异步计算改造

为了不阻塞主界面，传统上喜欢使用另一个线程来处理算法，并在完成后通知主线程。有一个处理类：

class fileDealer : public QObject
{
	Q_OBJECT
public:
	explicit fileDealer(QObject *parent = nullptr);
	//dealFile 计算哈希，存储在 result 里
	void dealFile(QString filename);
public:
	QByteArray result;
private:
	std::thread * m_pThread = nullptr;
signals:
	void sig_done();
};

void fileDealer::dealFile(QString filename)
{
	m_pThread = new std::thread([filename,this]()->void{		
		QFile fp(filename);
		char buf[1024];
		unsigned long long hashfile = 0;
		if (fp.open(QIODevice::ReadOnly))
		{			
			int rlen = fp.read(buf,1024);
			while (rlen>0)
			{
				for (int i=0;i<rlen;++i)
				{
					unsigned char c = hashfile>>56;
					hashfile <<=8;
					hashfile ^= (buf[i] ^ c );
				}				
				rlen = fp.read(buf,1024);
			}
		}
		emit sig_done();		
	});
}

这个类会开启一个独立的线程，做完后触发信号sig_done。上述代码是主干功能，相应的new,delete维护部分略去。如此一来，则需要在按钮响应函数里改造异步调用：



void DlgCT::on_pushButton_thread_clicked()
{
	fileDealer * dealer = new fileDealer(this);
	connect(dealer,&fileDealer::sig_done,[dealer,this]()->void{
		dealer->deleteLater();
	});
	dealer->dealFile(ui->lineEdit_file->text());
}

即可完成非阻塞处理。

3. 使用协程 co_await 同步风格编程

如果使用C++协程，当然希望直接可以实现同步风格的异步调用：

void DlgCT::on_pushButton_file_clicked()
{
	dealFile(ui->lineEdit_file->text());
}
FileTask DlgCT::dealFile(QString filename)
{
	QByteArray res = co_await awDealFile(filename);
	//注意！若协程库开发不周到，此时有可能已经不是在主界面线程了！一定注意操作界面控件的线程安全性。
	showMsg(res);
}

在 co_await 语句后，返回主消息循环，此时定时器等依旧顺利工作。直到文件计算完毕后，才返回 showMsg(res);。为了达到上述效果，需要如下两步骤：

3.1 添加协程代码

首先，添加协程返回对象结构体. 本示例只使用 co_await关键词，所以大部分的必备函数入口都是默认值，啥也不做。

/*!
 * \brief The FileTask class	协程结构体
 */
struct FileTask
{
	struct promise_type;
	using handle_type = std::coroutine_handle<promise_type>;
	FileTask(handle_type h)
	{}
	FileTask(FileTask&& s)
	{}

	struct promise_type {
		promise_type() = default;
		~promise_type() = default;
		auto get_return_object() noexcept {
			return FileTask{handle_type::from_promise(*this)};
		}
		auto initial_suspend() noexcept {
			//一创建立刻执行
			return std::suspend_never{};
		}
		auto final_suspend() noexcept {
			return std::suspend_always{};
		}
		void unhandled_exception() {
			exit(1);
		}
		void return_void()
		{}
	};

};

3.2 创建 await 辅助类

关键实现await功能的就是下面这个类：

/*!
 * \brief The awDealFile class	协程 await 对象
 */
class awDealFile : public QObject
{
	Q_OBJECT
public:
	awDealFile(QString filename, QObject *parent = nullptr)
		:QObject(parent)
		,m_fn(filename)
		,m_pDealer(new fileDealer)
	{
		//处理完毕的信号，会在处理线程里发出，所以用QueuedConnection确保协程返回时，保持线程不变。
		QObject::connect(m_pDealer,&fileDealer::sig_done,this, &awDealFile::slot_done,Qt::QueuedConnection);

	}
	~awDealFile()
	{
			if (m_pDealer)
			m_pDealer->deleteLater();
		m_pDealer = nullptr;
	}
	bool await_ready() {	return false;	}
	/*!
	 * \brief await_resume	这个函数的返回值决定了 await 关键词可以返回什么类型的东西
	 * \return 哈希结果
	 */
	QByteArray await_resume() {
		return m_pDealer->result;
	}
	/*!
	 * \brief await_suspend	co_await 时，会调用这个函数。此时，启动处理，并在处理完毕后resume
	 * \param h
	 */
	void await_suspend(FileTask::handle_type h) {
		hd = h;
		//处理
		m_pDealer->dealFile(m_fn);
	}
private slots:
	void slot_done()
	{
		if (hd)	hd.resume();
	}
private:
	QString m_fn;
	fileDealer * m_pDealer = nullptr;
	FileTask::handle_type hd;
};

有了上述代码，则可实现同步调用。

4. 关于线程切换的风险

协程的co_await 实际上提供了一个无栈的暂停-恢复框架。关键是要在确保处理完毕后，及时调用 resume 恢复执行。值得注意的是，对于从一个 std::thread内直接 resume的方法，会导致线程切换！此行为务必引起重视。在哪个线程调用的resume，协程函数恢复后，就回到哪个线程。这对操作GUI控件的代码带来了隐晦的风险！

可以看到，在例子里使用Qt的跨线程队列槽 (Qt::QueuedConnection)确保恢复后的协程执行序依旧位于主线程。虽然在实验中，多线程操作控件似乎也没有报错，但这不是推荐的控件操作方法。

	//处理完毕的信号，会在处理线程里发出，所以用QueuedConnection确保协程返回时，保持线程不变。
		QObject::connect(m_pDealer,&fileDealer::sig_done,this, &awDealFile::slot_done,Qt::QueuedConnection);

   void slot_done()
	{
		if (hd)	hd.resume();
	}