AbstractIndex.java

Go to the documentation of this file.

/*
 * Gridarta MMORPG map editor for Crossfire, Daimonin and similar games.
 * Copyright (C) 2000-2015 The Gridarta Developers.
 *
 * 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 2 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, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

package net.sf.gridarta.model.index;

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import net.sf.gridarta.utils.EventListenerList2;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public class AbstractIndex<V extends Serializable> implements Index<V> {

    @NotNull
    private final Object sync = new Object();

    @NotNull
    private final EventListenerList2<IndexListener<V>> indexListeners = new EventListenerList2<>(IndexListener.class);

    @NotNull
    private final Map<V, Long> timestamps = new HashMap<>();

    @NotNull
    private final Map<V, String> names = new HashMap<>();

    @NotNull
    private final Collection<V> pending = new HashSet<>();

    private boolean modified;

    private boolean transaction;

    @NotNull
    private final Collection<V> transactionDelete = new HashSet<>();

    @Override
    public int size() {
        return timestamps.size();
    }

    @NotNull
    @Override
    public Collection<V> findPartialName(@NotNull final String name) {
        synchronized (sync) {
            final String nameLowerCase = name.toLowerCase();
            final Collection<V> result = new HashSet<>();
            for (final Entry<V, String> e : names.entrySet()) {
                if (e.getValue().toLowerCase().contains(nameLowerCase)) {
                    result.add(e.getKey());
                }
            }
            return result;
        }
    }

    @Override
    public void beginUpdate() {
        synchronized (sync) {
            transaction = true;
            transactionDelete.clear();
            transactionDelete.addAll(timestamps.keySet());
        }
    }

    @Override
    public void endUpdate() {
        synchronized (sync) {
            if (!transaction) {
                throw new IllegalStateException();
            }
            transaction = false;

            final Collection<V> tmp = new ArrayList<>(transactionDelete);
            transactionDelete.clear();
            if (!tmp.isEmpty()) {
                modified = true;
                for (final V value : tmp) {
                    timestamps.remove(value);
                    pending.remove(value);
                    names.remove(value);
                }
                for (final V value : tmp) {
                    for (final IndexListener<V> listener : indexListeners.getListeners()) {
                        listener.valueRemoved(value);
                        listener.nameChanged();
                    }
                }
            }
        }
    }

    @Override
    public void add(@NotNull final V value, final long timestamp) {
        synchronized (sync) {
            final Long oldTimestamp = timestamps.put(value, timestamp);
            final boolean notifyPending;
            final boolean notifyAdded;
            if (oldTimestamp == null || oldTimestamp != timestamp) {
                notifyPending = pending.add(value) && pending.size() == 1;
                notifyAdded = oldTimestamp == null;
                modified = true;
            } else {
                notifyPending = false;
                notifyAdded = false;
            }
            transactionDelete.remove(value);
            if (notifyAdded) {
                for (final IndexListener<V> listener : indexListeners.getListeners()) {
                    listener.valueAdded(value);
                }
            }
            if (notifyPending) {
                for (final IndexListener<V> listener : indexListeners.getListeners()) {
                    listener.pendingChanged();
                }
            }
        }
    }

    @Override
    public void remove(@NotNull final V value) {
        synchronized (sync) {
            if (timestamps.remove(value) == null) {
                return;
            }
            modified = true;
            pending.remove(value);
            names.remove(value);
            for (final IndexListener<V> listener : indexListeners.getListeners()) {
                listener.valueRemoved(value);
                listener.nameChanged();
            }
        }
    }

    @Override
    public void setPending(@NotNull final V value) {
        synchronized (sync) {
            final boolean notifyPending = pending.add(value) && pending.size() == 1;
            if (!timestamps.containsKey(value)) {
                modified = true;
                timestamps.put(value, 0L);
                for (final IndexListener<V> listener : indexListeners.getListeners()) {
                    listener.valueAdded(value);
                }
            }
            if (notifyPending) {
                for (final IndexListener<V> listener : indexListeners.getListeners()) {
                    listener.pendingChanged();
                }
            }
        }
    }

    @Override
    public void setName(@NotNull final V value, final long timestamp, @NotNull final String name) {
        synchronized (sync) {
            timestamps.put(value, timestamp);
            final String oldName = names.put(value, name);
            if (oldName != null && oldName.equals(name)) {
                return;
            }
            modified = true;
            for (final IndexListener<V> listener : indexListeners.getListeners()) {
                listener.nameChanged();
            }
        }
    }

    @Nullable
    @Override
    public String getName(@NotNull final V value) {
        synchronized (sync) {
            return names.get(value);
        }
    }

    @Nullable
    @Override
    public V removePending() {
        synchronized (sync) {
            final Iterator<V> it = pending.iterator();
            if (!it.hasNext()) {
                return null;
            }

            final V result = it.next();
            it.remove();
            return result;
        }
    }

    @Override
    public boolean hasPending() {
        synchronized (sync) {
            return !pending.isEmpty();
        }
    }

    @Override
    public boolean isModified() {
        synchronized (sync) {
            return modified;
        }
    }

    @Override
    public void addIndexListener(@NotNull final IndexListener<V> listener) {
        indexListeners.add(listener);
    }

    @Override
    public void removeIndexListener(@NotNull final IndexListener<V> listener) {
        indexListeners.remove(listener);
    }

    @Override
    public void save(@NotNull final ObjectOutputStream objectOutputStream) throws IOException {
        synchronized (sync) {
            objectOutputStream.writeObject(timestamps);
            objectOutputStream.writeObject(names);
            modified = false;
        }
    }

    @Override
    @SuppressWarnings("unchecked")
    public void load(@NotNull final ObjectInputStream objectInputStream) throws IOException {
        synchronized (sync) {
            final Map<V, Long> tmpTimestamps;
            final Map<V, String> tmpNames;
            try {
                tmpTimestamps = (Map<V, Long>) objectInputStream.readObject();
                tmpNames = (Map<V, String>) objectInputStream.readObject();
            } catch (final ClassNotFoundException ex) {
                throw new IOException(ex.getMessage(), ex);
            }
            if (transaction) {
                throw new IOException("cannot restore state within active transaction");
            }

            // drop excess elements from tmpNames to force a consistent state
            tmpNames.keySet().retainAll(tmpTimestamps.keySet());

            clear();
            timestamps.putAll(tmpTimestamps);
            names.putAll(tmpNames);
        }
    }

    @Override
    public void clear() {
        synchronized (sync) {
            timestamps.clear();
            names.clear();
            pending.clear();
            transactionDelete.clear();
            modified = false;
        }
    }

    @Override
    public void indexingFinished() {
        for (final IndexListener<V> listener : indexListeners.getListeners()) {
            listener.indexingFinished();
        }
    }

}