org.shoulder.code.store.impl.SessionValidateCodeRepository Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of shoulder-code Show documentation

The newest version!

package org.shoulder.code.store.impl;

import org.shoulder.code.dto.ValidateCodeDTO;
import org.shoulder.code.store.ValidateCodeStore;
import org.springframework.web.context.request.ServletWebRequest;

/**
 * 基于session的验证码存取器
 * @author lym
 */
public class SessionValidateCodeRepository implements ValidateCodeStore {

	/**
	 * 验证码放入session时的前缀
	 */
	private static final String DEFAULT_KEY_PREFIX = "CAPTCHA_CODE:";
	
	@Override
	public void save(ServletWebRequest request, ValidateCodeDTO code, String validateCodeType) {
		request.getRequest().getSession().setAttribute(builderSessionKey(validateCodeType), code);
	}
	

	@Override
	public ValidateCodeDTO get(ServletWebRequest request, String validateCodeType) {
		return (ValidateCodeDTO) request.getRequest().getSession().getAttribute(builderSessionKey(validateCodeType));
	}

	@Override
	public void remove(ServletWebRequest request, String codeType) {
		request.getRequest().getSession().removeAttribute(builderSessionKey(codeType));
	}

	/**
	 * 验证码放入 session 的 key
	 */
	protected String builderSessionKey(String validateCodeType) {
		return DEFAULT_KEY_PREFIX + validateCodeType.toUpperCase();
	}

}