org.mapdb.IndexTreeList.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mapdb Show documentation
MapDB provides concurrent Maps, Sets and Queues backed by disk storage or off-heap memory. It is a fast, scalable and easy to use embedded Java database.
There is a newer version: 3.1.0
Show newest version
package org.mapdb

import org.eclipse.collections.api.map.primitive.MutableLongLongMap
import java.util.*
import java.util.concurrent.locks.ReadWriteLock
import java.util.concurrent.locks.ReentrantReadWriteLock

/**
 * [ArrayList] like structure backed by tree
 */
class IndexTreeList (
        val store:Store,
        val serializer:Serializer,
        val map: MutableLongLongMap,
        val counterRecid:Long,
        val isThreadSafe:Boolean
        ) : AbstractList() {

    val lock = if(isThreadSafe) ReentrantReadWriteLock() else null

    override fun add(element: E?): Boolean {
        Utils.lockWrite(lock) {
            val index = size++
            val recid = store.put(element, serializer)
            map.put(index.toLong(), recid)
            return true
        }
    }

    override fun add(index: Int, element: E?) {
        Utils.lockWrite(lock) {
            checkIndex(index)
            //make space
            for (i in size - 1 downTo index) {
                val recid = map.get(i.toLong())
                if (recid == 0L)
                    continue;
                map.remove(i.toLong())
                map.put((i + 1).toLong(), recid)
            }
            size++

            val recid = map[index.toLong()]
            if (recid == 0L) {
                map.put(index.toLong(), store.put(element, serializer))
            } else {
                store.update(recid, element, serializer)
            }
        }
    }

    override fun clear() {
        Utils.lockWrite(lock) {
            size = 0;
            //TODO iterate over map and clear in in single pass if IndexTreeLongLongMap
            map.forEachValue { recid -> store.delete(recid, serializer) }
            map.clear()
        }
    }

    override fun removeAt(index: Int): E? {
        Utils.lockWrite(lock) {
            checkIndex(index)
            val recid = map[index.toLong()]
            val ret = if (recid == 0L) {
                null;
            } else {
                val ret = store.get(recid, serializer)
                store.delete(recid, serializer)
                map.remove(index.toLong())
                ret
            }
            //move down rest of the list
            for (i in index + 1 until size) {
                val recid = map.get(i.toLong())
                if (recid == 0L)
                    continue;
                map.remove(i.toLong())
                map.put((i - 1).toLong(), recid)
            }
            size--
            return ret;
        }
    }

    override fun set(index: Int, element: E?): E? {
        Utils.lockWrite(lock) {
            checkIndex(index)
            val recid = map[index.toLong()]
            if (recid == 0L) {
                map.put(index.toLong(), store.put(element, serializer))
                return null;
            } else {
                val ret = store.get(recid, serializer)
                store.update(recid, element, serializer)
                return ret
            }
        }
    }

    fun checkIndex(index:Int){
        if(index<0 || index>=size)
            throw IndexOutOfBoundsException()
    }

    override fun get(index: Int): E? {
        Utils.lockRead(lock) {
            checkIndex(index)

            val recid = map[index.toLong()]
            if (recid == 0L) {
                return null;
            }
            return store.get(recid, serializer)
        }
    }

    override fun isEmpty(): Boolean {
        return size==0
    }

    //TODO PERF iterate over Map and fill gaps, should be faster. But careful if map is HashMap or not sorted other way
    override fun iterator(): MutableIterator {
        return object:MutableIterator{

            @Volatile var index = 0;
            @Volatile var indexToRemove:Int?=null;
            override fun hasNext(): Boolean {
                Utils.lockRead(lock) {
                    return index < [email protected]
                }
            }

            override fun next(): E? {
                Utils.lockRead(lock) {
                    if (index >= [email protected])
                        throw NoSuchElementException()
                    indexToRemove = index
                    val ret = this@IndexTreeList[index]
                    index++;
                    return ret;
                }
            }

            override fun remove() {
                Utils.lockWrite(lock) {
                    removeAt(indexToRemove ?: throw IllegalStateException())
                    index--
                    indexToRemove = null
                }
            }

        }
    }


    override var size: Int
        get() = store.get(counterRecid, Serializer.LONG_PACKED)!!.toInt()
        protected set(size:Int) {
            store.update(counterRecid, size.toLong(), Serializer.LONG_PACKED)
        }


}