Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.fmap.dal;

import java.util.HashMap;

import java.util.Map;

/**

 *

 * @author fdiaz

 */

public class SupportTransactionsHelper implements SupportTransactions {

    private Map<Long, DataTransaction> transactions;

    @Override

    public void setTransaction(DataTransaction transaction) {

        if (this.transactions == null) {

            this.transactions = new HashMap<>();

        }

        long threadId = Thread.currentThread().getId();

        if (transaction == null) {

            this.transactions.remove(threadId);

            return;

        }

        DataTransaction t = transactions.get(threadId);

        if (t == null) {

            transactions.put(threadId, transaction);

            return;

        }

        if (t == transaction) {

            return;

        }

        throw new IllegalArgumentException("Can't add the object to more than one transaction in the same thread");

    }

    @Override

    public DataTransaction getTransaction() {

        if (this.transactions == null) {

            this.transactions = new HashMap<>();

        }

        long threadId = Thread.currentThread().getId();

        DataTransaction t = transactions.get(threadId);

        return t;

    }

}