manager/orion-visor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加 spring-boot-mybatis starter.

Browse Source
This commit is contained in:
ljh01459796
2023-06-25 16:42:49 +08:00
parent 00487ac4ef
commit 1f3be99b3b
8 changed files with 220 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
orion-ops-framework/orion-ops-common/src/main/java/com/orion/ops/framework/common/constant/FilterOrderConst.java
View File

@@ -13,4 +13,6 @@ public interface FilterOrderConst {
int TRICE_ID_FILTER = Integer.MIN_VALUE + 10;
int MYBATIS_CACHE_CLEAR_FILTER = Integer.MIN_VALUE + 100000;
}

33
orion-ops-framework/orion-ops-common/src/main/java/com/orion/ops/framework/common/filter/FilterCreator.java Normal file
View File

@@ -0,0 +1,33 @@
package com.orion.ops.framework.common.filter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import javax.servlet.Filter;
/**
* 过滤器构造器
*
* @author Jiahang Li
* @version 1.0.0
* @since 2023/6/25 15:05
*/
public class FilterCreator {
private FilterCreator() {
}
/**
* 创建过滤器
*
* @param filter filter
* @param order order
* @param <T> type
* @return filter bean
*/
public static <T extends Filter> FilterRegistrationBean<T> create(T filter, Integer order) {
FilterRegistrationBean<T> bean = new FilterRegistrationBean<>(filter);
bean.setOrder(order);
return bean;
}
}

76
orion-ops-framework/orion-ops-spring-boot-starter-mybatis/src/main/java/com/orion/ops/framework/mybatis/cache/CacheQuery.java vendored Normal file
View File

@@ -0,0 +1,76 @@
package com.orion.ops.framework.mybatis.cache;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.orion.lang.define.wrapper.Store;
import com.orion.lang.utils.Valid;
import java.io.Serializable;
/**
* 缓存查询器
* <p>
* 查询会存入缓存
* <p>
* TODO test
*
* @author Jiahang Li
* @version 1.0.0
* @since 2023/6/25 13:05
*/
public class CacheQuery<T> {
private final BaseMapper<T> dao;
private Serializable id;
private CacheQuery(BaseMapper<T> dao, Serializable id) {
this.dao = dao;
this.id = id;
}
public static <T> CacheQuery<T> of(BaseMapper<T> dao) {
Valid.notNull(dao, "dao is null");
return new CacheQuery<>(dao, null);
}
public static <T> CacheQuery<T> of(BaseMapper<T> dao, Serializable id) {
Valid.notNull(dao, "dao is null");
Valid.notNull(id, "id is null");
return new CacheQuery<>(dao, id);
}
/**
* 设置 id
*
* @param id id
* @return this
*/
public CacheQuery<T> id(Serializable id) {
this.id = id;
return this;
}
@SuppressWarnings("unchecked")
public <R> R get(Class<R> c) {
T row = this.get();
// TODO FIXME mapstruct
return (R) row;
}
@SuppressWarnings("unchecked")
public T get() {
Class<? extends BaseMapper<T>> mapperClass = (Class<? extends BaseMapper<T>>) dao.getClass();
// 从缓存中获取
Store<T> store = RowCacheHolder.get(mapperClass, id);
// 设置过缓存
if (store != null) {
return store.get();
}
// 查询
T row = dao.selectById(id);
// 设置缓存
RowCacheHolder.set(mapperClass, id, row);
return row;
}
}

32
orion-ops-framework/orion-ops-spring-boot-starter-mybatis/src/main/java/com/orion/ops/framework/mybatis/cache/RowCacheClearFilter.java vendored Normal file
View File

@@ -0,0 +1,32 @@
package com.orion.ops.framework.mybatis.cache;
import org.springframework.web.filter.OncePerRequestFilter;
import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
/**
* mybatis 缓存清理过滤器
*
* @author Jiahang Li
* @version 1.0.0
* @since 2023/6/25 15:14
*/
public class RowCacheClearFilter extends OncePerRequestFilter {
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
try {
// 执行请求
filterChain.doFilter(request, response);
} finally {
// 清理缓存
// TODO test
RowCacheHolder.remove();
}
}
}

61
orion-ops-framework/orion-ops-spring-boot-starter-mybatis/src/main/java/com/orion/ops/framework/mybatis/cache/RowCacheHolder.java vendored Normal file
View File

