Yii2开发技巧 使用类似闭包的方式封装事务

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

$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);

    });

大功告成,代码看起来有没有更好看呢?

如有问题,欢迎指教。

相关推荐