zhaoqiaobo
11 months ago
77 changed files with 2435 additions and 557 deletions
@ -0,0 +1,79 @@
|
||||
package com.logpm.factorydata.olo.config; |
||||
|
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.slf4j.MDC; |
||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.task.TaskDecorator; |
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.scheduling.annotation.EnableAsync; |
||||
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
||||
import org.springframework.web.context.request.RequestAttributes; |
||||
import org.springframework.web.context.request.RequestContextHolder; |
||||
|
||||
import javax.annotation.Nonnull; |
||||
import java.util.Map; |
||||
import java.util.concurrent.Executor; |
||||
import java.util.concurrent.ThreadPoolExecutor; |
||||
|
||||
@Configuration |
||||
@Slf4j |
||||
@EnableAsync |
||||
public class ExecutorConfig { |
||||
|
||||
@Bean |
||||
public Executor asyncExecutor() { |
||||
log.info("start async executor"); |
||||
ThreadPoolTaskExecutor threadPoolTaskExecutor = new ThreadPoolTaskExecutor(); |
||||
// 配置核心线程数
|
||||
threadPoolTaskExecutor.setCorePoolSize(10); |
||||
// 配置最大线程数
|
||||
threadPoolTaskExecutor.setMaxPoolSize(20); |
||||
// 配置队列大小
|
||||
threadPoolTaskExecutor.setQueueCapacity(50); |
||||
// 配置线程池中线程的名称前缀
|
||||
threadPoolTaskExecutor.setThreadNamePrefix("ASYNC_THREAD_"); |
||||
// HelloWorldServiceImpl rejection-policy: 当pool已经达到max size时，如何处理新任务：
|
||||
// CallerRunsPolicy: 不在新线程中执行任务，而是由调用者所在的线程来执行；
|
||||
// AbortPolicy: 拒绝执行新任务，并抛出RejectedExecutionException异常；
|
||||
// DiscardPolicy：丢弃当前将要加入队列的任务；
|
||||
// DiscardOldestPolicy：丢弃任务队列中最旧的任务；
|
||||
threadPoolTaskExecutor.setRejectedExecutionHandler( |
||||
new ThreadPoolExecutor.CallerRunsPolicy() |
||||
); |
||||
threadPoolTaskExecutor.setTaskDecorator(new ContextCopyingDecorator()); |
||||
threadPoolTaskExecutor.initialize(); |
||||
return threadPoolTaskExecutor; |
||||
} |
||||
|
||||
|
||||
static class ContextCopyingDecorator implements TaskDecorator { |
||||
@Nonnull |
||||
@Override |
||||
public Runnable decorate(@Nonnull Runnable runnable) { |
||||
RequestAttributes context = RequestContextHolder.currentRequestAttributes(); |
||||
HttpHeaders headers = (HttpHeaders) ThreadLocalUtil.get("bladeContext"); |
||||
Map<String, String> mdcMap = MDC.getCopyOfContextMap(); |
||||
return () -> { |
||||
try { |
||||
ThreadLocalUtil.put("bladeContext", headers); |
||||
if (mdcMap != null && !mdcMap.isEmpty()) { |
||||
MDC.setContextMap(mdcMap); |
||||
} |
||||
RequestContextHolder.setRequestAttributes(context); |
||||
|
||||
runnable.run(); |
||||
} finally { |
||||
RequestContextHolder.resetRequestAttributes(); |
||||
if (mdcMap != null) { |
||||
mdcMap.clear(); |
||||
} |
||||
ThreadLocalUtil.clear(); |
||||
MDC.clear(); |
||||
} |
||||
}; |
||||
} |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,32 @@
|
||||
package com.logpm.factorydata.olo.config; |
||||
|
||||
import com.logpm.factorydata.olo.interceptor.FactoryAccountsInterceptor; |
||||
import com.logpm.factorydata.olo.interceptor.LocalServerLoginAccountsInterceptor; |
||||
import com.logpm.factorydata.olo.service.IFactoryTokenService; |
||||
import lombok.AllArgsConstructor; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.env.Environment; |
||||
import org.springframework.web.servlet.config.annotation.InterceptorRegistry; |
||||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; |
||||
|
||||
@Configuration |
||||
@AllArgsConstructor |
||||
public class InterceptorAdapterConfig implements WebMvcConfigurer { |
||||
|
||||
private final IFactoryTokenService factoryTokenService; |
||||
|
||||
private final BladeRedis redis; |
||||
|
||||
private final Environment environment; |
||||
|
||||
|
||||
@Override |
||||
public void addInterceptors(InterceptorRegistry interceptorRegistry) { |
||||
// interceptorRegistry.addInterceptor(new FactoryAccountsInterceptor(factoryTokenService))
|
||||
// .addPathPatterns("/**").order(1);
|
||||
interceptorRegistry.addInterceptor(new LocalServerLoginAccountsInterceptor(redis, environment)) |
||||
.addPathPatterns("/**").order(2); |
||||
|
||||
} |
||||
} |
||||
@ -0,0 +1,32 @@
|
||||
package com.logpm.factorydata.olo.entity; |
||||
|
||||
import com.baomidou.mybatisplus.annotation.TableName; |
||||
import com.fasterxml.jackson.annotation.JsonFormat; |
||||
import io.swagger.annotations.ApiModel; |
||||
import io.swagger.annotations.ApiModelProperty; |
||||
import lombok.Data; |
||||
import org.springblade.core.mp.base.BaseEntity; |
||||
|
||||
import java.util.Date; |
||||
|
||||
@Data |
||||
@TableName("factory_token") |
||||
@ApiModel(value = "FactoryToken对象", description = "工厂账户token") |
||||
public class FactoryToken extends BaseEntity { |
||||
|
||||
@ApiModelProperty("token") |
||||
private String token; |
||||
|
||||
@ApiModelProperty("token摘要") |
||||
private String tokenAbst; |
||||
|
||||
@JsonFormat( |
||||
pattern = "yyyy-MM-dd HH:mm:ss" |
||||
) |
||||
@ApiModelProperty("过期时间") |
||||
private Date expireTime; |
||||
|
||||
@ApiModelProperty("企业id") |
||||
private String corpid; |
||||
|
||||
} |
||||
@ -0,0 +1,87 @@
|
||||
package com.logpm.factorydata.olo.interceptor; |
||||
|
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.logpm.factorydata.olo.service.IFactoryTokenService; |
||||
import com.logpm.factorydata.olo.wrapper.RequestWrapper; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.log4j.Log4j2; |
||||
import org.springblade.common.exception.CustomerException; |
||||
import org.springblade.core.tool.api.R; |
||||
import org.springblade.core.tool.utils.StringUtil; |
||||
import org.springframework.web.servlet.HandlerInterceptor; |
||||
import org.springframework.web.servlet.ModelAndView; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.io.PrintWriter; |
||||
|
||||
@Log4j2 |
||||
@AllArgsConstructor |
||||
public class FactoryAccountsInterceptor implements HandlerInterceptor { |
||||
|
||||
private final IFactoryTokenService factoryTokenService; |
||||
|
||||
@Override |
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws CustomerException { |
||||
try { |
||||
|
||||
RequestWrapper myRequestWrapper = new RequestWrapper(request); |
||||
String body = myRequestWrapper.getBody(); |
||||
JSONObject jsonObject = JSONObject.parseObject(body); |
||||
//获取头中参数
|
||||
String token = request.getHeader("Authorization"); |
||||
String corpId = request.getHeader("corpid"); |
||||
if(StringUtil.isBlank(corpId)){ |
||||
corpId = jsonObject.getString("corpid"); |
||||
} |
||||
if(!StringUtil.hasLength(token)){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,token有误"))); |
||||
return false; |
||||
} |
||||
if(!StringUtil.hasLength(corpId)){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,corpId有误"))); |
||||
return false; |
||||
} |
||||
|
||||
log.info("##########preHandle: token={}",token); |
||||
//验证token
|
||||
boolean b = factoryTokenService.verifyToken(token,corpId); |
||||
if(!b){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,token不存在或已过期"))); |
||||
return false; |
||||
} |
||||
return true; |
||||
} catch (Exception e) { |
||||
returnJson(response,JSONObject.toJSONString(R.fail(500,"服务异常，请联系管理员"))); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void returnJson(HttpServletResponse response, String json) { |
||||
PrintWriter writer = null; |
||||
response.setCharacterEncoding("UTF-8"); |
||||
response.setContentType("application/json"); |
||||
try { |
||||
writer = response.getWriter(); |
||||
writer.print(json); |
||||
|
||||
} catch (IOException e) { |
||||
System.out.println(e.getMessage()); |
||||
} finally { |
||||
if (writer != null){ |
||||
writer.close(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { |
||||
HandlerInterceptor.super.postHandle(request, response, handler, modelAndView); |
||||
} |
||||
|
||||
@Override |
||||
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { |
||||
HandlerInterceptor.super.afterCompletion(request, response, handler, ex); |
||||
} |
||||
} |
||||
@ -0,0 +1,100 @@
|
||||
package com.logpm.factorydata.olo.interceptor; |
||||
|
||||
import cn.hutool.http.HttpRequest; |
||||
import cn.hutool.http.HttpResponse; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.logpm.factorydata.olo.wrapper.CustomHttpServletRequestWrapper; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.log4j.Log4j2; |
||||
import org.springblade.common.cache.CacheNames; |
||||
import org.springblade.common.constant.LauncherConstant; |
||||
import org.springblade.common.exception.CustomerException; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springblade.core.tool.api.R; |
||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||
import org.springframework.core.env.Environment; |
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.web.context.request.RequestContextHolder; |
||||
import org.springframework.web.context.request.ServletRequestAttributes; |
||||
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.io.PrintWriter; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
import java.util.Objects; |
||||
|
||||
@Log4j2 |
||||
@AllArgsConstructor |
||||
public class LocalServerLoginAccountsInterceptor extends HandlerInterceptorAdapter { |
||||
|
||||
private final BladeRedis bladeRedis; |
||||
private final Environment environment; |
||||
@Override |
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws CustomerException { |
||||
|
||||
|
||||
try { |
||||
CustomHttpServletRequestWrapper wrappedRequest = new CustomHttpServletRequestWrapper(request); |
||||
String account ="shujutongbu"; |
||||
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||
if(Objects.isNull(data)){ |
||||
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||
HttpRequest urlRequest = HttpRequest.post(url); |
||||
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||
urlRequest.header("Tenant-Id", "627683"); |
||||
|
||||
Map<String, Object> params = new HashMap<>(); |
||||
params.put("grant_type", "local_server"); |
||||
params.put("scope", "all"); |
||||
params.put("username", account); |
||||
params.put("tenantId", "627683"); |
||||
HttpResponse execute = urlRequest.form(params).execute(); |
||||
String body = execute.body(); |
||||
data = JSONObject.parseObject(body); |
||||
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||
} |
||||
|
||||
// 修改或添加header
|
||||
|
||||
// 发送登录请求
|
||||
log.info("##########preHandle: request={}", request); |
||||
|
||||
wrappedRequest.addHeader("Blade-Auth", "bearer "+data.getString("access_token")); |
||||
HttpHeaders httpHeaders = new HttpHeaders(); |
||||
httpHeaders.add("Blade-Auth","bearer "+data.get("access_token") ); |
||||
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||
|
||||
|
||||
// 用包装后的request替换原始request
|
||||
request = wrappedRequest; |
||||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(request)); |
||||
return true; |
||||
} catch (Exception e) { |
||||
returnJson(response, JSONObject.toJSONString(R.fail(500, "服务异常，请联系管理员"))); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void returnJson(HttpServletResponse response, String json) { |
||||
PrintWriter writer = null; |
||||
response.setCharacterEncoding("UTF-8"); |
||||
response.setContentType("application/json"); |
||||
try { |
||||
writer = response.getWriter(); |
||||
writer.print(json); |
||||
|
||||
} catch (IOException e) { |
||||
System.out.println(e.getMessage()); |
||||
} finally { |
||||
if (writer != null) { |
||||
writer.close(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,31 @@
|
||||
/* |
||||
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* |
||||
* Redistributions of source code must retain the above copyright notice, |
||||
* this list of conditions and the following disclaimer. |
||||
* Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* Neither the name of the dreamlu.net developer nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* Author: Chill 庄骞 (smallchill@163.com) |
||||
*/ |
||||
package com.logpm.factorydata.olo.mapper; |
||||
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
||||
import com.logpm.factorydata.olo.entity.FactoryToken; |
||||
|
||||
/** |
||||
* 工厂账号接口日志 Mapper 接口 |
||||
* |
||||
* @author zhy |
||||
* @since 2023-03-28 |
||||
*/ |
||||
public interface FactoryTokenMapper extends BaseMapper<FactoryToken> { |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,20 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> |
||||
<mapper namespace="com.logpm.factorydata.olo.mapper.FactoryTokenMapper"> |
||||
|
||||
<!-- 通用查询映射结果 --> |
||||
<!-- <resultMap id="orderLogResultMap" type="com.logpm.factory.snm.entity.FactoryOrder">--> |
||||
<!-- <result column="id" property="id"/>--> |
||||
<!-- <result column="req_args" property="reqArgs"/>--> |
||||
<!-- <result column="res_body" property="resBody"/>--> |
||||
<!-- <result column="type" property="type"/>--> |
||||
<!-- <result column="create_user" property="createUser"/>--> |
||||
<!-- <result column="create_time" property="createTime"/>--> |
||||
<!-- <result column="update_user" property="updateUser"/>--> |
||||
<!-- <result column="update_time" property="updateTime"/>--> |
||||
<!-- <result column="status" property="status"/>--> |
||||
<!-- <result column="is_deleted" property="isDeleted"/>--> |
||||
<!-- <result column="create_dept" property="createDept"/>--> |
||||
<!-- </resultMap>--> |
||||
|
||||
</mapper> |
||||
@ -0,0 +1,15 @@
|
||||
package com.logpm.factorydata.olo.service; |
||||
|
||||
import com.logpm.factorydata.olo.entity.FactoryToken; |
||||
import org.springblade.core.mp.base.BaseService; |
||||
|
||||
import java.security.NoSuchAlgorithmException; |
||||
|
||||
/** |
||||
* 工厂推送数据接口 |
||||
*/ |
||||
public interface IFactoryTokenService extends BaseService<FactoryToken> { |
||||
|
||||
|
||||
boolean verifyToken(String token, String corpId) throws NoSuchAlgorithmException; |
||||
} |
||||
@ -0,0 +1,62 @@
|
||||
package com.logpm.factorydata.olo.service.impl; |
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
||||
import com.logpm.factorydata.olo.entity.FactoryToken; |
||||
import com.logpm.factorydata.olo.mapper.FactoryTokenMapper; |
||||
import com.logpm.factorydata.olo.service.IFactoryTokenService; |
||||
import lombok.AllArgsConstructor; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
import org.springblade.core.mp.base.BaseServiceImpl; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springframework.stereotype.Service; |
||||
|
||||
import java.security.NoSuchAlgorithmException; |
||||
import java.util.Objects; |
||||
|
||||
@AllArgsConstructor |
||||
@Service |
||||
public class FactoryTokenServiceImpl extends BaseServiceImpl<FactoryTokenMapper, FactoryToken> implements IFactoryTokenService { |
||||
|
||||
private static final Logger logger = LoggerFactory.getLogger(FactoryTokenServiceImpl.class); |
||||
|
||||
private final BladeRedis bladeRedis; |
||||
|
||||
@Override |
||||
public boolean verifyToken(String token, String corpId) throws NoSuchAlgorithmException { |
||||
logger.info("#########verifyToken: 验证token开始"); |
||||
//先生成token摘要
|
||||
// String tokenAbst = MD5Utils.md5Hex(token.getBytes(StandardCharsets.UTF_8));
|
||||
|
||||
//编写查询条件
|
||||
String key = "corpId:" + corpId; |
||||
FactoryToken factoryToken = bladeRedis.get(key); |
||||
|
||||
if (Objects.isNull(factoryToken)) { |
||||
QueryWrapper<FactoryToken> queryWrapper = new QueryWrapper<>(); |
||||
queryWrapper.eq("token", token); |
||||
factoryToken = baseMapper.selectOne(queryWrapper); |
||||
} |
||||
|
||||
|
||||
if (Objects.isNull(factoryToken)) { |
||||
logger.error("#########verifyToken: token1验证不通过 token={}，corpid={}", token, corpId); |
||||
return false; |
||||
} |
||||
logger.info("#########verifyToken: 验证token 传入token {} 验证token {}", token, factoryToken.getToken()); |
||||
if (!factoryToken.getToken().equals(token)) { |
||||
logger.error("#########verifyToken: token2验证不通过 token={}，corpid={}", token, corpId); |
||||
return false; |
||||
} |
||||
|
||||
|
||||
Long expireTimeLong = factoryToken.getExpireTime().getTime(); |
||||
Long now = System.currentTimeMillis(); |
||||
//判断是否过期
|
||||
if (now > expireTimeLong) { |
||||
logger.error("#########verifyToken: token验证不通过 已过期 token={}，corpId={}", token, corpId); |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
} |
||||
@ -1,43 +0,0 @@
|
||||
package com.logpm.factorydata.olo.util; |
||||
|
||||
import cn.hutool.core.thread.ExecutorBuilder; |
||||
import cn.hutool.core.thread.ThreadFactoryBuilder; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
|
||||
import java.util.concurrent.ExecutorService; |
||||
|
||||
/** |
||||
* @author zhaoqiaobo |
||||
* @create 2024-03-14 10:26 |
||||
*/ |
||||
@Slf4j |
||||
public class ThreadPoolUtil { |
||||
|
||||
private ThreadPoolUtil() { |
||||
} |
||||
|
||||
/** |
||||
* 线程池 |
||||
*/ |
||||
public static ExecutorService executor = null; |
||||
|
||||
static { |
||||
init(); |
||||
} |
||||
|
||||
synchronized public static void init() { |
||||
if (null != executor) { |
||||
executor.shutdownNow(); |
||||
} |
||||
executor = ExecutorBuilder.create() |
||||
.setCorePoolSize(8) |
||||
.setMaxPoolSize(20) |
||||
.setThreadFactory(ThreadFactoryBuilder.create().setNamePrefix("logpm-factory-data-sfy-").build()) |
||||
.useSynchronousQueue().build(); |
||||
} |
||||
|
||||
public static ExecutorService getThreadPool() { |
||||
return executor; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,38 @@
|
||||
package com.logpm.factorydata.olo.wrapper; |
||||
|
||||
import org.springframework.http.HttpHeaders; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletRequestWrapper; |
||||
import java.util.Collections; |
||||
import java.util.Enumeration; |
||||
import java.util.List; |
||||
|
||||
public class CustomHttpServletRequestWrapper extends HttpServletRequestWrapper { |
||||
|
||||
private final HttpHeaders headers; |
||||
|
||||
public CustomHttpServletRequestWrapper(HttpServletRequest request) { |
||||
super(request); |
||||
headers = new HttpHeaders(); |
||||
} |
||||
|
||||
@Override |
||||
public String getHeader(String name) { |
||||
String headerValue = headers.getFirst(name); |
||||
return headerValue != null ? headerValue : super.getHeader(name); |
||||
} |
||||
|
||||
@Override |
||||
public Enumeration<String> getHeaderNames() { |
||||
List<String> names = Collections.list(super.getHeaderNames()); |
||||
headers.forEach((key, value) -> names.add(key)); |
||||
return Collections.enumeration(names); |
||||
} |
||||
|
||||
// 其他需要覆盖的方法...
|
||||
|
||||
public void addHeader(String name, String value) { |
||||
headers.add(name, value); |
||||
} |
||||
} |
||||
@ -1,20 +0,0 @@
|
||||
package com.logpm.factorydata; |
||||
|
||||
import org.springblade.common.constant.ModuleNameConstant; |
||||
import org.springblade.core.cloud.client.BladeCloudApplication; |
||||
import org.springblade.core.launch.BladeApplication; |
||||
|
||||
/** |
||||
* 工厂数据基础服务启动类 |
||||
* |
||||
* @Author zqb |
||||
* @Date 2024/3/14 |
||||
**/ |
||||
@BladeCloudApplication |
||||
public class PianoFactoryDataApplication { |
||||
|
||||
public static void main(String[] args) { |
||||
BladeApplication.run(ModuleNameConstant.LOGPM_FACTORY_DATA_BASE_NAME, PianoFactoryDataApplication.class, args); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,79 @@
|
||||
package com.logpm.factorydata.suofeiya.config; |
||||
|
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.slf4j.MDC; |
||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.task.TaskDecorator; |
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.scheduling.annotation.EnableAsync; |
||||
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
||||
import org.springframework.web.context.request.RequestAttributes; |
||||
import org.springframework.web.context.request.RequestContextHolder; |
||||
|
||||
import javax.annotation.Nonnull; |
||||
import java.util.Map; |
||||
import java.util.concurrent.Executor; |
||||
import java.util.concurrent.ThreadPoolExecutor; |
||||
|
||||
@Configuration |
||||
@Slf4j |
||||
@EnableAsync |
||||
public class ExecutorConfig { |
||||
|
||||
@Bean |
||||
public Executor asyncExecutor() { |
||||
log.info("start async executor"); |
||||
ThreadPoolTaskExecutor threadPoolTaskExecutor = new ThreadPoolTaskExecutor(); |
||||
// 配置核心线程数
|
||||
threadPoolTaskExecutor.setCorePoolSize(10); |
||||
// 配置最大线程数
|
||||
threadPoolTaskExecutor.setMaxPoolSize(20); |
||||
// 配置队列大小
|
||||
threadPoolTaskExecutor.setQueueCapacity(50); |
||||
// 配置线程池中线程的名称前缀
|
||||
threadPoolTaskExecutor.setThreadNamePrefix("ASYNC_THREAD_"); |
||||
// HelloWorldServiceImpl rejection-policy: 当pool已经达到max size时，如何处理新任务：
|
||||
// CallerRunsPolicy: 不在新线程中执行任务，而是由调用者所在的线程来执行；
|
||||
// AbortPolicy: 拒绝执行新任务，并抛出RejectedExecutionException异常；
|
||||
// DiscardPolicy：丢弃当前将要加入队列的任务；
|
||||
// DiscardOldestPolicy：丢弃任务队列中最旧的任务；
|
||||
threadPoolTaskExecutor.setRejectedExecutionHandler( |
||||
new ThreadPoolExecutor.CallerRunsPolicy() |
||||
); |
||||
threadPoolTaskExecutor.setTaskDecorator(new ContextCopyingDecorator()); |
||||
threadPoolTaskExecutor.initialize(); |
||||
return threadPoolTaskExecutor; |
||||
} |
||||
|
||||
|
||||
static class ContextCopyingDecorator implements TaskDecorator { |
||||
@Nonnull |
||||
@Override |
||||
public Runnable decorate(@Nonnull Runnable runnable) { |
||||
RequestAttributes context = RequestContextHolder.currentRequestAttributes(); |
||||
HttpHeaders headers = (HttpHeaders) ThreadLocalUtil.get("bladeContext"); |
||||
Map<String, String> mdcMap = MDC.getCopyOfContextMap(); |
||||
return () -> { |
||||
try { |
||||
ThreadLocalUtil.put("bladeContext", headers); |
||||
if (mdcMap != null && !mdcMap.isEmpty()) { |
||||
MDC.setContextMap(mdcMap); |
||||
} |
||||
RequestContextHolder.setRequestAttributes(context); |
||||
|
||||
runnable.run(); |
||||
} finally { |
||||
RequestContextHolder.resetRequestAttributes(); |
||||
if (mdcMap != null) { |
||||
mdcMap.clear(); |
||||
} |
||||
ThreadLocalUtil.clear(); |
||||
MDC.clear(); |
||||
} |
||||
}; |
||||
} |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,28 @@
|
||||
package com.logpm.factorydata.suofeiya.config; |
||||
|
||||
import com.logpm.factorydata.suofeiya.interceptor.FactoryAccountsInterceptor; |
||||
import com.logpm.factorydata.suofeiya.interceptor.LocalServerLoginAccountsInterceptor; |
||||
import com.logpm.factorydata.suofeiya.service.IFactoryTokenService; |
||||
import lombok.AllArgsConstructor; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.env.Environment; |
||||
import org.springframework.web.servlet.config.annotation.InterceptorRegistry; |
||||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; |
||||
|
||||
@Configuration |
||||
@AllArgsConstructor |
||||
public class InterceptorAdapterConfig implements WebMvcConfigurer { |
||||
|
||||
private final IFactoryTokenService factoryTokenService; |
||||
private final BladeRedis redis; |
||||
private final Environment environment; |
||||
|
||||
@Override |
||||
public void addInterceptors(InterceptorRegistry interceptorRegistry) { |
||||
// interceptorRegistry.addInterceptor(new FactoryAccountsInterceptor(factoryTokenService))
|
||||
// .addPathPatterns("/**").order(1);
|
||||
interceptorRegistry.addInterceptor(new LocalServerLoginAccountsInterceptor(redis, environment)) |
||||
.addPathPatterns("/**").order(2); |
||||
} |
||||
} |
||||
@ -0,0 +1,32 @@
|
||||
package com.logpm.factorydata.suofeiya.entity; |
||||
|
||||
import com.baomidou.mybatisplus.annotation.TableName; |
||||
import com.fasterxml.jackson.annotation.JsonFormat; |
||||
import io.swagger.annotations.ApiModel; |
||||
import io.swagger.annotations.ApiModelProperty; |
||||
import lombok.Data; |
||||
import org.springblade.core.mp.base.BaseEntity; |
||||
|
||||
import java.util.Date; |
||||
|
||||
@Data |
||||
@TableName("factory_token") |
||||
@ApiModel(value = "FactoryToken对象", description = "工厂账户token") |
||||
public class FactoryToken extends BaseEntity { |
||||
|
||||
@ApiModelProperty("token") |
||||
private String token; |
||||
|
||||
@ApiModelProperty("token摘要") |
||||
private String tokenAbst; |
||||
|
||||
@JsonFormat( |
||||
pattern = "yyyy-MM-dd HH:mm:ss" |
||||
) |
||||
@ApiModelProperty("过期时间") |
||||
private Date expireTime; |
||||
|
||||
@ApiModelProperty("企业id") |
||||
private String corpid; |
||||
|
||||
} |
||||
@ -0,0 +1,87 @@
|
||||
package com.logpm.factorydata.suofeiya.interceptor; |
||||
|
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.logpm.factorydata.suofeiya.service.IFactoryTokenService; |
||||
import com.logpm.factorydata.suofeiya.wrapper.RequestWrapper; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.log4j.Log4j2; |
||||
import org.springblade.common.exception.CustomerException; |
||||
import org.springblade.core.tool.api.R; |
||||
import org.springblade.core.tool.utils.StringUtil; |
||||
import org.springframework.web.servlet.HandlerInterceptor; |
||||
import org.springframework.web.servlet.ModelAndView; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.io.PrintWriter; |
||||
|
||||
@Log4j2 |
||||
@AllArgsConstructor |
||||
public class FactoryAccountsInterceptor implements HandlerInterceptor { |
||||
|
||||
private final IFactoryTokenService factoryTokenService; |
||||
|
||||
@Override |
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws CustomerException { |
||||
try { |
||||
|
||||
RequestWrapper myRequestWrapper = new RequestWrapper(request); |
||||
String body = myRequestWrapper.getBody(); |
||||
JSONObject jsonObject = JSONObject.parseObject(body); |
||||
//获取头中参数
|
||||
String token = request.getHeader("Authorization"); |
||||
String corpId = request.getHeader("corpid"); |
||||
if(StringUtil.isBlank(corpId)){ |
||||
corpId = jsonObject.getString("corpid"); |
||||
} |
||||
if(!StringUtil.hasLength(token)){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,token有误"))); |
||||
return false; |
||||
} |
||||
if(!StringUtil.hasLength(corpId)){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,corpId有误"))); |
||||
return false; |
||||
} |
||||
|
||||
log.info("##########preHandle: token={}",token); |
||||
//验证token
|
||||
boolean b = factoryTokenService.verifyToken(token,corpId); |
||||
if(!b){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,token不存在或已过期"))); |
||||
return false; |
||||
} |
||||
return true; |
||||
} catch (Exception e) { |
||||
returnJson(response,JSONObject.toJSONString(R.fail(500,"服务异常，请联系管理员"))); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void returnJson(HttpServletResponse response, String json) { |
||||
PrintWriter writer = null; |
||||
response.setCharacterEncoding("UTF-8"); |
||||
response.setContentType("application/json"); |
||||
try { |
||||
writer = response.getWriter(); |
||||
writer.print(json); |
||||
|
||||
} catch (IOException e) { |
||||
System.out.println(e.getMessage()); |
||||
} finally { |
||||
if (writer != null){ |
||||
writer.close(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { |
||||
HandlerInterceptor.super.postHandle(request, response, handler, modelAndView); |
||||
} |
||||
|
||||
@Override |
||||
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { |
||||
HandlerInterceptor.super.afterCompletion(request, response, handler, ex); |
||||
} |
||||
} |
||||
@ -0,0 +1,100 @@
|
||||
package com.logpm.factorydata.suofeiya.interceptor; |
||||
|
||||
import cn.hutool.http.HttpRequest; |
||||
import cn.hutool.http.HttpResponse; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.logpm.factorydata.suofeiya.wrapper.CustomHttpServletRequestWrapper; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.log4j.Log4j2; |
||||
import org.springblade.common.cache.CacheNames; |
||||
import org.springblade.common.constant.LauncherConstant; |
||||
import org.springblade.common.exception.CustomerException; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springblade.core.tool.api.R; |
||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||
import org.springframework.core.env.Environment; |
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.web.context.request.RequestContextHolder; |
||||
import org.springframework.web.context.request.ServletRequestAttributes; |
||||
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.io.PrintWriter; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
import java.util.Objects; |
||||
|
||||
@Log4j2 |
||||
@AllArgsConstructor |
||||
public class LocalServerLoginAccountsInterceptor extends HandlerInterceptorAdapter { |
||||
|
||||
private final BladeRedis bladeRedis; |
||||
private final Environment environment; |
||||
@Override |
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws CustomerException { |
||||
|
||||
|
||||
try { |
||||
CustomHttpServletRequestWrapper wrappedRequest = new CustomHttpServletRequestWrapper(request); |
||||
String account ="shujutongbu"; |
||||
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||
if(Objects.isNull(data)){ |
||||
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||
HttpRequest urlRequest = HttpRequest.post(url); |
||||
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||
urlRequest.header("Tenant-Id", "627683"); |
||||
|
||||
Map<String, Object> params = new HashMap<>(); |
||||
params.put("grant_type", "local_server"); |
||||
params.put("scope", "all"); |
||||
params.put("username", account); |
||||
params.put("tenantId", "627683"); |
||||
HttpResponse execute = urlRequest.form(params).execute(); |
||||
String body = execute.body(); |
||||
data = JSONObject.parseObject(body); |
||||
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||
} |
||||
|
||||
// 修改或添加header
|
||||
|
||||
// 发送登录请求
|
||||
log.info("##########preHandle: request={}", request); |
||||
|
||||
wrappedRequest.addHeader("Blade-Auth", "bearer "+data.getString("access_token")); |
||||
HttpHeaders httpHeaders = new HttpHeaders(); |
||||
httpHeaders.add("Blade-Auth","bearer "+data.get("access_token") ); |
||||
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||
|
||||
|
||||
// 用包装后的request替换原始request
|
||||
request = wrappedRequest; |
||||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(request)); |
||||
return true; |
||||
} catch (Exception e) { |
||||
returnJson(response, JSONObject.toJSONString(R.fail(500, "服务异常，请联系管理员"))); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void returnJson(HttpServletResponse response, String json) { |
||||
PrintWriter writer = null; |
||||
response.setCharacterEncoding("UTF-8"); |
||||
response.setContentType("application/json"); |
||||
try { |
||||
writer = response.getWriter(); |
||||
writer.print(json); |
||||
|
||||
} catch (IOException e) { |
||||
System.out.println(e.getMessage()); |
||||
} finally { |
||||
if (writer != null) { |
||||
writer.close(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,31 @@
|
||||
/* |
||||
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* |
||||
* Redistributions of source code must retain the above copyright notice, |
||||
* this list of conditions and the following disclaimer. |
||||
* Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* Neither the name of the dreamlu.net developer nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* Author: Chill 庄骞 (smallchill@163.com) |
||||
*/ |
||||
package com.logpm.factorydata.suofeiya.mapper; |
||||
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
||||
import com.logpm.factorydata.suofeiya.entity.FactoryToken; |
||||
|
||||
/** |
||||
* 工厂账号接口日志 Mapper 接口 |
||||
* |
||||
* @author zhy |
||||
* @since 2023-03-28 |
||||
*/ |
||||
public interface FactoryTokenMapper extends BaseMapper<FactoryToken> { |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,20 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> |
||||
<mapper namespace="com.logpm.factorydata.suofeiya.mapper.FactoryTokenMapper"> |
||||
|
||||
<!-- 通用查询映射结果 --> |
||||
<!-- <resultMap id="orderLogResultMap" type="com.logpm.factory.snm.entity.FactoryOrder">--> |
||||
<!-- <result column="id" property="id"/>--> |
||||
<!-- <result column="req_args" property="reqArgs"/>--> |
||||
<!-- <result column="res_body" property="resBody"/>--> |
||||
<!-- <result column="type" property="type"/>--> |
||||
<!-- <result column="create_user" property="createUser"/>--> |
||||
<!-- <result column="create_time" property="createTime"/>--> |
||||
<!-- <result column="update_user" property="updateUser"/>--> |
||||
<!-- <result column="update_time" property="updateTime"/>--> |
||||
<!-- <result column="status" property="status"/>--> |
||||
<!-- <result column="is_deleted" property="isDeleted"/>--> |
||||
<!-- <result column="create_dept" property="createDept"/>--> |
||||
<!-- </resultMap>--> |
||||
|
||||
</mapper> |
||||
@ -0,0 +1,15 @@
|
||||
package com.logpm.factorydata.suofeiya.service; |
||||
|
||||
import com.logpm.factorydata.suofeiya.entity.FactoryToken; |
||||
import org.springblade.core.mp.base.BaseService; |
||||
|
||||
import java.security.NoSuchAlgorithmException; |
||||
|
||||
/** |
||||
* 工厂推送数据接口 |
||||
*/ |
||||
public interface IFactoryTokenService extends BaseService<FactoryToken> { |
||||
|
||||
|
||||
boolean verifyToken(String token, String corpId) throws NoSuchAlgorithmException; |
||||
} |
||||
@ -0,0 +1,62 @@
|
||||
package com.logpm.factorydata.suofeiya.service.impl; |
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
||||
import com.logpm.factorydata.suofeiya.entity.FactoryToken; |
||||
import com.logpm.factorydata.suofeiya.mapper.FactoryTokenMapper; |
||||
import com.logpm.factorydata.suofeiya.service.IFactoryTokenService; |
||||
import lombok.AllArgsConstructor; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
import org.springblade.core.mp.base.BaseServiceImpl; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springframework.stereotype.Service; |
||||
|
||||
import java.security.NoSuchAlgorithmException; |
||||
import java.util.Objects; |
||||
|
||||
@AllArgsConstructor |
||||
@Service |
||||
public class FactoryTokenServiceImpl extends BaseServiceImpl<FactoryTokenMapper, FactoryToken> implements IFactoryTokenService { |
||||
|
||||
private static final Logger logger = LoggerFactory.getLogger(FactoryTokenServiceImpl.class); |
||||
|
||||
private final BladeRedis bladeRedis; |
||||
|
||||
@Override |
||||
public boolean verifyToken(String token, String corpId) throws NoSuchAlgorithmException { |
||||
logger.info("#########verifyToken: 验证token开始"); |
||||
//先生成token摘要
|
||||
// String tokenAbst = MD5Utils.md5Hex(token.getBytes(StandardCharsets.UTF_8));
|
||||
|
||||
//编写查询条件
|
||||
String key = "corpId:" + corpId; |
||||
FactoryToken factoryToken = bladeRedis.get(key); |
||||
|
||||
if (Objects.isNull(factoryToken)) { |
||||
QueryWrapper<FactoryToken> queryWrapper = new QueryWrapper<>(); |
||||
queryWrapper.eq("token", token); |
||||
factoryToken = baseMapper.selectOne(queryWrapper); |
||||
} |
||||
|
||||
|
||||
if (Objects.isNull(factoryToken)) { |
||||
logger.error("#########verifyToken: token1验证不通过 token={}，corpid={}", token, corpId); |
||||
return false; |
||||
} |
||||
logger.info("#########verifyToken: 验证token 传入token {} 验证token {}", token, factoryToken.getToken()); |
||||
if (!factoryToken.getToken().equals(token)) { |
||||
logger.error("#########verifyToken: token2验证不通过 token={}，corpid={}", token, corpId); |
||||
return false; |
||||
} |
||||
|
||||
|
||||
Long expireTimeLong = factoryToken.getExpireTime().getTime(); |
||||
Long now = System.currentTimeMillis(); |
||||
//判断是否过期
|
||||
if (now > expireTimeLong) { |
||||
logger.error("#########verifyToken: token验证不通过 已过期 token={}，corpId={}", token, corpId); |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
} |
||||
@ -1,43 +0,0 @@
|
||||
package com.logpm.factorydata.suofeiya.util; |
||||
|
||||
import cn.hutool.core.thread.ExecutorBuilder; |
||||
import cn.hutool.core.thread.ThreadFactoryBuilder; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
|
||||
import java.util.concurrent.ExecutorService; |
||||
|
||||
/** |
||||
* @author zhaoqiaobo |
||||
* @create 2024-03-14 10:26 |
||||
*/ |
||||
@Slf4j |
||||
public class ThreadPoolUtil { |
||||
|
||||
private ThreadPoolUtil() { |
||||
} |
||||
|
||||
/** |
||||
* 线程池 |
||||
*/ |
||||
public static ExecutorService executor = null; |
||||
|
||||
static { |
||||
init(); |
||||
} |
||||
|
||||
synchronized public static void init() { |
||||
if (null != executor) { |
||||
executor.shutdownNow(); |
||||
} |
||||
executor = ExecutorBuilder.create() |
||||
.setCorePoolSize(8) |
||||
.setMaxPoolSize(20) |
||||
.setThreadFactory(ThreadFactoryBuilder.create().setNamePrefix("logpm-factory-data-sfy-").build()) |
||||
.useSynchronousQueue().build(); |
||||
} |
||||
|
||||
public static ExecutorService getThreadPool() { |
||||
return executor; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,38 @@
|
||||
package com.logpm.factorydata.suofeiya.wrapper; |
||||
|
||||
import org.springframework.http.HttpHeaders; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletRequestWrapper; |
||||
import java.util.Collections; |
||||
import java.util.Enumeration; |
||||
import java.util.List; |
||||
|
||||
public class CustomHttpServletRequestWrapper extends HttpServletRequestWrapper { |
||||
|
||||
private final HttpHeaders headers; |
||||
|
||||
public CustomHttpServletRequestWrapper(HttpServletRequest request) { |
||||
super(request); |
||||
headers = new HttpHeaders(); |
||||
} |
||||
|
||||
@Override |
||||
public String getHeader(String name) { |
||||
String headerValue = headers.getFirst(name); |
||||
return headerValue != null ? headerValue : super.getHeader(name); |
||||
} |
||||
|
||||
@Override |
||||
public Enumeration<String> getHeaderNames() { |
||||
List<String> names = Collections.list(super.getHeaderNames()); |
||||
headers.forEach((key, value) -> names.add(key)); |
||||
return Collections.enumeration(names); |
||||
} |
||||
|
||||
// 其他需要覆盖的方法...
|
||||
|
||||
public void addHeader(String name, String value) { |
||||
headers.add(name, value); |
||||
} |
||||
} |
||||
@ -0,0 +1,75 @@
|
||||
package com.logpm.factorydata.suofeiya.wrapper; |
||||
|
||||
import javax.servlet.ReadListener; |
||||
import javax.servlet.ServletInputStream; |
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletRequestWrapper; |
||||
import java.io.BufferedReader; |
||||
import java.io.ByteArrayInputStream; |
||||
import java.io.IOException; |
||||
import java.io.InputStream; |
||||
import java.io.InputStreamReader; |
||||
|
||||
public class RequestWrapper extends HttpServletRequestWrapper { |
||||
private final String body; |
||||
public RequestWrapper(HttpServletRequest request) throws IOException { |
||||
super(request); |
||||
StringBuilder stringBuilder = new StringBuilder(); |
||||
BufferedReader bufferedReader = null; |
||||
try { |
||||
InputStream inputStream = request.getInputStream(); |
||||
if (inputStream != null) { |
||||
bufferedReader = new BufferedReader(new InputStreamReader(inputStream)); |
||||
char[] charBuffer = new char[128]; |
||||
int bytesRead = -1; |
||||
while ((bytesRead = bufferedReader.read(charBuffer)) > 0) { |
||||
stringBuilder.append(charBuffer, 0, bytesRead); |
||||
} |
||||
} else { |
||||
stringBuilder.append(""); |
||||
} |
||||
} catch (IOException ex) { |
||||
throw ex; |
||||
} finally { |
||||
if (bufferedReader != null) { |
||||
try { |
||||
bufferedReader.close(); |
||||
} catch (IOException ex) { |
||||
throw ex; |
||||
} |
||||
} |
||||
} |
||||
body = stringBuilder.toString(); |
||||
} |
||||
|
||||
@Override |
||||
public ServletInputStream getInputStream() throws IOException { |
||||
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes()); |
||||
ServletInputStream servletInputStream = new ServletInputStream() { |
||||
@Override |
||||
public boolean isFinished() { |
||||
return false; |
||||
} |
||||
@Override |
||||
public boolean isReady() { |
||||
return false; |
||||
} |
||||
@Override |
||||
public void setReadListener(ReadListener readListener) {} |
||||
@Override |
||||
public int read() throws IOException { |
||||
return byteArrayInputStream.read(); |
||||
} |
||||
}; |
||||
return servletInputStream; |
||||
|
||||
} |
||||
@Override |
||||
public BufferedReader getReader() throws IOException { |
||||
return new BufferedReader(new InputStreamReader(this.getInputStream())); |
||||
} |
||||
public String getBody() { |
||||
return this.body; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,79 @@
|
||||
package com.logpm.factorydata.zbom.config; |
||||
|
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.slf4j.MDC; |
||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.task.TaskDecorator; |
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.scheduling.annotation.EnableAsync; |
||||
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
||||
import org.springframework.web.context.request.RequestAttributes; |
||||
import org.springframework.web.context.request.RequestContextHolder; |
||||
|
||||
import javax.annotation.Nonnull; |
||||
import java.util.Map; |
||||
import java.util.concurrent.Executor; |
||||
import java.util.concurrent.ThreadPoolExecutor; |
||||
|
||||
@Configuration |
||||
@Slf4j |
||||
@EnableAsync |
||||
public class ExecutorConfig { |
||||
|
||||
@Bean |
||||
public Executor asyncExecutor() { |
||||
log.info("start async executor"); |
||||
ThreadPoolTaskExecutor threadPoolTaskExecutor = new ThreadPoolTaskExecutor(); |
||||
// 配置核心线程数
|
||||
threadPoolTaskExecutor.setCorePoolSize(10); |
||||
// 配置最大线程数
|
||||
threadPoolTaskExecutor.setMaxPoolSize(20); |
||||
// 配置队列大小
|
||||
threadPoolTaskExecutor.setQueueCapacity(50); |
||||
// 配置线程池中线程的名称前缀
|
||||
threadPoolTaskExecutor.setThreadNamePrefix("ASYNC_THREAD_"); |
||||
// HelloWorldServiceImpl rejection-policy: 当pool已经达到max size时，如何处理新任务：
|
||||
// CallerRunsPolicy: 不在新线程中执行任务，而是由调用者所在的线程来执行；
|
||||
// AbortPolicy: 拒绝执行新任务，并抛出RejectedExecutionException异常；
|
||||
// DiscardPolicy：丢弃当前将要加入队列的任务；
|
||||
// DiscardOldestPolicy：丢弃任务队列中最旧的任务；
|
||||
threadPoolTaskExecutor.setRejectedExecutionHandler( |
||||
new ThreadPoolExecutor.CallerRunsPolicy() |
||||
); |
||||
threadPoolTaskExecutor.setTaskDecorator(new ContextCopyingDecorator()); |
||||
threadPoolTaskExecutor.initialize(); |
||||
return threadPoolTaskExecutor; |
||||
} |
||||
|
||||
|
||||
static class ContextCopyingDecorator implements TaskDecorator { |
||||
@Nonnull |
||||
@Override |
||||
public Runnable decorate(@Nonnull Runnable runnable) { |
||||
RequestAttributes context = RequestContextHolder.currentRequestAttributes(); |
||||
HttpHeaders headers = (HttpHeaders) ThreadLocalUtil.get("bladeContext"); |
||||
Map<String, String> mdcMap = MDC.getCopyOfContextMap(); |
||||
return () -> { |
||||
try { |
||||
ThreadLocalUtil.put("bladeContext", headers); |
||||
if (mdcMap != null && !mdcMap.isEmpty()) { |
||||
MDC.setContextMap(mdcMap); |
||||
} |
||||
RequestContextHolder.setRequestAttributes(context); |
||||
|
||||
runnable.run(); |
||||
} finally { |
||||
RequestContextHolder.resetRequestAttributes(); |
||||
if (mdcMap != null) { |
||||
mdcMap.clear(); |
||||
} |
||||
ThreadLocalUtil.clear(); |
||||
MDC.clear(); |
||||
} |
||||
}; |
||||
} |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,29 @@
|
||||
package com.logpm.factorydata.zbom.config; |
||||
|
||||
import com.logpm.factorydata.zbom.interceptor.LocalServerLoginAccountsInterceptor; |
||||
import com.logpm.factorydata.zbom.interceptor.ZbFactoryAccountsInterceptor; |
||||
import com.logpm.factorydata.zbom.pros.ZbFactoryProperties; |
||||
import com.logpm.factorydata.zbom.service.IFactoryTokenService; |
||||
import lombok.AllArgsConstructor; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.env.Environment; |
||||
import org.springframework.web.servlet.config.annotation.InterceptorRegistry; |
||||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; |
||||
|
||||
@Configuration |
||||
@AllArgsConstructor |
||||
public class InterceptorAdapterConfig implements WebMvcConfigurer { |
||||
|
||||
private final BladeRedis redis; |
||||
private final ZbFactoryProperties zbFactoryProperties; |
||||
private final Environment environment; |
||||
|
||||
@Override |
||||
public void addInterceptors(InterceptorRegistry interceptorRegistry) { |
||||
interceptorRegistry.addInterceptor(new ZbFactoryAccountsInterceptor(zbFactoryProperties)) |
||||
.addPathPatterns("/**").order(2); |
||||
interceptorRegistry.addInterceptor(new LocalServerLoginAccountsInterceptor(redis, environment)) |
||||
.addPathPatterns("/**").order(3); |
||||
} |
||||
} |
||||
@ -0,0 +1,58 @@
|
||||
package com.logpm.factorydata.zbom.entity; |
||||
|
||||
import com.baomidou.mybatisplus.annotation.TableName; |
||||
import io.swagger.annotations.ApiModel; |
||||
import io.swagger.annotations.ApiModelProperty; |
||||
import lombok.Data; |
||||
import lombok.EqualsAndHashCode; |
||||
import org.springblade.core.mp.base.BaseEntity; |
||||
|
||||
@Data |
||||
@TableName("zb_factory_log") |
||||
@ApiModel(value = "接收工厂订单日志", description = "接收工厂订单日志") |
||||
@EqualsAndHashCode(callSuper = true) |
||||
public class FactoryLogEntity extends BaseEntity { |
||||
|
||||
/** |
||||
* 预留1 |
||||
*/ |
||||
@ApiModelProperty(name = "预留1", notes = "") |
||||
private String reserve1; |
||||
/** |
||||
* 预留2 |
||||
*/ |
||||
@ApiModelProperty(name = "预留2", notes = "") |
||||
private String reserve2; |
||||
/** |
||||
* 预留3 |
||||
*/ |
||||
@ApiModelProperty(name = "预留3", notes = "") |
||||
private String reserve3; |
||||
/** |
||||
* 预留4 |
||||
*/ |
||||
@ApiModelProperty(name = "预留4", notes = "") |
||||
private String reserve4; |
||||
/** |
||||
* 预留5 |
||||
*/ |
||||
@ApiModelProperty(name = "预留5", notes = "") |
||||
private String reserve5; |
||||
/** |
||||
* 日志文件存放地址 |
||||
*/ |
||||
@ApiModelProperty(name = "日志文件存放地址", notes = "") |
||||
private String logUrl; |
||||
/** |
||||
* 数据类型 1.入库数据 |
||||
*/ |
||||
@ApiModelProperty(name = "数据类型 1.入库数据", notes = "") |
||||
private Integer type; |
||||
/** |
||||
* 解析状态 0 未解析 1 解析失败 2 已解析 |
||||
*/ |
||||
@ApiModelProperty(name = "解析状态 0 未解析 1 解析失败 2 已解析", notes = "") |
||||
private Integer saxStatus; |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,32 @@
|
||||
package com.logpm.factorydata.zbom.entity; |
||||
|
||||
import com.baomidou.mybatisplus.annotation.TableName; |
||||
import com.fasterxml.jackson.annotation.JsonFormat; |
||||
import io.swagger.annotations.ApiModel; |
||||
import io.swagger.annotations.ApiModelProperty; |
||||
import lombok.Data; |
||||
import org.springblade.core.mp.base.BaseEntity; |
||||
|
||||
import java.util.Date; |
||||
|
||||
@Data |
||||
@TableName("factory_token") |
||||
@ApiModel(value = "FactoryToken对象", description = "工厂账户token") |
||||
public class FactoryToken extends BaseEntity { |
||||
|
||||
@ApiModelProperty("token") |
||||
private String token; |
||||
|
||||
@ApiModelProperty("token摘要") |
||||
private String tokenAbst; |
||||
|
||||
@JsonFormat( |
||||
pattern = "yyyy-MM-dd HH:mm:ss" |
||||
) |
||||
@ApiModelProperty("过期时间") |
||||
private Date expireTime; |
||||
|
||||
@ApiModelProperty("企业id") |
||||
private String corpid; |
||||
|
||||
} |
||||
@ -0,0 +1,87 @@
|
||||
package com.logpm.factorydata.zbom.interceptor; |
||||
|
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.logpm.factorydata.zbom.service.IFactoryTokenService; |
||||
import com.logpm.factorydata.zbom.wrapper.RequestWrapper; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.log4j.Log4j2; |
||||
import org.springblade.common.exception.CustomerException; |
||||
import org.springblade.core.tool.api.R; |
||||
import org.springblade.core.tool.utils.StringUtil; |
||||
import org.springframework.web.servlet.HandlerInterceptor; |
||||
import org.springframework.web.servlet.ModelAndView; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.io.PrintWriter; |
||||
|
||||
@Log4j2 |
||||
@AllArgsConstructor |
||||
public class FactoryAccountsInterceptor implements HandlerInterceptor { |
||||
|
||||
private final IFactoryTokenService factoryTokenService; |
||||
|
||||
@Override |
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws CustomerException { |
||||
try { |
||||
|
||||
RequestWrapper myRequestWrapper = new RequestWrapper(request); |
||||
String body = myRequestWrapper.getBody(); |
||||
JSONObject jsonObject = JSONObject.parseObject(body); |
||||
//获取头中参数
|
||||
String token = request.getHeader("Authorization"); |
||||
String corpId = request.getHeader("corpid"); |
||||
if(StringUtil.isBlank(corpId)){ |
||||
corpId = jsonObject.getString("corpid"); |
||||
} |
||||
if(!StringUtil.hasLength(token)){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,token有误"))); |
||||
return false; |
||||
} |
||||
if(!StringUtil.hasLength(corpId)){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,corpId有误"))); |
||||
return false; |
||||
} |
||||
|
||||
log.info("##########preHandle: token={}",token); |
||||
//验证token
|
||||
boolean b = factoryTokenService.verifyToken(token,corpId); |
||||
if(!b){ |
||||
returnJson(response,JSONObject.toJSONString(R.fail(203,"认证不通过,token不存在或已过期"))); |
||||
return false; |
||||
} |
||||
return true; |
||||
} catch (Exception e) { |
||||
returnJson(response,JSONObject.toJSONString(R.fail(500,"服务异常，请联系管理员"))); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void returnJson(HttpServletResponse response, String json) { |
||||
PrintWriter writer = null; |
||||
response.setCharacterEncoding("UTF-8"); |
||||
response.setContentType("application/json"); |
||||
try { |
||||
writer = response.getWriter(); |
||||
writer.print(json); |
||||
|
||||
} catch (IOException e) { |
||||
System.out.println(e.getMessage()); |
||||
} finally { |
||||
if (writer != null){ |
||||
writer.close(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { |
||||
HandlerInterceptor.super.postHandle(request, response, handler, modelAndView); |
||||
} |
||||
|
||||
@Override |
||||
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { |
||||
HandlerInterceptor.super.afterCompletion(request, response, handler, ex); |
||||
} |
||||
} |
||||
@ -0,0 +1,100 @@
|
||||
package com.logpm.factorydata.zbom.interceptor; |
||||
|
||||
import cn.hutool.http.HttpRequest; |
||||
import cn.hutool.http.HttpResponse; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.logpm.factorydata.zbom.wrapper.CustomHttpServletRequestWrapper; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.log4j.Log4j2; |
||||
import org.springblade.common.cache.CacheNames; |
||||
import org.springblade.common.constant.LauncherConstant; |
||||
import org.springblade.common.exception.CustomerException; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springblade.core.tool.api.R; |
||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||
import org.springframework.core.env.Environment; |
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.web.context.request.RequestContextHolder; |
||||
import org.springframework.web.context.request.ServletRequestAttributes; |
||||
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.io.PrintWriter; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
import java.util.Objects; |
||||
|
||||
@Log4j2 |
||||
@AllArgsConstructor |
||||
public class LocalServerLoginAccountsInterceptor extends HandlerInterceptorAdapter { |
||||
|
||||
private final BladeRedis bladeRedis; |
||||
private final Environment environment; |
||||
@Override |
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws CustomerException { |
||||
|
||||
|
||||
try { |
||||
CustomHttpServletRequestWrapper wrappedRequest = new CustomHttpServletRequestWrapper(request); |
||||
String account ="shujutongbu"; |
||||
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||
if(Objects.isNull(data)){ |
||||
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||
HttpRequest urlRequest = HttpRequest.post(url); |
||||
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||
urlRequest.header("Tenant-Id", "627683"); |
||||
|
||||
Map<String, Object> params = new HashMap<>(); |
||||
params.put("grant_type", "local_server"); |
||||
params.put("scope", "all"); |
||||
params.put("username", account); |
||||
params.put("tenantId", "627683"); |
||||
HttpResponse execute = urlRequest.form(params).execute(); |
||||
String body = execute.body(); |
||||
data = JSONObject.parseObject(body); |
||||
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||
} |
||||
|
||||
// 修改或添加header
|
||||
|
||||
// 发送登录请求
|
||||
log.info("##########preHandle: request={}", request); |
||||
|
||||
wrappedRequest.addHeader("Blade-Auth", "bearer "+data.getString("access_token")); |
||||
HttpHeaders httpHeaders = new HttpHeaders(); |
||||
httpHeaders.add("Blade-Auth","bearer "+data.get("access_token") ); |
||||
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||
|
||||
|
||||
// 用包装后的request替换原始request
|
||||
request = wrappedRequest; |
||||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(request)); |
||||
return true; |
||||
} catch (Exception e) { |
||||
returnJson(response, JSONObject.toJSONString(R.fail(500, "服务异常，请联系管理员"))); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void returnJson(HttpServletResponse response, String json) { |
||||
PrintWriter writer = null; |
||||
response.setCharacterEncoding("UTF-8"); |
||||
response.setContentType("application/json"); |
||||
try { |
||||
writer = response.getWriter(); |
||||
writer.print(json); |
||||
|
||||
} catch (IOException e) { |
||||
System.out.println(e.getMessage()); |
||||
} finally { |
||||
if (writer != null) { |
||||
writer.close(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,32 @@
|
||||
/* |
||||
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* |
||||
* Redistributions of source code must retain the above copyright notice, |
||||
* this list of conditions and the following disclaimer. |
||||
* Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* Neither the name of the dreamlu.net developer nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* Author: Chill 庄骞 (smallchill@163.com) |
||||
*/ |
||||
package com.logpm.factorydata.zbom.mapper; |
||||
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
||||
import com.logpm.factorydata.zbom.entity.FactoryLogEntity; |
||||
import org.apache.ibatis.annotations.Mapper; |
||||
|
||||
/** |
||||
* 接收发货单日志 Mapper |
||||
* |
||||
* @author zqb |
||||
* @since 2024-03-26 |
||||
*/ |
||||
@Mapper |
||||
public interface FactoryLogMapper extends BaseMapper<FactoryLogEntity> { |
||||
|
||||
} |
||||
@ -0,0 +1,5 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> |
||||
<mapper namespace="com.logpm.factorydata.zbom.mapper.FactoryLogMapper"> |
||||
|
||||
</mapper> |
||||
@ -0,0 +1,31 @@
|
||||
/* |
||||
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* |
||||
* Redistributions of source code must retain the above copyright notice, |
||||
* this list of conditions and the following disclaimer. |
||||
* Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* Neither the name of the dreamlu.net developer nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* Author: Chill 庄骞 (smallchill@163.com) |
||||
*/ |
||||
package com.logpm.factorydata.zbom.mapper; |
||||
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
||||
import com.logpm.factorydata.zbom.entity.FactoryToken; |
||||
|
||||
/** |
||||
* 工厂账号接口日志 Mapper 接口 |
||||
* |
||||
* @author zhy |
||||
* @since 2023-03-28 |
||||
*/ |
||||
public interface FactoryTokenMapper extends BaseMapper<FactoryToken> { |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,20 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> |
||||
<mapper namespace="com.logpm.factorydata.zbom.mapper.FactoryTokenMapper"> |
||||
|
||||
<!-- 通用查询映射结果 --> |
||||
<!-- <resultMap id="orderLogResultMap" type="com.logpm.factory.snm.entity.FactoryOrder">--> |
||||
<!-- <result column="id" property="id"/>--> |
||||
<!-- <result column="req_args" property="reqArgs"/>--> |
||||
<!-- <result column="res_body" property="resBody"/>--> |
||||
<!-- <result column="type" property="type"/>--> |
||||
<!-- <result column="create_user" property="createUser"/>--> |
||||
<!-- <result column="create_time" property="createTime"/>--> |
||||
<!-- <result column="update_user" property="updateUser"/>--> |
||||
<!-- <result column="update_time" property="updateTime"/>--> |
||||
<!-- <result column="status" property="status"/>--> |
||||
<!-- <result column="is_deleted" property="isDeleted"/>--> |
||||
<!-- <result column="create_dept" property="createDept"/>--> |
||||
<!-- </resultMap>--> |
||||
|
||||
</mapper> |
||||
@ -0,0 +1,14 @@
|
||||
package com.logpm.factorydata.zbom.service; |
||||
|
||||
import com.logpm.factorydata.zbom.entity.FactoryLogEntity; |
||||
import org.springblade.core.mp.base.BaseService; |
||||
|
||||
/** |
||||
* 工厂订单日志 服务类 |
||||
* |
||||
* @Author zqb |
||||
* @Date 2024/4/26 |
||||
**/ |
||||
public interface FactoryLogService extends BaseService<FactoryLogEntity> { |
||||
|
||||
} |
||||
@ -0,0 +1,15 @@
|
||||
package com.logpm.factorydata.zbom.service; |
||||
|
||||
import com.logpm.factorydata.zbom.entity.FactoryToken; |
||||
import org.springblade.core.mp.base.BaseService; |
||||
|
||||
import java.security.NoSuchAlgorithmException; |
||||
|
||||
/** |
||||
* 工厂推送数据接口 |
||||
*/ |
||||
public interface IFactoryTokenService extends BaseService<FactoryToken> { |
||||
|
||||
|
||||
boolean verifyToken(String token, String corpId) throws NoSuchAlgorithmException; |
||||
} |
||||
@ -0,0 +1,22 @@
|
||||
package com.logpm.factorydata.zbom.service.impl; |
||||
|
||||
import com.logpm.factorydata.zbom.entity.FactoryLogEntity; |
||||
import com.logpm.factorydata.zbom.mapper.FactoryLogMapper; |
||||
import com.logpm.factorydata.zbom.service.FactoryLogService; |
||||
import lombok.AllArgsConstructor; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.springblade.core.mp.base.BaseServiceImpl; |
||||
import org.springframework.stereotype.Service; |
||||
|
||||
/** |
||||
* 工厂订单日志 业务实现类 |
||||
* |
||||
* @Author zqb |
||||
* @Date 2024/4/26 |
||||
**/ |
||||
@Slf4j |
||||
@Service |
||||
@AllArgsConstructor |
||||
public class FactoryLogServiceImpl extends BaseServiceImpl<FactoryLogMapper, FactoryLogEntity> implements FactoryLogService { |
||||
|
||||
} |
||||
@ -0,0 +1,62 @@
|
||||
package com.logpm.factorydata.zbom.service.impl; |
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
||||
import com.logpm.factorydata.zbom.entity.FactoryToken; |
||||
import com.logpm.factorydata.zbom.mapper.FactoryTokenMapper; |
||||
import com.logpm.factorydata.zbom.service.IFactoryTokenService; |
||||
import lombok.AllArgsConstructor; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
import org.springblade.core.mp.base.BaseServiceImpl; |
||||
import org.springblade.core.redis.cache.BladeRedis; |
||||
import org.springframework.stereotype.Service; |
||||
|
||||
import java.security.NoSuchAlgorithmException; |
||||
import java.util.Objects; |
||||
|
||||
@AllArgsConstructor |
||||
@Service |
||||
public class FactoryTokenServiceImpl extends BaseServiceImpl<FactoryTokenMapper, FactoryToken> implements IFactoryTokenService { |
||||
|
||||
private static final Logger logger = LoggerFactory.getLogger(FactoryTokenServiceImpl.class); |
||||
|
||||
private final BladeRedis bladeRedis; |
||||
|
||||
@Override |
||||
public boolean verifyToken(String token, String corpId) throws NoSuchAlgorithmException { |
||||
logger.info("#########verifyToken: 验证token开始"); |
||||
//先生成token摘要
|
||||
// String tokenAbst = MD5Utils.md5Hex(token.getBytes(StandardCharsets.UTF_8));
|
||||
|
||||
//编写查询条件
|
||||
String key = "corpId:" + corpId; |
||||
FactoryToken factoryToken = bladeRedis.get(key); |
||||
|
||||
if (Objects.isNull(factoryToken)) { |
||||
QueryWrapper<FactoryToken> queryWrapper = new QueryWrapper<>(); |
||||
queryWrapper.eq("token", token); |
||||
factoryToken = baseMapper.selectOne(queryWrapper); |
||||
} |
||||
|
||||
|
||||
if (Objects.isNull(factoryToken)) { |
||||
logger.error("#########verifyToken: token1验证不通过 token={}，corpid={}", token, corpId); |
||||
return false; |
||||
} |
||||
logger.info("#########verifyToken: 验证token 传入token {} 验证token {}", token, factoryToken.getToken()); |
||||
if (!factoryToken.getToken().equals(token)) { |
||||
logger.error("#########verifyToken: token2验证不通过 token={}，corpid={}", token, corpId); |
||||
return false; |
||||
} |
||||
|
||||
|
||||
Long expireTimeLong = factoryToken.getExpireTime().getTime(); |
||||
Long now = System.currentTimeMillis(); |
||||
//判断是否过期
|
||||
if (now > expireTimeLong) { |
||||
logger.error("#########verifyToken: token验证不通过 已过期 token={}，corpId={}", token, corpId); |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
} |
||||
@ -1,43 +0,0 @@
|
||||
package com.logpm.factorydata.zbom.util; |
||||
|
||||
import cn.hutool.core.thread.ExecutorBuilder; |
||||
import cn.hutool.core.thread.ThreadFactoryBuilder; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
|
||||
import java.util.concurrent.ExecutorService; |
||||
|
||||
/** |
||||
* @author zhaoqiaobo |
||||
* @create 2024-03-14 10:26 |
||||
*/ |
||||
@Slf4j |
||||
public class ThreadPoolUtil { |
||||
|
||||
private ThreadPoolUtil() { |
||||
} |
||||
|
||||
/** |
||||
* 线程池 |
||||
*/ |
||||
public static ExecutorService executor = null; |
||||
|
||||
static { |
||||
init(); |
||||
} |
||||
|
||||
synchronized public static void init() { |
||||
if (null != executor) { |
||||
executor.shutdownNow(); |
||||
} |
||||
executor = ExecutorBuilder.create() |
||||
.setCorePoolSize(8) |
||||
.setMaxPoolSize(20) |
||||
.setThreadFactory(ThreadFactoryBuilder.create().setNamePrefix("logpm-factory-data-zbom-").build()) |
||||
.useSynchronousQueue().build(); |
||||
} |
||||
|
||||
public static ExecutorService getThreadPool() { |
||||
return executor; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,38 @@
|
||||
package com.logpm.factorydata.zbom.wrapper; |
||||
|
||||
import org.springframework.http.HttpHeaders; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletRequestWrapper; |
||||
import java.util.Collections; |
||||
import java.util.Enumeration; |
||||
import java.util.List; |
||||
|
||||
public class CustomHttpServletRequestWrapper extends HttpServletRequestWrapper { |
||||
|
||||
private final HttpHeaders headers; |
||||
|
||||
public CustomHttpServletRequestWrapper(HttpServletRequest request) { |
||||
super(request); |
||||
headers = new HttpHeaders(); |
||||
} |
||||
|
||||
@Override |
||||
public String getHeader(String name) { |
||||
String headerValue = headers.getFirst(name); |
||||
return headerValue != null ? headerValue : super.getHeader(name); |
||||
} |
||||
|
||||
@Override |
||||
public Enumeration<String> getHeaderNames() { |
||||
List<String> names = Collections.list(super.getHeaderNames()); |
||||
headers.forEach((key, value) -> names.add(key)); |
||||
return Collections.enumeration(names); |
||||
} |
||||
|
||||
// 其他需要覆盖的方法...
|
||||
|
||||
public void addHeader(String name, String value) { |
||||
headers.add(name, value); |
||||
} |
||||
} |
||||
@ -0,0 +1,75 @@
|
||||
package com.logpm.factorydata.zbom.wrapper; |
||||
|
||||
import javax.servlet.ReadListener; |
||||
import javax.servlet.ServletInputStream; |
||||
import javax.servlet.http.HttpServletRequest; |
||||
import javax.servlet.http.HttpServletRequestWrapper; |
||||
import java.io.BufferedReader; |
||||
import java.io.ByteArrayInputStream; |
||||
import java.io.IOException; |
||||
import java.io.InputStream; |
||||
import java.io.InputStreamReader; |
||||
|
||||
public class RequestWrapper extends HttpServletRequestWrapper { |
||||
private final String body; |
||||
public RequestWrapper(HttpServletRequest request) throws IOException { |
||||
super(request); |
||||
StringBuilder stringBuilder = new StringBuilder(); |
||||
BufferedReader bufferedReader = null; |
||||
try { |
||||
InputStream inputStream = request.getInputStream(); |
||||
if (inputStream != null) { |
||||
bufferedReader = new BufferedReader(new InputStreamReader(inputStream)); |
||||
char[] charBuffer = new char[128]; |
||||
int bytesRead = -1; |
||||
while ((bytesRead = bufferedReader.read(charBuffer)) > 0) { |
||||
stringBuilder.append(charBuffer, 0, bytesRead); |
||||
} |
||||
} else { |
||||
stringBuilder.append(""); |
||||
} |
||||
} catch (IOException ex) { |
||||
throw ex; |
||||
} finally { |
||||
if (bufferedReader != null) { |
||||
try { |
||||
bufferedReader.close(); |
||||
} catch (IOException ex) { |
||||
throw ex; |
||||
} |
||||
} |
||||
} |
||||
body = stringBuilder.toString(); |
||||
} |
||||
|
||||
@Override |
||||
public ServletInputStream getInputStream() throws IOException { |
||||
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes()); |
||||
ServletInputStream servletInputStream = new ServletInputStream() { |
||||
@Override |
||||
public boolean isFinished() { |
||||
return false; |
||||
} |
||||
@Override |
||||
public boolean isReady() { |
||||
return false; |
||||
} |
||||
@Override |
||||
public void setReadListener(ReadListener readListener) {} |
||||
@Override |
||||
public int read() throws IOException { |
||||
return byteArrayInputStream.read(); |
||||
} |
||||
}; |
||||
return servletInputStream; |
||||
|
||||
} |
||||
@Override |
||||
public BufferedReader getReader() throws IOException { |
||||
return new BufferedReader(new InputStreamReader(this.getInputStream())); |
||||
} |
||||
public String getBody() { |
||||
return this.body; |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue