增加与ai大模型对话

huimv-employment/fe-api/src/main/java/com/huimv/employment/controller/mp/ConversationController.java

@@ -1,14 +1,20 @@
 package com.huimv.employment.controller.mp;
 
+import com.huimv.employment.common.exception.BizException;
+import com.huimv.employment.common.exception.ErrorCode;
 import com.huimv.employment.common.web.R;
 import com.huimv.employment.security.LoginUser;
 import com.huimv.employment.security.LoginUserHolder;
 import com.huimv.employment.service.conversation.ConversationService;
+import com.huimv.employment.service.conversation.dto.ConversationCreateRequest;
 import com.huimv.employment.service.conversation.dto.ConversationSummaryResponse;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.security.SecurityRequirement;
 import io.swagger.v3.oas.annotations.tags.Tag;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.GetMapping;
+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.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
@@ -22,16 +28,18 @@ import java.util.List;
  * 从 JWT 解析 {@code userId}，仅返回当前用户自己的会话，不支持跨用户查询。
  * </p>
  * <p>
- * 典型用法：对话中心侧边栏加载历史会话列表；点击某条后带 {@code conversationNo}（即 AI {@code session_id}）
- * 继续多轮对话。
+ * 典型用法：点击「新对话」调用 {@code POST /conversations} 获取 {@code conversationNo}，
+ * 再作为 {@code session_id} 调用 {@code POST /ai/console/chat} 进行多轮对话。
  * </p>
  */
 @RestController
 @RequestMapping("/api/v1/mp/conversations")
