top.lshaci.framework.web.service.impl.RedisPreventRepeat Maven / Gradle / Ivy

Go to download
package top.lshaci.framework.web.service.impl;

import lombok.AllArgsConstructor;
import org.springframework.data.redis.core.StringRedisTemplate;
import top.lshaci.framework.web.service.PreventRepeat;

import static java.util.concurrent.TimeUnit.MILLISECONDS;

/**
 * Redis-based prevent repeat submit


 *
 * 1.0.7:This method getAndSet add parameter timeout
 *
 * @author lshaci
 * @since 1.0.5
 * @version 1.0.7
 */
@AllArgsConstructor
public class RedisPreventRepeat implements PreventRepeat {

    /**
     * The timeout of the submit key
     */
    private final long timeout;

    /**
     * The String Redis Template
     */
    private final StringRedisTemplate redisTemplate;

    @Override
    public String getAndSet(String key, long timeout) {
        String value = redisTemplate.opsForValue().getAndSet(key, VALUE);
        redisTemplate.expire(key, timeout > 0 ? timeout : this.timeout, MILLISECONDS);
        return value;
    }

    @Override
    public void remove(String key) {
        redisTemplate.delete(key);
    }
}