nbcp.myoql.db.redis.proxy.RedisSortedSetProxy.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.AsString
import nbcp.myoql.db.redis.BaseRedisProxy
import org.springframework.data.redis.core.DefaultTypedTuple

/**
 * 有序集合，是按 分值排序。
 */
open class RedisSortedSetProxy @JvmOverloads constructor(
        key: String,
        defaultCacheSeconds: Int = 0,
        autoRenewal:Boolean = false
) :  BaseRedisProxy(key, defaultCacheSeconds,autoRenewal) {

    fun add(  member: String, score: Double) {
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        stringCommand.opsForZSet().add(cacheKey, member, score)
    }

    fun add(  vararg value: Pair) {
        if (value.any() == false) return
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        val set = value.map { DefaultTypedTuple(it.first, it.second)  }.toSet()
        stringCommand.opsForZSet().add(cacheKey, set)
    }

    fun size(): Int {
        val cacheKey = getFullKey(key);

        return stringCommand.opsForZSet().size(cacheKey).AsInt();
    }

    fun isMember(member: String): Boolean {
        val cacheKey = getFullKey(key);
        return stringCommand.opsForZSet().score(cacheKey, member) != null
    }

    /**
     *
     */
    fun getItem(minScore: Double, maxScore: Double): String {
        var cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForZSet().rangeByScore(cacheKey, minScore, maxScore, 0L, 1L).firstOrNull().AsString()

    }

    /**
     * 按分值获取区间
     */
    fun getListByScore(minScore: Double, maxScore: Double): List {
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForZSet().rangeByScore(cacheKey, minScore, maxScore).map { it.AsString() }
    }

    /**
     * 按索引获取区间
     */
    fun getListByIndex(start: Int, end: Int): List {
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForZSet().range(cacheKey, start.toLong(), end.toLong()).map { it.AsString() }
    }

    /**
     * 按索引取第一个
     */
    fun getItem(): String {
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForZSet().range(cacheKey, 0L, 0L).firstOrNull().AsString()
    }

    /**
     * 获取分值
     */
    fun getScore(member: String): Double {
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        return stringCommand.opsForZSet().score(cacheKey, member)
    }

//    fun zscan(key: String, member: String, limit: Int): List {
//        var cacheKey = getFullKey(key);
//        return stringCommand.opsForZSet()
//                .scan(cacheKey, ScanOptions.scanOptions().match(member).count(limit).build())
//                .
//            it.zscan(cacheKey, ScanArgs.Builder.matches(member).limit(limit.toLong())).values.map { it.value }
//        }
//    }

//    fun existMember(member: String) = zscore(member) != 0.toDouble()

    /**
     * 移除
     */
    fun removeItems(vararg members: String): Long {
        val cacheKey = getFullKey(key);

        if(autoRenewal){
            renewalKey()
        }
        return  stringCommand.opsForZSet().remove(cacheKey, *members)
    }
}