TxByActionMethods.java

/**
 * Copyright (c) 2011-2013, James Zhan 詹波 (jfinal@126.com).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.tooan.ynpay.third.jfinal.plugin.activerecord.tx;

import net.tooan.ynpay.third.jfinal.aop.Interceptor;
import net.tooan.ynpay.third.jfinal.core.ActionInvocation;
import net.tooan.ynpay.third.jfinal.plugin.activerecord.Db;
import net.tooan.ynpay.third.jfinal.plugin.activerecord.IAtom;

import java.sql.SQLException;
import java.util.HashSet;
import java.util.Set;

/**
 * TxByActionMethods
 */
public class TxByActionMethods implements Interceptor {

    private Set<String> actionMethodSet = new HashSet<String>();

    public TxByActionMethods(String... actionMethods) {
        if (actionMethods == null || actionMethods.length == 0)
            throw new IllegalArgumentException("actionMethods can not be blank.");

        for (String actionMethod : actionMethods)
            actionMethodSet.add(actionMethod.trim());
    }

    public void intercept(final ActionInvocation ai) {
        if (actionMethodSet.contains(ai.getMethodName())) {
            Db.tx(new IAtom() {
                public boolean run() throws SQLException {
                    ai.invoke();
                    return true;
                }
            });
        } else {
            ai.invoke();
        }
    }
}