sa-token/sa-token-doc/doc/use/dao-extend.md
2020-12-14 23:01:15 +08:00

3.6 KiB
Raw Blame History

持久层扩展


  • 每次重启项目就得重新登录一遍,我想把登录数据都放在redis里,这样重启项目也不用重新登录,行不行?
  • 行!
  • 你需要做的就是重写sa-token的dao层实现方式参考以下方案

1. 首先在pom.xml引入依赖

	<!-- SpringBoot整合redis -->
	<dependency>
		<groupId>org.springframework.boot</groupId>
		<artifactId>spring-boot-starter-redis</artifactId>
		<version>RELEASE</version>
	</dependency>

2. 实现SaTokenDao接口

新建文件SaTokenDaoRedis.java,实现接口SaTokenDao, 并加上注解@Component保证此类被springboot扫描到

  • 代码参考:

package com.pj.satoken;

import java.util.concurrent.TimeUnit;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.serializer.JdkSerializationRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;
import org.springframework.stereotype.Component;

import cn.dev33.satoken.dao.SaTokenDao;
import cn.dev33.satoken.session.SaSession;

/**
 * sa-token持久层的实现类 , 基于redis 
 */
@Component	// 打开此注解保证此类被springboot扫描即可完成sa-token与redis的集成 
public class SaTokenDaoRedis implements SaTokenDao {


	// string专用
	@Autowired
	StringRedisTemplate stringRedisTemplate;	

	// SaSession专用 
	RedisTemplate<String, SaSession> redisTemplate;
	@Autowired
	@SuppressWarnings({ "rawtypes", "unchecked" })
	public void setRedisTemplate(RedisTemplate redisTemplate) {
	    redisTemplate.setKeySerializer(new StringRedisSerializer());
	    redisTemplate.setValueSerializer(new JdkSerializationRedisSerializer());
		this.redisTemplate = redisTemplate;
	}
	
	
	// 根据key获取value ,如果没有,则返回空 
	@Override
	public String getValue(String key) {
		return stringRedisTemplate.opsForValue().get(key);
	}

	// 写入指定key-value键值对并设定过期时间(单位:秒)
	@Override
	public void setValue(String key, String value, long timeout) {
		stringRedisTemplate.opsForValue().set(key, value, timeout, TimeUnit.SECONDS);
	}

	// 更新指定key-value键值对 (过期时间取原来的值)
	@Override
	public void updateValue(String key, String value) {
		long expire = redisTemplate.getExpire(key);
		if(expire == -2) {	// -2 = 无此键 
			return;
		}
		stringRedisTemplate.opsForValue().set(key, value, expire, TimeUnit.SECONDS);
	}
	
	// 删除一个指定的key 
	@Override
	public void delKey(String key) {
		stringRedisTemplate.delete(key);
	}

	
	// 根据指定key的session如果没有则返回空 
	@Override
	public SaSession getSaSession(String sessionId) {
		return redisTemplate.opsForValue().get(sessionId);
	}

	// 将指定session持久化 
	@Override
	public void saveSaSession(SaSession session, long timeout) {
		redisTemplate.opsForValue().set(session.getId(), session, timeout, TimeUnit.SECONDS);
	}

	// 更新指定session 
	@Override
	public void updateSaSession(SaSession session) {
		long expire = redisTemplate.getExpire(session.getId());
		if(expire == -2) {	// -2 = 无此键 
			return;
		}
		redisTemplate.opsForValue().set(session.getId(), session, expire, TimeUnit.SECONDS);
	}

	// 删除一个指定的session 
	@Override
	public void deleteSaSession(String sessionId) {
		redisTemplate.delete(sessionId);
	}

}