mirror of
https://github.com/FatttSnake/Pinnacle-OA.git
synced 2026-04-04 22:41:24 +08:00
Added spring security
This commit is contained in:
115
Pinnacle/src/main/java/com/cfive/pinnacle/utils/JwtUtil.java
Normal file
115
Pinnacle/src/main/java/com/cfive/pinnacle/utils/JwtUtil.java
Normal file
@@ -0,0 +1,115 @@
|
||||
package com.cfive.pinnacle.utils;
|
||||
|
||||
import com.auth0.jwt.JWT;
|
||||
import com.auth0.jwt.JWTVerifier;
|
||||
import com.auth0.jwt.algorithms.Algorithm;
|
||||
import com.auth0.jwt.interfaces.DecodedJWT;
|
||||
|
||||
import javax.crypto.SecretKey;
|
||||
import javax.crypto.spec.SecretKeySpec;
|
||||
import java.util.Base64;
|
||||
import java.util.Date;
|
||||
import java.util.UUID;
|
||||
|
||||
/**
|
||||
* JWT 工具类
|
||||
*/
|
||||
public class JwtUtil {
|
||||
|
||||
// 有效期
|
||||
public static final Long JWT_TTL = 60 * 60 * 1000L; // 60 * 60 * 1000 一个小时
|
||||
// 秘钥明文
|
||||
public static final String JWT_KEY = "pinnacle";
|
||||
//签发者
|
||||
public static final String ISSUER = "cfive";
|
||||
|
||||
public static String getUUID() {
|
||||
return UUID.randomUUID().toString().replaceAll("-", "");
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 生成加密后的秘钥 secretKey
|
||||
*
|
||||
* @return 密钥
|
||||
*/
|
||||
public static SecretKey generalKey() {
|
||||
byte[] encodedKey = Base64.getDecoder().decode(JwtUtil.JWT_KEY);
|
||||
return new SecretKeySpec(encodedKey, 0, encodedKey.length, "AES");
|
||||
}
|
||||
|
||||
public static Algorithm algorithm() {
|
||||
SecretKey secretKey = generalKey();
|
||||
return Algorithm.HMAC256(secretKey.toString());
|
||||
}
|
||||
|
||||
private static String getJwt(String subject, Long ttlMillis, String uuid) {
|
||||
long nowMillis = System.currentTimeMillis();
|
||||
Date now = new Date(nowMillis);
|
||||
if (ttlMillis == null) {
|
||||
ttlMillis = JwtUtil.JWT_TTL;
|
||||
}
|
||||
long expMillis = nowMillis + ttlMillis;
|
||||
Date expDate = new Date(expMillis);
|
||||
return JWT.create()
|
||||
.withJWTId(uuid)
|
||||
.withSubject(subject)
|
||||
.withIssuer(ISSUER)
|
||||
.withIssuedAt(now)
|
||||
.withExpiresAt(expDate)
|
||||
.sign(algorithm());
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成 jwt
|
||||
*
|
||||
* @param subject token 中要存放的数据（json 格式）
|
||||
* @return jwt 字符串
|
||||
*/
|
||||
public static String createJWT(String subject) {
|
||||
return getJwt(subject, null, getUUID());
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成 jwt
|
||||
*
|
||||
* @param subject token 中要存放的数据（json格式）
|
||||
* @param ttlMillis token 超时时间
|
||||
* @return jwt 字符串
|
||||
*/
|
||||
public static String createJWT(String subject, Long ttlMillis) {
|
||||
return getJwt(subject, ttlMillis, getUUID());
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建 token
|
||||
*
|
||||
* @param id 唯一的 ID
|
||||
* @param subject token 中要存放的数据（json 格式）
|
||||
* @param ttlMillis token 超时时间
|
||||
* @return jwt 字符串
|
||||
*/
|
||||
public static String createJWT(String id, String subject, Long ttlMillis) {
|
||||
return getJwt(subject, ttlMillis, id);
|
||||
}
|
||||
|
||||
/**
|
||||
* 解析 jwt
|
||||
*
|
||||
* @param jwt 字符串
|
||||
* @return 解析内容
|
||||
*/
|
||||
public static DecodedJWT parseJWT(String jwt) {
|
||||
JWTVerifier jwtVerifier = JWT.require(algorithm())
|
||||
.build();
|
||||
return jwtVerifier.verify(jwt);
|
||||
}
|
||||
|
||||
public static void main(String[] args) {
|
||||
// String token = "eyJhbGciOiJIUzI1NiJ9.eyJqdGkiOiJjYWM2ZDVhZi1mNjVlLTQ0MDAtYjcxMi0zYWEwOGIyOTIwYjQiLCJzdWIiOiJzZyIsImlzcyI6InNnIiwiaWF0IjoxNjM4MTA2NzEyLCJleHAiOjE2MzgxMTAzMTJ9.JVsSbkP94wuczb4QryQbAke3ysBDIL5ou8fWsbt_ebg";
|
||||
// Claims claims = parseJWT(token);
|
||||
|
||||
System.out.println(parseJWT("eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJzdWIiOiJxd2UiLCJpc3MiOiJjZml2ZSIsImV4cCI6MTY4MzE5MzkyOSwiaWF0IjoxNjgzMTkwMzI5LCJqdGkiOiIzOWY5YTcxYTllY2E0Mjg1OGVjNGExODU2ZmQwYjk4OCJ9.4YOOILGWxlnmToWTdo4YoCbfXqvzdJF_Ds4zulDWX1o")
|
||||
.getClaims());
|
||||
}
|
||||
}
|
||||
222
Pinnacle/src/main/java/com/cfive/pinnacle/utils/RedisCache.java
Normal file
222
Pinnacle/src/main/java/com/cfive/pinnacle/utils/RedisCache.java
Normal file
@@ -0,0 +1,222 @@
|
||||
package com.cfive.pinnacle.utils;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.data.redis.core.BoundSetOperations;
|
||||
import org.springframework.data.redis.core.HashOperations;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.data.redis.core.ValueOperations;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import java.util.*;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
@SuppressWarnings(value = {"unchecked", "rawtypes"})
|
||||
@Component
|
||||
public class RedisCache {
|
||||
private RedisTemplate redisTemplate;
|
||||
|
||||
@Autowired
|
||||
public void setRedisTemplate(RedisTemplate redisTemplate) {
|
||||
this.redisTemplate = redisTemplate;
|
||||
}
|
||||
|
||||
/**
|
||||
* 缓存基本的对象，Integer、String、实体类等
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @param value 缓存的值
|
||||
*/
|
||||
public <T> void setCacheObject(final String key, final T value) {
|
||||
redisTemplate.opsForValue().set(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 缓存基本的对象，Integer、String、实体类等
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @param value 缓存的值
|
||||
* @param timeout 超时时间
|
||||
* @param timeUnit 时间颗粒度
|
||||
*/
|
||||
public <T> void setCacheObject(final String key, final T value, final Integer timeout, final TimeUnit timeUnit) {
|
||||
redisTemplate.opsForValue().set(key, value, timeout, timeUnit);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置有效时间
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @param timeout 超时时间
|
||||
* @return true=设置成功；false=设置失败
|
||||
*/
|
||||
public boolean expire(final String key, final long timeout) {
|
||||
return expire(key, timeout, TimeUnit.SECONDS);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置有效时间
|
||||
*
|
||||
* @param key Redis 键
|
||||
* @param timeout 超时时间
|
||||
* @param unit 时间颗粒度
|
||||
* @return true=设置成功；false=设置失败
|
||||
*/
|
||||
public boolean expire(final String key, final long timeout, final TimeUnit unit) {
|
||||
return Boolean.TRUE.equals(redisTemplate.expire(key, timeout, unit));
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得缓存的基本对象
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @return 缓存的值
|
||||
*/
|
||||
public <T> T getCacheObject(final String key) {
|
||||
ValueOperations<String, T> operation = redisTemplate.opsForValue();
|
||||
return operation.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除单个对象
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @return true=删除成功；false=删除失败
|
||||
*/
|
||||
public boolean deleteObject(final String key) {
|
||||
return Boolean.TRUE.equals(redisTemplate.delete(key));
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除集合对象
|
||||
*
|
||||
* @param collection 多个键
|
||||
* @return 删除个数
|
||||
*/
|
||||
public long deleteObject(final Collection collection) {
|
||||
return redisTemplate.delete(collection);
|
||||
}
|
||||
|
||||
/**
|
||||
* 缓存 List 数据
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @param dataList 缓存的 List 数据
|
||||
* @return 缓存的个数
|
||||
*/
|
||||
public <T> long setCacheList(final String key, final List<T> dataList) {
|
||||
Long count = redisTemplate.opsForList().rightPushAll(key, dataList);
|
||||
return count == null ? 0 : count;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得缓存的 List 数据
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @return 缓存的键对应的 List 数据
|
||||
*/
|
||||
public <T> List<T> getCacheList(final String key) {
|
||||
return redisTemplate.opsForList().range(key, 0, -1);
|
||||
}
|
||||
|
||||
/**
|
||||
* 缓存 Set 数据
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @param dataSet 缓存的 Set 数据
|
||||
* @return 缓存数据的对象
|
||||
*/
|
||||
public <T> BoundSetOperations<String, T> setCacheSet(final String key, final Set<T> dataSet) {
|
||||
BoundSetOperations<String, T> setOperation = redisTemplate.boundSetOps(key);
|
||||
for (T t : dataSet) {
|
||||
setOperation.add(t);
|
||||
}
|
||||
return setOperation;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得缓存的 Set 数据
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @return 缓存的键对应的 Set 数据
|
||||
*/
|
||||
public <T> Set<T> getCacheSet(final String key) {
|
||||
return redisTemplate.opsForSet().members(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 缓存 Map 数据
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @param dataMap 缓存的 Map 数据
|
||||
*/
|
||||
public <T> void setCacheMap(final String key, final Map<String, T> dataMap) {
|
||||
if (dataMap != null) {
|
||||
redisTemplate.opsForHash().putAll(key, dataMap);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得缓存的 Map 据
|
||||
*
|
||||
* @param key 缓存的键
|
||||
* @return 缓存的键对应的 Map 数据
|
||||
*/
|
||||
public <T> Map<String, T> getCacheMap(final String key) {
|
||||
return redisTemplate.opsForHash().entries(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 往 Hash 中存入数据
|
||||
*
|
||||
* @param key Redis 键
|
||||
* @param hKey Hash 键
|
||||
* @param value 值
|
||||
*/
|
||||
public <T> void setCacheMapValue(final String key, final String hKey, final T value) {
|
||||
redisTemplate.opsForHash().put(key, hKey, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取 Hash 中的数据
|
||||
*
|
||||
* @param key Redis 键
|
||||
* @param hKey Hash 键
|
||||
* @return Hash 中的对象
|
||||
*/
|
||||
public <T> T getCacheMapValue(final String key, final String hKey) {
|
||||
HashOperations<String, String, T> opsForHash = redisTemplate.opsForHash();
|
||||
return opsForHash.get(key, hKey);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除 Hash 中的数据
|
||||
*
|
||||
* @param key Redis 键
|
||||
* @param hkey Hash 键
|
||||
*/
|
||||
public void delCacheMapValue(final String key, final String hkey) {
|
||||
HashOperations hashOperations = redisTemplate.opsForHash();
|
||||
hashOperations.delete(key, hkey);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取多个 Hash 中的数据
|
||||
*
|
||||
* @param key Redis 键
|
||||
* @param hKeys Hash 键集合
|
||||
* @return Hash 对象集合
|
||||
*/
|
||||
public <T> List<T> getMultiCacheMapValue(final String key, final Collection<Object> hKeys) {
|
||||
return redisTemplate.opsForHash().multiGet(key, hKeys);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得缓存的基本对象列表
|
||||
*
|
||||
* @param pattern 字符串前缀
|
||||
* @return 对象列表
|
||||
*/
|
||||
public Collection<String> keys(final String pattern) {
|
||||
return redisTemplate.keys(pattern);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user