241 lines
9.2 KiB
Java
241 lines
9.2 KiB
Java
package com.emotion.controller;
|
|
|
|
import com.emotion.common.Result;
|
|
import com.emotion.dto.request.AiChatRequest;
|
|
import com.emotion.dto.request.AiSummaryRequest;
|
|
import com.emotion.dto.request.ChatStatsRequest;
|
|
import com.emotion.dto.request.GuestChatRequest;
|
|
import com.emotion.dto.request.ConversationCreateRequest;
|
|
import com.emotion.dto.response.AiChatResponse;
|
|
import com.emotion.dto.response.AiSummaryResponse;
|
|
import com.emotion.dto.response.AiStatusResponse;
|
|
import com.emotion.dto.response.ChatStatsResponse;
|
|
import com.emotion.dto.response.GuestChatResponse;
|
|
import com.emotion.dto.response.GuestUserInfoResponse;
|
|
import com.emotion.dto.response.ConversationResponse;
|
|
import com.emotion.entity.Conversation;
|
|
import com.emotion.service.AIChatService;
|
|
import com.emotion.service.MessageService;
|
|
import com.emotion.service.ConversationService;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.beans.BeanUtils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.validation.Valid;
|
|
import java.time.format.DateTimeFormatter;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* AI聊天控制器
|
|
*
|
|
* @author emotion-museum
|
|
* @date 2025-07-23
|
|
*/
|
|
@Slf4j
|
|
@RestController
|
|
@RequestMapping("/api/ai")
|
|
public class AiChatController {
|
|
|
|
@Autowired
|
|
private AIChatService aiChatService;
|
|
|
|
@Autowired
|
|
private MessageService messageService;
|
|
|
|
@Autowired
|
|
private ConversationService conversationService;
|
|
|
|
private static final DateTimeFormatter DATE_TIME_FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
|
|
|
|
/**
|
|
* 发送聊天消息
|
|
*/
|
|
@PostMapping("/chat")
|
|
public Result<AiChatResponse> sendChatMessage(@Valid @RequestBody AiChatRequest request) {
|
|
log.info("收到AI聊天请求: conversationId={}, userId={}, message={}",
|
|
request.getConversationId(), request.getUserId(), request.getMessage());
|
|
|
|
// 调用AI服务
|
|
String aiReply = aiChatService.sendChatMessage(request.getConversationId(), request.getMessage(),
|
|
request.getUserId());
|
|
|
|
// 构建响应
|
|
AiChatResponse response = new AiChatResponse();
|
|
response.setConversationId(request.getConversationId());
|
|
response.setUserMessage(request.getMessage());
|
|
response.setAiReply(aiReply);
|
|
response.setUserId(request.getUserId());
|
|
response.setTimestamp(System.currentTimeMillis());
|
|
|
|
return Result.success(response);
|
|
}
|
|
|
|
/**
|
|
* 生成对话总结
|
|
*/
|
|
@PostMapping("/summary")
|
|
public Result<AiSummaryResponse> generateSummary(@Valid @RequestBody AiSummaryRequest request) {
|
|
log.info("收到对话总结请求: conversationId={}, userId={}", request.getConversationId(), request.getUserId());
|
|
|
|
// 调用AI总结服务
|
|
String summary = aiChatService.generateConversationSummary(request.getConversationId(),
|
|
request.getUserId());
|
|
|
|
// 构建响应
|
|
AiSummaryResponse response = new AiSummaryResponse();
|
|
response.setConversationId(request.getConversationId());
|
|
response.setSummary(summary);
|
|
response.setUserId(request.getUserId());
|
|
response.setTimestamp(System.currentTimeMillis());
|
|
|
|
return Result.success(response);
|
|
}
|
|
|
|
/**
|
|
* 获取AI服务状态
|
|
*/
|
|
@GetMapping("/status")
|
|
public Result<AiStatusResponse> getServiceStatus() {
|
|
log.info("获取AI服务状态");
|
|
|
|
boolean available = aiChatService.isServiceAvailable();
|
|
String status = aiChatService.getServiceStatus();
|
|
|
|
// 构建响应
|
|
AiStatusResponse response = new AiStatusResponse();
|
|
response.setAvailable(available);
|
|
response.setStatus(status);
|
|
response.setTimestamp(System.currentTimeMillis());
|
|
|
|
return Result.success(response);
|
|
}
|
|
|
|
/**
|
|
* 获取聊天记录统计
|
|
*/
|
|
@GetMapping("/stats")
|
|
public Result<ChatStatsResponse> getChatStats(@RequestParam(required = false) String userId,
|
|
@RequestParam(required = false) String conversationId) {
|
|
log.info("获取聊天统计: userId={}, conversationId={}", userId, conversationId);
|
|
|
|
// 构建响应
|
|
ChatStatsResponse response = new ChatStatsResponse();
|
|
|
|
if (userId != null && !userId.trim().isEmpty()) {
|
|
Long userConversationCount = conversationService.countByUserId(userId);
|
|
Long activeConversationCount = conversationService.countActiveByUserId(userId);
|
|
|
|
response.setUserConversationCount(userConversationCount);
|
|
response.setActiveConversationCount(activeConversationCount);
|
|
}
|
|
|
|
if (conversationId != null && !conversationId.trim().isEmpty()) {
|
|
Long conversationMessageCount = messageService.countByConversationId(conversationId);
|
|
response.setConversationMessageCount(conversationMessageCount);
|
|
}
|
|
|
|
response.setTimestamp(System.currentTimeMillis());
|
|
|
|
return Result.success(response);
|
|
}
|
|
|
|
/**
|
|
* 访客聊天（不需要登录）
|
|
*/
|
|
@PostMapping("/guest/chat")
|
|
public Result<GuestChatResponse> guestChat(@Valid @RequestBody GuestChatRequest request,
|
|
HttpServletRequest httpRequest) {
|
|
String clientIp = getClientIp(httpRequest);
|
|
log.info("访客聊天请求: {}, IP: {}", request.getMessage(), clientIp);
|
|
|
|
// 调用AI服务
|
|
Map<String, Object> aiResponse = aiChatService.guestChat(request.getMessage(), clientIp);
|
|
|
|
// 构建响应
|
|
GuestChatResponse response = new GuestChatResponse();
|
|
response.setMessage((String) aiResponse.get("message"));
|
|
response.setMessageId((String) aiResponse.get("messageId"));
|
|
response.setTimestamp((Long) aiResponse.get("timestamp"));
|
|
response.setError((Boolean) aiResponse.get("error"));
|
|
|
|
return Result.success("发送成功", response);
|
|
}
|
|
|
|
/**
|
|
* 获取访客用户信息
|
|
*/
|
|
@GetMapping("/guest/user/info")
|
|
public Result<GuestUserInfoResponse> getGuestUserInfo(HttpServletRequest request) {
|
|
String clientIp = getClientIp(request);
|
|
log.info("获取访客用户信息: IP={}", clientIp);
|
|
|
|
// 调用AI服务
|
|
Map<String, Object> userInfo = aiChatService.getGuestUserInfo(clientIp);
|
|
|
|
// 构建响应
|
|
GuestUserInfoResponse response = new GuestUserInfoResponse();
|
|
response.setId((String) userInfo.get("id"));
|
|
response.setUsername((String) userInfo.get("username"));
|
|
response.setNickname((String) userInfo.get("nickname"));
|
|
response.setType((String) userInfo.get("type"));
|
|
response.setClientIp((String) userInfo.get("clientIp"));
|
|
response.setUserCreateTime((Long) userInfo.get("createTime"));
|
|
|
|
return Result.success(response);
|
|
}
|
|
|
|
/**
|
|
* 创建对话
|
|
*/
|
|
@PostMapping("/conversation/create")
|
|
public Result<ConversationResponse> createConversation(@Valid @RequestBody ConversationCreateRequest request,
|
|
HttpServletRequest httpRequest) {
|
|
log.info("创建对话请求: userId={}, title={}", request.getUserId(), request.getTitle());
|
|
|
|
String clientIp = getClientIp(httpRequest);
|
|
|
|
// 调用AI服务创建对话
|
|
Map<String, Object> aiConversation = aiChatService.createConversation(request.getUserId(),
|
|
request.getTitle());
|
|
|
|
// 创建数据库对话记录
|
|
Conversation conversation = conversationService.createConversation(
|
|
request.getUserId(),
|
|
request.getTitle(),
|
|
request.getType() != null ? request.getType() : "user");
|
|
|
|
// 构建响应
|
|
ConversationResponse response = new ConversationResponse();
|
|
BeanUtils.copyProperties(conversation, response);
|
|
response.setId(conversation.getId());
|
|
if (conversation.getCreateTime() != null) {
|
|
response.setCreateTime(conversation.getCreateTime().format(DATE_TIME_FORMATTER));
|
|
}
|
|
if (conversation.getUpdateTime() != null) {
|
|
response.setUpdateTime(conversation.getUpdateTime().format(DATE_TIME_FORMATTER));
|
|
}
|
|
|
|
return Result.success("创建成功", response);
|
|
}
|
|
|
|
/**
|
|
* 获取客户端IP
|
|
*/
|
|
private String getClientIp(HttpServletRequest request) {
|
|
String xForwardedFor = request.getHeader("X-Forwarded-For");
|
|
if (xForwardedFor != null && !xForwardedFor.isEmpty() && !"unknown".equalsIgnoreCase(xForwardedFor)) {
|
|
return xForwardedFor.split(",")[0].trim();
|
|
}
|
|
|
|
String xRealIp = request.getHeader("X-Real-IP");
|
|
if (xRealIp != null && !xRealIp.isEmpty() && !"unknown".equalsIgnoreCase(xRealIp)) {
|
|
return xRealIp;
|
|
}
|
|
|
|
return request.getRemoteAddr();
|
|
}
|
|
}
|