dorkbox.util.storage.MemoryStorage Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2014 dorkbox, llc
 *
 * 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.
 */
package dorkbox.util.storage;

import java.io.File;
import java.util.HashMap;
import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;

/**
 * Storage that is in memory only (and is not persisted to disk)
 */
class MemoryStorage implements Storage {


    // must be volatile
    @SuppressWarnings("MismatchedQueryAndUpdateOfCollection")
    private volatile HashMap storage = new HashMap();

    private final Object singleWriterLock = new Object[0];

    // Recommended for best performance while adhering to the "single writer principle". Must be static-final
    private static final AtomicReferenceFieldUpdater storageREF =
            AtomicReferenceFieldUpdater.newUpdater(MemoryStorage.class, HashMap.class, "storage");

    private int version;


    MemoryStorage() {}


    /**
     * Returns the number of objects in the database.
     */
    @Override
    public
    int size() {
        // access a snapshot of the storage (single-writer-principle)
        HashMap storage = storageREF.get(this);
        return storage.size();
    }

    /**
     * Checks if there is a object corresponding to the given key.
     */
    @Override
    public
    boolean contains(final StorageKey key) {
        // access a snapshot of the storage (single-writer-principle)
        HashMap storage = storageREF.get(this);
        return storage.containsKey(key);
    }

    /**
     * Reads a object using the specific key, and casts it to the expected class
     */
    @SuppressWarnings("unchecked")
    @Override
    public
     T get(final StorageKey key) {
        // access a snapshot of the storage (single-writer-principle)
        HashMap storage = storageREF.get(this);
        return (T) storage.get(key);
    }

    @SuppressWarnings("unchecked")
    @Override
    public
     T get(final StorageKey key, final T data) {
        // access a snapshot of the storage (single-writer-principle)
        HashMap storage = storageREF.get(this);

        final Object o = storage.get(key);
        if (o == null) {
            storage.put(key, data);
            return data;
        }
        return (T) o;
    }

    /**
     * Saves the given data to storage with the associated key.
     * 
     * Also will update existing data. If the new contents do not fit in the original space, then the update is handled by
     * deleting the old data and adding the new.
     */
    @Override
    public
    void put(final StorageKey key, final Object object) {
        // synchronized is used here to ensure the "single writer principle", and make sure that ONLY one thread at a time can enter this
        // section. Because of this, we can have unlimited reader threads all going at the same time, without contention.
        synchronized (singleWriterLock) {
            storage.put(key, object);
        }
    }

    /**
     * Deletes an object from storage.
     *
     * @return true if the delete was successful. False if there were problems deleting the data.
     */
    @Override
    public synchronized
    boolean delete(final StorageKey key) {
        // synchronized is used here to ensure the "single writer principle", and make sure that ONLY one thread at a time can enter this
        // section. Because of this, we can have unlimited reader threads all going at the same time, without contention.
        synchronized (singleWriterLock) {
            storage.remove(key);
        }

        return true;
    }

    /**
     * @return null. There is no file that backs this storage
     */
    @Override
    public
    File getFile() {
        return null;
    }

    /**
     * Gets the backing file size.
     *
     * @return 0. There is no file that backs this storage
     */
    @Override
    public
    long getFileSize() {
        return 0;
    }

    /**
     * @return false. Writes to in-memory storage are immediate.
     */
    @Override
    public
    boolean hasWriteWaiting() {
        return false;
    }

    /**
     * @return 0. There is no file that backs this storage
     */
    @Override
    public
    long getSaveDelay() {
        return 0L;
    }


    /**
     * @return the version of data stored in the database
     */
    @Override
    public synchronized
    int getVersion() {
        return version;
    }

    /**
     * Sets the version of data stored in the database
     */
    @Override
    public synchronized
    void setVersion(final int version) {
        this.version = version;
    }

    /**
     * There is no file that backs this storage, so writes are immediate and saves do nothgin
     */
    @Override
    public
    void save() {
        // no-op
    }

    /**
     * In-memory storage systems do not have a backing file, so there is nothing to close
     */
    @Override
    public
    void close() {
        StorageSystem.close(this);
    }
}