@@ -0,0 +1,61 @@
package com.orion.ops.framework.mybatis.cache;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.orion.lang.define.collect.MultiHashMap;
import com.orion.lang.define.wrapper.Store;
import java.io.Serializable;
/**
* 缓存行持有者
*
* @author Jiahang Li
* @version 1.0.0
* @since 2023/6/25 14:21
*/
public class RowCacheHolder {
private RowCacheHolder() {
}
/**
* 缓存
* <p>
* key: mapperClass
* value: id > row
*/
private static final ThreadLocal<MultiHashMap<Class<? extends BaseMapper<?>>, Serializable, Store<?>>> HOLDER = ThreadLocal.withInitial(MultiHashMap::new);
/**
* 获取缓存
*
* @param mapperClass mapperClass
* @param id id
* @param <T> domain
* @return cacheWrapper
*/
@SuppressWarnings("unchecked")
public static <T> Store<T> get(Class<? extends BaseMapper<?>> mapperClass, Serializable id) {
return (Store<T>) HOLDER.get().get(mapperClass, id);
}
/**
* 设置缓存
*
* @param mapperClass mapperClass
* @param id id
* @param row row
* @param <T> domainClass
*/
public static <T> void set(Class<? extends BaseMapper<T>> mapperClass, Serializable id, T row) {
HOLDER.get().put(mapperClass, id, new Store<>(row));
}
/**
* 清空缓存
*/
public static void remove() {
HOLDER.remove();
}
}

12
orion-ops-framework/orion-ops-spring-boot-starter-mybatis/src/main/java/com/orion/ops/framework/mybatis/config/OrionMybatisAutoConfiguration.java
View File

@@ -1,9 +1,13 @@
package com.orion.ops.framework.mybatis.config;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.orion.ops.framework.common.constant.FilterOrderConst;
import com.orion.ops.framework.common.filter.FilterCreator;
import com.orion.ops.framework.mybatis.cache.RowCacheClearFilter;
import com.orion.ops.framework.mybatis.handler.FieldFillHandler;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.autoconfigure.AutoConfiguration;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
/**
@@ -26,4 +30,12 @@ public class OrionMybatisAutoConfiguration {
return new FieldFillHandler();
}
/**
* @return mybatis 缓存清理过滤器
*/
@Bean
public FilterRegistrationBean<RowCacheClearFilter> rowCacheClearFilterBean() {
return FilterCreator.create(new RowCacheClearFilter(), FilterOrderConst.MYBATIS_CACHE_CLEAR_FILTER);
}
}

20
orion-ops-framework/orion-ops-spring-boot-starter-web/src/main/java/com/orion/ops/framework/web/config/OrionWebAutoConfiguration.java
View File

@@ -5,6 +5,7 @@ import com.alibaba.fastjson.support.config.FastJsonConfig;
import com.alibaba.fastjson.support.spring.FastJsonHttpMessageConverter;
import com.orion.lang.utils.collect.Lists;
import com.orion.ops.framework.common.constant.FilterOrderConst;
import com.orion.ops.framework.common.filter.FilterCreator;
import com.orion.ops.framework.web.core.filter.TraceIdFilter;
import com.orion.ops.framework.web.core.handler.GlobalExceptionHandler;
import com.orion.ops.framework.web.core.handler.WrapperResultHandler;
@@ -25,7 +26,6 @@ import org.springframework.web.filter.CorsFilter;
import org.springframework.web.servlet.config.annotation.PathMatchConfigurer;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import javax.servlet.Filter;
import java.util.ArrayList;
import java.util.List;
@@ -124,7 +124,7 @@ public class OrionWebAutoConfiguration implements WebMvcConfigurer {
// 创建 UrlBasedCorsConfigurationSource 对象
UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource();
source.registerCorsConfiguration("/**", config);
return createFilterBean(new CorsFilter(source), FilterOrderConst.CORS_FILTER);
return FilterCreator.create(new CorsFilter(source), FilterOrderConst.CORS_FILTER);
}
/**
@@ -132,21 +132,7 @@ public class OrionWebAutoConfiguration implements WebMvcConfigurer {
*/
@Bean
public FilterRegistrationBean<TraceIdFilter> traceIdFilterBean() {
return createFilterBean(new TraceIdFilter(), FilterOrderConst.TRICE_ID_FILTER);
}
/**
* 创建过滤器
*
* @param filter filter
* @param order order
* @param <T> type
* @return filter bean
*/
public static <T extends Filter> FilterRegistrationBean<T> createFilterBean(T filter, Integer order) {
FilterRegistrationBean<T> bean = new FilterRegistrationBean<>(filter);
bean.setOrder(order);
return bean;
return FilterCreator.create(new TraceIdFilter(), FilterOrderConst.TRICE_ID_FILTER);
}
}

1
orion-ops-framework/orion-ops-spring-boot-starter-web/src/main/java/com/orion/ops/framework/web/core/filter/TraceIdFilter.java
View File

@@ -32,6 +32,7 @@ public class TraceIdFilter extends OncePerRequestFilter {
// 执行请求
filterChain.doFilter(request, response);
} finally {
// 清理缓存
TraceIdHolder.remove();
}
}