1.添加 生成数字人销售话术接口

2. 调整认证过滤器代码
master
gitee 1 month ago
parent 2c22b4304a
commit 1961adb6ae

@ -0,0 +1,30 @@
package com.supervision.livedigitalavatarmanage.constant;
/**
*
*/
public class PromptTemplate {
/**
*
*/
public static final String GENERATE_SALESPITCH_TEMPLATE = """
1.
2. 500
3.
4. ********
5. 10****
6. jsonArray["话术1","话术2"]/no_think
""";
private String aa_back = """
广
112X43cm100 216X54cm120
便?FAS* FAS)
·VDESIGNCONCEPTI/no_think
""";
}

@ -3,13 +3,10 @@ package com.supervision.livedigitalavatarmanage.controller;
import cn.hutool.core.lang.Assert;
import com.supervision.livedigitalavatarmanage.domain.Users;
import com.supervision.livedigitalavatarmanage.dto.R;
import com.supervision.livedigitalavatarmanage.dto.UserDetail;
import com.supervision.livedigitalavatarmanage.service.AuthService;
import com.supervision.livedigitalavatarmanage.util.UserUtil;
import com.supervision.livedigitalavatarmanage.vo.LoginReqVo;
import lombok.RequiredArgsConstructor;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.web.bind.annotation.*;
@RestController
@ -31,17 +28,6 @@ public class AuthController {
return R.ok(token);
}
/**
*
* @return
*/
@GetMapping("/heartbeat")
public R<Void> heartbeat() {
UserDetail userDetail = UserUtil.currentUser();
authService.heartbeat(userDetail.getUserId());
return R.ok();
}
@GetMapping("/me")
public R<Users> getCurrentUserDetails() {

@ -0,0 +1,31 @@
package com.supervision.livedigitalavatarmanage.controller;
import com.supervision.livedigitalavatarmanage.dto.R;
import com.supervision.livedigitalavatarmanage.service.LiveDigitalService;
import com.supervision.livedigitalavatarmanage.vo.SalesPitchReqVo;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;
@RestController
@RequestMapping("/liveDigital")
@RequiredArgsConstructor
public class LiveDigitalController {
private final LiveDigitalService liveDigitalService;
/**
*
* @param salespitchReqVo
* @return
*/
@PostMapping("/generate/salespitch")
public R<List<String>> generateSalesPitch(@RequestBody SalesPitchReqVo salespitchReqVo) {
List<String> salesPitchResVos = liveDigitalService.generateSalesPitch(salespitchReqVo);
return R.ok(salesPitchResVos);
}
}

@ -10,6 +10,7 @@ import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import java.io.IOException;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.security.core.context.SecurityContextHolder;
@ -20,6 +21,7 @@ import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.web.authentication.WebAuthenticationDetailsSource;
@Slf4j
@Component
@RequiredArgsConstructor
public class JwtAuthenticationFilter extends OncePerRequestFilter {
@ -41,6 +43,7 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
String authHeader = request.getHeader("Authorization");
//2根本没有 Authorization 头
if (authHeader == null || !authHeader.startsWith("Bearer ")) {
log.info("用户未携带token信息请求路径:{}", request.getRequestURI());
if (permitAllRequest) {
filterChain.doFilter(request, response);
return;
@ -55,6 +58,7 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
try {
username = jwtUtils.getUsernameFromToken(token);
} catch (Exception e) {
log.error("token认证失败请求路径:{},permitRequest is {}", request.getRequestURI(),permitAllRequest,e);
if (permitAllRequest) {
filterChain.doFilter(request, response);
return;
@ -77,13 +81,14 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
new UsernamePasswordAuthenticationToken(userDetails, null, userDetails.getAuthorities());
authToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
SecurityContextHolder.getContext().setAuthentication(authToken);
log.info("用户 {} token认证成功请求路径:{}", userDetails.getUsername(), request.getRequestURI());
//成功后放行,进入 Controller
filterChain.doFilter(request, response);
return;
} else {
//Token 过期 或 登录态不一致
log.info("用户 {} token认证失败,请求路径:{},permitRequest is {}", userDetails.getUsername(), request.getRequestURI(),permitAllRequest);
if (permitAllRequest) {
filterChain.doFilter(request, response);
return;
@ -93,6 +98,7 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
}
} catch (UsernameNotFoundException e) {
log.error("token认证失败请求路径:{},permitRequest is {}", request.getRequestURI(),permitAllRequest,e);
if (permitAllRequest) {
filterChain.doFilter(request, response);
return;
@ -100,6 +106,7 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
writeTokenErrorResponse(response, "用户不存在");
return;
} catch (Exception e) {
log.error("token认证失败请求路径:{},permitRequest is {}", request.getRequestURI(),permitAllRequest,e);
if (permitAllRequest) {
filterChain.doFilter(request, response);
return;

@ -0,0 +1,10 @@
package com.supervision.livedigitalavatarmanage.service;
import com.supervision.livedigitalavatarmanage.vo.SalesPitchReqVo;
import java.util.List;
public interface LiveDigitalService {
List<String> generateSalesPitch(SalesPitchReqVo salespitchReqVo);
}

@ -3,7 +3,6 @@ package com.supervision.livedigitalavatarmanage.service.impl;
import com.supervision.livedigitalavatarmanage.domain.Users;
import com.supervision.livedigitalavatarmanage.dto.UserDetail;
import com.supervision.livedigitalavatarmanage.enums.UserDisabledEnum;
import com.supervision.livedigitalavatarmanage.exception.BusinessException;
import com.supervision.livedigitalavatarmanage.service.AuthService;
import com.supervision.livedigitalavatarmanage.service.UsersService;
import com.supervision.livedigitalavatarmanage.util.JwtUtils;
@ -13,9 +12,7 @@ import org.springframework.security.authentication.BadCredentialsException;
import org.springframework.security.authentication.DisabledException;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Service;
import java.time.LocalDateTime;
@Service
@ -28,9 +25,6 @@ public class AuthServiceImpl implements AuthService {
private final UsersService userService;
// 掉线时间,单位:秒
private long offlineTime = 90; // 默认90秒
/**
* JWT Token
*/
@ -43,10 +37,6 @@ public class AuthServiceImpl implements AuthService {
if (UserDisabledEnum.DISABLED.getCode().equals(user.getDisabled())) {
throw new RuntimeException("用户已被禁用");
}
LocalDateTime lastActive = user.getLastActive();
if (null != lastActive && lastActive.plusSeconds(offlineTime).isAfter(LocalDateTime.now())){
throw new BusinessException("一个账号同时只能登录一个设备,请先下线其他设备");
}
try {
Authentication authentication = authenticationManager.authenticate(
new UsernamePasswordAuthenticationToken(username, password)

@ -0,0 +1,51 @@
package com.supervision.livedigitalavatarmanage.service.impl;
import cn.hutool.core.lang.Assert;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.supervision.livedigitalavatarmanage.constant.PromptTemplate;
import com.supervision.livedigitalavatarmanage.service.LiveDigitalService;
import com.supervision.livedigitalavatarmanage.vo.SalesPitchReqVo;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.chat.messages.SystemMessage;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.ai.chat.model.ChatResponse;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.ollama.OllamaChatModel;
import org.springframework.stereotype.Service;
import java.util.List;
@Slf4j
@Service
@RequiredArgsConstructor
public class LiveDigitalServiceImpl implements LiveDigitalService {
private final OllamaChatModel ollamaChatModel;
@Override
public List<String> generateSalesPitch(SalesPitchReqVo salespitchReqVo) {
Assert.notEmpty(salespitchReqVo.getProductName(), "产品名称不能为空");
Assert.notEmpty(salespitchReqVo.getSpecifications(), "产品规格不能为空");
Assert.notEmpty(salespitchReqVo.getDetail(), "产品详情不能为空");
for (int i = 0; i < 3; i++) {
try {
String salesPitchTemplate = PromptTemplate.GENERATE_SALESPITCH_TEMPLATE;
SystemMessage systemMessage = new SystemMessage(salesPitchTemplate);
UserMessage userMessage = new UserMessage(salespitchReqVo.buildTemplate());
Prompt prompt = new Prompt(systemMessage, userMessage);
ChatResponse call = ollamaChatModel.call(prompt);
String text = call.getResult().getOutput().getText();
// 去除think标签
if (StrUtil.isNotBlank(text) && text.contains("<think>") && text.contains("</think>")) {
text = text.replaceAll("(?s)<think>.*?</think>", "").trim();
}
return JSONUtil.toList(text, String.class);
}catch (Exception e) {
log.error("生成销售话术失败,尝试第 {} 次。请求内容:{}", i + 1, salespitchReqVo.buildTemplate(),e);
}
}
throw new RuntimeException("生成销售话术失败,请稍后重试");
}
}

@ -0,0 +1,17 @@
package com.supervision.livedigitalavatarmanage.vo;
import lombok.Data;
@Data
public class ProductSpecification {
/**
*
*/
private String size;
/**
*
*/
private String price;
}

@ -0,0 +1,44 @@
package com.supervision.livedigitalavatarmanage.vo;
import lombok.Data;
import java.util.List;
/**
* @author Administrator
* @description
* @createDate 2025-07-28 14:11:27
*/
@Data
public class SalesPitchReqVo {
/**
*
*/
private String productName;
/**
*
*/
private List<ProductSpecification> specifications;
/**
*
*/
private String detail;
public String buildTemplate() {
StringBuilder template = new StringBuilder();
template.append("请根据产品信息生成销售话术:");
template.append("\n");
template.append("产品名称:").append(productName).append("\n");
template.append("产品规格:");
for (ProductSpecification specification : specifications) {
template.append("产品类型:").append(specification.getSize()).append("\n");
template.append("产品价格:").append(specification.getPrice()).append("\n");
}
template.append("产品详情:").append(detail).append("/no_think");
return template.toString();
}
}

@ -0,0 +1,160 @@
<?xml version="1.0" encoding="UTF-8"?>
<configuration scan="true" scanPeriod="10 seconds">
<contextName>logback</contextName>
<!-- name的值是变量的名称value的值时变量定义的值。通过定义的值会被插入到logger上下文中。定义后可以使“${}”来使用变量。 -->
<property name="log.path" value="./log" />
<!--0. 日志格式和颜色渲染 -->
<!-- 彩色日志依赖的渲染类 -->
<conversionRule conversionWord="clr" converterClass="org.springframework.boot.logging.logback.ColorConverter" />
<conversionRule conversionWord="wex" converterClass="org.springframework.boot.logging.logback.WhitespaceThrowableProxyConverter" />
<conversionRule conversionWord="wEx" converterClass="org.springframework.boot.logging.logback.ExtendedWhitespaceThrowableProxyConverter" />
<!-- 彩色日志格式 -->
<property name="CONSOLE_LOG_PATTERN" value="${CONSOLE_LOG_PATTERN:-%clr(%d{yyyy-MM-dd HH:mm:ss.SSS}){faint} %clr(${LOG_LEVEL_PATTERN:-%5p}) %clr(${PID:- }){magenta} %clr(---){faint} %clr([%15.15t]){faint} %clr(%-40.40logger{39}){cyan} %clr(:){faint} %m%n${LOG_EXCEPTION_CONVERSION_WORD:-%wEx}}"/>
<!--1. 输出到控制台-->
<appender name="CONSOLE" class="ch.qos.logback.core.ConsoleAppender">
<!--此日志appender是为开发使用只配置最底级别控制台输出的日志级别是大于或等于此级别的日志信息-->
<filter class="ch.qos.logback.classic.filter.ThresholdFilter">
<level>debug</level>
</filter>
<encoder>
<Pattern>%d{yyyy-MM-dd HH:mm:ss.SSS}---[%thread] %-5level %logger{50} - %msg%n</Pattern>
<!-- 设置字符集 -->
<charset>UTF-8</charset>
</encoder>
</appender>
<!--2. 输出到文档-->
<!-- 2.1 level为 DEBUG 日志,时间滚动输出 -->
<appender name="DEBUG_FILE" class="ch.qos.logback.core.rolling.RollingFileAppender">
<!-- 正在记录的日志文档的路径及文档名 -->
<file>${log.path}/web_debug.log</file>
<!--日志文档输出格式-->
<encoder>
<pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n</pattern>
<charset>UTF-8</charset> <!-- 设置字符集 -->
</encoder>
<!-- 日志记录器的滚动策略,按日期,按大小记录 -->
<rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy">
<!-- 日志归档 -->
<fileNamePattern>${log.path}/web-debug-%d{yyyy-MM-dd}.%i.log</fileNamePattern>
<timeBasedFileNamingAndTriggeringPolicy class="ch.qos.logback.core.rolling.SizeAndTimeBasedFNATP">
<maxFileSize>100MB</maxFileSize>
</timeBasedFileNamingAndTriggeringPolicy>
<!--日志文档保留天数-->
<maxHistory>30</maxHistory>
<totalSizeCap>500MB</totalSizeCap>
</rollingPolicy>
<!-- 此日志文档只记录debug级别的 -->
<filter class="ch.qos.logback.classic.filter.LevelFilter">
<level>debug</level>
<onMatch>ACCEPT</onMatch>
<onMismatch>DENY</onMismatch>
</filter>
</appender>
<!-- 2.2 level为 INFO 日志,时间滚动输出 -->
<appender name="INFO_FILE" class="ch.qos.logback.core.rolling.RollingFileAppender">
<!-- 正在记录的日志文档的路径及文档名 -->
<file>${log.path}/web_info.log</file>
<!--日志文档输出格式-->
<encoder>
<pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n</pattern>
<charset>UTF-8</charset>
</encoder>
<!-- 日志记录器的滚动策略,按日期,按大小记录 -->
<rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy">
<!-- 每天日志归档路径以及格式 -->
<fileNamePattern>${log.path}/web-info-%d{yyyy-MM-dd}.%i.log</fileNamePattern>
<timeBasedFileNamingAndTriggeringPolicy class="ch.qos.logback.core.rolling.SizeAndTimeBasedFNATP">
<maxFileSize>100MB</maxFileSize>
</timeBasedFileNamingAndTriggeringPolicy>
<!--日志文档保留天数-->
<maxHistory>30</maxHistory>
<totalSizeCap>1GB</totalSizeCap>
</rollingPolicy>
<!-- 此日志文档只记录info级别的 -->
<filter class="ch.qos.logback.classic.filter.LevelFilter">
<level>info</level>
<onMatch>ACCEPT</onMatch>
<onMismatch>DENY</onMismatch>
</filter>
</appender>
<!-- 2.3 level为 WARN 日志,时间滚动输出 -->
<appender name="WARN_FILE" class="ch.qos.logback.core.rolling.RollingFileAppender">
<!-- 正在记录的日志文档的路径及文档名 -->
<file>${log.path}/web_warn.log</file>
<!--日志文档输出格式-->
<encoder>
<pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n</pattern>
<charset>UTF-8</charset> <!-- 此处设置字符集 -->
</encoder>
<!-- 日志记录器的滚动策略,按日期,按大小记录 -->
<rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy">
<fileNamePattern>${log.path}/web-warn-%d{yyyy-MM-dd}.%i.log</fileNamePattern>
<timeBasedFileNamingAndTriggeringPolicy class="ch.qos.logback.core.rolling.SizeAndTimeBasedFNATP">
<maxFileSize>100MB</maxFileSize>
</timeBasedFileNamingAndTriggeringPolicy>
<!--日志文档保留天数-->
<maxHistory>30</maxHistory>
<totalSizeCap>500MB</totalSizeCap>
</rollingPolicy>
<!-- 此日志文档只记录warn级别的 -->
<filter class="ch.qos.logback.classic.filter.LevelFilter">
<level>warn</level>
<onMatch>ACCEPT</onMatch>
<onMismatch>DENY</onMismatch>
</filter>
</appender>
<!-- 2.4 level为 ERROR 日志,时间滚动输出 -->
<appender name="ERROR_FILE" class="ch.qos.logback.core.rolling.RollingFileAppender">
<!-- 正在记录的日志文档的路径及文档名 -->
<file>${log.path}/web_error.log</file>
<!--日志文档输出格式-->
<encoder>
<pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n</pattern>
<charset>UTF-8</charset> <!-- 此处设置字符集 -->
</encoder>
<!-- 日志记录器的滚动策略,按日期,按大小记录 -->
<rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy">
<fileNamePattern>${log.path}/web-error-%d{yyyy-MM-dd}.%i.log</fileNamePattern>
<timeBasedFileNamingAndTriggeringPolicy class="ch.qos.logback.core.rolling.SizeAndTimeBasedFNATP">
<maxFileSize>100MB</maxFileSize>
</timeBasedFileNamingAndTriggeringPolicy>
<!--日志文档保留天数-->
<maxHistory>30</maxHistory>
<totalSizeCap>500MB</totalSizeCap>
</rollingPolicy>
<!-- 此日志文档只记录ERROR级别的 -->
<filter class="ch.qos.logback.classic.filter.LevelFilter">
<level>ERROR</level>
<onMatch>ACCEPT</onMatch>
<onMismatch>DENY</onMismatch>
</filter>
</appender>
<root level="info">
<appender-ref ref="CONSOLE" />
<appender-ref ref="DEBUG_FILE" />
<appender-ref ref="INFO_FILE" />
<appender-ref ref="WARN_FILE" />
<appender-ref ref="ERROR_FILE" />
</root>
<!-- 4.2 生产环境:输出到文档
<springProfile name="pro">
<root level="info">
<appender-ref ref="CONSOLE" />
<appender-ref ref="DEBUG_FILE" />
<appender-ref ref="INFO_FILE" />
<appender-ref ref="ERROR_FILE" />
<appender-ref ref="WARN_FILE" />
</root>
</springProfile> -->
</configuration>
Loading…
Cancel
Save