-@Tag(name = "AI 会话", description = "对话会话列表，需 JWT 鉴权")
+@Tag(name = "AI 会话", description = "对话会话管理，需 JWT 鉴权")
 @SecurityRequirement(name = "Authorization")
 public class ConversationController {
 
+    private static final String USER_TYPE_ENTERPRISE = "enterprise";
+
     private final ConversationService conversationService;
 
     public ConversationController(ConversationService conversationService) {
@@ -40,15 +48,10 @@ public class ConversationController {
 
     /**
      * 查询当前登录用户的全部 AI 对话会话。
-     *
-     * @param status       可选，{@code active} / {@code archived}
-     * @param chatType     可选，如 {@code employment}
-     * @param enterpriseId 可选，多企业时按企业筛选
      */
     @GetMapping
     @Operation(summary = "查询当前用户的 AI 对话会话列表",
-            description = "按 last_message_at 倒序返回。可选按 status、chatType、enterpriseId 筛选。"
-                    + "数据来源 fe_conversation，需先有会话落库记录。")
+            description = "按 last_message_at 倒序返回。可选按 status、chatType、enterpriseId 筛选。")
     public R<List<ConversationSummaryResponse>> list(
             @RequestParam(required = false) String status,
             @RequestParam(required = false) String chatType,
@@ -57,4 +60,25 @@ public class ConversationController {
         return R.ok(conversationService.listByUser(
                 loginUser.getUserId(), status, chatType, enterpriseId));
     }
+
+    /**
+     * 新建 AI 对话会话。
+     * <p>返回的 {@code conversationNo} 即后续 AI 对话的 {@code session_id}。</p>
+     */
+    @PostMapping
+    @Operation(summary = "新建 AI 对话会话",
+            description = "企业主用户创建新会话。conversationNo 不传时服务端自动生成；"
+                    + "chatType 默认 employment。需已完成企业登记。")
+    public R<ConversationSummaryResponse> create(@Validated @RequestBody(required = false) ConversationCreateRequest request) {
+        LoginUser loginUser = requireEnterprise();
+        return R.ok(conversationService.create(loginUser.getUserId(), request));
+    }
+
+    private static LoginUser requireEnterprise() {
+        LoginUser loginUser = LoginUserHolder.require();
+        if (!USER_TYPE_ENTERPRISE.equals(loginUser.getUserType())) {
+            throw new BizException(ErrorCode.ENTERPRISE_ONLY);
+        }
+        return loginUser;
+    }
 }

huimv-employment/fe-service/src/main/java/com/huimv/employment/service/conversation/ConversationService.java

@@ -1,9 +1,12 @@
 package com.huimv.employment.service.conversation;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.huimv.employment.common.exception.BizException;
+import com.huimv.employment.common.exception.ErrorCode;
 import com.huimv.employment.dao.entity.FeConversation;
 import com.huimv.employment.dao.entity.FeEnterprise;
 import com.huimv.employment.dao.mapper.FeConversationMapper;
+import com.huimv.employment.service.conversation.dto.ConversationCreateRequest;
 import com.huimv.employment.service.conversation.dto.ConversationSummaryResponse;
 import com.huimv.employment.service.enterprise.EnterpriseService;
 import org.slf4j.Logger;
@@ -14,15 +17,18 @@ import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.StringUtils;
 
 import java.time.LocalDateTime;
+import java.util.Arrays;
 import java.util.Collections;
+import java.util.HashSet;
 import java.util.List;
+import java.util.Set;
+import java.util.UUID;
 import java.util.stream.Collectors;
 
 /**
  * AI 对话会话业务层。
  * <p>
- * 一期提供按登录用户查询会话列表；{@link #touchFromChat(Long, String)} 供后续在对话入口落库时调用
- *（当前未接入 {@code AiProxyController}）。
+ * 一期提供会话列表查询与手动新建；{@link #touchFromChat(Long, String)} 供后续在对话入口自动刷新时调用。
  * </p>
  */
 @Service
@@ -33,6 +39,10 @@ public class ConversationService {
     private static final String STATUS_ACTIVE = "active";
     private static final String CHAT_TYPE_EMPLOYMENT = "employment";
 
+    private static final Set<String> ALLOWED_CHAT_TYPES = Collections.unmodifiableSet(new HashSet<>(Arrays.asList(
+            "employment", "knowledge", "order", "cost", "attendance", "emergency"
+    )));
+
     /** 与表字段 {@code conversation_no} VARCHAR(32) 一致 */
     private static final int CONVERSATION_NO_MAX_LEN = 32;
 
@@ -70,6 +80,41 @@ public class ConversationService {
                 .collect(Collectors.toList());
     }
 
+    /**
+     * 新建 AI 对话会话。
+     * <p>
+     * {@code conversationNo} 不传时自动生成 32 位 hex，作为后续 {@code /ai/console/chat} 的 {@code session_id}。
+     * 要求用户已完成企业登记。
+     * </p>
+     */
+    @Transactional(rollbackFor = Exception.class)
+    public ConversationSummaryResponse create(Long userId, ConversationCreateRequest request) {
+        FeEnterprise enterprise = resolveEnterprise(userId, request != null ? request.getEnterpriseId() : null);
+        String chatType = resolveChatType(request != null ? request.getChatType() : null);
+        String conversationNo = resolveNewConversationNo(request != null ? request.getConversationNo() : null);
+        assertConversationNoAvailable(conversationNo);
+
+        LocalDateTime now = LocalDateTime.now();
+        FeConversation created = new FeConversation();
+        created.setConversationNo(conversationNo);
+        created.setEnterpriseId(enterprise.getId());
+        created.setUserId(userId);
+        if (request != null && StringUtils.hasText(request.getTitle())) {
+            created.setTitle(request.getTitle().trim());
+        }
+        created.setChatType(chatType);
+        created.setStatus(STATUS_ACTIVE);
+        created.setLastMessageAt(now);
+        created.setCreateTime(now);
+        created.setUpdateTime(now);
+        try {
+            feConversationMapper.insert(created);
+        } catch (DuplicateKeyException ex) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "会话编号已存在，请更换后重试");
+        }
+        return toSummary(created);
+    }
+
     /**
      * 用户发起 AI 对话时刷新会话记录。
      * <p>
@@ -104,11 +149,58 @@ public class ConversationService {
             log.debug("用户 {} 未绑定企业，跳过会话落库 sessionId={}", userId, conversationNo);
             return;
         }
+        insertConversation(userId, enterprise.getId(), conversationNo, null, CHAT_TYPE_EMPLOYMENT, now);
+    }
+
+    private FeEnterprise resolveEnterprise(Long userId, Long enterpriseId) {
+        FeEnterprise bound = enterpriseService.findEnterpriseByUserId(userId);
+        if (bound == null) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "请先完成企业登记");
+        }
+        if (enterpriseId != null && !enterpriseId.equals(bound.getId())) {
+            throw new BizException(ErrorCode.FORBIDDEN, "无权在该企业下创建会话");
+        }
+        return bound;
+    }
+
+    private String resolveChatType(String chatType) {
+        if (!StringUtils.hasText(chatType)) {
+            return CHAT_TYPE_EMPLOYMENT;
+        }
+        String normalized = chatType.trim();
+        if (!ALLOWED_CHAT_TYPES.contains(normalized)) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "不支持的对话类型：" + normalized);
+        }
+        return normalized;
+    }
+
+    private String resolveNewConversationNo(String conversationNo) {
+        if (!StringUtils.hasText(conversationNo)) {
+            return UUID.randomUUID().toString().replace("-", "");
+        }
+        return normalizeConversationNo(conversationNo.trim());
+    }
+
+    private void assertConversationNoAvailable(String conversationNo) {
+        Long count = feConversationMapper.selectCount(new LambdaQueryWrapper<FeConversation>()
+                .eq(FeConversation::getConversationNo, conversationNo));
+        if (count != null && count > 0) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "会话编号已存在，请更换后重试");
+        }
+    }
+
+    private void insertConversation(Long userId,
+                                    Long enterpriseId,
+                                    String conversationNo,
+                                    String title,
+                                    String chatType,
+                                    LocalDateTime now) {
         FeConversation created = new FeConversation();
         created.setConversationNo(conversationNo);
-        created.setEnterpriseId(enterprise.getId());
+        created.setEnterpriseId(enterpriseId);
         created.setUserId(userId);
-        created.setChatType(CHAT_TYPE_EMPLOYMENT);
+        created.setTitle(title);
+        created.setChatType(chatType);
         created.setStatus(STATUS_ACTIVE);
         created.setLastMessageAt(now);
         created.setCreateTime(now);
@@ -116,7 +208,6 @@ public class ConversationService {
         try {
             feConversationMapper.insert(created);
         } catch (DuplicateKeyException ex) {
-            // conversation_no 全局唯一，并发创建时回退为更新
             FeConversation retry = feConversationMapper.selectOne(new LambdaQueryWrapper<FeConversation>()
                     .eq(FeConversation::getConversationNo, conversationNo)
                     .eq(FeConversation::getUserId, userId)
@@ -127,7 +218,9 @@ public class ConversationService {
                 update.setLastMessageAt(now);
                 update.setUpdateTime(now);
                 feConversationMapper.updateById(update);
+                return;
             }
+            throw new BizException(ErrorCode.BAD_REQUEST, "会话编号已存在，请更换后重试");
         }
     }
 

huimv-employment/fe-service/src/main/java/com/huimv/employment/service/conversation/dto/ConversationCreateRequest.java

@@ -0,0 +1,60 @@
+package com.huimv.employment.service.conversation.dto;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+
+import javax.validation.constraints.Size;
+
+/**
+ * 新建 AI 对话会话请求。
+ */
+@Schema(description = "新建 AI 对话会话请求")
+public class ConversationCreateRequest {
+
+    @Schema(description = "会话标题（可选，首条消息后可再更新）", example = "新对话")
+    @Size(max = 200, message = "标题不能超过 200 字")
+    private String title;
+
+    @Schema(description = "对话类型，默认 employment", example = "employment",
+            allowableValues = {"employment", "knowledge", "order", "cost", "attendance", "emergency"})
+    private String chatType;
+
+    @Schema(description = "自定义会话编号（可选，须全局唯一；不传则服务端生成，用作 AI session_id）",
+            example = "a1b2c3d4e5f6789012345678901234")
+    @Size(max = 32, message = "会话编号不能超过 32 字符")
+    private String conversationNo;
+
+    @Schema(description = "企业 ID（可选，默认当前用户绑定的企业）", example = "10")
+    private Long enterpriseId;
+
+    public String getTitle() {
+        return title;
+    }
+
+    public void setTitle(String title) {
+        this.title = title;
+    }
+
+    public String getChatType() {
+        return chatType;
+    }
+
+    public void setChatType(String chatType) {
+        this.chatType = chatType;
+    }
+
+    public String getConversationNo() {
+        return conversationNo;
+    }
+
+    public void setConversationNo(String conversationNo) {
+        this.conversationNo = conversationNo;
+    }
+
+    public Long getEnterpriseId() {
+        return enterpriseId;
+    }
+
+    public void setEnterpriseId(Long enterpriseId) {
+        this.enterpriseId = enterpriseId;
+    }
+}