com.thinkbiganalytics.metadata.config.OperationalMetadataTransactionTemplateMetadataAccess Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kylo-operational-metadata-jpa Show documentation
There is a newer version: 0.10.0
Show newest version
/**
 *
 */
package com.thinkbiganalytics.metadata.config;

/*-
 * #%L
 * thinkbig-operational-metadata-jpa
 * %%
 * Copyright (C) 2017 ThinkBig Analytics
 * %%
 * 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.
 * #L%
 */


import com.thinkbiganalytics.metadata.api.MetadataAccess;
import com.thinkbiganalytics.metadata.api.MetadataAction;
import com.thinkbiganalytics.metadata.api.MetadataCommand;
import com.thinkbiganalytics.metadata.api.MetadataExecutionException;
import com.thinkbiganalytics.metadata.api.MetadataRollbackAction;
import com.thinkbiganalytics.metadata.api.MetadataRollbackCommand;

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.TransactionCallback;
import org.springframework.transaction.support.TransactionTemplate;

import java.security.Principal;

import javax.inject.Inject;

/**
 * Transaction template for querying and committing to the database
 */
public class OperationalMetadataTransactionTemplateMetadataAccess implements MetadataAccess {

    private TransactionTemplate template;

    /**
     * Set the Transaction manager, wiring in the one configured with Hibernate
     */
    @Inject
    public void setTransactionManager(@Qualifier("operationalMetadataTransactionManager") PlatformTransactionManager transactionMgr) {
        this.template = new TransactionTemplate(transactionMgr);
    }


    /**
     * Perform a command and commit the transaction
     *
     * @param cmd        the command to execute
     * @param principals one or more principals, or none to use the current security context
     * @return an object resulting from the commit
     */
    @Override
    public  R commit(MetadataCommand cmd, Principal... principals) {
        return commit(cmd);
    }

    /**
     * Perform a command and commit the transaction
     * Rollback is not supported here
     *
     * @param cmd         the command to execute
     * @param rollbackCmd the command to execute if an exception occurs and the transaction is rolled back
     * @param principals  one or more principals, or none to use the current security context
     */
    @Override
    public  R commit(MetadataCommand cmd, MetadataRollbackCommand rollbackCmd, Principal... principals) {
        // TODO Rollback command currently not supported
        return commit(cmd);
    }

    /**
     * Perform a command and commit the transaction
     *
     * @param action     the command to execute
     * @param principals one or more principals, or none to use the current security context
     */
    @Override
    public void commit(MetadataAction action, Principal... principals) {
        commit(action);
    }

    /**
     * Perform a command and commit the transaction
     *
     * @param action     the command to execute
     * @param principals one or more principals, or none to use the current security context
     */
    @Override
    public void commit(MetadataAction action, MetadataRollbackAction rollbackAction, Principal... principals) {
        // TODO Rollback command currently not supported
        commit(action);
    }


    /**
     * Perform a read only action
     *
     * @param cmd        the command to execute
     * @param principals one or more principals, or none to use the current security context
     */
    @Override
    public  R read(MetadataCommand cmd, Principal... principals) {
        return read(cmd);
    }


    /**
     * Perform a read only action
     *
     * @param cmd        the command to execute
     * @param principals one or more principals, or none to use the current security context
     */
    @Override
    public void read(MetadataAction cmd, Principal... principals) {
        read(cmd);
    }

    /**
     * Perform a command and commit the transaction
     */
    protected  R commit(MetadataCommand cmd) {
        return template.execute(createCallback(cmd, false));
    }

    /**
     * Perform a read only command
     */
    protected  R read(MetadataCommand cmd) {
        return template.execute(createCallback(cmd, true));
    }

    /**
     * Perform a command and commit the transaction
     */
    protected void commit(MetadataAction action) {
        template.execute(createCallback(action, false));
    }

    /**
     * Perform a read only command
     */
    protected void read(MetadataAction action) {
        template.execute(createCallback(action, true));
    }

    /**
     * Return the callback after executing the passed in action.
     */
    private TransactionCallback