com.skillw.pouvoir.internal.feature.database.sql.UserContainer.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of Pouvoir Show documentation

Bukkit Script Engine Plugin.

There is a newer version: 1.6.7-beta-6

package com.skillw.pouvoir.internal.feature.database.sql

import com.skillw.pouvoir.api.feature.database.ContainerHolder
import com.skillw.pouvoir.api.feature.database.UserBased
import com.skillw.pouvoir.api.feature.database.sql.IPouTable
import taboolib.common.platform.function.submitAsync
import taboolib.module.database.ColumnBuilder
import taboolib.module.database.Host
import java.util.concurrent.ConcurrentHashMap

/**
 * @className NormalContainer
 *
 * @author Glom
 * @date 2023/1/12 20:51 Copyright 2024 Glom.
 */
class UserContainer, E : ColumnBuilder>(
    holder: ContainerHolder>,
    table: IPouTable,
) : NormalContainer(holder, table), UserBased {
    private val cache = ConcurrentHashMap>()
    private val database by lazy {
        UserDatabase(this)
    }

    override fun database(): UserDatabase {
        return database
    }

    override fun onEnable() {
        database.loadAll(cache)
        submitAsync(period = holder.userContainerSyncTime) {
            database.saveAll(cache)
        }
    }

    override fun onDisable() {
        cache.forEach { (user, map) ->
            map.forEach { (key, value) ->
                database[user, key] = value
            }
        }
        close()
    }

    override fun get(user: String, key: String): String? {
        return cache[user]?.get(key)
    }

    override fun delete(user: String, key: String) {
        cache[user]?.remove(key)
    }

    override fun set(user: String, key: String, value: String?) {
        cache.computeIfAbsent(user) { ConcurrentHashMap() }[key] = value ?: return
    }

    override fun contains(user: String, key: String): Boolean {
        return cache.containsKey(user) && cache[user]!!.containsKey(key)
    }
}