Application: gvSIG desktop

/*

 * gvSIG. Desktop Geographic Information System.

 * 

 * Copyright (C) 2007-2020 gvSIG Association.

 * 

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 * 

 * You should have received a copy of the GNU General Public License 

 * along with this program. If not, see <https://www.gnu.org/licenses/>. 

 * 

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.dal.impl;

import java.util.ArrayList;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import java.util.UUID;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataTransaction;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.tools.dispose.Disposable;

import org.gvsig.tools.dispose.DisposeUtils;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class DefaultTransaction implements DataTransaction {

    private final String code;

    private Set<DataStore> stores;

    private boolean inProgress;

    private List<Disposable> disposables;

    public DefaultTransaction() {

        this.code = UUID.randomUUID().toString().replace("-", "");

        this.stores = new HashSet<>();

        this.disposables = new ArrayList<>();

        this.inProgress = false;

    }

    @Override

    public String getCode() {

        return this.code;

    }

    @Override

    public void begin() throws DataException {

        this.inProgress = true;

    }

    @Override

    public void commit() throws DataException {

        if( !this.isInProgress() ) {

            return;

        }

        for (DataStore store : stores) {

            if( store instanceof FeatureStore ) {

                FeatureStore fstore = (FeatureStore) store;

                if( fstore.isAppending() || fstore.isEditing() ) {

                    fstore.finishEditing();

                }

            }

        }

        this.inProgress = false;

    }

    @Override

    public void rollback() throws DataException {

        if( !this.isInProgress() ) {

            return;

        }

        for (DataStore store : stores) {

            if( store instanceof FeatureStore ) {

                FeatureStore fstore = (FeatureStore) store;

                if( fstore.isAppending() || fstore.isEditing() ) {

                    fstore.cancelEditing();

                }

            }

        }

        this.inProgress = false;

    }

    @Override

    public void rollbackQuietly() {

        try {

            this.rollback();

        } catch(Exception ex) {

            // Do nothing

        }

    }

    @Override

    public void add(DataStore store) {

//        if( !store.setTransaction(this) ) {

//            throw new IllegalStateException("Can't add store to transaction.");

//        }

        DisposeUtils.bind(store);

        this.stores.add(store);

    }

    @Override

    public void add(Disposable resource) throws DataException {

        this.disposables.add(resource);

    }

    @Override

    public void remove(DataStore store) {

        if( this.inProgress ) {

            throw new IllegalStateException("Can't remove store from a in progress transaction.");

        }

//        store.setTransaction(null);

        this.stores.remove(store);

        DisposeUtils.dispose(store);

    }

    @Override

    public boolean isInProgress() {

        return inProgress;

    }

    @Override

    public void dispose() {

        if( this.inProgress ) {

            this.rollbackQuietly();

        }

        for (DataStore store : stores) {

//            store.setTransaction(null);

            DisposeUtils.disposeQuietly(store);

        }

        for (Disposable resource : disposables) {

//            store.setTransaction(null);

            DisposeUtils.disposeQuietly(resource);

        }

        this.disposables = null;

        this.stores = null;

    }

    @Override

    public void close() throws Exception {

        this.dispose();

    }

}