com.maxifier.mxcache.impl.caches.def.TupleObjectWeakTroveStorage Maven / Gradle / Ivy

/*
 * Copyright (c) 2008-2014 Maxifier Ltd. All Rights Reserved.
 */
package com.maxifier.mxcache.impl.caches.def;

import com.maxifier.mxcache.tuple.Tuple;
import com.maxifier.mxcache.transform.SmartReference;
import gnu.trove.strategy.HashingStrategy;

import java.util.*;

/**
 * TupleObjectWeakTroveStorage
 *
 * @author Andrey Yakoushin ([email protected])
 * @author Alexander Kochurov ([email protected])
 */
public class TupleObjectWeakTroveStorage extends ObjectObjectTroveStorage {
    private final List removed = Collections.synchronizedList(new ArrayList());

    private final int[] tupleIndices;

    public TupleObjectWeakTroveStorage() {
        throw new UnsupportedOperationException("TupleObjectWeakTroveStorage actually is not externalizable");
    }

    public TupleObjectWeakTroveStorage(int[] tupleIndices) {
        this.tupleIndices = tupleIndices;
    }

    public TupleObjectWeakTroveStorage(HashingStrategy strategy, int[] tupleIndices) {
        super(strategy);
        this.tupleIndices = tupleIndices;
    }

    private void cleanup() {
        for (Tuple t : removed) {
            remove(t);
        }
        removed.clear();
    }

    @Override
    public Object load(Tuple o) {
        cleanup();
        return super.load(o);
    }

    @Override
    public void save(Tuple o, Object t) {
        cleanup();
        Callback callback = new Callback(o);
        for (int index : tupleIndices) {
            ((SmartReference) o.get(index)).setCallback(callback);
        }
        super.save(o, t);
    }

    private class Callback implements Runnable {
        private final Tuple o;

        public Callback(Tuple o) {
            this.o = o;
        }

        @Override
        public void run() {
            removed.add(o);
        }
    }
}