aiforge/modules/redis/redis_client/client.go

package redis_client

import (
	"code.gitea.io/gitea/modules/labelmsg"
	"fmt"
	"github.com/gomodule/redigo/redis"
	"math"
	"strconv"
	"time"
)

func Setex(key, value string, timeout time.Duration) (bool, error) {
	redisClient := labelmsg.Get()
	defer redisClient.Close()

	seconds := int(math.Floor(timeout.Seconds()))
	reply, err := redisClient.Do("SETEX", key, seconds, value)
	if err != nil {
		return false, err
	}
	if reply != "OK" {
		return false, nil
	}
	return true, nil

}

func Setnx(key, value string, timeout time.Duration) (bool, error) {
	redisClient := labelmsg.Get()
	defer redisClient.Close()

	seconds := int(math.Floor(timeout.Seconds()))
	reply, err := redisClient.Do("SET", key, value, "NX", "EX", seconds)
	if err != nil {
		return false, err
	}
	if reply != "OK" {
		return false, nil
	}
	return true, nil

}

func SETNX(conn redis.Conn, key, value string, seconds int) (bool, error) {
	reply, err := conn.Do("SET", key, value, "NX", "EX", seconds)
	return redis.Bool(reply, err)

}

func SET(conn redis.Conn, key, value string, seconds int) (bool, error) {
	reply, err := conn.Do("SETEX", key, seconds, value)
	return redis.Bool(reply, err)

}

func HSETNX(conn redis.Conn, key, subKey string, value interface{}) error {
	_, err := conn.Do("HSETNX", key, subKey, value)
	return err

}

func HGET(conn redis.Conn, key, subKey string) (interface{}, error) {
	return conn.Do("HGET", key, subKey)
}
func EXISTS(conn redis.Conn, key string) (bool, error) {

	reply, err := conn.Do("EXISTS", key)
	return redis.Bool(reply, err)

}

func HEXISTS(conn redis.Conn, key string, subKey string) (bool, error) {

	reply, err := conn.Do("HEXISTS", key, subKey)
	return redis.Bool(reply, err)

}

func Expire(conn redis.Conn, key string, seconds int) error {
	_, err := conn.Do("EXPIRE", key, seconds)
	return err

}

func HINCRBY(conn redis.Conn, key, subKey string, value int) error {
	_, err := conn.Do("HINCRBY", key, subKey, value)
	return err
}
func GET(conn redis.Conn, key string) (interface{}, error) {
	return conn.Do("GET", key)
}

func Ttl(conn redis.Conn, key string) (int, error) {

	reply, err := conn.Do("TTL", key)
	if err != nil {
		return 0, err
	}
	n, _ := strconv.Atoi(fmt.Sprint(reply))
	return n, nil

}

func Get(key string) (string, error) {
	redisClient := labelmsg.Get()
	defer redisClient.Close()

	reply, err := redisClient.Do("GET", key)
	if err != nil {
		return "", err
	}
	if reply == nil {
		return "", err
	}
	s, _ := redis.String(reply, nil)
	return s, nil

}

func Del(key string) (int, error) {
	redisClient := labelmsg.Get()
	defer redisClient.Close()

	reply, err := redisClient.Do("DEL", key)
	if err != nil {
		return 0, err
	}
	if reply == nil {
		return 0, err
	}
	s, _ := redis.Int(reply, nil)
	return s, nil

}

func TTL(key string) (int, error) {
	redisClient := labelmsg.Get()
	defer redisClient.Close()

	reply, err := redisClient.Do("TTL", key)
	if err != nil {
		return 0, err
	}
	n, _ := strconv.Atoi(fmt.Sprint(reply))
	return n, nil

}