使用类似闭包的方式封装事务

发表于 | 分类于

在控制器中执行事务的时候，一般的代码如下：

$transaction = Yii::$app->db->beginTransaction();
try {
    //一些业务代码
    $transaction->commit();
} catch (\Exception $e) {
    $transaction->rollBack();
    throw $e;
}

于是我在想，这个代码结构，只有//一些业务代码 这一部分是不一样，却要重复很多遍，这一不是很冗余吗？ 而且 不！好！看！，于是我试着寻找解决方法，一开始在stackflow找到一个类似的提问，有方案是在model里做封装，但是这样做有一定问题，如产生嵌套事务等，有兴趣的可以点击这里查看该问答。

我们的Yii框架给出了一个方法transaction，乍一看好像不能解决传参的问题，我们先不管，往下看，该方法调用方式如下：

Yii::$app->db->transaction(function() {
    //一些业务代码
});

我们来看一下这个方法的源码

/**
 * Executes callback provided in a transaction.
 *
 * @param callable $callback a valid PHP callback that performs the job. Accepts connection instance as parameter.
 * @param string|null $isolationLevel The isolation level to use for this transaction.
 * See [[Transaction::begin()]] for details.
 * @throws \Exception|\Throwable if there is any exception during query. In this case the transaction will be rolled back.
 * @return mixed result of callback function
 */
public function transaction(callable $callback, $isolationLevel = null)
{
    $transaction = $this->beginTransaction($isolationLevel);
    $level = $transaction->level;

    try {
        $result = call_user_func($callback, $this);
        if ($transaction->isActive && $transaction->level === $level) {
            $transaction->commit();
        }
    } catch (\Exception $e) {
        $this->rollbackTransactionOnLevel($transaction, $level);
        throw $e;
    } catch (\Throwable $e) {
        $this->rollbackTransactionOnLevel($transaction, $level);
        throw $e;
    }

    return $result;
}

这个方法接受一个回调函数和事务的隔离级别，
从这里我们看出，这个方法虽然解决重复代码，却还有几个问题没有解决：
第一，这个方法抛出的异常我们需要在接收外面处理，我们不可能直接抛出，这样对客户端很不友好。
第二：没有记录日志的行为，即使出了问题也不容易排除。
第三：其实还是第一个问题，如果我们需要对每个异常做处理，在transaction方法外再嵌套一层try…catch…，那么和没有封装好像没什么区别？

根据方法可扩展不可修改的原则，我们应该在自己公共方法里对这个方法进行重载，重载代码如下：

public static function TransactionExecute(callable $function,$level=null)
{
    try{
        \Yii::$app->db->transaction($function,$level);
}catch (\Exception $e){
        //记录日志
        \Yii::error($e->getMessage());
        //这里可以理解成抛出自定义的异常类。
        (new self())->returnWayTip(1004, 'trans异常错误');
    }
}

然后回到如何传参的问题，我们可以使用闭包，贴一段伪代码，如下：

//执行事务
PublicFunction::TransactionExecute(function () use ($token_reward, $reward_info) {
        //业务代码
        $token_reward->save(0);
    MsgHelper::send($reward_info['post_id'], MsgHelper::SOMEONE_FINISH_REWARD, $reward_info);

    });

大功告成，代码看起来有没有更好看呢？

如有问题，欢迎指教。