nbcp.myoql.db.redis.proxy.RedisSetProxy.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ktmyoql Show documentation
kotlin orm -- mysql,mongo , just like ktorm
The newest version!
package nbcp.myoql.db.redis.proxy

import nbcp.base.extend.AsInt
import nbcp.base.extend.AsLong
import nbcp.myoql.db.redis.BaseRedisProxy
import org.springframework.data.redis.core.ScanOptions

/**
 * Created by yuxh on 2018/6/7
 */

open class RedisSetProxy @JvmOverloads constructor(
    key: String,
    defaultCacheSeconds: Int = 0,
    autoRenewal:Boolean = false
) :
    BaseRedisProxy(key, defaultCacheSeconds,autoRenewal) {


    /**
     * 成员数量
     */
    fun size(): Int {
        val cacheKey = getFullKey(key);
        return stringCommand.opsForSet().size(cacheKey).toInt().AsInt()
    }

    fun isMember(member: String): Boolean {
        val cacheKey = getFullKey(key);
        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForSet().isMember(cacheKey, member)
    }

    /**
     * 删除成员
     * 返回删除的成员个数。
     */
    fun removeItems(vararg members: String): Long {
        if (!members.any()) return 0;
        val cacheKey = getFullKey(key);
        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForSet().remove(cacheKey, *members);
    }


    /**
     * 添加
     */
    fun add(vararg value: String) {
        if (value.any() == false) return
        var cacheKey = getFullKey(key);
        if(autoRenewal){
            renewalKey()
        }
        stringCommand.opsForSet().add(cacheKey, *value)
    }

    /**
     * 获取成员
     */
    fun getListString(): List {
        var cacheKey = getFullKey(key);
        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForSet().members(cacheKey).map { it.toString() }
    }

    /**
     * 移除并返回一个随机元素
     */
    fun spop(): String? {
        var cacheKey = getFullKey(key);
        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForSet().pop(cacheKey)?.toString()
    }

    /**
     * 扫描 key
     */
    fun sscan(pattern: String, limit: Int = 999): Set {
        var cacheKey = getFullKey(key);

        stringCommand.opsForSet()
            .scan(
                cacheKey, ScanOptions.scanOptions()
                    .match(pattern)
                    .count(limit.AsLong())
                    .build()
            )
            .use { result ->
                var list = mutableSetOf()
                while (result.hasNext()) {
                    list.add(result.next().toString())
                }
                return list;
            }
    }


}