C++ 팁 - transaction 처리

CUJ 2004 년 1월호에 재밌는 글이 있어 소개합니다.

constructor 와 destructor를 이용한 scoped lock, trace등의 기술은 많이
익숙한데, 여기에 소개하는 것은 transaction 처리를 자동으로 하도록 하는
방법입니다.

macro, for 문의 절묘한 사용, temporary object, function template 등의
특성등을 잘 조합하여 직관적인 방법이 만들어 지는 군요.

scoped_transaction.h

//
// from CUJ Jan. 2004
//
class ScopedTransactionBase
{
public:
	ScopedTransactionBase() 
		: aborted_( true ) {}

	operator bool() const
	{ 
		return aborted_; 
	}
	void finish() const
	{
		aborted_ = false;
	}
	mutable bool aborted_;
};

template<class TransactionT>
class ScopedTransaction : public ScopedTransactionBase
{
public:
	ScopedTransaction( TransactionT& tran ) :
		tran_( tran )
	{}

	~ScopedTransaction()
	{
		if ( aborted_ ) {
			tran_.rollback();
		} else {
			tran_.commit();
		}
	}
private:
	TransactionT& tran_;
};


template< class TranT >
ScopedTransaction< TranT> makeScopedTransaction( TranT& tran )
{
	return ScopedTransaction<TranT>(tran);
}

#define transaction(tran) for ( const ScopedTransactionBase& tran_##__LINE__ =	\
									makeScopedTransaction( tran );		\
								tran_##__LINE__;							\
								tran_##__LINE__.finish() ) 

다음은 예제 입니다.
Session classs는 commit() 과 rollback()을 멤버 함수로 가지고 있는 임의의 class가 될 수 있으며,

transaction(sess)
{
//
// do something
//
}

위에서 transaction block 이 정상적으로 종료되면 sess의
commit() 이 호출되고 그렇지 않으면( break, return, exception등에 의해) rollback()이 호출 됩니다.

#include <iostream>
#include "scoped_transaction.h"

using namespace std;

//
// example
//
class AnySession
{
public:
	void commit() 
	{
		cout << "commit()" << endl;
	}

	void rollback() 
	{
		cout << "rollback()" << endl;
	}
};

int main()
{
	AnySession sess;

	cout << "1. normal success" << endl;
	transaction( sess ) 
	{
		cout << "do something" << endl;
	}

	cout << "2. fail in processing" << endl;
	transaction( sess ) 
	{
		cout << "do domething" << endl;

		cout << "do domething(which failed)" << endl;
		break;
	}
}

결과

1. normal success
do something
commit()
2. fail in processing
do domething
do domething(which failed)
rollback()