com.thebund1st.daming.redis.RedisSmsVerificationRepository Maven / Gradle / Ivy

Go to download
package com.thebund1st.daming.redis;

import com.thebund1st.daming.core.MobilePhoneNumber;
import com.thebund1st.daming.core.SmsVerification;
import com.thebund1st.daming.core.SmsVerificationRepository;
import com.thebund1st.daming.core.SmsVerificationScope;
import com.thebund1st.daming.core.exceptions.MobileIsNotUnderVerificationException;
import com.thebund1st.daming.events.TooManyFailureSmsVerificationAttemptsEvent;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.event.EventListener;
import org.springframework.data.redis.core.RedisTemplate;

@Slf4j
@Setter
public class RedisSmsVerificationRepository implements SmsVerificationRepository {

    private final RedisTemplate redisTemplate;
    private String keyPrefix = "sms.verification.";

    public RedisSmsVerificationRepository(RedisTemplate redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    @Override
    public void store(SmsVerification smsVerification) {
        redisTemplate.opsForValue()
                .set(toKey(smsVerification.getMobile(), smsVerification.getScope()),
                        smsVerification,
                        smsVerification.getExpires());
    }

    private String toKey(MobilePhoneNumber mobile, SmsVerificationScope scope) {
        return String.format("%s.%s.%s", keyPrefix, mobile.getValue(), scope.getValue());
    }

    @Override
    public boolean exists(MobilePhoneNumber mobile, SmsVerificationScope scope) {
        Boolean aBoolean = redisTemplate.hasKey(toKey(mobile, scope));
        return aBoolean == null ? false : aBoolean;
    }

    @Override
    public SmsVerification shouldFindBy(MobilePhoneNumber mobile, SmsVerificationScope scope) {
        SmsVerification codeMaybe = redisTemplate.opsForValue().get(toKey(mobile, scope));
        if (codeMaybe == null) {
            throw new MobileIsNotUnderVerificationException(mobile, scope);
        } else {
            return codeMaybe;
        }
    }

    @Override
    public void remove(SmsVerification smsVerification) {
        remove(smsVerification.getMobile(), smsVerification.getScope());
    }

    private void remove(MobilePhoneNumber mobile, SmsVerificationScope scope) {
        redisTemplate.delete(toKey(mobile, scope));
    }

    @EventListener
    public void on(TooManyFailureSmsVerificationAttemptsEvent event) {
        // TODO: Should the listener be explicit?
        log.debug("Receiving {}", event);
        remove(event.getMobile(), event.getScope());
        log.debug("The [{}] code for [{}] is removed due to too many failure verification attempts",
                event.getScope(), event.getMobile());
    }

}