zhangyeguang
1 month ago
69 changed files with 3734 additions and 140 deletions
@ -0,0 +1,199 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
<parent> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-parent</artifactId> |
|||
<version>2.2.6.RELEASE</version> |
|||
<relativePath/> <!-- lookup parent from repository --> |
|||
</parent> |
|||
<groupId>com.jiagutech</groupId> |
|||
<artifactId>siliqua_recognition</artifactId> |
|||
<version>0.0.1-SNAPSHOT</version> |
|||
<name>siliqua_recognition</name> |
|||
<description>角果识别</description> |
|||
|
|||
|
|||
<properties> |
|||
<java.version>8</java.version> |
|||
<maven.compiler.source>8</maven.compiler.source> |
|||
<maven.compiler.target>8</maven.compiler.target> |
|||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> |
|||
<mapstruct.version>1.6.1</mapstruct.version> |
|||
<lombok.version>1.18.30</lombok.version> |
|||
<hutool.version>5.8.22</hutool.version> |
|||
|
|||
</properties> |
|||
<dependencies> |
|||
<dependency> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-data-redis</artifactId> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-web</artifactId> |
|||
</dependency> |
|||
|
|||
<!-- <dependency>--> |
|||
<!-- <groupId>org.springframework.boot</groupId>--> |
|||
<!-- <artifactId>spring-boot-devtools</artifactId>--> |
|||
<!-- <scope>runtime</scope>--> |
|||
<!-- <optional>true</optional>--> |
|||
<!-- </dependency>--> |
|||
<dependency> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-data-jdbc</artifactId> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.mybatis.spring.boot</groupId> |
|||
<artifactId>mybatis-spring-boot-starter</artifactId> |
|||
<version>2.3.2</version> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>com.baomidou</groupId> |
|||
<artifactId>mybatis-plus-boot-starter</artifactId> |
|||
<version>3.5.0</version> |
|||
<exclusions> |
|||
<exclusion> |
|||
<groupId>org.mybatis</groupId> |
|||
<artifactId>mybatis</artifactId> |
|||
</exclusion> |
|||
</exclusions> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>com.mysql</groupId> |
|||
<artifactId>mysql-connector-j</artifactId> |
|||
<version>8.2.0</version> |
|||
<scope>runtime</scope> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-validation</artifactId> |
|||
</dependency> |
|||
|
|||
|
|||
<dependency> |
|||
<groupId>org.projectlombok</groupId> |
|||
<artifactId>lombok</artifactId> |
|||
<optional>true</optional> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.mapstruct</groupId> |
|||
<artifactId>mapstruct</artifactId> |
|||
<version>${mapstruct.version}</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>com.alibaba.fastjson2</groupId> |
|||
<artifactId>fastjson2</artifactId> |
|||
<version>2.0.43</version> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>org.apache.commons</groupId> |
|||
<artifactId>commons-lang3</artifactId> |
|||
<version>3.12.0</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>com.google.code.gson</groupId> |
|||
<artifactId>gson</artifactId> |
|||
<version>2.10.1</version> |
|||
</dependency> |
|||
|
|||
|
|||
<dependency> |
|||
<groupId>com.alibaba</groupId> |
|||
<artifactId>easyexcel</artifactId> |
|||
<version>4.0.2</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>commons-io</groupId> |
|||
<artifactId>commons-io</artifactId> |
|||
<version>2.16.1</version> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>cn.dev33</groupId> |
|||
<artifactId>sa-token-spring-boot-starter</artifactId> |
|||
<version>1.37.0</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>cn.dev33</groupId> |
|||
<artifactId>sa-token-jwt</artifactId> |
|||
<version>1.37.0</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>cn.dev33</groupId> |
|||
<artifactId>sa-token-redis-jackson</artifactId> |
|||
<version>1.37.0</version> |
|||
</dependency> |
|||
|
|||
|
|||
<!-- hutool 的依赖配置--> |
|||
<dependency> |
|||
<groupId>cn.hutool</groupId> |
|||
<artifactId>hutool-bom</artifactId> |
|||
<version>${hutool.version}</version> |
|||
<type>pom</type> |
|||
<scope>import</scope> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>com.huaweicloud</groupId> |
|||
<artifactId>esdk-obs-java</artifactId> |
|||
<version>3.20.6.1</version> |
|||
<scope>compile</scope> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>org.apache.logging.log4j</groupId> |
|||
<artifactId>log4j-core</artifactId> |
|||
<version>2.23.1</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.apache.logging.log4j</groupId> |
|||
<artifactId>log4j-api</artifactId> |
|||
<version>2.23.1</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>io.springfox</groupId> |
|||
<artifactId>springfox-swagger2</artifactId> |
|||
<version>2.7.0</version> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>io.springfox</groupId> |
|||
<artifactId>springfox-swagger-ui</artifactId> |
|||
<version>2.7.0</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>com.github.xiaoymin</groupId> |
|||
<artifactId>knife4j-spring-boot-starter</artifactId> |
|||
<version>2.0.8</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-starter-mail</artifactId> |
|||
</dependency> |
|||
|
|||
</dependencies> |
|||
|
|||
<build> |
|||
<plugins> |
|||
<plugin> |
|||
<groupId>org.springframework.boot</groupId> |
|||
<artifactId>spring-boot-maven-plugin</artifactId> |
|||
<configuration> |
|||
<excludes> |
|||
<exclude> |
|||
<groupId>org.projectlombok</groupId> |
|||
<artifactId>lombok</artifactId> |
|||
</exclude> |
|||
</excludes> |
|||
</configuration> |
|||
</plugin> |
|||
|
|||
</plugins> |
|||
</build> |
|||
|
|||
</project> |
|||
@ -0,0 +1,15 @@ |
|||
package com.jiagutech; |
|||
|
|||
import org.mybatis.spring.annotation.MapperScan; |
|||
import org.springframework.boot.SpringApplication; |
|||
import org.springframework.boot.autoconfigure.SpringBootApplication; |
|||
import org.springframework.scheduling.annotation.EnableAsync; |
|||
|
|||
@MapperScan("com.jiagutech.mapper") |
|||
@EnableAsync |
|||
@SpringBootApplication |
|||
public class RecognitionRun { |
|||
public static void main(String[] args) { |
|||
SpringApplication.run(RecognitionRun.class, args); |
|||
} |
|||
} |
|||
@ -0,0 +1,61 @@ |
|||
package com.jiagutech.client; |
|||
|
|||
import com.alibaba.fastjson2.JSON; |
|||
import com.alibaba.fastjson2.JSONObject; |
|||
import com.jiagutech.dto.ModelPredictResponse; |
|||
import com.jiagutech.dto.RecordModel; |
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.beans.factory.annotation.Value; |
|||
import org.springframework.core.io.FileSystemResource; |
|||
import org.springframework.http.*; |
|||
import org.springframework.stereotype.Component; |
|||
import org.springframework.util.CollectionUtils; |
|||
import org.springframework.util.LinkedMultiValueMap; |
|||
import org.springframework.util.MultiValueMap; |
|||
import org.springframework.web.client.RestTemplate; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import java.io.File; |
|||
import java.util.Collections; |
|||
import java.util.List; |
|||
|
|||
@Slf4j |
|||
@Component |
|||
@RequiredArgsConstructor |
|||
public class RecognitionModelClient { |
|||
|
|||
@Value("${recognition.model.url}") |
|||
private String modelUrl; |
|||
|
|||
private final RestTemplate restTemplate; |
|||
|
|||
|
|||
public List<ModelPredictResponse> predict(File file, String modeName, float lou, float conf) { |
|||
MultiValueMap<String, Object> body = new LinkedMultiValueMap<>(); |
|||
FileSystemResource fileResource = new FileSystemResource(file); |
|||
if (!fileResource.exists()) { |
|||
throw new IllegalArgumentException("文件不存在: " + file.getAbsolutePath()); |
|||
} |
|||
body.add("image", fileResource); |
|||
body.add("model_name", modeName); |
|||
body.add("lou", lou); |
|||
body.add("conf", conf); |
|||
|
|||
// 设置请求头
|
|||
HttpHeaders headers = new HttpHeaders(); |
|||
headers.setContentType(MediaType.parseMediaType("multipart/form-data; charset=UTF-8")); |
|||
HttpEntity<MultiValueMap<String, Object>> requestEntity = new HttpEntity<>(body, headers); |
|||
ResponseEntity<String> result = restTemplate.exchange(modelUrl + "/api/predict", HttpMethod.POST, requestEntity, String.class); |
|||
if (result.getStatusCode() == HttpStatus.OK) { |
|||
return JSON.parseArray(result.getBody(), ModelPredictResponse.class); |
|||
} |
|||
return Collections.emptyList(); |
|||
} |
|||
|
|||
public RecordModel models() { |
|||
return restTemplate.getForObject(modelUrl + "/api/models", RecordModel.class); |
|||
} |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,93 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
/** |
|||
* 返回状态码 |
|||
* |
|||
* @author Lion Li |
|||
*/ |
|||
public interface HttpStatus { |
|||
/** |
|||
* 操作成功 |
|||
*/ |
|||
int SUCCESS = 200; |
|||
|
|||
/** |
|||
* 对象创建成功 |
|||
*/ |
|||
int CREATED = 201; |
|||
|
|||
/** |
|||
* 请求已经被接受 |
|||
*/ |
|||
int ACCEPTED = 202; |
|||
|
|||
/** |
|||
* 操作已经执行成功，但是没有返回数据 |
|||
*/ |
|||
int NO_CONTENT = 204; |
|||
|
|||
/** |
|||
* 资源已被移除 |
|||
*/ |
|||
int MOVED_PERM = 301; |
|||
|
|||
/** |
|||
* 重定向 |
|||
*/ |
|||
int SEE_OTHER = 303; |
|||
|
|||
/** |
|||
* 资源没有被修改 |
|||
*/ |
|||
int NOT_MODIFIED = 304; |
|||
|
|||
/** |
|||
* 参数列表错误（缺少，格式不匹配） |
|||
*/ |
|||
int BAD_REQUEST = 400; |
|||
|
|||
/** |
|||
* 未授权 |
|||
*/ |
|||
int UNAUTHORIZED = 401; |
|||
|
|||
/** |
|||
* 访问受限，授权过期 |
|||
*/ |
|||
int FORBIDDEN = 403; |
|||
|
|||
/** |
|||
* 资源，服务未找到 |
|||
*/ |
|||
int NOT_FOUND = 404; |
|||
|
|||
/** |
|||
* 不允许的http方法 |
|||
*/ |
|||
int BAD_METHOD = 405; |
|||
|
|||
/** |
|||
* 资源冲突，或者资源被锁 |
|||
*/ |
|||
int CONFLICT = 409; |
|||
|
|||
/** |
|||
* 不支持的数据，媒体类型 |
|||
*/ |
|||
int UNSUPPORTED_TYPE = 415; |
|||
|
|||
/** |
|||
* 系统内部错误 |
|||
*/ |
|||
int ERROR = 500; |
|||
|
|||
/** |
|||
* 接口未实现 |
|||
*/ |
|||
int NOT_IMPLEMENTED = 501; |
|||
|
|||
/** |
|||
* 系统警告消息 |
|||
*/ |
|||
int WARN = 601; |
|||
} |
|||
@ -0,0 +1,50 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
import com.github.xiaoymin.knife4j.spring.annotations.EnableKnife4j; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.context.annotation.Profile; |
|||
import springfox.documentation.builders.ApiInfoBuilder; |
|||
import springfox.documentation.builders.PathSelectors; |
|||
import springfox.documentation.builders.RequestHandlerSelectors; |
|||
import springfox.documentation.service.ApiInfo; |
|||
import springfox.documentation.service.Contact; |
|||
import springfox.documentation.spi.DocumentationType; |
|||
import springfox.documentation.spring.web.plugins.Docket; |
|||
import springfox.documentation.swagger2.annotations.EnableSwagger2; |
|||
|
|||
/** |
|||
* 自定义 Knife4j 接口文档的配置 |
|||
*/ |
|||
@Configuration |
|||
@EnableSwagger2 |
|||
@EnableKnife4j |
|||
public class Knife4jConfig { |
|||
|
|||
@Bean(value = "defaultApi2") |
|||
public Docket defaultApi2() { |
|||
return new Docket(DocumentationType.SWAGGER_2) |
|||
.apiInfo(apiInfo()) |
|||
.select() |
|||
// 项目controller的位置
|
|||
.apis(RequestHandlerSelectors.basePackage("com.jiagutech.controller")) |
|||
.paths(PathSelectors.any()) |
|||
.build(); |
|||
} |
|||
|
|||
/** |
|||
* api 信息 |
|||
* |
|||
* @return |
|||
*/ |
|||
private ApiInfo apiInfo() { |
|||
return new ApiInfoBuilder() |
|||
.title("角果识别") |
|||
.description("角果识别接口文档") |
|||
.termsOfServiceUrl("http://recognition.jiagutech.com") |
|||
.contact(new Contact("yeguangzhang", "http://recognition.jiagutech.com", "yeguangzhang@126.com")) |
|||
.version("1.0.0") |
|||
.build(); |
|||
} |
|||
} |
|||
|
|||
@ -0,0 +1,23 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
|
|||
import com.baomidou.mybatisplus.annotation.DbType; |
|||
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor; |
|||
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
|
|||
@Configuration |
|||
public class MybatisPlusConfig { |
|||
|
|||
/** |
|||
* 添加分页插件 |
|||
*/ |
|||
@Bean |
|||
public MybatisPlusInterceptor mybatisPlusInterceptor() { |
|||
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); |
|||
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));//如果配置多个插件,切记分页最后添加
|
|||
return interceptor; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,39 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
|
|||
import com.obs.services.ObsClient; |
|||
import com.obs.services.ObsConfiguration; |
|||
import lombok.Data; |
|||
import lombok.NoArgsConstructor; |
|||
import org.springframework.boot.context.properties.ConfigurationProperties; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.context.annotation.Scope; |
|||
|
|||
@Data |
|||
@Configuration |
|||
@NoArgsConstructor |
|||
@ConfigurationProperties(prefix = "huawei.obs") |
|||
public class ObsConfig { |
|||
|
|||
|
|||
private String endPoint; |
|||
private String ak; |
|||
private String sk; |
|||
|
|||
private String bucketName; |
|||
private String pathPrefix; |
|||
|
|||
@Bean |
|||
@Scope("singleton") |
|||
public ObsClient obsClient() { |
|||
// ObsConfiguration config = new ObsConfiguration();
|
|||
// config.setConnectionTimeout(3000);
|
|||
// config.setSocketTimeout(5000);
|
|||
// config.setMaxErrorRetry(2);
|
|||
// config.setEndPoint(endPoint);
|
|||
|
|||
ObsClient obsClient = new ObsClient(ak, sk, endPoint); |
|||
return obsClient; |
|||
} |
|||
} |
|||
@ -0,0 +1,41 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
import org.springframework.beans.factory.annotation.Value; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.http.client.SimpleClientHttpRequestFactory; |
|||
import org.springframework.http.converter.StringHttpMessageConverter; |
|||
import org.springframework.web.client.RestTemplate; |
|||
|
|||
import java.nio.charset.StandardCharsets; |
|||
import java.util.Collections; |
|||
|
|||
|
|||
/** |
|||
* RestTemplate工具类，主要用来提供RestTemplate对象 |
|||
*/ |
|||
@Configuration |
|||
public class RestTemplateConfig { |
|||
|
|||
|
|||
|
|||
@Value("${http.template.connect-timeout}") |
|||
private int connectionTimeout; |
|||
@Value("${http.template.read-timeout}") |
|||
private int readTimeout; |
|||
|
|||
/** |
|||
* 创建RestTemplate对象 |
|||
*/ |
|||
@Bean |
|||
public RestTemplate restTemplate() { |
|||
SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory(); |
|||
factory.setConnectTimeout(connectionTimeout); // 设置连接超时时间（毫秒）
|
|||
factory.setReadTimeout(readTimeout); // 设置读取超时时间（毫秒）
|
|||
RestTemplate restTemplate = new RestTemplate(factory); |
|||
restTemplate.setInterceptors(Collections.singletonList(new RestTemplateInterceptor())); |
|||
restTemplate.getMessageConverters().set(1,new StringHttpMessageConverter(StandardCharsets.UTF_8)); // 支持中文编码
|
|||
return restTemplate; |
|||
} |
|||
} |
|||
|
|||
@ -0,0 +1,33 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.http.HttpRequest; |
|||
import org.springframework.http.client.ClientHttpRequestExecution; |
|||
import org.springframework.http.client.ClientHttpRequestInterceptor; |
|||
import org.springframework.http.client.ClientHttpResponse; |
|||
|
|||
import java.io.IOException; |
|||
import java.nio.charset.StandardCharsets; |
|||
|
|||
@Slf4j |
|||
public class RestTemplateInterceptor implements ClientHttpRequestInterceptor { |
|||
|
|||
|
|||
@Override |
|||
public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException { |
|||
|
|||
// 打印出请求的详细信息
|
|||
log.info("URI: {}", request.getURI()); |
|||
|
|||
// 执行请求并获取响应
|
|||
ClientHttpResponse response = execution.execute(request, body); |
|||
|
|||
// 打印出响应的详细信息
|
|||
log.info("Response Status: {}", response.getStatusCode()); |
|||
log.info("Response Headers: {}", response.getHeaders()); |
|||
response.getBody(); // 读取响应体，确保内容被完全读取，这样连接才能被关闭
|
|||
|
|||
return response; |
|||
|
|||
} |
|||
} |
|||
@ -0,0 +1,143 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
/** |
|||
* 用户常量信息 |
|||
* |
|||
* @author ruoyi |
|||
*/ |
|||
public interface UserConstants { |
|||
|
|||
/** |
|||
* 平台内系统用户的唯一标志 |
|||
*/ |
|||
String SYS_SESSION = "CURRENT_USER"; |
|||
|
|||
|
|||
/** |
|||
* 正常状态 |
|||
*/ |
|||
String NORMAL = "0"; |
|||
|
|||
/** |
|||
* 异常状态 |
|||
*/ |
|||
String EXCEPTION = "1"; |
|||
|
|||
/** |
|||
* 用户正常状态 |
|||
*/ |
|||
String USER_NORMAL = "0"; |
|||
|
|||
/** |
|||
* 用户封禁状态 |
|||
*/ |
|||
String USER_DISABLE = "1"; |
|||
|
|||
/** |
|||
* 角色正常状态 |
|||
*/ |
|||
String ROLE_NORMAL = "0"; |
|||
|
|||
/** |
|||
* 角色封禁状态 |
|||
*/ |
|||
String ROLE_DISABLE = "1"; |
|||
|
|||
/** |
|||
* 部门正常状态 |
|||
*/ |
|||
String DEPT_NORMAL = "0"; |
|||
|
|||
/** |
|||
* 部门停用状态 |
|||
*/ |
|||
String DEPT_DISABLE = "1"; |
|||
|
|||
/** |
|||
* 岗位正常状态 |
|||
*/ |
|||
String POST_NORMAL = "0"; |
|||
|
|||
/** |
|||
* 岗位停用状态 |
|||
*/ |
|||
String POST_DISABLE = "1"; |
|||
|
|||
/** |
|||
* 字典正常状态 |
|||
*/ |
|||
String DICT_NORMAL = "0"; |
|||
|
|||
/** |
|||
* 是否为系统默认（是） |
|||
*/ |
|||
String YES = "Y"; |
|||
|
|||
/** |
|||
* 是否菜单外链（是） |
|||
*/ |
|||
String YES_FRAME = "0"; |
|||
|
|||
/** |
|||
* 是否菜单外链（否） |
|||
*/ |
|||
String NO_FRAME = "1"; |
|||
|
|||
/** |
|||
* 菜单正常状态 |
|||
*/ |
|||
String MENU_NORMAL = "0"; |
|||
|
|||
/** |
|||
* 菜单停用状态 |
|||
*/ |
|||
String MENU_DISABLE = "1"; |
|||
|
|||
/** |
|||
* 菜单类型（目录） |
|||
*/ |
|||
String TYPE_DIR = "M"; |
|||
|
|||
/** |
|||
* 菜单类型（菜单） |
|||
*/ |
|||
String TYPE_MENU = "C"; |
|||
|
|||
/** |
|||
* 菜单类型（按钮） |
|||
*/ |
|||
String TYPE_BUTTON = "F"; |
|||
|
|||
/** |
|||
* Layout组件标识 |
|||
*/ |
|||
String LAYOUT = "Layout"; |
|||
|
|||
/** |
|||
* ParentView组件标识 |
|||
*/ |
|||
String PARENT_VIEW = "ParentView"; |
|||
|
|||
/** |
|||
* InnerLink组件标识 |
|||
*/ |
|||
String INNER_LINK = "InnerLink"; |
|||
|
|||
/** |
|||
* 用户名长度限制 |
|||
*/ |
|||
int USERNAME_MIN_LENGTH = 2; |
|||
int USERNAME_MAX_LENGTH = 20; |
|||
|
|||
/** |
|||
* 密码长度限制 |
|||
*/ |
|||
int PASSWORD_MIN_LENGTH = 5; |
|||
int PASSWORD_MAX_LENGTH = 20; |
|||
|
|||
/** |
|||
* 超级管理员ID |
|||
*/ |
|||
Long SUPER_ADMIN_ID = 1L; |
|||
|
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
package com.jiagutech.common; |
|||
|
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.web.cors.CorsConfiguration; |
|||
import org.springframework.web.cors.UrlBasedCorsConfigurationSource; |
|||
import org.springframework.web.filter.CorsFilter; |
|||
|
|||
@Configuration |
|||
public class WebCrosConfig { |
|||
|
|||
|
|||
@Bean |
|||
public CorsFilter corsFilter() { |
|||
CorsConfiguration config = new CorsConfiguration(); |
|||
config.addAllowedOrigin("*"); |
|||
config.addAllowedMethod("GET"); |
|||
config.addAllowedMethod("POST"); |
|||
config.addAllowedMethod("PUT"); |
|||
config.addAllowedMethod("DELETE"); |
|||
config.addAllowedMethod("OPTIONS"); |
|||
config.addAllowedHeader("*"); |
|||
config.setAllowCredentials(false); |
|||
config.setMaxAge(3600L); |
|||
|
|||
UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource(); |
|||
source.registerCorsConfiguration("/**", config); |
|||
|
|||
return new CorsFilter(source); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,63 @@ |
|||
package com.jiagutech.common.exception; |
|||
|
|||
import lombok.Getter; |
|||
|
|||
@Getter |
|||
public enum BizCode implements CodeInterface { |
|||
General_Success(200, "接口调用成功"), |
|||
ServerError(10001, "服务器异常"), |
|||
General_Failure(10004, "接口调用失败"), |
|||
General_DBError(10005, "DB错误"), |
|||
General_ParameterInvalid(12001, "参数校验失败"), |
|||
/** |
|||
* 通用类 |
|||
*/ |
|||
USER_UNREGISTERED(12001, "用户未注册"), |
|||
NOT_FOUND(12002, "数据不存在"), |
|||
LOGIN_TYPE_ERROR(12005, "登录类型错误"), |
|||
|
|||
|
|||
/** |
|||
* 文件服务 |
|||
*/ |
|||
FILE_DOWNLOAD_ERROR(90001, "文件下载错误"), |
|||
|
|||
FILE_UPLOAD_ERROR(90002, "文件上传失败"), |
|||
|
|||
|
|||
|
|||
USER_NOT_FOUND(40004, "用户不存在"), |
|||
|
|||
CAPTCHA_ERROR(40003, "验证码错误"), |
|||
|
|||
PASSWORD_ERROR(40002, "密码错误"), |
|||
|
|||
TOKEN_TIMEOUT_ERROR(40006, "token已过期"), |
|||
|
|||
USER_STOP_FOUND(40008, "用户被禁用"), |
|||
USER_STATUS_ERROR(40009,"用户未审核通过"), |
|||
|
|||
USER_NOT_LOGIN(40011, "用户未登录,请登录后再操作"), |
|||
|
|||
ACCESS_TOKEN_INVALID(40040, "用户token无效,请重新登录"), |
|||
|
|||
ACCESS_NOT_ALLOWABLE(50001, "服务不允许直接访问"), |
|||
|
|||
PERMISSION_NOT_FOUND(40001,"该用户无操作权限"), |
|||
|
|||
USER_PHONE_EXIST(40012, "用户手机号已存在"), |
|||
|
|||
|
|||
; |
|||
|
|||
|
|||
private final Integer code; |
|||
|
|||
private final String msg; |
|||
|
|||
BizCode(Integer code, String msg) { |
|||
this.code = code; |
|||
this.msg = msg; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,53 @@ |
|||
package com.jiagutech.common.exception; |
|||
|
|||
import lombok.Data; |
|||
|
|||
@Data |
|||
public class BusinessException extends RuntimeException { |
|||
|
|||
private int code; |
|||
|
|||
private String message; |
|||
|
|||
public BusinessException( int code,String message) { |
|||
super(message); |
|||
this.code = code; |
|||
this.message = message; |
|||
} |
|||
|
|||
public BusinessException(String message, Throwable cause, int code, String message1) { |
|||
super(message, cause); |
|||
this.code = code; |
|||
this.message = message1; |
|||
} |
|||
|
|||
public BusinessException(Throwable cause, int code, String message) { |
|||
super(cause); |
|||
this.code = code; |
|||
this.message = message; |
|||
} |
|||
|
|||
public BusinessException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace, int code, String message1) { |
|||
super(message, cause, enableSuppression, writableStackTrace); |
|||
this.code = code; |
|||
this.message = message1; |
|||
} |
|||
|
|||
public BusinessException(CodeInterface biz) { |
|||
super(); |
|||
this.code = biz.getCode(); |
|||
this.message = biz.getMsg(); |
|||
} |
|||
|
|||
public BusinessException(String message) { |
|||
super(message); |
|||
this.code= BizCode.General_ParameterInvalid.getCode(); |
|||
this.message =message; |
|||
} |
|||
|
|||
public BusinessException(BizCode biz, String msg) { |
|||
super(); |
|||
this.code = biz.getCode(); |
|||
this.message = msg; |
|||
} |
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
package com.jiagutech.common.exception; |
|||
|
|||
/** |
|||
* 业务异常接口 |
|||
*/ |
|||
public interface CodeInterface { |
|||
|
|||
/** |
|||
* 消息 |
|||
* |
|||
* @return String |
|||
*/ |
|||
String getMsg(); |
|||
|
|||
/** |
|||
* 业务代码 |
|||
* |
|||
* @return int |
|||
*/ |
|||
Integer getCode(); |
|||
|
|||
} |
|||
@ -0,0 +1,80 @@ |
|||
package com.jiagutech.common.exception; |
|||
|
|||
|
|||
import cn.dev33.satoken.exception.NotLoginException; |
|||
import cn.dev33.satoken.exception.NotRoleException; |
|||
import com.alibaba.fastjson2.JSON; |
|||
import com.jiagutech.dto.common.R; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.http.HttpStatus; |
|||
import org.springframework.http.ResponseEntity; |
|||
import org.springframework.validation.BindException; |
|||
import org.springframework.validation.FieldError; |
|||
import org.springframework.web.bind.MethodArgumentNotValidException; |
|||
import org.springframework.web.bind.annotation.ExceptionHandler; |
|||
import org.springframework.web.bind.annotation.RestControllerAdvice; |
|||
import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException; |
|||
|
|||
import java.sql.SQLIntegrityConstraintViolationException; |
|||
import java.util.HashMap; |
|||
import java.util.Map; |
|||
|
|||
@Slf4j |
|||
@RestControllerAdvice |
|||
public class GlobalExceptionHandler { |
|||
|
|||
@ExceptionHandler(BindException.class) |
|||
public ResponseEntity<R> handleBindExceptions(BindException ex) { |
|||
log.error("参数绑定失败", ex); |
|||
Map<String, String> errorMap = new HashMap<>(); |
|||
for (FieldError fieldError : ex.getFieldErrors()) { |
|||
errorMap.put(fieldError.getField(), fieldError.getDefaultMessage()); |
|||
} |
|||
return new ResponseEntity<>(R.fail(BizCode.General_ParameterInvalid.getCode(), JSON.toJSONString(errorMap)), HttpStatus.BAD_REQUEST); |
|||
} |
|||
|
|||
@ExceptionHandler(MethodArgumentNotValidException.class) |
|||
public ResponseEntity<R> handleValidationExceptions(MethodArgumentNotValidException ex) { |
|||
log.error("参数校验失败", ex); |
|||
Map<String, String> errorMap = new HashMap<>(); |
|||
for (FieldError fieldError : ex.getBindingResult().getFieldErrors()) { |
|||
errorMap.put(fieldError.getField(), fieldError.getDefaultMessage()); |
|||
} |
|||
return new ResponseEntity<>(R.fail(BizCode.General_ParameterInvalid.getCode(), JSON.toJSONString(errorMap)), HttpStatus.BAD_REQUEST); |
|||
} |
|||
|
|||
@ExceptionHandler(BusinessException.class) |
|||
public ResponseEntity<R> handleBusinessException(BusinessException ex) { |
|||
return new ResponseEntity<>(R.fail(ex.getCode(), ex.getMessage()), HttpStatus.BAD_REQUEST); |
|||
} |
|||
|
|||
@ExceptionHandler(MethodArgumentTypeMismatchException.class) |
|||
public ResponseEntity<R> handleMethodArgumentTypeMismatchException(MethodArgumentTypeMismatchException ex) { |
|||
log.error("参数类型错误", ex); |
|||
return new ResponseEntity<>(R.fail(new BusinessException("参数类型错误")), HttpStatus.BAD_REQUEST); |
|||
} |
|||
|
|||
@ExceptionHandler(NotLoginException.class) |
|||
public ResponseEntity<R> handleLoginException(Exception ex) { |
|||
log.error("系统异常", ex); |
|||
return new ResponseEntity<>(R.fail(BizCode.USER_NOT_LOGIN.getCode(), BizCode.USER_NOT_LOGIN.getMsg()), HttpStatus.UNAUTHORIZED); |
|||
} |
|||
|
|||
@ExceptionHandler(NotRoleException.class) |
|||
public ResponseEntity<R> handlePermissionException(Exception ex) { |
|||
log.error("鉴权失败", ex); |
|||
return new ResponseEntity<>(R.fail(BizCode.PERMISSION_NOT_FOUND.getCode(), BizCode.PERMISSION_NOT_FOUND.getMsg()), HttpStatus.FORBIDDEN); |
|||
} |
|||
|
|||
@ExceptionHandler(SQLIntegrityConstraintViolationException.class) |
|||
public ResponseEntity<R> handleDuplicateException(Exception ex) { |
|||
log.error("数据重复", ex); |
|||
return new ResponseEntity<>(R.fail(BizCode.USER_PHONE_EXIST.getCode(), BizCode.USER_PHONE_EXIST.getMsg()), HttpStatus.BAD_REQUEST); |
|||
} |
|||
|
|||
@ExceptionHandler(Exception.class) |
|||
public ResponseEntity<R> handleCustomException(Exception ex) { |
|||
log.error("系统异常", ex); |
|||
return new ResponseEntity<>(R.fail(BizCode.General_Failure), HttpStatus.BAD_REQUEST); |
|||
} |
|||
} |
|||
@ -0,0 +1,38 @@ |
|||
package com.jiagutech.common.sa; |
|||
|
|||
import cn.dev33.satoken.stp.StpInterface; |
|||
import com.jiagutech.dto.LoginUser; |
|||
import com.jiagutech.utils.LoginUtil; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.Collections; |
|||
import java.util.List; |
|||
|
|||
/** |
|||
* sa-token 权限管理实现类 |
|||
* |
|||
* @author Lion Li |
|||
*/ |
|||
public class SaPermissionImpl implements StpInterface { |
|||
|
|||
/** |
|||
* 获取菜单权限列表 |
|||
*/ |
|||
@Override |
|||
public List<String> getPermissionList(Object loginId, String loginType) { |
|||
|
|||
return new ArrayList<>(); |
|||
} |
|||
|
|||
/** |
|||
* 获取角色权限列表 |
|||
*/ |
|||
@Override |
|||
public List<String> getRoleList(Object loginId, String loginType) { |
|||
LoginUser loginUser = LoginUtil.getLoginUser(); |
|||
if (loginUser == null) { |
|||
return new ArrayList<>(); |
|||
} |
|||
return Collections.emptyList(); |
|||
} |
|||
} |
|||
@ -0,0 +1,30 @@ |
|||
package com.jiagutech.common.sa; |
|||
|
|||
import cn.dev33.satoken.jwt.StpLogicJwtForSimple; |
|||
import cn.dev33.satoken.stp.StpInterface; |
|||
import cn.dev33.satoken.stp.StpLogic; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
|
|||
/** |
|||
* sa-token 配置 |
|||
*/ |
|||
@Configuration |
|||
public class SaTokenConfig { |
|||
|
|||
@Bean |
|||
public StpLogic getStpLogicJwt() { |
|||
// Sa-Token 整合 jwt (简单模式)
|
|||
return new StpLogicJwtForSimple(); |
|||
} |
|||
|
|||
/** |
|||
* 权限接口实现(使用bean注入方便用户替换) |
|||
*/ |
|||
@Bean |
|||
public StpInterface stpInterface() { |
|||
return new SaPermissionImpl(); |
|||
} |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,34 @@ |
|||
package com.jiagutech.common.sa; |
|||
|
|||
import cn.dev33.satoken.context.SaHolder; |
|||
import cn.dev33.satoken.interceptor.SaInterceptor; |
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.web.servlet.config.annotation.InterceptorRegistry; |
|||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; |
|||
|
|||
@Slf4j |
|||
@Configuration |
|||
public class SaTokenConfigure implements WebMvcConfigurer { |
|||
// 注册 Sa-Token 拦截器，打开注解式鉴权功能
|
|||
@Override |
|||
public void addInterceptors(InterceptorRegistry registry) { |
|||
// 注册 Sa-Token 拦截器，打开注解式鉴权功能
|
|||
registry.addInterceptor(new SaInterceptor(handle -> { |
|||
try { |
|||
log.info("-------- 前端访问path：" + SaHolder.getRequest().getRequestPath()); |
|||
StpUtil.checkLogin(); |
|||
log.info("-------- 此 path 校验成功：" + SaHolder.getRequest().getRequestPath()); |
|||
} catch (Exception e) { |
|||
log.info("-------- 此 path 校验失败：" + SaHolder.getRequest().getRequestPath()); |
|||
throw e; |
|||
} |
|||
})) |
|||
.addPathPatterns("/**") |
|||
.excludePathPatterns("/user/login", "user/logout", |
|||
"/user/register", "/record/export/**", "/favicon.ico","/error", |
|||
"/swagger-ui/**", "/swagger-resources/**", "/v2/api-docs/**", "/doc.html/**", "/webjars/**" |
|||
); |
|||
} |
|||
} |
|||
@ -0,0 +1,105 @@ |
|||
package com.jiagutech.common.thread; |
|||
|
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
|
|||
import java.util.concurrent.*; |
|||
|
|||
@Slf4j |
|||
public class MonitorThreadPoolExecutor extends ThreadPoolExecutor { |
|||
|
|||
private final ThreadLocal<Long> executeStartTime; |
|||
protected String poolName; |
|||
private final int slowTaskThreshold; |
|||
private final int queueTimeThreshold; |
|||
|
|||
private final int queueSizeThreshold; |
|||
|
|||
private static final int DEFAULT_SLOW_TASK_TIME = 1000; |
|||
|
|||
private static final int DEFAULT_QUEUE_TIME = 100; |
|||
|
|||
private static final int DEFAULT_QUEUE_SIZE = 450; |
|||
|
|||
|
|||
public MonitorThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue, String poolName) { |
|||
this(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, new NameThreadFactory(poolName), new AbortPolicy(), poolName); |
|||
} |
|||
|
|||
public MonitorThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue, ThreadFactory threadFactory, String poolName) { |
|||
this(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory, new AbortPolicy(), poolName); |
|||
} |
|||
|
|||
public MonitorThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue, RejectedExecutionHandler handler, String poolName) { |
|||
this(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, new NameThreadFactory(poolName), handler, poolName); |
|||
} |
|||
|
|||
public MonitorThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue, ThreadFactory threadFactory, RejectedExecutionHandler handler, String poolName) { |
|||
super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory, handler); |
|||
this.poolName = poolName; |
|||
executeStartTime = new ThreadLocal<>(); |
|||
slowTaskThreshold = 0; |
|||
queueTimeThreshold = 0; |
|||
queueSizeThreshold = 0; |
|||
} |
|||
|
|||
public MonitorThreadPoolExecutor(ThreadPoolConfig poolConfig) { |
|||
super(poolConfig.getCorePoolSize(), |
|||
poolConfig.getMaxPoolSize(), |
|||
poolConfig.getKeepAliveSeconds(), |
|||
TimeUnit.SECONDS, |
|||
new ArrayBlockingQueue<>(poolConfig.getQueueCapacity()), new NameThreadFactory(poolConfig.getThreadName()), |
|||
(r, executor1) -> log.info("任务被中断")); |
|||
this.poolName = poolConfig.getThreadName(); |
|||
executeStartTime = new ThreadLocal<>(); |
|||
this.slowTaskThreshold = poolConfig.getTaskThreshold() > 0 ? poolConfig.getTaskThreshold() : DEFAULT_SLOW_TASK_TIME; |
|||
this.queueTimeThreshold = poolConfig.getQueueTimeThreshold() > 0 ? poolConfig.getQueueTimeThreshold() : DEFAULT_QUEUE_TIME; |
|||
this.queueSizeThreshold = poolConfig.getQueueSizeThreshold() > 0 ? poolConfig.getQueueSizeThreshold() : DEFAULT_QUEUE_SIZE; |
|||
} |
|||
|
|||
@Override |
|||
public void execute(Runnable command) { |
|||
super.execute(command); |
|||
} |
|||
|
|||
@Override |
|||
protected void beforeExecute(Thread t, Runnable r) { |
|||
super.beforeExecute(t, r); |
|||
executeStartTime.set(System.nanoTime()); |
|||
} |
|||
|
|||
@Override |
|||
protected void afterExecute(Runnable r, Throwable t) { |
|||
MonitoredRunnable monitored = null; |
|||
try { |
|||
long executeEndNano = System.nanoTime(); |
|||
Long executeStartTime = this.executeStartTime.get(); |
|||
monitored = (MonitoredRunnable) r; |
|||
long queueNanoTime = monitored.getInQueueNanoTime(); |
|||
int queueTime = (int) ((executeStartTime - queueNanoTime) / 1000000L); |
|||
int executeTime = (int) ((executeEndNano - executeStartTime) / 1000000L); |
|||
ThreadStatistics packStatistics = null; |
|||
if (monitored.getThreadLocal() != null) { |
|||
packStatistics = monitored.getThreadLocal().get(); |
|||
} |
|||
if (executeTime > this.slowTaskThreshold || queueTime > this.queueTimeThreshold || getQueue().size() >= queueSizeThreshold) { |
|||
log.warn("线程池({}),触发告警，当前任务执行时间={}，当前任务排队时间={}，队列线程总数={} ", Thread.currentThread().getName(), executeTime, queueTime, getQueue().size()); |
|||
if (packStatistics != null) { |
|||
log.warn("当前任务各阶段耗时统计：{}", packStatistics); |
|||
} |
|||
} |
|||
|
|||
if (t != null) { |
|||
log.error("线程池名称 = {}, 执行异常的任务数+1", poolName); |
|||
} |
|||
} catch (Exception ignore) { |
|||
} finally { |
|||
executeStartTime.remove(); |
|||
if (monitored != null && monitored.getThreadLocal() != null) { |
|||
monitored.getThreadLocal().remove(); |
|||
} |
|||
} |
|||
} |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
package com.jiagutech.common.thread; |
|||
|
|||
|
|||
|
|||
public class MonitoredRunnable implements Runnable { |
|||
|
|||
private final Runnable runnable; |
|||
private ThreadLocal<ThreadStatistics> threadLocal; |
|||
private final long inQueueNanoTime; |
|||
|
|||
public MonitoredRunnable(Runnable runnable, ThreadLocal<ThreadStatistics> threadLocal) { |
|||
this.runnable = runnable; |
|||
this.threadLocal = threadLocal; |
|||
this.inQueueNanoTime = System.nanoTime(); |
|||
} |
|||
|
|||
|
|||
public ThreadLocal<ThreadStatistics> getThreadLocal() { |
|||
return threadLocal; |
|||
} |
|||
|
|||
public void setThreadLocal(ThreadLocal<ThreadStatistics> threadLocal) { |
|||
this.threadLocal = threadLocal; |
|||
} |
|||
|
|||
public long getInQueueNanoTime() { |
|||
return inQueueNanoTime; |
|||
} |
|||
|
|||
public MonitoredRunnable(Runnable runnable) { |
|||
this.runnable = runnable; |
|||
this.inQueueNanoTime = System.nanoTime(); |
|||
} |
|||
|
|||
|
|||
@Override |
|||
public void run() { |
|||
this.runnable.run(); |
|||
} |
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
package com.jiagutech.common.thread; |
|||
|
|||
import java.util.concurrent.ThreadFactory; |
|||
import java.util.concurrent.atomic.AtomicInteger; |
|||
|
|||
public class NameThreadFactory implements ThreadFactory { |
|||
|
|||
private String namePrefix; |
|||
|
|||
private AtomicInteger nextId = new AtomicInteger(1); |
|||
|
|||
public NameThreadFactory(String namePrefix) { |
|||
this.namePrefix = namePrefix; |
|||
} |
|||
|
|||
@Override |
|||
public Thread newThread(Runnable r) { |
|||
return new Thread(r, namePrefix + nextId.getAndIncrement()); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,31 @@ |
|||
package com.jiagutech.common.thread; |
|||
|
|||
import io.netty.util.concurrent.ThreadProperties; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
|
|||
import javax.annotation.Resource; |
|||
import java.util.concurrent.ThreadPoolExecutor; |
|||
|
|||
@Slf4j |
|||
@Configuration |
|||
@ConditionalOnProperty(prefix = "thread", name = "enabled", havingValue = "true") |
|||
public class ThreadExecutorConfig { |
|||
@Resource |
|||
private ThreadPoolConfig threadPoolConfig; |
|||
|
|||
|
|||
// @Primary
|
|||
@Bean(name = "customThreadPool") |
|||
public ThreadPoolExecutor getcustomExecutor() { |
|||
return generateExecutor(threadPoolConfig); |
|||
} |
|||
|
|||
private ThreadPoolExecutor generateExecutor(ThreadPoolConfig threadPoolConfig) { |
|||
return new MonitorThreadPoolExecutor(threadPoolConfig); |
|||
} |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,29 @@ |
|||
package com.jiagutech.common.thread; |
|||
|
|||
import lombok.Data; |
|||
import org.springframework.boot.context.properties.ConfigurationProperties; |
|||
import org.springframework.context.annotation.Configuration; |
|||
|
|||
@Data |
|||
@Configuration |
|||
@ConfigurationProperties(prefix = "thread.pool") |
|||
public class ThreadPoolConfig { |
|||
|
|||
private int corePoolSize; |
|||
|
|||
private int maxPoolSize; |
|||
|
|||
private int keepAliveSeconds; |
|||
|
|||
private int queueCapacity; |
|||
|
|||
private String threadName; |
|||
|
|||
private int taskThreshold; |
|||
|
|||
private int queueTimeThreshold; |
|||
|
|||
private int queueSizeThreshold; |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,118 @@ |
|||
package com.jiagutech.common.thread; |
|||
|
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
|
|||
import java.util.Map; |
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
public class ThreadStatistics { |
|||
|
|||
private String imei; |
|||
|
|||
private long totalCost; |
|||
|
|||
private long start; |
|||
|
|||
private long finish; |
|||
|
|||
private Map<StatisticsEnum, Item> phases = new ConcurrentHashMap<>(); |
|||
|
|||
/** |
|||
* 指定阶段处理统计 |
|||
* |
|||
* @param statisticsEnum |
|||
*/ |
|||
public void startPhase(StatisticsEnum statisticsEnum) { |
|||
phases.computeIfAbsent(statisticsEnum, key -> new Item()); |
|||
} |
|||
|
|||
/** |
|||
* 完成指定部分的处理统计 |
|||
* |
|||
* @param statisticsEnum 指定部分 |
|||
*/ |
|||
public void finishPhase(StatisticsEnum statisticsEnum) { |
|||
Item item = phases.getOrDefault(statisticsEnum, null); |
|||
if (item != null) { |
|||
item.finish(); |
|||
} |
|||
} |
|||
|
|||
public void finishStatistics() { |
|||
this.finish = System.currentTimeMillis(); |
|||
this.totalCost = this.finish - this.start; |
|||
} |
|||
|
|||
|
|||
public ThreadStatistics(long start) { |
|||
this.start = start; |
|||
} |
|||
|
|||
|
|||
@Data |
|||
public static class Item { |
|||
public Item() { |
|||
start = System.currentTimeMillis(); |
|||
} |
|||
|
|||
long start; |
|||
|
|||
volatile long finish; |
|||
|
|||
volatile long cost; |
|||
|
|||
/** |
|||
* 完成处理统计 |
|||
*/ |
|||
public void finish() { |
|||
if (finish != 0) { |
|||
return; |
|||
} |
|||
this.finish = System.currentTimeMillis(); |
|||
this.cost = this.finish - start; |
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return "Item{" + |
|||
"cost=" + cost + |
|||
'}'; |
|||
} |
|||
} |
|||
|
|||
|
|||
public enum StatisticsEnum { |
|||
/** |
|||
* 基础信息解析 |
|||
*/ |
|||
base_info, |
|||
/** |
|||
* 解密 |
|||
*/ |
|||
decrypt, |
|||
/** |
|||
* 数据包解析 |
|||
*/ |
|||
deserialization, |
|||
|
|||
/** |
|||
* 数据保存到ck |
|||
*/ |
|||
save; |
|||
|
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return "Statistics{" + |
|||
"imei='" + imei + '\'' + |
|||
", totalCost=" + totalCost + |
|||
", start=" + start + |
|||
", finish=" + finish + |
|||
", phases=" + phases + |
|||
'}'; |
|||
} |
|||
} |
|||
@ -0,0 +1,76 @@ |
|||
package com.jiagutech.controller; |
|||
|
|||
import cn.hutool.core.map.MapUtil; |
|||
import com.jiagutech.dto.CreateRecordRequest; |
|||
import com.jiagutech.dto.RecordItem; |
|||
import com.jiagutech.dto.RecordModel; |
|||
import com.jiagutech.dto.common.PageRequest; |
|||
import com.jiagutech.dto.common.PageResult; |
|||
import com.jiagutech.dto.common.R; |
|||
import com.jiagutech.dto.RecordDetail; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import com.jiagutech.service.RecordService; |
|||
import io.swagger.annotations.Api; |
|||
import io.swagger.annotations.ApiOperation; |
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.http.MediaType; |
|||
import org.springframework.web.bind.annotation.*; |
|||
|
|||
import javax.servlet.http.HttpServletResponse; |
|||
import javax.validation.Valid; |
|||
import java.util.List; |
|||
|
|||
@Api(tags = "识别记录") |
|||
@Slf4j |
|||
@RestController |
|||
@RequiredArgsConstructor |
|||
@RequestMapping("/record") |
|||
public class RecordController { |
|||
private final RecordService recordService; |
|||
|
|||
|
|||
@ApiOperation(value = "创建识别记录") |
|||
@PostMapping(value = "/createRecord", consumes = "multipart/form-data", produces = "application/json") |
|||
public R createRecord(@Valid @ModelAttribute CreateRecordRequest request) { |
|||
String taskId = recordService.createRecordAsync(request); |
|||
return R.ok(MapUtil.of("taskId", taskId)); |
|||
} |
|||
|
|||
@ApiOperation(value = "识别记录处理进度") |
|||
@GetMapping("/taskProgress/{taskId}") |
|||
public Float getTaskProgress(@PathVariable String taskId) { |
|||
return recordService.getTaskProgress(taskId); |
|||
} |
|||
|
|||
@ApiOperation(value = "获取识别记录详情") |
|||
@GetMapping("/getRecordDetail/{recordId}") |
|||
public R<RecordDetail> getRecordDetail(@PathVariable Long recordId) { |
|||
return R.ok(recordService.getRecordDetail(recordId)); |
|||
} |
|||
|
|||
@ApiOperation(value = "导出识别记录",notes = "下载文件(响应类型为文件流)") |
|||
@GetMapping(value = "/export/{recordId}", produces = MediaType.APPLICATION_OCTET_STREAM_VALUE) |
|||
public void exportRecord(@PathVariable Long recordId, HttpServletResponse response) { |
|||
recordService.exportRecord(recordId, response); |
|||
} |
|||
|
|||
@ApiOperation(value = "分页获取识别记录") |
|||
@PostMapping("/page") |
|||
public R<PageResult<RecordItem>> getPages(@RequestBody PageRequest pageRequest) { |
|||
return R.ok(recordService.getPages(pageRequest)); |
|||
} |
|||
|
|||
@ApiOperation(value = "更新识别记录内容") |
|||
@PutMapping("/updateRecordContent") |
|||
public void updateRecord(@RequestBody RecordContentEntity recordContentEntity) { |
|||
recordService.updateRecordContent(recordContentEntity); |
|||
} |
|||
|
|||
|
|||
@ApiOperation(value = "获取模型列表") |
|||
@GetMapping("/models") |
|||
public R<RecordModel> getModels() { |
|||
return R.ok(recordService.getModels()); |
|||
} |
|||
} |
|||
@ -0,0 +1,61 @@ |
|||
package com.jiagutech.controller; |
|||
|
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import com.jiagutech.dto.LoginRequest; |
|||
import com.jiagutech.dto.LoginResponse; |
|||
import com.jiagutech.dto.common.R; |
|||
import com.jiagutech.dto.UserRequest; |
|||
import com.jiagutech.service.UserService; |
|||
import io.swagger.annotations.Api; |
|||
import io.swagger.annotations.ApiOperation; |
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.validation.annotation.Validated; |
|||
import org.springframework.web.bind.annotation.*; |
|||
|
|||
@Api(tags = "用户") |
|||
@Slf4j |
|||
@RestController |
|||
@RequestMapping("/user") |
|||
@RequiredArgsConstructor |
|||
public class UserController { |
|||
|
|||
private final UserService userService; |
|||
|
|||
@GetMapping("/info") |
|||
public String info() { |
|||
log.info("info"); |
|||
return "info"; |
|||
} |
|||
|
|||
@ApiOperation(value = "登录") |
|||
@PostMapping("/login") |
|||
public R<LoginResponse> login(@Validated @RequestBody LoginRequest loginBody) { |
|||
LoginResponse loginVo = userService.login(loginBody); |
|||
return R.ok(loginVo); |
|||
} |
|||
|
|||
@ApiOperation(value = "登出") |
|||
@GetMapping("/logout") |
|||
@PostMapping("/logout") |
|||
public R<String> logout() { |
|||
StpUtil.logout(); |
|||
return R.ok("logout"); |
|||
} |
|||
|
|||
@ApiOperation(value = "注册") |
|||
@PostMapping(value = "/register", consumes = "application/json") |
|||
public R<Void> register(@Validated @RequestBody UserRequest user) { |
|||
userService.register(user); |
|||
return R.ok(); |
|||
} |
|||
|
|||
|
|||
@ApiOperation(value = "审批") |
|||
@PutMapping(value = "/approval/{userId}") |
|||
public R<Void> approval(@PathVariable Long userId) { |
|||
userService.approval(userId); |
|||
return R.ok(); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,38 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import io.swagger.annotations.ApiModel; |
|||
import io.swagger.annotations.ApiModelProperty; |
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Data; |
|||
import lombok.NoArgsConstructor; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import javax.validation.constraints.*; |
|||
|
|||
@Data |
|||
@AllArgsConstructor |
|||
@NoArgsConstructor |
|||
@ApiModel("创建记录请求") |
|||
public class CreateRecordRequest { |
|||
@ApiModelProperty(value = "图片列表", required = true) |
|||
@NotEmpty(message = "文件不能为空") |
|||
private MultipartFile[] files; |
|||
@ApiModelProperty(value = "模型名称", required = true) |
|||
@NotBlank(message = "模型名称不能为空") |
|||
private String modelName; |
|||
@ApiModelProperty(value = "重叠度", required = true) |
|||
@NotNull(message = "IOU 不能为空") |
|||
@Min(value = 0, message = "IOU 不能小于 0") |
|||
@Max(value = 1, message = "IOU 不能大于 1") |
|||
private Float iou; |
|||
@ApiModelProperty(value = "置信度", required = true) |
|||
@NotNull(message = "Conf 不能为空") |
|||
@Min(value = 0, message = "Conf 不能小于 0") |
|||
@Max(value = 1, message = "Conf 不能大于 1") |
|||
private Float conf; |
|||
@ApiModelProperty(value = "任务名称", required = true) |
|||
@NotBlank(message = "任务名称不能为空") |
|||
private String taskName; |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
|
|||
import com.jiagutech.common.UserConstants; |
|||
import lombok.Data; |
|||
import org.hibernate.validator.constraints.Length; |
|||
|
|||
import javax.validation.constraints.NotBlank; |
|||
import java.io.Serializable; |
|||
|
|||
@Data |
|||
public class LoginRequest implements Serializable { |
|||
|
|||
@NotBlank(message = "{user.phone.not.blank}") |
|||
private String phone; |
|||
@NotBlank(message = "{user.password.not.blank}") |
|||
@Length(min = UserConstants.PASSWORD_MIN_LENGTH, max = UserConstants.PASSWORD_MAX_LENGTH, message = "{user.password.length.valid}") |
|||
private String password; |
|||
|
|||
private String grantType = "password"; |
|||
} |
|||
@ -0,0 +1,37 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonProperty; |
|||
import lombok.Data; |
|||
|
|||
/** |
|||
* 登录验证信息 |
|||
*/ |
|||
@Data |
|||
public class LoginResponse { |
|||
|
|||
/** |
|||
* 授权令牌 |
|||
*/ |
|||
@JsonProperty("access_token") |
|||
private String accessToken; |
|||
|
|||
/** |
|||
* 刷新令牌 |
|||
*/ |
|||
@JsonProperty("refresh_token") |
|||
private String refreshToken; |
|||
|
|||
/** |
|||
* 授权令牌 access_token 的有效期 |
|||
*/ |
|||
@JsonProperty("expire_in") |
|||
private Long expireIn; |
|||
|
|||
/** |
|||
* 刷新令牌 refresh_token 的有效期 |
|||
*/ |
|||
@JsonProperty("refresh_expire_in") |
|||
private Long refreshExpireIn; |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,86 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonIgnore; |
|||
import com.fasterxml.jackson.databind.annotation.JsonSerialize; |
|||
import com.fasterxml.jackson.databind.ser.std.ToStringSerializer; |
|||
import lombok.Data; |
|||
import lombok.NoArgsConstructor; |
|||
|
|||
import java.io.Serializable; |
|||
import java.time.LocalDateTime; |
|||
import java.util.List; |
|||
|
|||
/** |
|||
* @ClassName LoginUser |
|||
* @author: zhangyeguang |
|||
* @create: 2024-09-01 09:28 |
|||
* @Version 1.0 |
|||
* @description: |
|||
**/ |
|||
@Data |
|||
@NoArgsConstructor |
|||
public class LoginUser implements Serializable { |
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
|
|||
|
|||
/** |
|||
* 用户ID |
|||
*/ |
|||
@JsonSerialize(using = ToStringSerializer.class) |
|||
private Long userId; |
|||
|
|||
/** |
|||
* 部门 |
|||
*/ |
|||
private String unitName; |
|||
|
|||
/** |
|||
* 用户唯一标识 |
|||
*/ |
|||
private String token; |
|||
|
|||
|
|||
/** |
|||
* 登录时间 |
|||
*/ |
|||
private Long loginTime; |
|||
|
|||
/** |
|||
* 过期时间 |
|||
*/ |
|||
private Long expireTime; |
|||
|
|||
/** |
|||
* 登录IP地址 |
|||
*/ |
|||
private String ipaddr; |
|||
|
|||
|
|||
/** |
|||
* 用户名 |
|||
*/ |
|||
private String username; |
|||
|
|||
/** |
|||
* 用户昵称 |
|||
*/ |
|||
private String realName; |
|||
/** |
|||
* 手机号 |
|||
*/ |
|||
private String phone; |
|||
@JsonIgnore |
|||
private String password; |
|||
|
|||
private String email; |
|||
/** |
|||
* 头像 |
|||
*/ |
|||
private String avatar; |
|||
|
|||
private LocalDateTime createTime; |
|||
|
|||
private Integer status; |
|||
|
|||
} |
|||
@ -0,0 +1,10 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import lombok.Data; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
public class ModelPredictData{ |
|||
private List<ModelPredictResponse> recognitionData; |
|||
} |
|||
@ -0,0 +1,14 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import lombok.Data; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
public class ModelPredictResponse { |
|||
private Integer classify; |
|||
private String name; |
|||
private int[][] points; |
|||
private Float length; |
|||
} |
|||
|
|||
@ -0,0 +1,31 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import com.alibaba.excel.annotation.ExcelProperty; |
|||
import lombok.Data; |
|||
|
|||
@Data |
|||
public class RecordContentExcel { |
|||
|
|||
@ExcelProperty("文件名") |
|||
private String name; |
|||
@ExcelProperty("总喙数") |
|||
private int beakNum; |
|||
@ExcelProperty("总柄数") |
|||
private int handleNum; |
|||
@ExcelProperty("总角果数") |
|||
private int rapeNum; |
|||
@ExcelProperty("置信度") |
|||
private float conf; |
|||
@ExcelProperty("重叠度") |
|||
private float iou; |
|||
@ExcelProperty("比例系数") |
|||
private int scale; |
|||
|
|||
@ExcelProperty("角果长") |
|||
private float rapeLength; |
|||
|
|||
@ExcelProperty("喙长") |
|||
private float beakLength; |
|||
@ExcelProperty("柄长") |
|||
private float handleLength; |
|||
} |
|||
@ -0,0 +1,27 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonFormat; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
|
|||
import java.time.LocalDateTime; |
|||
import java.util.List; |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
public class RecordDetail { |
|||
private Long id; |
|||
|
|||
private String name; |
|||
|
|||
private int imageNum; |
|||
|
|||
@JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss") |
|||
private LocalDateTime createTime; |
|||
; |
|||
@JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss") |
|||
private LocalDateTime updateTime; |
|||
|
|||
private List<RecordContentEntity> content; |
|||
} |
|||
@ -0,0 +1,31 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import com.alibaba.excel.annotation.ExcelProperty; |
|||
import lombok.Data; |
|||
|
|||
@Data |
|||
public class RecordExcel { |
|||
|
|||
@ExcelProperty("总图片数") |
|||
private int imageNum; |
|||
@ExcelProperty("总喙数") |
|||
private int beakNum; |
|||
@ExcelProperty("总柄数") |
|||
private int handleNum; |
|||
@ExcelProperty("总角果数") |
|||
private int rapeNum; |
|||
@ExcelProperty("平均角果数") |
|||
private float avgRapeNum; |
|||
@ExcelProperty("最大角果数") |
|||
private int maxRapeNum; |
|||
@ExcelProperty("最少角果数") |
|||
private int minRapeNum; |
|||
|
|||
@ExcelProperty("平均角果长度") |
|||
private float avgRapeLength; |
|||
|
|||
@ExcelProperty("平均喙长度") |
|||
private float avgBeakLength; |
|||
@ExcelProperty("平均柄长度") |
|||
private float avgHandleLength; |
|||
} |
|||
@ -0,0 +1,29 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonFormat; |
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
|
|||
import java.time.LocalDateTime; |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
public class RecordItem { |
|||
|
|||
private String recordName; |
|||
|
|||
private Long recordId; |
|||
|
|||
private int imageNum; |
|||
|
|||
private String userName; |
|||
|
|||
private Long userId; |
|||
|
|||
private String unitName; |
|||
|
|||
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") |
|||
private LocalDateTime createTime; |
|||
|
|||
private String phone; |
|||
} |
|||
@ -0,0 +1,26 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
import lombok.Data; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
public class RecordModel { |
|||
private List<String> loaded_models; |
|||
|
|||
private List<ModelDetail> models; |
|||
|
|||
@Data |
|||
public static class ModelDetail { |
|||
|
|||
private float conf; |
|||
|
|||
private float iou; |
|||
|
|||
private String name; |
|||
|
|||
private String desc; |
|||
|
|||
private String file; |
|||
} |
|||
} |
|||
@ -0,0 +1,43 @@ |
|||
package com.jiagutech.dto; |
|||
|
|||
|
|||
import lombok.Data; |
|||
import lombok.NoArgsConstructor; |
|||
|
|||
import javax.validation.constraints.Pattern; |
|||
import javax.validation.constraints.Size; |
|||
|
|||
/** |
|||
* @ClassName UserRequest |
|||
* @author: zhangyeguang |
|||
* @create: 2024-08-31 10:00 |
|||
* @Version 1.0 |
|||
* @description: |
|||
**/ |
|||
@Data |
|||
@NoArgsConstructor |
|||
public class UserRequest { |
|||
|
|||
/** |
|||
* 用户昵称 |
|||
*/ |
|||
@Size(min = 2, max = 30, message = "用户姓名长度不能超过{max}个字符") |
|||
private String realName; |
|||
|
|||
/** |
|||
* 手机号码 |
|||
*/ |
|||
@Pattern(regexp = "^1[34578]\\d{9}$", message = "手机号码格式不正确") |
|||
private String phone; |
|||
|
|||
|
|||
/** |
|||
* 密码 |
|||
*/ |
|||
private String password; |
|||
|
|||
|
|||
private String unitName; |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
package com.jiagutech.dto.common; |
|||
|
|||
import lombok.Data; |
|||
|
|||
/** |
|||
* @ClassName PageInfo |
|||
* @author: zhangyeguang |
|||
* @create: 2024-09-02 15:21 |
|||
* @Version 1.0 |
|||
* @description: |
|||
**/ |
|||
@Data |
|||
public class PageRequest<T> implements java.io.Serializable { |
|||
private static final long serialVersionUID = 1L; |
|||
|
|||
private int pageNum; |
|||
private int pageSize; |
|||
private String oderByColumn; |
|||
|
|||
private T request; |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,33 @@ |
|||
package com.jiagutech.dto.common; |
|||
|
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
|
|||
import java.util.List; |
|||
|
|||
/** |
|||
* @ClassName PageResult |
|||
* @author: zhangyeguang |
|||
* @create: 2024-09-02 15:22 |
|||
* @Version 1.0 |
|||
* @description: |
|||
**/ |
|||
@Data |
|||
@Accessors(chain = true) |
|||
public class PageResult<T> { |
|||
private int total; |
|||
private int pageSize; |
|||
private int pageNum; |
|||
private int pages; |
|||
private List<T> records; |
|||
|
|||
public static <T> PageResult<T> of(int total, int pageSize, int pageNum, List<T> records) { |
|||
PageResult<T> pageResult = new PageResult<>(); |
|||
pageResult.setTotal(total); |
|||
pageResult.setPageSize(pageSize); |
|||
pageResult.setPageNum(pageNum); |
|||
pageResult.setPages(total==0||pageSize==0?0:total % pageSize == 0 ? total / pageSize : total / pageSize + 1); |
|||
pageResult.setRecords(records); |
|||
return pageResult; |
|||
} |
|||
} |
|||
@ -0,0 +1,110 @@ |
|||
package com.jiagutech.dto.common; |
|||
|
|||
|
|||
|
|||
import com.jiagutech.common.HttpStatus; |
|||
import lombok.Data; |
|||
import lombok.NoArgsConstructor; |
|||
import java.io.Serializable; |
|||
|
|||
/** |
|||
* 响应信息主体 |
|||
* |
|||
* @author Lion Li |
|||
*/ |
|||
@Data |
|||
@NoArgsConstructor |
|||
public class R<T> implements Serializable { |
|||
|
|||
|
|||
private static final long serialVersionUID = 1L; |
|||
|
|||
/** |
|||
* 成功 |
|||
*/ |
|||
public static final int SUCCESS = 200; |
|||
|
|||
/** |
|||
* 失败 |
|||
*/ |
|||
public static final int FAIL = 500; |
|||
|
|||
private int code; |
|||
|
|||
private String msg; |
|||
|
|||
private T data; |
|||
|
|||
public static <T> R<T> ok() { |
|||
return restResult(null, SUCCESS, "操作成功"); |
|||
} |
|||
|
|||
public static <T> R<T> ok(T data) { |
|||
return restResult(data, SUCCESS, "操作成功"); |
|||
} |
|||
|
|||
public static <T> R<T> ok(String msg) { |
|||
return restResult(null, SUCCESS, msg); |
|||
} |
|||
|
|||
public static <T> R<T> ok(String msg, T data) { |
|||
return restResult(data, SUCCESS, msg); |
|||
} |
|||
|
|||
public static <T> R<T> fail() { |
|||
return restResult(null, FAIL, "操作失败"); |
|||
} |
|||
|
|||
public static <T> R<T> fail(String msg) { |
|||
return restResult(null, FAIL, msg); |
|||
} |
|||
|
|||
public static <T> R<T> fail(T data) { |
|||
return restResult(data, FAIL, "操作失败"); |
|||
} |
|||
|
|||
public static <T> R<T> fail(String msg, T data) { |
|||
return restResult(data, FAIL, msg); |
|||
} |
|||
|
|||
public static <T> R<T> fail(int code, String msg) { |
|||
return restResult(null, code, msg); |
|||
} |
|||
|
|||
/** |
|||
* 返回警告消息 |
|||
* |
|||
* @param msg 返回内容 |
|||
* @return 警告消息 |
|||
*/ |
|||
public static <T> R<T> warn(String msg) { |
|||
return restResult(null, HttpStatus.WARN, msg); |
|||
} |
|||
|
|||
/** |
|||
* 返回警告消息 |
|||
* |
|||
* @param msg 返回内容 |
|||
* @param data 数据对象 |
|||
* @return 警告消息 |
|||
*/ |
|||
public static <T> R<T> warn(String msg, T data) { |
|||
return restResult(data, HttpStatus.WARN, msg); |
|||
} |
|||
|
|||
private static <T> R<T> restResult(T data, int code, String msg) { |
|||
R<T> r = new R<>(); |
|||
r.setCode(code); |
|||
r.setData(data); |
|||
r.setMsg(msg); |
|||
return r; |
|||
} |
|||
|
|||
public static <T> Boolean isError(R<T> ret) { |
|||
return !isSuccess(ret); |
|||
} |
|||
|
|||
public static <T> Boolean isSuccess(R<T> ret) { |
|||
return R.SUCCESS == ret.getCode(); |
|||
} |
|||
} |
|||
@ -0,0 +1,42 @@ |
|||
package com.jiagutech.entity; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.annotation.TableField; |
|||
import com.baomidou.mybatisplus.annotation.TableId; |
|||
import com.baomidou.mybatisplus.annotation.TableName; |
|||
import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler; |
|||
import com.jiagutech.dto.ModelPredictResponse; |
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
@TableName(value = "record_content", autoResultMap = true) |
|||
public class RecordContentEntity { |
|||
@TableId(type = IdType.AUTO) |
|||
private Long id; |
|||
|
|||
private Long recordId; |
|||
|
|||
private String imageName; |
|||
|
|||
private String imageUrl; |
|||
@TableField(typeHandler = JacksonTypeHandler.class) |
|||
private List<ModelPredictResponse> recognitionData; |
|||
|
|||
private int siliquaNum; |
|||
|
|||
private int handleNum; |
|||
|
|||
private int beakNum; |
|||
|
|||
private int size; |
|||
//比例
|
|||
private int scale; |
|||
|
|||
private float conf; |
|||
|
|||
private float iou; |
|||
} |
|||
@ -0,0 +1,29 @@ |
|||
package com.jiagutech.entity; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.annotation.TableId; |
|||
import com.baomidou.mybatisplus.annotation.TableName; |
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
import org.springframework.data.annotation.Id; |
|||
|
|||
import java.time.LocalDateTime; |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
@TableName("record_info") |
|||
public class RecordEntity { |
|||
@TableId(type = IdType.AUTO) |
|||
private Long id; |
|||
|
|||
private String name; |
|||
|
|||
private Long userId; |
|||
|
|||
private int imageNum; |
|||
|
|||
private LocalDateTime createTime; |
|||
|
|||
private LocalDateTime updateTime; |
|||
|
|||
} |
|||
@ -0,0 +1,28 @@ |
|||
package com.jiagutech.entity; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.TableId; |
|||
import com.baomidou.mybatisplus.annotation.TableName; |
|||
import lombok.Data; |
|||
import lombok.experimental.Accessors; |
|||
|
|||
import java.time.LocalDateTime; |
|||
|
|||
@Data |
|||
@TableName("siliqua_user") |
|||
@Accessors(chain = true) |
|||
public class UserEntity { |
|||
@TableId |
|||
private long id; |
|||
|
|||
private String phone; |
|||
private String password; |
|||
private String realName; |
|||
|
|||
private String unitName; |
|||
|
|||
private LocalDateTime createTime; |
|||
|
|||
private int status; |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,26 @@ |
|||
package com.jiagutech.handler; |
|||
|
|||
import com.alibaba.excel.write.handler.SheetWriteHandler; |
|||
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder; |
|||
import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder; |
|||
import org.apache.poi.ss.usermodel.Sheet; |
|||
import org.apache.poi.ss.util.CellRangeAddress; |
|||
|
|||
import java.util.Collections; |
|||
import java.util.List; |
|||
|
|||
public class AddCellRangeWriteHandler implements SheetWriteHandler { |
|||
|
|||
private final List<CellRangeAddress> rangeCellList; |
|||
|
|||
public AddCellRangeWriteHandler(List<CellRangeAddress> rangeCellList) { |
|||
this.rangeCellList = (rangeCellList == null) ? Collections.emptyList() : rangeCellList; |
|||
} |
|||
|
|||
public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) { |
|||
Sheet sheet = writeSheetHolder.getSheet(); |
|||
for (CellRangeAddress cellRangeAddress : this.rangeCellList) { |
|||
sheet.addMergedRegionUnsafe(cellRangeAddress); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,103 @@ |
|||
package com.jiagutech.handler; |
|||
|
|||
import com.alibaba.excel.metadata.Head; |
|||
import com.alibaba.excel.metadata.data.WriteCellData; |
|||
import com.alibaba.excel.write.handler.CellWriteHandler; |
|||
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder; |
|||
import com.alibaba.excel.write.metadata.holder.WriteTableHolder; |
|||
|
|||
import java.util.List; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.apache.poi.ss.usermodel.Cell; |
|||
import org.apache.poi.ss.usermodel.CellType; |
|||
import org.apache.poi.ss.usermodel.Row; |
|||
import org.apache.poi.ss.usermodel.Sheet; |
|||
import org.apache.poi.ss.util.CellRangeAddress; |
|||
@Slf4j |
|||
public class ExcelFillCellMergeStrategy implements CellWriteHandler { |
|||
|
|||
// 需要创建合并区的列
|
|||
private int[] mergeColumnIndex; |
|||
// 从第几行后开始合并，取列头行
|
|||
private int mergeRowIndex; |
|||
|
|||
public ExcelFillCellMergeStrategy() { |
|||
} |
|||
|
|||
public ExcelFillCellMergeStrategy(int mergeRowIndex, int[] mergeColumnIndex) { |
|||
this.mergeRowIndex = mergeRowIndex; |
|||
this.mergeColumnIndex = mergeColumnIndex; |
|||
} |
|||
|
|||
@Override |
|||
public void beforeCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Row row, Head head, Integer columnIndex, |
|||
Integer relativeRowIndex, Boolean isHead) { |
|||
|
|||
} |
|||
|
|||
@Override |
|||
public void afterCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Cell cell, Head head, Integer relativeRowIndex, |
|||
Boolean isHead) { |
|||
|
|||
} |
|||
|
|||
@Override |
|||
public void afterCellDispose(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, |
|||
List<WriteCellData<?>> list, Cell cell, Head head, |
|||
Integer integer, Boolean aBoolean) { |
|||
if (aBoolean){ |
|||
log.info("表头，不处理"); |
|||
return; |
|||
} |
|||
int curRowIndex = cell.getRowIndex(); |
|||
int curColIndex = cell.getColumnIndex(); |
|||
if (curRowIndex > mergeRowIndex) { |
|||
for (int i = 0; i < mergeColumnIndex.length; i++) { |
|||
// 需合并的列
|
|||
if (curColIndex == mergeColumnIndex[i]) { |
|||
mergeWithPrevRow(writeSheetHolder, cell, curRowIndex, curColIndex); |
|||
break; |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 当前单元格向上合并 |
|||
* |
|||
* @param writeSheetHolder |
|||
* @param cell 当前单元格 |
|||
* @param curRowIndex 当前行 |
|||
* @param curColIndex 当前列 |
|||
*/ |
|||
private void mergeWithPrevRow(WriteSheetHolder writeSheetHolder, Cell cell, int curRowIndex, int curColIndex) { |
|||
Object curData = cell.getCellType() == CellType.STRING ? cell.getStringCellValue() : cell.getNumericCellValue(); |
|||
Cell preCell = cell.getSheet().getRow(curRowIndex - 1).getCell(curColIndex); |
|||
Object preData = preCell.getCellType() == CellType.STRING ? preCell.getStringCellValue() : preCell.getNumericCellValue(); |
|||
// 将当前单元格数据与上一个单元格数据比较
|
|||
Boolean dataBool = preData.equals(curData); |
|||
//此处需要注意：因为我是按照序号确定是否需要合并的，所以获取每一行第一列数据和上一行第一列数据进行比较，如果相等合并
|
|||
Boolean bool = cell.getRow().getCell(0).getNumericCellValue() == cell.getSheet().getRow(curRowIndex - 1).getCell(0).getNumericCellValue(); |
|||
if (dataBool && bool) { |
|||
Sheet sheet = writeSheetHolder.getSheet(); |
|||
List<CellRangeAddress> mergeRegions = sheet.getMergedRegions(); |
|||
boolean isMerged = false; |
|||
for (int i = 0; i < mergeRegions.size() && !isMerged; i++) { |
|||
CellRangeAddress cellRangeAddr = mergeRegions.get(i); |
|||
// 若上一个单元格已经被合并，则先移出原有的合并单元，再重新添加合并单元
|
|||
if (cellRangeAddr.isInRange(curRowIndex - 1, curColIndex)) { |
|||
sheet.removeMergedRegion(i); |
|||
cellRangeAddr.setLastRow(curRowIndex); |
|||
sheet.addMergedRegion(cellRangeAddr); |
|||
isMerged = true; |
|||
} |
|||
} |
|||
// 若上一个单元格未被合并，则新增合并单元
|
|||
if (!isMerged) { |
|||
CellRangeAddress cellRangeAddress = new CellRangeAddress(curRowIndex - 1, curRowIndex, curColIndex, curColIndex); |
|||
sheet.addMergedRegion(cellRangeAddress); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,27 @@ |
|||
package com.jiagutech.handler; |
|||
|
|||
import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler; |
|||
import com.fasterxml.jackson.core.type.TypeReference; |
|||
import com.jiagutech.dto.ModelPredictResponse; |
|||
import com.fasterxml.jackson.databind.ObjectMapper; |
|||
|
|||
import java.sql.ResultSet; |
|||
import java.sql.SQLException; |
|||
import java.util.List; |
|||
|
|||
public class ModelPredictResponseListTypeHandler extends JacksonTypeHandler { |
|||
public ModelPredictResponseListTypeHandler(Class<?> type) { |
|||
super(type); |
|||
} |
|||
|
|||
@Override |
|||
public Object getResult(ResultSet rs, String columnName) throws SQLException { |
|||
String json = rs.getString(columnName); |
|||
try { |
|||
ObjectMapper objectMapper = new ObjectMapper(); |
|||
return objectMapper.readValue(json, new TypeReference<List<ModelPredictResponse>>(){}); |
|||
} catch (Exception e) { |
|||
throw new SQLException("Error deserializing recognition_data", e); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,13 @@ |
|||
package com.jiagutech.mapper; |
|||
|
|||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import org.apache.ibatis.annotations.Param; |
|||
|
|||
import java.util.List; |
|||
|
|||
|
|||
public interface RecordContentMapper extends BaseMapper<RecordContentEntity> { |
|||
|
|||
List<RecordContentEntity> selectListByRecordId(@Param("recordId") Long recordId); |
|||
} |
|||
@ -0,0 +1,14 @@ |
|||
package com.jiagutech.mapper; |
|||
|
|||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
|||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
|||
import com.jiagutech.dto.RecordDetail; |
|||
import com.jiagutech.dto.RecordItem; |
|||
import com.jiagutech.entity.RecordEntity; |
|||
|
|||
public interface RecordMapper extends BaseMapper<RecordEntity> { |
|||
|
|||
RecordDetail getRecordDetail(long id); |
|||
|
|||
Page<RecordItem> pageList(Page<RecordItem> page, Object o); |
|||
} |
|||
@ -0,0 +1,10 @@ |
|||
package com.jiagutech.mapper; |
|||
|
|||
import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
|||
import com.jiagutech.dto.LoginUser; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import com.jiagutech.entity.UserEntity; |
|||
|
|||
public interface UserMapper extends BaseMapper<UserEntity> { |
|||
LoginUser selectLoginUserByPhone(String phone); |
|||
} |
|||
@ -0,0 +1,93 @@ |
|||
package com.jiagutech.record; |
|||
|
|||
import com.jiagutech.common.thread.MonitoredRunnable; |
|||
import com.jiagutech.dto.CreateRecordRequest; |
|||
import com.jiagutech.dto.LoginUser; |
|||
import com.jiagutech.utils.LoginUtil; |
|||
import lombok.Data; |
|||
import lombok.SneakyThrows; |
|||
import lombok.experimental.Accessors; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import java.io.File; |
|||
import java.net.URL; |
|||
import java.util.concurrent.ThreadPoolExecutor; |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
@Slf4j |
|||
public class RecordTaskHandler { |
|||
|
|||
private RecordTask recordTask; |
|||
private ThreadPoolExecutor theadPoolExecutor; |
|||
private RecordTaskProcessor processor; |
|||
|
|||
|
|||
|
|||
public RecordTaskHandler(ThreadPoolExecutor threadPoolExecutor, RecordTaskProcessor processor) { |
|||
this.theadPoolExecutor = threadPoolExecutor; |
|||
this.processor = processor; |
|||
} |
|||
|
|||
protected void processTask() { |
|||
theadPoolExecutor.execute(new MonitoredRunnable(() -> { |
|||
try { |
|||
processor.handle(recordTask); |
|||
log.info("任务完成，任务ID:{}", recordTask.getTaskId()); |
|||
} finally { |
|||
for (File file : recordTask.getFiles()) { |
|||
log.info("删除临时文件:{}", file.getAbsolutePath()); |
|||
if (file.exists()) { |
|||
if (!file.delete()) |
|||
log.error("删除临时文件失败:{}", file.getAbsolutePath()); |
|||
} |
|||
} |
|||
RecordTaskManager.removeRecordTask(recordTask.getTaskId()); |
|||
recordTask = null; |
|||
|
|||
} |
|||
})); |
|||
} |
|||
|
|||
@SneakyThrows |
|||
public void createTask(CreateRecordRequest request, String taskId) { |
|||
URL resource = RecordTaskHandler.class.getClassLoader().getResource(""); |
|||
assert resource != null; |
|||
String path = resource.getPath(); |
|||
File[] files = new File[request.getFiles().length]; |
|||
for (int i = 0; i < request.getFiles().length; i++) { |
|||
MultipartFile file = request.getFiles()[i]; |
|||
File file1 = new File(path + file.getOriginalFilename()); |
|||
file.transferTo(file1); |
|||
files[i] = file1; |
|||
} |
|||
|
|||
this.recordTask = new RecordTask() |
|||
.setFiles(files) |
|||
.setModelName(request.getModelName()) |
|||
.setIou(request.getIou()) |
|||
.setConf(request.getConf()) |
|||
.setProgress(5f) |
|||
.setTaskName(request.getTaskName()) |
|||
.setTaskId(taskId) |
|||
.setLoginUser(LoginUtil.getLoginUser()); |
|||
processTask(); |
|||
} |
|||
|
|||
@Data |
|||
@Accessors(chain = true) |
|||
public static class RecordTask { |
|||
private String taskId; |
|||
private File[] files; |
|||
private String modelName; |
|||
private Float iou; |
|||
private Float conf; |
|||
private Float progress; |
|||
private String taskName; |
|||
private LoginUser loginUser; |
|||
|
|||
} |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
package com.jiagutech.record; |
|||
|
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
|
|||
public class RecordTaskManager { |
|||
|
|||
private static final ConcurrentHashMap<String, RecordTaskHandler> recordTasks = new ConcurrentHashMap<>(); |
|||
|
|||
public static void addRecordTask(String taskId, RecordTaskHandler handler) { |
|||
recordTasks.put(taskId, handler); |
|||
} |
|||
|
|||
public static RecordTaskHandler getRecordTask(String taskId) { |
|||
return recordTasks.get(taskId); |
|||
} |
|||
|
|||
public static void removeRecordTask(String taskId) { |
|||
recordTasks.remove(taskId); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,112 @@ |
|||
package com.jiagutech.record; |
|||
|
|||
import com.jiagutech.client.RecognitionModelClient; |
|||
import com.jiagutech.dto.ModelPredictResponse; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import com.jiagutech.entity.RecordEntity; |
|||
import com.jiagutech.mapper.RecordContentMapper; |
|||
import com.jiagutech.mapper.RecordMapper; |
|||
import com.jiagutech.utils.HuaweiObs; |
|||
import lombok.NonNull; |
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.stereotype.Component; |
|||
import org.springframework.transaction.annotation.Transactional; |
|||
import org.springframework.util.CollectionUtils; |
|||
|
|||
import java.io.File; |
|||
import java.math.BigDecimal; |
|||
import java.math.RoundingMode; |
|||
import java.time.LocalDateTime; |
|||
import java.util.List; |
|||
|
|||
@Slf4j |
|||
@Component |
|||
@RequiredArgsConstructor |
|||
public class RecordTaskProcessor { |
|||
private final RecordMapper recordMapper; |
|||
|
|||
private final RecordContentMapper recordContentMapper; |
|||
|
|||
private final RecognitionModelClient modelClient; |
|||
|
|||
private final HuaweiObs huaweiObs; |
|||
|
|||
@Transactional |
|||
public void handle(@NonNull RecordTaskHandler.RecordTask recordTask) { |
|||
log.info("开始处理任务：{}", recordTask.getTaskId()); |
|||
float progress = 5f; |
|||
RecordEntity recordEntity = new RecordEntity().setName(recordTask.getTaskName()) |
|||
.setImageNum(recordTask.getFiles().length) |
|||
.setUserId(recordTask.getLoginUser().getUserId()) |
|||
.setCreateTime(LocalDateTime.now()) |
|||
.setUpdateTime(LocalDateTime.now()); |
|||
recordMapper.insert(recordEntity); |
|||
recordTask.setProgress(progress); |
|||
for (File file : recordTask.getFiles()) { |
|||
List<ModelPredictResponse> predict = modelClient.predict(file, recordTask.getModelName(), recordTask.getIou(), recordTask.getConf()); |
|||
int handleNum = 0; |
|||
int rapeNum = 0; |
|||
int beakNum = 0; |
|||
for (ModelPredictResponse item : predict) { |
|||
item.setLength(getMaxLength(item.getPoints())); |
|||
switch (item.getClassify()) { |
|||
case 0: |
|||
rapeNum++; |
|||
break; |
|||
case 1: |
|||
beakNum++; |
|||
break; |
|||
default: |
|||
handleNum++; |
|||
} |
|||
} |
|||
progress += (float) (100 / recordTask.getFiles().length - 10); |
|||
recordTask.setProgress(progress); |
|||
if (!CollectionUtils.isEmpty(predict)) { |
|||
String imageUrl; |
|||
try { |
|||
imageUrl = huaweiObs.uploadFile(file); |
|||
progress += 2f; |
|||
} catch (Exception e) { |
|||
throw new RuntimeException(e); |
|||
} finally { |
|||
file.deleteOnExit(); |
|||
} |
|||
RecordContentEntity recordContentEntity = new RecordContentEntity() |
|||
.setRecordId(recordEntity.getId()).setImageUrl(imageUrl) |
|||
.setRecognitionData(predict) |
|||
.setImageName(file.getName()) |
|||
.setConf(recordTask.getConf()) |
|||
.setIou(recordTask.getIou()) |
|||
.setHandleNum(handleNum) |
|||
.setBeakNum(beakNum) |
|||
.setSiliquaNum(rapeNum); |
|||
recordContentMapper.insert(recordContentEntity); |
|||
progress += 3f; |
|||
} |
|||
|
|||
} |
|||
|
|||
} |
|||
|
|||
|
|||
private static double calculateDistance(int x1, int y1, int x2, int y2) { |
|||
return Math.sqrt(Math.pow(x2 - x1, 2) + Math.pow(y2 - y1, 2)); |
|||
} |
|||
|
|||
private float getMaxLength(int[][] points) { |
|||
// 计算四条边的长度
|
|||
double ab = calculateDistance(points[0][0], points[0][1], points[1][0], points[1][1]); // AB
|
|||
double bc = calculateDistance(points[1][0], points[1][1], points[2][0], points[2][1]); // BC
|
|||
double cd = calculateDistance(points[2][0], points[2][1], points[3][0], points[3][1]); // CD
|
|||
double da = calculateDistance(points[3][0], points[3][1], points[0][0], points[0][1]); // DA
|
|||
double max = Math.max(Math.max(ab, bc), Math.max(cd, da)); |
|||
BigDecimal b = new BigDecimal(max); |
|||
// 找出最大边长
|
|||
return b.setScale(2, RoundingMode.HALF_UP).floatValue(); |
|||
|
|||
|
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,33 @@ |
|||
package com.jiagutech.service; |
|||
|
|||
import com.jiagutech.dto.CreateRecordRequest; |
|||
import com.jiagutech.dto.RecordDetail; |
|||
import com.jiagutech.dto.RecordItem; |
|||
import com.jiagutech.dto.RecordModel; |
|||
import com.jiagutech.dto.common.PageRequest; |
|||
import com.jiagutech.dto.common.PageResult; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
|||
|
|||
import javax.servlet.http.HttpServletResponse; |
|||
|
|||
public interface RecordService { |
|||
|
|||
SseEmitter createRecord(MultipartFile[] files, String modelName, Float lou, Float conf); |
|||
|
|||
String createRecordAsync(CreateRecordRequest request); |
|||
|
|||
Float getTaskProgress(String taskId); |
|||
|
|||
RecordDetail getRecordDetail(Long recordId); |
|||
|
|||
void exportRecord(Long recordId, HttpServletResponse response); |
|||
|
|||
PageResult<RecordItem> getPages(PageRequest pageRequest); |
|||
|
|||
void updateRecordContent(RecordContentEntity recordContentEntity); |
|||
|
|||
RecordModel getModels(); |
|||
|
|||
} |
|||
@ -0,0 +1,286 @@ |
|||
package com.jiagutech.service; |
|||
|
|||
import com.alibaba.excel.EasyExcel; |
|||
import com.alibaba.excel.ExcelWriter; |
|||
import com.alibaba.excel.util.MapUtils; |
|||
import com.alibaba.excel.write.builder.ExcelWriterSheetBuilder; |
|||
import com.alibaba.excel.write.merge.OnceAbsoluteMergeStrategy; |
|||
import com.alibaba.excel.write.metadata.WriteSheet; |
|||
import com.alibaba.excel.write.metadata.style.WriteCellStyle; |
|||
import com.alibaba.excel.write.style.HorizontalCellStyleStrategy; |
|||
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy; |
|||
import com.alibaba.fastjson2.JSON; |
|||
import com.baomidou.mybatisplus.core.conditions.Wrapper; |
|||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
|||
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; |
|||
import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
|||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
|||
import com.jiagutech.client.RecognitionModelClient; |
|||
import com.jiagutech.dto.*; |
|||
import com.jiagutech.dto.common.PageRequest; |
|||
import com.jiagutech.dto.common.PageResult; |
|||
import com.jiagutech.entity.RecordContentEntity; |
|||
import com.jiagutech.handler.ExcelFillCellMergeStrategy; |
|||
import com.jiagutech.mapper.RecordContentMapper; |
|||
import com.jiagutech.mapper.RecordMapper; |
|||
import com.jiagutech.record.RecordTaskHandler; |
|||
import com.jiagutech.record.RecordTaskManager; |
|||
import com.jiagutech.record.RecordTaskProcessor; |
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.SneakyThrows; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.apache.commons.math3.analysis.function.Max; |
|||
import org.apache.poi.ss.usermodel.HorizontalAlignment; |
|||
import org.apache.poi.ss.usermodel.VerticalAlignment; |
|||
import org.springframework.stereotype.Service; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
|||
|
|||
import javax.servlet.ServletOutputStream; |
|||
import javax.servlet.http.HttpServletResponse; |
|||
import java.io.IOException; |
|||
import java.io.UnsupportedEncodingException; |
|||
import java.math.BigDecimal; |
|||
import java.math.RoundingMode; |
|||
import java.net.URLEncoder; |
|||
import java.text.DecimalFormat; |
|||
import java.util.*; |
|||
import java.util.concurrent.ThreadPoolExecutor; |
|||
|
|||
@Slf4j |
|||
@Service |
|||
@RequiredArgsConstructor |
|||
public class RecordServiceImpl implements RecordService { |
|||
|
|||
private final ThreadPoolExecutor customThreadPool; |
|||
|
|||
private final RecordTaskProcessor taskProcessor; |
|||
private final RecordMapper recordMapper; |
|||
private final RecordContentMapper recordContentMapper; |
|||
private final RecognitionModelClient recognitionModelClient; |
|||
|
|||
@Override |
|||
public SseEmitter createRecord(MultipartFile[] files, String modelName, Float lou, Float conf) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public String createRecordAsync(CreateRecordRequest request) { |
|||
String taskId = UUID.randomUUID().toString(); |
|||
RecordTaskHandler recordTaskHandler = new RecordTaskHandler(customThreadPool, taskProcessor); |
|||
recordTaskHandler.createTask(request, taskId); |
|||
RecordTaskManager.addRecordTask(taskId, recordTaskHandler); |
|||
return taskId; |
|||
} |
|||
|
|||
|
|||
@Override |
|||
public Float getTaskProgress(String taskId) { |
|||
RecordTaskHandler recordTask = RecordTaskManager.getRecordTask(taskId); |
|||
if (recordTask == null) { |
|||
return 100f; |
|||
} |
|||
if (recordTask.getRecordTask() == null) { |
|||
return 100f; |
|||
} |
|||
|
|||
return recordTask.getRecordTask().getProgress(); |
|||
} |
|||
|
|||
@Override |
|||
public RecordDetail getRecordDetail(Long recordId) { |
|||
return recordMapper.getRecordDetail(recordId); |
|||
} |
|||
|
|||
@Override |
|||
public void exportRecord(Long recordId, HttpServletResponse response) { |
|||
RecordDetail recordDetail = this.getRecordDetail(recordId); |
|||
List<RecordContentEntity> content = recordDetail.getContent(); |
|||
if (CollectionUtils.isNotEmpty(content)) { |
|||
RecordExcel recordExcel = new RecordExcel(); |
|||
recordExcel.setImageNum(recordDetail.getImageNum()); |
|||
int beakNum = 0; |
|||
int rapeNum = 0; |
|||
int handleNum = 0; |
|||
int rapeMax = 0; |
|||
int rapeMin = 10000; |
|||
List<Double> avgBeakLen = new ArrayList<>(); |
|||
List<Double> avgRapeLen = new ArrayList<>(); |
|||
List<Double> avgHandleLen = new ArrayList<>(); |
|||
List<RecordContentExcel> recordContentExcelList = new ArrayList<>(); |
|||
List<OnceAbsoluteMergeStrategy> mergeStrategies = new ArrayList<>(); |
|||
int startRowIndex = 1; |
|||
int endRowIndex = 0; |
|||
//处理记录详情数据
|
|||
for (RecordContentEntity record : content) { |
|||
//统计各分类总数
|
|||
beakNum += record.getBeakNum(); |
|||
rapeNum += record.getSiliquaNum(); |
|||
handleNum += record.getHandleNum(); |
|||
if (record.getSiliquaNum() > rapeMax) { |
|||
rapeMax = record.getSiliquaNum(); |
|||
} |
|||
if (record.getSiliquaNum() < rapeMin) { |
|||
rapeMin = record.getSiliquaNum(); |
|||
} |
|||
List<Float> rapeLenList = new ArrayList<>(); |
|||
List<Float> beakLenList = new ArrayList<>(); |
|||
List<Float> handleLenList = new ArrayList<>(); |
|||
|
|||
List<ModelPredictResponse> recognitionData = record.getRecognitionData(); |
|||
int size = record.getSize(); |
|||
int scale = record.getScale(); |
|||
//把识别数据进行分类
|
|||
for (ModelPredictResponse item : recognitionData) { |
|||
float len; |
|||
if (size > 0 && scale > 0) { |
|||
BigDecimal b = new BigDecimal(item.getLength() * size / scale); |
|||
len = b.setScale(2, RoundingMode.HALF_UP).floatValue(); |
|||
} else { |
|||
len = item.getLength(); |
|||
} |
|||
|
|||
switch (item.getClassify()) { |
|||
case 0: |
|||
rapeLenList.add(len); |
|||
break; |
|||
case 1: |
|||
beakLenList.add(len); |
|||
break; |
|||
default: |
|||
handleLenList.add(len); |
|||
} |
|||
} |
|||
int len = Math.max(rapeLenList.size(), Math.max(handleLenList.size(), beakLenList.size())); |
|||
//创建表格行记录
|
|||
for (int i = 0; i < len; i++) { |
|||
endRowIndex++; |
|||
RecordContentExcel contentExcel = new RecordContentExcel(); |
|||
//需要合并的列开始
|
|||
contentExcel.setName(record.getImageName()); |
|||
contentExcel.setBeakNum(record.getBeakNum()); |
|||
contentExcel.setRapeNum(record.getSiliquaNum()); |
|||
contentExcel.setHandleNum(record.getHandleNum()); |
|||
contentExcel.setScale(record.getScale()); |
|||
contentExcel.setConf(record.getConf()); |
|||
contentExcel.setIou(record.getIou()); |
|||
//需要合并的列结束
|
|||
recordContentExcelList.add(contentExcel); |
|||
if (i <= rapeLenList.size() - 1) { |
|||
contentExcel.setRapeLength(rapeLenList.get(i)); |
|||
} |
|||
if (i <= beakLenList.size() - 1) { |
|||
contentExcel.setBeakLength(beakLenList.get(i)); |
|||
} |
|||
if (i <= handleLenList.size() - 1) { |
|||
contentExcel.setHandleLength(handleLenList.get(i)); |
|||
} |
|||
} |
|||
//创建单元格的合并策略, 前7列数据需要合并
|
|||
for (int i = 0; i <= 6; i++) { |
|||
OnceAbsoluteMergeStrategy mergeStrategy = new OnceAbsoluteMergeStrategy( |
|||
startRowIndex, |
|||
endRowIndex, |
|||
i, i); |
|||
mergeStrategies.add(mergeStrategy); |
|||
} |
|||
|
|||
startRowIndex += len; |
|||
avgHandleLen.add(handleLenList.stream().mapToDouble(Float::doubleValue).average().orElse(0.0)); |
|||
avgRapeLen.add(rapeLenList.stream().mapToDouble(Float::doubleValue).average().orElse(0.0)); |
|||
avgBeakLen.add(beakLenList.stream().mapToDouble(Float::doubleValue).average().orElse(0.0)); |
|||
} |
|||
recordExcel.setBeakNum(beakNum); |
|||
recordExcel.setRapeNum(rapeNum); |
|||
recordExcel.setHandleNum(handleNum); |
|||
recordExcel.setMaxRapeNum(rapeMax); |
|||
recordExcel.setMinRapeNum(rapeMin); |
|||
DecimalFormat dfnum = new DecimalFormat("##0.00"); |
|||
float avgRapeNum = Float.parseFloat(dfnum.format((float) rapeNum / content.size())); |
|||
recordExcel.setAvgRapeNum(avgRapeNum); |
|||
double v = avgBeakLen.stream().mapToDouble(Double::doubleValue).average().orElse(0.0); |
|||
BigDecimal b = new BigDecimal(v); |
|||
recordExcel.setAvgBeakLength(b.setScale(2, RoundingMode.HALF_UP).floatValue()); |
|||
double r = avgRapeLen.stream().mapToDouble(Double::doubleValue).average().orElse(0.0); |
|||
BigDecimal b1 = new BigDecimal(r); |
|||
recordExcel.setAvgRapeLength(b1.setScale(2, RoundingMode.HALF_UP).floatValue()); |
|||
double h = avgHandleLen.stream().mapToDouble(Double::doubleValue).average().orElse(0.0); |
|||
BigDecimal b2 = new BigDecimal(h); |
|||
recordExcel.setAvgHandleLength(b2.setScale(2, RoundingMode.HALF_UP).floatValue()); |
|||
ServletOutputStream outputStream = null; |
|||
try { |
|||
outputStream = response.getOutputStream(); |
|||
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); |
|||
response.setCharacterEncoding("utf-8"); |
|||
// 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
|
|||
String fileName = URLEncoder.encode(recordDetail.getName(), "UTF-8").replaceAll("\\+", "%20"); |
|||
response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx"); |
|||
//创建 Cell 样式
|
|||
WriteCellStyle writeCellStyle = new WriteCellStyle(); |
|||
//设置垂直对齐方式
|
|||
writeCellStyle.setVerticalAlignment(VerticalAlignment.CENTER); |
|||
//设置水平对齐方式
|
|||
writeCellStyle.setHorizontalAlignment(HorizontalAlignment.CENTER); |
|||
ExcelWriter writer = EasyExcel.write(outputStream).autoCloseStream(false).build(); |
|||
//总览表格的sheet
|
|||
WriteSheet recordSheet = EasyExcel.writerSheet("总览").head(RecordExcel.class).build(); |
|||
//详情表格的sheet
|
|||
ExcelWriterSheetBuilder sheetBuilder = EasyExcel.writerSheet("详情") |
|||
.registerWriteHandler(new HorizontalCellStyleStrategy(null, writeCellStyle)) |
|||
.registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) |
|||
.head(RecordContentExcel.class); |
|||
//添加单元格合并策略
|
|||
for (OnceAbsoluteMergeStrategy mergeStrategy : mergeStrategies) { |
|||
sheetBuilder.registerWriteHandler(mergeStrategy); |
|||
} |
|||
WriteSheet contentSheet = sheetBuilder.build(); |
|||
writer.write(Collections.singletonList(recordExcel), recordSheet); |
|||
writer.write(recordContentExcelList, contentSheet); |
|||
writer.finish(); |
|||
} catch (Exception e) { |
|||
log.error("下载文件失败", e); |
|||
response.reset(); |
|||
response.setContentType("application/json"); |
|||
response.setCharacterEncoding("utf-8"); |
|||
Map<String, String> map = MapUtils.newHashMap(); |
|||
map.put("status", "failure"); |
|||
map.put("message", "下载文件失败" + e.getMessage()); |
|||
try { |
|||
response.getWriter().println(JSON.toJSONString(map)); |
|||
} catch (IOException ex) { |
|||
throw new RuntimeException(ex); |
|||
} |
|||
} finally { |
|||
try { |
|||
if (outputStream != null) { |
|||
outputStream.close(); |
|||
} |
|||
} catch (IOException e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
} |
|||
} |
|||
|
|||
|
|||
} |
|||
|
|||
|
|||
@Override |
|||
public PageResult<RecordItem> getPages(PageRequest pageRequest) { |
|||
Page<RecordItem> page = new Page(pageRequest.getPageNum(), pageRequest.getPageSize()); |
|||
Page<RecordItem> result = recordMapper.pageList(page, null); |
|||
return PageResult.of((int) result.getTotal(), (int) result.getSize(), (int) result.getCurrent(), result.getRecords()); |
|||
|
|||
} |
|||
|
|||
|
|||
@Override |
|||
public void updateRecordContent(RecordContentEntity recordContentEntity) { |
|||
recordContentMapper.updateById(recordContentEntity); |
|||
} |
|||
|
|||
@Override |
|||
public RecordModel getModels() { |
|||
return recognitionModelClient.models(); |
|||
} |
|||
} |
|||
@ -0,0 +1,36 @@ |
|||
package com.jiagutech.service; |
|||
|
|||
import org.springframework.stereotype.Service; |
|||
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
|||
|
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
|
|||
@Service |
|||
public class TaskProgressSSEService { |
|||
|
|||
private final ConcurrentHashMap<String, SseEmitter> sseEmitters = new ConcurrentHashMap<>(); |
|||
public SseEmitter createProgressEmitter(String taskId) { |
|||
SseEmitter emitter = new SseEmitter(); |
|||
sseEmitters.put(taskId, emitter); |
|||
return emitter; |
|||
} |
|||
|
|||
public void sendProgress(String taskId, int progress) throws Exception { |
|||
SseEmitter emitter = sseEmitters.get(taskId); |
|||
if (emitter != null) { |
|||
emitter.send(progress); |
|||
if (progress >= 100) { |
|||
emitter.complete(); |
|||
sseEmitters.remove(taskId); |
|||
} |
|||
} |
|||
} |
|||
|
|||
public void completeEmitter(String taskId) { |
|||
SseEmitter emitter = sseEmitters.get(taskId); |
|||
if (emitter != null) { |
|||
emitter.complete(); |
|||
sseEmitters.remove(taskId); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,13 @@ |
|||
package com.jiagutech.service; |
|||
|
|||
import com.jiagutech.dto.LoginRequest; |
|||
import com.jiagutech.dto.LoginResponse; |
|||
import com.jiagutech.dto.UserRequest; |
|||
|
|||
public interface UserService { |
|||
LoginResponse login(LoginRequest loginBody); |
|||
|
|||
void register(UserRequest user); |
|||
|
|||
void approval(Long userId); |
|||
} |
|||
@ -0,0 +1,71 @@ |
|||
package com.jiagutech.service; |
|||
|
|||
import cn.dev33.satoken.secure.BCrypt; |
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import com.jiagutech.common.UserConstants; |
|||
import com.jiagutech.common.exception.BizCode; |
|||
import com.jiagutech.common.exception.BusinessException; |
|||
import com.jiagutech.dto.LoginRequest; |
|||
import com.jiagutech.dto.LoginResponse; |
|||
import com.jiagutech.dto.LoginUser; |
|||
import com.jiagutech.dto.UserRequest; |
|||
import com.jiagutech.entity.UserEntity; |
|||
import com.jiagutech.mapper.UserMapper; |
|||
import com.jiagutech.utils.EmailUtil; |
|||
import lombok.RequiredArgsConstructor; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import java.time.LocalDateTime; |
|||
|
|||
@Service |
|||
@RequiredArgsConstructor |
|||
public class UserServiceImpl implements UserService { |
|||
private final UserMapper userMapper; |
|||
private final EmailUtil emailUtil; |
|||
|
|||
|
|||
@Override |
|||
public LoginResponse login(LoginRequest loginBody) { |
|||
String phone = loginBody.getPhone(); |
|||
String password = loginBody.getPassword(); |
|||
LoginUser user = userMapper.selectLoginUserByPhone(phone); |
|||
if (user == null) { |
|||
throw new BusinessException(BizCode.USER_NOT_FOUND); |
|||
} |
|||
if (user.getStatus() != 1) { |
|||
throw new BusinessException(BizCode.USER_STATUS_ERROR); |
|||
} |
|||
boolean checkPw = BCrypt.checkpw(password, user.getPassword()); |
|||
if (!checkPw) { |
|||
throw new BusinessException(BizCode.PASSWORD_ERROR); |
|||
} |
|||
StpUtil.login(user.getUserId()); |
|||
user.setLoginTime(System.currentTimeMillis()); |
|||
StpUtil.getSession().set(UserConstants.SYS_SESSION, user); |
|||
LoginResponse loginVO = new LoginResponse(); |
|||
loginVO.setAccessToken(StpUtil.getTokenValue()); |
|||
loginVO.setExpireIn(StpUtil.getTokenTimeout()); |
|||
return loginVO; |
|||
} |
|||
|
|||
@Override |
|||
public void register(UserRequest user) { |
|||
UserEntity userEntity = new UserEntity() |
|||
.setCreateTime(LocalDateTime.now()) |
|||
.setStatus(2) |
|||
.setRealName(user.getRealName()) |
|||
.setUnitName(user.getUnitName()) |
|||
.setPhone(user.getPhone()) |
|||
.setPassword(BCrypt.hashpw(user.getPassword())); |
|||
userMapper.insert(userEntity); |
|||
emailUtil.sendEmail(null, user.getPhone(), user.getUnitName(), user.getRealName()); |
|||
|
|||
} |
|||
|
|||
@Override |
|||
public void approval(Long userId) { |
|||
UserEntity userEntity = userMapper.selectById(userId); |
|||
userEntity.setStatus(1); |
|||
userMapper.updateById(userEntity); |
|||
} |
|||
} |
|||
@ -0,0 +1,45 @@ |
|||
package com.jiagutech.utils; |
|||
|
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.beans.factory.annotation.Value; |
|||
import org.springframework.mail.javamail.JavaMailSender; |
|||
import org.springframework.mail.javamail.MimeMessageHelper; |
|||
import org.springframework.stereotype.Component; |
|||
|
|||
import javax.mail.MessagingException; |
|||
import javax.mail.internet.MimeMessage; |
|||
|
|||
@Slf4j |
|||
@Component |
|||
@RequiredArgsConstructor |
|||
public class EmailUtil { |
|||
|
|||
private final JavaMailSender mailSender; |
|||
@Value("${email.to}") |
|||
private String MAILTO; |
|||
|
|||
public void sendEmail(String to, String phone, String unitName, String realName) { |
|||
try { |
|||
// 构建邮件
|
|||
MimeMessage message = mailSender.createMimeMessage(); |
|||
MimeMessageHelper helper = new MimeMessageHelper(message, true, "UTF-8"); |
|||
|
|||
helper.setFrom("noreply@pushmail.jiagutech.com"); |
|||
String receiver = to == null ? MAILTO : to; |
|||
helper.setTo(receiver); |
|||
helper.setSubject("新注册用户待审核"); |
|||
|
|||
String text = String.format("【用户待审核】现有<b>%s</b>用户<b>%s(%s)</b><br>已注册成为角果识别系统的新用户" + |
|||
"，请您尽快审核！<br><br>", unitName, realName, phone); |
|||
helper.setText(text, true); // true 表示 HTML 格式
|
|||
|
|||
// 发送邮件
|
|||
mailSender.send(message); |
|||
log.info("邮件发送成功: {}", receiver); |
|||
|
|||
} catch (MessagingException e) { |
|||
log.error("邮件发送失败: ", e); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,52 @@ |
|||
package com.jiagutech.utils; |
|||
|
|||
import cn.hutool.core.convert.Convert; |
|||
import cn.hutool.core.util.ObjectUtil; |
|||
import com.alibaba.excel.converters.Converter; |
|||
import com.alibaba.excel.enums.CellDataTypeEnum; |
|||
import com.alibaba.excel.metadata.GlobalConfiguration; |
|||
import com.alibaba.excel.metadata.data.ReadCellData; |
|||
import com.alibaba.excel.metadata.data.WriteCellData; |
|||
import com.alibaba.excel.metadata.property.ExcelContentProperty; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
|
|||
import java.math.BigDecimal; |
|||
|
|||
/** |
|||
* 大数值转换 |
|||
* Excel 数值长度位15位 大于15位的数值转换位字符串 |
|||
* |
|||
* @author Lion Li |
|||
*/ |
|||
@Slf4j |
|||
public class ExcelBigNumberConvert implements Converter<Long> { |
|||
|
|||
@Override |
|||
public Class<Long> supportJavaTypeKey() { |
|||
return Long.class; |
|||
} |
|||
|
|||
@Override |
|||
public CellDataTypeEnum supportExcelTypeKey() { |
|||
return CellDataTypeEnum.STRING; |
|||
} |
|||
|
|||
@Override |
|||
public Long convertToJavaData(ReadCellData<?> cellData, ExcelContentProperty contentProperty, GlobalConfiguration globalConfiguration) { |
|||
return Convert.toLong(cellData.getData()); |
|||
} |
|||
|
|||
@Override |
|||
public WriteCellData<Object> convertToExcelData(Long object, ExcelContentProperty contentProperty, GlobalConfiguration globalConfiguration) { |
|||
if (ObjectUtil.isNotNull(object)) { |
|||
String str = Convert.toStr(object); |
|||
if (str.length() > 15) { |
|||
return new WriteCellData<>(str); |
|||
} |
|||
} |
|||
WriteCellData<Object> cellData = new WriteCellData<>(new BigDecimal(object)); |
|||
cellData.setType(CellDataTypeEnum.NUMBER); |
|||
return cellData; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,53 @@ |
|||
package com.jiagutech.utils; |
|||
|
|||
import cn.hutool.core.util.IdUtil; |
|||
import com.alibaba.excel.EasyExcel; |
|||
import com.alibaba.excel.write.builder.ExcelWriterSheetBuilder; |
|||
import com.alibaba.excel.write.handler.impl.DefaultRowWriteHandler; |
|||
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy; |
|||
|
|||
import javax.servlet.ServletOutputStream; |
|||
import javax.servlet.http.HttpServletResponse; |
|||
import java.io.IOException; |
|||
import java.net.URLEncoder; |
|||
import java.nio.charset.StandardCharsets; |
|||
import java.util.List; |
|||
|
|||
/** |
|||
* @ClassName ExcelUtil |
|||
* @author: zhangyeguang |
|||
* @create: 2024-09-04 10:28 |
|||
* @Version 1.0 |
|||
* @description: |
|||
**/ |
|||
public class ExcelUtil { |
|||
|
|||
|
|||
public static <T> void exportExcel(List<T> list, String sheetName, Class<T> clazz, HttpServletResponse response) { |
|||
try { |
|||
String filename = IdUtil.fastSimpleUUID() + "_" + sheetName + ".xlsx"; |
|||
String percentEncodedFileName = URLEncoder.encode(filename, String.valueOf(StandardCharsets.UTF_8)).replaceAll("\\+", "%20"); |
|||
String contentDispositionValue = String.format( |
|||
"attachment; filename=%s;filename*=utf-8''%s", |
|||
percentEncodedFileName, |
|||
percentEncodedFileName |
|||
); |
|||
response.addHeader("Access-Control-Expose-Headers", "Content-Disposition,download-filename"); |
|||
response.setHeader("Content-disposition", contentDispositionValue); |
|||
response.setHeader("download-filename", percentEncodedFileName); |
|||
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=UTF-8"); |
|||
ServletOutputStream os = response.getOutputStream(); |
|||
ExcelWriterSheetBuilder builder = EasyExcel.write(os, clazz) |
|||
.autoCloseStream(false) |
|||
// 自动适配
|
|||
.registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) |
|||
// 大数值自动转换 防止失真
|
|||
.registerConverter(new ExcelBigNumberConvert()) |
|||
.sheet(sheetName); |
|||
builder.registerWriteHandler(new DefaultRowWriteHandler()); |
|||
builder.doWrite(list); |
|||
} catch (IOException e) { |
|||
throw new RuntimeException("导出Excel异常"); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,300 @@ |
|||
package com.jiagutech.utils; |
|||
|
|||
import com.alibaba.fastjson2.JSON; |
|||
import com.obs.services.ObsClient; |
|||
import com.obs.services.exception.ObsException; |
|||
import com.obs.services.model.*; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.apache.commons.io.IOUtils; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Component; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import java.io.*; |
|||
import java.util.Calendar; |
|||
import java.util.HashMap; |
|||
import java.util.Map; |
|||
import java.util.UUID; |
|||
|
|||
/** |
|||
* @author Lenovo |
|||
*/ |
|||
@Slf4j |
|||
@Component |
|||
public class HuaweiObs { |
|||
|
|||
//Access Key Id
|
|||
//private String ak = "Access Key Id";
|
|||
|
|||
//Secret Access Key
|
|||
//private String sk = "Secret Access Key";
|
|||
|
|||
//桶名称
|
|||
private String bucketName = "jg-iot"; |
|||
|
|||
// 终端节点访问Endpoint
|
|||
//private String endpoint = "Endpoint";
|
|||
|
|||
// 文件目录
|
|||
private String prifix = "/recognition"; |
|||
|
|||
// 访问域名 在域名后面或文件目录前加“/”
|
|||
private String path = "/recognition/"; |
|||
|
|||
|
|||
@Autowired |
|||
private ObsClient obsClient; |
|||
|
|||
/** |
|||
* 文件上传 |
|||
* |
|||
* @param file |
|||
* @return |
|||
* @throws IOException |
|||
*/ |
|||
public String upload(MultipartFile file) throws Exception { |
|||
//ObsClient obsClient = null;
|
|||
|
|||
Calendar cal = Calendar.getInstance(); |
|||
int year = cal.get(Calendar.YEAR); |
|||
int month = cal.get(Calendar.MONTH); |
|||
int day = cal.get(Calendar.DATE); |
|||
|
|||
String fileName = Md5Util.calculateMD5(file.getInputStream()); |
|||
|
|||
String objectName = prifix + "/" + year + "/" + month + "/" + day + "/" + fileName; |
|||
//obsClient = new ObsClient(ak, sk, endpoint);
|
|||
PutObjectResult response = obsClient.putObject(bucketName, objectName, file.getInputStream()); |
|||
log.info(JSON.toJSONString(response)); |
|||
// 可选：调用成功后，记录调用成功的HTTP状态码和服务端请求ID
|
|||
int statusCode = response.getStatusCode(); |
|||
|
|||
if (200 == statusCode) { |
|||
|
|||
return response.getObjectUrl(); |
|||
} |
|||
|
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* 上传文件--流式 |
|||
* |
|||
* @param fileName 上传文件名称 |
|||
* @param is 文件流 |
|||
* @return |
|||
*/ |
|||
public String uploadFile(String fileName, InputStream is) { |
|||
try { |
|||
if (is.available() == 0) { |
|||
log.error("InputStream is empty."); |
|||
return null; |
|||
} |
|||
} catch (IOException e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
|
|||
Calendar cal = Calendar.getInstance(); |
|||
int year = cal.get(Calendar.YEAR); |
|||
int month = cal.get(Calendar.MONTH); |
|||
int day = cal.get(Calendar.DATE); |
|||
String objectName = prifix + "/" + year + "/" + month + "/" + day + "/" + fileName; |
|||
PutObjectResult response = obsClient.putObject(bucketName, objectName, is); |
|||
log.info(JSON.toJSONString(response)); |
|||
// 可选：调用成功后，记录调用成功的HTTP状态码和服务端请求ID
|
|||
int statusCode = response.getStatusCode(); |
|||
|
|||
if (200 == statusCode) { |
|||
|
|||
return response.getObjectUrl(); |
|||
} |
|||
|
|||
return null; |
|||
} |
|||
|
|||
public String uploadFile(File file) { |
|||
|
|||
try { |
|||
String subbfix = file.getName().substring(file.getName().indexOf(".")); |
|||
String fileName = UUID.randomUUID().toString().replaceAll("-", "") + subbfix; |
|||
Calendar cal = Calendar.getInstance(); |
|||
int year = cal.get(Calendar.YEAR); |
|||
int month = cal.get(Calendar.MONTH); |
|||
int day = cal.get(Calendar.DATE); |
|||
String objectName = prifix + "/" + year + "/" + month + "/" + fileName; |
|||
PutObjectRequest putObjectRequest = new PutObjectRequest(); |
|||
putObjectRequest.setFile(file); |
|||
putObjectRequest.setBucketName(bucketName); |
|||
putObjectRequest.setObjectKey(objectName); |
|||
PutObjectResult response = obsClient.putObject(putObjectRequest); |
|||
log.info(JSON.toJSONString(response)); |
|||
// 可选：调用成功后，记录调用成功的HTTP状态码和服务端请求ID
|
|||
int statusCode = response.getStatusCode(); |
|||
if (200 == statusCode) { |
|||
return response.getObjectUrl(); |
|||
} |
|||
} catch (ObsException e) { |
|||
log.error("文件上传失败：http code={},error code={},error message={}", e.getResponseCode(), e.getErrorCode(), e.getErrorMessage(), e); |
|||
} catch (Exception e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
|
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* 上传文件--字节数组 |
|||
* |
|||
* @param fileName 上传文件名称 |
|||
* @param fileType 文件路径 |
|||
* @param is 文件流 |
|||
* @return |
|||
*/ |
|||
public boolean uploadFileByte(String fileName, FileType fileType, byte[] is) { |
|||
//ObsClient obsClient = null;
|
|||
try { |
|||
String objectName = fileType.getType().concat("/").concat(fileName); |
|||
//obsClient = new ObsClient(ak, sk, endpoint);
|
|||
HeaderResponse response = obsClient.putObject(bucketName, objectName, new ByteArrayInputStream(is)); |
|||
// 可选：调用成功后，记录调用成功的HTTP状态码和服务端请求ID
|
|||
int statusCode = response.getStatusCode(); |
|||
if (200 == statusCode) { |
|||
return true; |
|||
} |
|||
obsClient.close(); |
|||
} catch (IOException e) { |
|||
log.info("文件上传失败：{}", e.getMessage(), e); |
|||
} |
|||
return false; |
|||
} |
|||
|
|||
/** |
|||
* 下载文件 |
|||
* |
|||
* @param fileName 文件名称 |
|||
* @param fileType 文件路径 |
|||
* @return |
|||
*/ |
|||
public String getDownloadUrl(String fileName, FileType fileType) { |
|||
//ObsClient obsClient = null;
|
|||
//obsClient = new ObsClient(ak, sk, endpoint);
|
|||
// URL有效期，3600秒.5分钟
|
|||
long expireSeconds = 3600L; |
|||
String objectName = fileType.getType().concat("/").concat(fileName); |
|||
TemporarySignatureRequest request = new TemporarySignatureRequest(HttpMethodEnum.GET, expireSeconds); |
|||
request.setBucketName(bucketName); |
|||
request.setObjectKey(objectName); |
|||
TemporarySignatureResponse response = obsClient.createTemporarySignature(request); |
|||
return response.getSignedUrl(); |
|||
} |
|||
|
|||
/** |
|||
* 获取上传地址 |
|||
* |
|||
* @param fileName 文件名称 |
|||
* @param fileType 文件路径 |
|||
* @return |
|||
*/ |
|||
public String getUploadUrl(String fileName, FileType fileType) { |
|||
try { |
|||
// 创建ObsClient实例
|
|||
//ObsClient obsClient = new ObsClient(ak, sk, endpoint);
|
|||
// URL有效期，3600秒
|
|||
long expireSeconds = 3600L; |
|||
Map<String, String> headers = new HashMap<String, String>(); |
|||
headers.put("Content-Type", "application/octet-stream"); |
|||
String objectName = fileType.getType().concat("/").concat(fileName); |
|||
TemporarySignatureRequest request = new TemporarySignatureRequest(HttpMethodEnum.PUT, expireSeconds); |
|||
request.setBucketName(bucketName); |
|||
request.setObjectKey(objectName); |
|||
request.setHeaders(headers); |
|||
TemporarySignatureResponse response = obsClient.createTemporarySignature(request); |
|||
return response.getSignedUrl(); |
|||
} catch (Exception e) { |
|||
log.error("获取上传地址异常：{}", e.getMessage(), e); |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* 下载文件返回字节数组 |
|||
* |
|||
* @param fileName 文件名称 |
|||
* @param fileType 文件路径 |
|||
* @return |
|||
*/ |
|||
public byte[] downloadFile(String fileName, FileType fileType) { |
|||
try { |
|||
InputStream inputStream = downloadFileInputStream(fileName, fileType); |
|||
byte[] bytes = IOUtils.toByteArray(inputStream); |
|||
return bytes; |
|||
} catch (Exception e) { |
|||
log.error("下载文件异常：{}", e.getMessage(), e); |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* 上传视频 |
|||
* |
|||
* @param fileName 文件名称 |
|||
* @param fileType 文件路径 |
|||
* @return |
|||
*/ |
|||
public boolean uploadFileVideo(String fileName, FileType fileType, InputStream is) { |
|||
try { |
|||
String objectName = fileType.getType().concat("/").concat(fileName); |
|||
//ObsClient obsClient = new ObsClient(ak, sk, endpoint);
|
|||
// 添加 ContentType (添加后可在浏览器中直接浏览，而非下载链接)
|
|||
obsClient.putObject(bucketName, objectName, is); |
|||
obsClient.close(); |
|||
return true; |
|||
} catch (Exception e) { |
|||
log.error("上传视频文件异常：{}", e.getMessage(), e); |
|||
} |
|||
return false; |
|||
} |
|||
|
|||
/** |
|||
* 下载文件返回流式 |
|||
* |
|||
* @param fileName 文件名称 |
|||
* @param fileType 文件路径 |
|||
* @return |
|||
*/ |
|||
public InputStream downloadFileInputStream(String fileName, FileType fileType) { |
|||
try { |
|||
String objectName = fileType.getType().concat("/").concat(fileName); |
|||
// 用户拿到STS临时凭证后，通过其中的安全令牌（SecurityToken）和临时访问密钥（AccessKeyId和AccessKeySecret）生成OSSClient。
|
|||
//ObsClient obsClient = new ObsClient(ak, sk, endpoint);
|
|||
ObsObject obsObject = obsClient.getObject(bucketName, objectName); |
|||
obsClient.close(); |
|||
return obsObject.getObjectContent(); |
|||
} catch (Exception e) { |
|||
log.error("下载文件异常：{}", e.getMessage(), e); |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
|
|||
public enum FileType { |
|||
TEST("TEST", "测试"); |
|||
|
|||
private String type; |
|||
private String desc; |
|||
|
|||
FileType(String type, String desc) { |
|||
this.type = type; |
|||
this.desc = desc; |
|||
} |
|||
|
|||
public String getType() { |
|||
return type; |
|||
} |
|||
|
|||
public String getDesc() { |
|||
return desc; |
|||
} |
|||
} |
|||
} |
|||
|
|||
@ -0,0 +1,14 @@ |
|||
package com.jiagutech.utils; |
|||
|
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import com.jiagutech.common.UserConstants; |
|||
import com.jiagutech.dto.LoginUser; |
|||
|
|||
public class LoginUtil { |
|||
|
|||
|
|||
public static LoginUser getLoginUser() { |
|||
return (LoginUser) StpUtil.getSession().get(UserConstants.SYS_SESSION); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,34 @@ |
|||
package com.jiagutech.utils; |
|||
|
|||
import java.io.InputStream; |
|||
import java.security.MessageDigest; |
|||
|
|||
public class Md5Util { |
|||
public static String calculateMD5(InputStream inputStream) throws Exception { |
|||
// 创建 MD5 消息摘要实例
|
|||
MessageDigest digest = MessageDigest.getInstance("MD5"); |
|||
|
|||
byte[] buffer = new byte[1024]; // 缓冲区大小
|
|||
int bytesRead; |
|||
|
|||
// 读取 InputStream 并更新 MD5 摘要
|
|||
while ((bytesRead = inputStream.read(buffer)) != -1) { |
|||
digest.update(buffer, 0, bytesRead); |
|||
} |
|||
|
|||
// 获取 MD5 摘要结果
|
|||
byte[] md5Bytes = digest.digest(); |
|||
|
|||
// 转换为 16 进制字符串
|
|||
StringBuilder hexString = new StringBuilder(); |
|||
for (byte b : md5Bytes) { |
|||
String hex = Integer.toHexString(0xff & b); |
|||
if (hex.length() == 1) { |
|||
hexString.append('0'); |
|||
} |
|||
hexString.append(hex); |
|||
} |
|||
|
|||
return hexString.toString(); |
|||
} |
|||
} |
|||
@ -0,0 +1,101 @@ |
|||
server: |
|||
port: 9196 |
|||
spring: |
|||
application: |
|||
name: siliqua_recognition |
|||
|
|||
datasource: |
|||
driver-class-name: com.mysql.cj.jdbc.Driver |
|||
url: jdbc:mysql://101.34.243.138:7306/siliqua_recognition?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8&autoReconnect=true&rewriteBatchedStatements=true&allowPublicKeyRetrieval=true |
|||
username: ams |
|||
password: ams@1234 |
|||
type: com.zaxxer.hikari.HikariDataSource |
|||
hikari: |
|||
maximum-pool-size: 10 |
|||
minimum-idle: 5 |
|||
idle-timeout: 30000 |
|||
connection-timeout: 30000 |
|||
max-lifetime: 1800000 |
|||
pool-name: siliqua-recognition |
|||
|
|||
jackson: |
|||
date-format: yyyy-MM-dd HH:mm:ss |
|||
time-zone: GMT+8 |
|||
|
|||
servlet: |
|||
multipart: |
|||
enabled: true |
|||
max-file-size: 20MB |
|||
max-request-size: 500MB |
|||
redis: |
|||
host: 192.168.10.111 |
|||
port: 6379 |
|||
database: 8 |
|||
mail: |
|||
host: smtpdm.aliyun.com |
|||
port: 465 |
|||
username: noreply@pushmail.jiagutech.com |
|||
password: Uv4eG2qUxFOmzX |
|||
protocol: smtp |
|||
properties: |
|||
mail: |
|||
smtp: |
|||
auth: true |
|||
ssl: |
|||
enable: true |
|||
email: |
|||
to: yeguangzhang@126.com |
|||
mybatis-plus: |
|||
mapper-locations: classpath*:mapper/*.xml |
|||
type-aliases-package: com.jiagutech.entity |
|||
configuration: |
|||
map-underscore-to-camel-case: true |
|||
log-impl: org.apache.ibatis.logging.stdout.StdOutImpl |
|||
|
|||
recognition: |
|||
model: |
|||
url: http://api.camacloud.org.cn/recognition |
|||
|
|||
sa-token: |
|||
# token名称 (同时也是cookie名称) |
|||
token-name: Authorization |
|||
token-prefix: Bearer |
|||
# 是否允许同一账号并发登录 (为true时允许一起登录, 为false时新登录挤掉旧登录) |
|||
is-concurrent: true |
|||
# 在多人登录同一账号时，是否共用一个token (为true时所有登录共用一个token, 为false时每次登录新建一个token) |
|||
is-share: false |
|||
# jwt秘钥 |
|||
jwt-secret-key: abcdefghijklmnopqrstuvwxyz |
|||
is-print: false |
|||
is-log: false |
|||
|
|||
huawei: |
|||
obs: |
|||
end-point: "obs.cn-east-2.myhuaweicloud.com" |
|||
ak: "NMUAWG3TN50OSJESCIRV" |
|||
sk: "qG0VFwhiTAtiCuUaMAxJL3zsusp3W4GHs7THR9pC" |
|||
bucket-name: iot |
|||
path-prefix: recognition |
|||
|
|||
|
|||
thread: |
|||
enabled: true |
|||
pool: |
|||
core-pool-size: 5 |
|||
max-pool-size: 8 |
|||
queue-capacity: 500 |
|||
thread-name: siliqua-recognition |
|||
task-threshold: 300000 |
|||
queue-size-threshold: 200 |
|||
queue-time-threshold: 1000 |
|||
|
|||
http: |
|||
template: |
|||
# 连接超时时间 |
|||
connect-timeout: 3000 |
|||
# 读取超时时间 |
|||
read-timeout: 20000 |
|||
logging: |
|||
level: |
|||
com.jiagutech.mapper: debug |
|||
root: info |
|||
@ -0,0 +1,23 @@ |
|||
<?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.jiagutech.mapper.RecordContentMapper"> |
|||
<resultMap id="RecordContentMap" type="com.jiagutech.entity.RecordContentEntity"> |
|||
<id column="id" property="id"/> |
|||
<result column="record_id" property="recordId"/> |
|||
<result column="image_name" property="imageName"/> |
|||
<result column="image_url" property="imageUrl"/> |
|||
<result column="recognition_data" property="recognitionData" |
|||
typeHandler="com.jiagutech.handler.ModelPredictResponseListTypeHandler" |
|||
javaType="java.util.List"/> |
|||
<result column="siliqua_num" property="siliquaNum"/> |
|||
<result column="handle_num" property="handleNum"/> |
|||
<result column="beak_num" property="beakNum"/> |
|||
|
|||
</resultMap> |
|||
<select id="selectListByRecordId" resultMap="RecordContentMap"> |
|||
select * |
|||
from record_content |
|||
where record_id = #{recordId} |
|||
order by id asc |
|||
</select> |
|||
</mapper> |
|||
@ -0,0 +1,30 @@ |
|||
<?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.jiagutech.mapper.RecordMapper"> |
|||
<resultMap id="RecordDetailMap" type="com.jiagutech.dto.RecordDetail"> |
|||
<id property="id" column="id"/> |
|||
<collection property="content" javaType="java.util.List" column="id" |
|||
select="com.jiagutech.mapper.RecordContentMapper.selectListByRecordId"/> |
|||
|
|||
</resultMap> |
|||
|
|||
<select id="getRecordDetail" resultMap="RecordDetailMap"> |
|||
select * |
|||
from record_info |
|||
where id = #{id} |
|||
</select> |
|||
<select id="pageList" resultType="com.jiagutech.dto.RecordItem"> |
|||
select r.id as recordId, |
|||
r.name as recordName, |
|||
r.user_id, |
|||
r.image_num as imageNum, |
|||
r.create_time as createTime, |
|||
u.real_name as userName, |
|||
u.unit_name as unitName, |
|||
u.phone |
|||
from record_info r |
|||
left join siliqua_user u on r.user_id = u.id |
|||
order by r.create_time desc |
|||
|
|||
</select> |
|||
</mapper> |
|||
@ -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.jiagutech.mapper.UserMapper"> |
|||
|
|||
|
|||
<select id="selectLoginUserByPhone" resultType="com.jiagutech.dto.LoginUser"> |
|||
select id as user_id, |
|||
real_name, |
|||
phone, |
|||
create_time, |
|||
password, |
|||
unit_name, |
|||
status |
|||
from siliqua_user |
|||
where phone = #{phone} |
|||
and status != 0 |
|||
</select> |
|||
|
|||
|
|||
</mapper> |
|||
Loading…
Reference in new issue