TASK:1.request outfit and retrieve history session regenerate 2.customer check-in

src/main/java/com/aida/lanecarford/controller/CustomerController.java

@@ -2,6 +2,7 @@ package com.aida.lanecarford.controller;
 
 import com.aida.lanecarford.common.ApiResponse;
 import com.aida.lanecarford.dto.BaseRequest;
+import com.aida.lanecarford.entity.Customer;
 import com.aida.lanecarford.service.CustomerService;
 import com.aida.lanecarford.vo.CustomerCheckInVO;
 import com.aida.lanecarford.vo.CustomerVO;
@@ -25,11 +26,11 @@ public class CustomerController {
 
     @Operation(
         summary = "顾客入店登记", 
-        description = "验证顾客身份并创建入店记录，如果是新顾客则自动注册到系统中。"
+        description = "验证顾客身份并创建入店记录"
     )
     @GetMapping("/checkIn")
-    public ApiResponse<CustomerCheckInVO> customerCheckIn(@RequestParam String vipId) {
+    public ApiResponse<CustomerCheckInVO> customerCheckIn(@RequestParam String nickname) {
-        return ApiResponse.success(customerService.customerCheckIn(vipId));
+        return ApiResponse.success(customerService.customerCheckIn(nickname));
     }
 
     @PostMapping("/getAllCustomer")
@@ -37,4 +38,15 @@ public class CustomerController {
         return ApiResponse.success(customerService.getAllCustomer(request));
     }
 
+    @Operation(
+            summary = "新增顾客",
+            description = "根据用户提供的vipId和昵称与当前sales绑定，创建账号"
+    )
+    @GetMapping("/createCustomer")
+    public ApiResponse<Customer> createCustomer(@RequestParam String vipId, @RequestParam String nickname) {
+        return ApiResponse.success(customerService.createCustomer(vipId, nickname));
+    }
+
+
+
 }

src/main/java/com/aida/lanecarford/controller/StyleController.java

@@ -78,4 +78,15 @@ public class StyleController {
         return ApiResponse.success(styleService.getOutfitResult(requestIDs));
     }
 
+    @Operation(
+            summary = "回溯历史对话，重新生成搭配图",
+            description = "根据当前的穿搭结果，回溯历史穿搭请求数据及历史对话，重新生成搭配"
+    )
+    @GetMapping("/retrieveAndRegenerate")
+    public ApiResponse<List<String>> retrieveAndRegenerate(
+            @Parameter(description = "tryOn后的图片id", required = true, example = "1369")
+            @RequestParam Long tryOnEffectsId) {
+        return ApiResponse.success(styleService.retrieveAndRegenerate(tryOnEffectsId));
+    }
+
 }

src/main/java/com/aida/lanecarford/dto/OutfitCallbackDTO.java

@@ -16,4 +16,8 @@ public class OutfitCallbackDTO {
     private String path;
 
     private List<Map<String, String>> items;
+
+    private String request_summary;
+
+    private List<String> occasions;
 }

src/main/java/com/aida/lanecarford/dto/RequestOutfitDTO.java

@@ -1,9 +1,12 @@
 package com.aida.lanecarford.dto;
 
+import io.swagger.v3.oas.annotations.Hidden;
 import io.swagger.v3.oas.annotations.media.Schema;
 import jakarta.validation.constraints.NotNull;
 import lombok.Data;
 
+import java.util.List;
+
 
 @Data
 @Schema(description = "AI穿搭推荐请求参数")
@@ -59,4 +62,10 @@ public class RequestOutfitDTO {
     )
     private String sessionId;
 
+    @Hidden
+    private String summary;
+
+    @Hidden
+    private List<String> occasion;
+
 }

src/main/java/com/aida/lanecarford/entity/Customer.java

@@ -20,6 +20,12 @@ import java.time.LocalDateTime;
 @TableName("customers")
 public class Customer extends BaseEntity {
 
+    /**
+     * 导购id,即user表的主键id
+     */
+    @TableField("user_id")
+    private Long salesId;
+
     /**
      * vip ID
      */

src/main/java/com/aida/lanecarford/entity/OutfitRequest.java

@@ -31,4 +31,9 @@ public class OutfitRequest extends BaseEntity{
      * 当前任务状态
      */
     private int status;
+
+    /**
+     * 会话记录id
+     */
+    private Long sessionRecordId;
 }

src/main/java/com/aida/lanecarford/entity/SessionRecord.java

@@ -0,0 +1,51 @@
+package com.aida.lanecarford.entity;
+
+import com.baomidou.mybatisplus.annotation.TableName;
+import com.fasterxml.jackson.annotation.JsonIgnore;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.Accessors;
+import org.apache.commons.lang3.StringUtils;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+@Accessors(chain = true)
+@EqualsAndHashCode(callSuper = true)
+@TableName("session_record")
+public class SessionRecord extends BaseEntity{
+
+    private Long visitRecordId;
+
+    private String sessionId;
+
+    private String requestSummary;
+
+    private String occasions;
+
+    // 获取时转换为List
+    @JsonIgnore
+    public List<String> getOccasionsList() {
+        if (StringUtils.isBlank(this.occasions)) {
+            return new ArrayList<>();
+        }
+        // 使用特定分隔符，确保不会出现在内容中
+        return Arrays.asList(this.occasions.split("\\|\\|"));
+    }
+
+    // 设置时转换为String
+    public void setOccasionsList(List<String> occasionsList) {
+        if (occasionsList == null || occasionsList.isEmpty()) {
+            this.occasions = null;
+        } else {
+            this.occasions = String.join("||", occasionsList);
+        }
+    }
+
+}

src/main/java/com/aida/lanecarford/mapper/SessionRecordMapper.java

@@ -0,0 +1,9 @@
+package com.aida.lanecarford.mapper;
+
+import com.aida.lanecarford.entity.SessionRecord;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+
+@Mapper
+public interface SessionRecordMapper extends BaseMapper<SessionRecord> {
+}

src/main/java/com/aida/lanecarford/service/CustomerService.java

@@ -16,4 +16,6 @@ public interface CustomerService extends IService<Customer> {
 
     IPage<CustomerVO> getAllCustomer(BaseRequest request);
 
+    Customer createCustomer(String vipId, String nickname);
+
 }

src/main/java/com/aida/lanecarford/service/StyleService.java

@@ -19,4 +19,6 @@ public interface StyleService extends IService<Style> {
 
     List<OutfitResultVO> getOutfitResult(List<String> requestIDs);
 
+    List<String> retrieveAndRegenerate(Long tryOnEffectsId);
+
 }

src/main/java/com/aida/lanecarford/service/impl/CustomerServiceImpl.java

@@ -1,5 +1,6 @@
 package com.aida.lanecarford.service.impl;
 
+import com.aida.lanecarford.common.response.ResultEnum;
 import com.aida.lanecarford.common.security.context.UserContext;
 import com.aida.lanecarford.dto.BaseRequest;
 import com.aida.lanecarford.entity.Customer;
@@ -15,8 +16,8 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import io.netty.util.internal.StringUtil;
 import lombok.RequiredArgsConstructor;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Service;
 
 import java.time.LocalDateTime;
@@ -32,13 +33,16 @@ public class CustomerServiceImpl extends ServiceImpl<CustomerMapper, Customer> i
     private final VisitRecordService visitRecordService;
 
     // 选择顾客登录并添加入店记录
-    public CustomerCheckInVO customerCheckIn(String vipId) {
+    public CustomerCheckInVO customerCheckIn(String nickname) {
-        if (StringUtil.isNullOrEmpty(vipId)) {
+        // sales ID即当前的用户id
-            throw new BusinessException("Please enter a VIP ID.");
+        Long salesId = UserContext.getUserHolder().getId();
+        if (StringUtils.isBlank(nickname)) {
+            throw new BusinessException("Please enter a nickname.", ResultEnum.PROMPT.getCode());
         }
         // 1. 判断当前顾客信息在数据库中是否有存储
         LambdaQueryWrapper<Customer> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(Customer::getVipId, vipId);
+        queryWrapper.eq(Customer::getName, nickname)
+                .eq(Customer::getSalesId, salesId);
 
         Customer customer = getOne(queryWrapper);
 
@@ -46,26 +50,28 @@ public class CustomerServiceImpl extends ServiceImpl<CustomerMapper, Customer> i
         if (Objects.isNull(customer)) {
             // todo 从连卡佛数据库查数据
             // 先假设都找不到
-            // throw new BusinessException("This customer does not currently have a registered VIP account.");
+             throw new BusinessException("This customer does not currently have a registered VIP account.");
             // 如果找到了，则添加到数据库
             // 3. 添加当前顾客到本系统数据库
-            customer = new Customer();
+//            customer = new Customer();
-            customer.setVipId(vipId);
+//            customer.setVipId(vipId);
-            customer.setCreatedTime(LocalDateTime.now());
+//            customer.setCreatedTime(LocalDateTime.now());
-
+//
-            save(customer);
+//            save(customer);
         }
 
         // 4. 添加入店记录
-        VisitRecord visitRecord = visitRecordService.addRecord(customer.getId(), UserContext.getUserHolder().getId());
+        VisitRecord visitRecord = visitRecordService.addRecord(customer.getId(), salesId);
 
         return new CustomerCheckInVO(customer.getId(), visitRecord.getId());
     }
 
     // 获取所有的顾客名单
     public IPage<CustomerVO> getAllCustomer(BaseRequest request) {
+        Long salesId = UserContext.getUserHolder().getId();
         LambdaQueryWrapper<Customer> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.select(Customer::getName, Customer::getEmail);
+        queryWrapper.eq(Customer::getSalesId, salesId);
+        queryWrapper.select(Customer::getName, Customer::getVipId);
 
         Page<Customer> page = page(new Page<>(request.getCurrent(), request.getSize()), queryWrapper);
 
@@ -77,4 +83,34 @@ public class CustomerServiceImpl extends ServiceImpl<CustomerMapper, Customer> i
         });
     }
 
+    public Customer createCustomer(String vipId, String nickname) {
+        Long salesId = UserContext.getUserHolder().getId();
+        // 1. 确认nickname是否有重复，有，返回提示信息
+        boolean nicknameExists = lambdaQuery().eq(Customer::getName, nickname)
+                .eq(Customer::getSalesId, salesId).count() > 0;
+        if (nicknameExists) {
+            throw new BusinessException("'" + nickname + "' already exists. Please choose a different nickname.");
+        }
+
+        // 2. 确认输入的vipId是否已存在账号
+        LambdaQueryWrapper<Customer> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(Customer::getVipId, vipId)
+                .eq(Customer::getSalesId, salesId);
+
+        Customer customer = getOne(queryWrapper);
+
+        // 3. 不存在则新建
+        if (Objects.isNull(customer)) {
+            customer = new Customer();
+            customer.setVipId(vipId);
+            customer.setName(nickname);
+            customer.setSalesId(salesId);
+            customer.setCreatedTime(LocalDateTime.now());
+
+            save(customer);
+        }
+
+        return customer;
+    }
+
 }

src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java

@@ -7,10 +7,14 @@ import com.aida.lanecarford.common.enums.StylistPathEnum;
 import com.aida.lanecarford.dto.OutfitCallbackDTO;
 import com.aida.lanecarford.dto.RequestOutfitDTO;
 import com.aida.lanecarford.entity.OutfitRequest;
+import com.aida.lanecarford.entity.SessionRecord;
 import com.aida.lanecarford.entity.Style;
+import com.aida.lanecarford.entity.TryOnEffect;
 import com.aida.lanecarford.exception.BusinessException;
 import com.aida.lanecarford.mapper.OutfitRequestMapper;
+import com.aida.lanecarford.mapper.SessionRecordMapper;
 import com.aida.lanecarford.mapper.StyleMapper;
+import com.aida.lanecarford.mapper.TryOnEffectMapper;
 import com.aida.lanecarford.service.StyleService;
 import com.aida.lanecarford.util.CacheUtil;
 import com.aida.lanecarford.util.MinioUtil;
@@ -20,12 +24,16 @@ import com.aida.lanecarford.vo.OutfitResultVO;
 import com.alibaba.fastjson.JSONObject;
 import com.alibaba.fastjson2.JSON;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import io.netty.util.internal.StringUtil;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.jetbrains.annotations.NotNull;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
 
 import java.time.LocalDateTime;
 import java.util.*;
@@ -41,6 +49,8 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
     private final CacheUtil cacheUtil;
     private final MinioUtil minioUtil;
     private final OutfitRequestMapper outfitRequestMapper;
+    private final SessionRecordMapper sessionRecordMapper;
+    private final TryOnEffectMapper tryOnEffectMapper;
 
     @Value("${webhook.domain}")
     private String webhookDomain;
@@ -51,14 +61,16 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
         // 请求需要顾客id, 生成的数量，风格
 
         StylistPathEnum stylistPathEnum = StylistPathEnum.of(requestOutfitDTO.getStylist());
-        Map<String, Object> params = setRequestOutfitParams(requestOutfitDTO.getCustomerId(), requestOutfitDTO.getNum(),
+        Map<String, Object> params = setRequestOutfitParams(requestOutfitDTO, stylistPathEnum);
-                stylistPathEnum.getName(), requestOutfitDTO.getGender(), requestOutfitDTO.getSessionId());
+
+        SessionRecord sessionRecord = saveOrUpdateSession(requestOutfitDTO.getSessionId(), requestOutfitDTO.getCheckInId(), null, null);
 
         OutfitRequest outfitRequest = new OutfitRequest();
         outfitRequest.setCustomerId(requestOutfitDTO.getCustomerId());
         outfitRequest.setVisitRecordId(requestOutfitDTO.getCheckInId());
         outfitRequest.setStylist(requestOutfitDTO.getStylist());
         outfitRequest.setGender(requestOutfitDTO.getGender());
+        outfitRequest.setSessionRecordId(sessionRecord.getId());
         outfitRequestMapper.insert(outfitRequest);
 
         log.info("agent request params: {}", JSON.toJSONString(params));
@@ -94,16 +106,22 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
 
     }
 
-    private Map<String, Object> setRequestOutfitParams(Long customerId, int num, String stylistPath, String gender, String sessionId) {
+    private Map<String, Object> setRequestOutfitParams(RequestOutfitDTO requestOutfitDTO, StylistPathEnum stylistPathEnum) {
         HashMap<String, Object> params = new HashMap<>();
-        params.put("user_id", customerId.toString());
+        params.put("user_id", requestOutfitDTO.getCustomerId().toString());
-        params.put("num_outfits", num);
+        params.put("num_outfits", requestOutfitDTO.getNum());
-        params.put("stylist_path", stylistPath);
+        params.put("stylist_path", stylistPathEnum.getName());
         params.put("callback_url", webhookDomain);
-        params.put("gender", gender);
+        params.put("gender", requestOutfitDTO.getGender());
 //        params.put("max_len", 5);
-        params.put("session_id", sessionId);
+        params.put("session_id", requestOutfitDTO.getSessionId());
         params.put("batch_sources", Collections.singleton("2025_q4"));
+        if (StringUtils.isNotBlank(requestOutfitDTO.getSummary())) {
+            params.put("request_summary", requestOutfitDTO.getSummary());
+        }
+        if (!CollectionUtils.isEmpty(requestOutfitDTO.getOccasion())) {
+            params.put("occasions", requestOutfitDTO.getOccasion());
+        }
 
         return params;
     }
@@ -166,6 +184,13 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
             outfit.setItems(itemsJson);
 
             updateById(outfit);
+            OutfitRequest outfitRequest = outfitRequestMapper.selectById(outfit.getOutfitRequestId());
+            if (Objects.nonNull(outfitRequest) && Objects.nonNull(outfitRequest.getSessionRecordId())) {
+                SessionRecord sessionRecord = sessionRecordMapper.selectById(outfitRequest.getSessionRecordId());
+                saveOrUpdateSession(sessionRecord.getSessionId(), sessionRecord.getVisitRecordId(),
+                        callbackDTO.getRequest_summary(), callbackDTO.getOccasions());
+            }
+
         }
     }
 
@@ -206,4 +231,77 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
         return resultVOS;
     }
 
+    public SessionRecord saveOrUpdateSession(String sessionId, Long visitsId, String summary, List<String> occasion) {
+        // 判断同一次进店记录中，当前会话id是否已存在
+        QueryWrapper<SessionRecord> queryWrapper = new QueryWrapper<>();
+        queryWrapper.lambda().eq(SessionRecord::getVisitRecordId, visitsId)
+                .eq(SessionRecord::getSessionId, sessionId);
+        SessionRecord sessionRecord = sessionRecordMapper.selectOne(queryWrapper);
+        if (Objects.isNull(sessionRecord)) {
+            sessionRecord = new SessionRecord();
+            sessionRecord.setVisitRecordId(visitsId);
+            sessionRecord.setSessionId(sessionId);
+            sessionRecord.setRequestSummary(summary);
+            sessionRecord.setOccasionsList(occasion);
+            sessionRecord.setCreatedTime(LocalDateTime.now());
+            int insert = sessionRecordMapper.insert(sessionRecord);
+            log.info("新增session record，影响{}条记录", insert);
+        } else {
+            sessionRecord.setRequestSummary(summary);
+            sessionRecord.setOccasionsList(occasion);
+            sessionRecord.setUpdatedTime(LocalDateTime.now());
+            int row = sessionRecordMapper.updateById(sessionRecord);
+            log.info("更新session record，影响{}条记录", row);
+        }
+        return sessionRecord;
+    }
+
+    public List<String> retrieveAndRegenerate(Long tryOnEffectsId) {
+        // 1. 判断id是否有效
+        TryOnEffect tryOnEffect = tryOnEffectMapper.selectById(tryOnEffectsId);
+        if (Objects.isNull(tryOnEffect)) {
+            log.error("无效id: {}", tryOnEffectsId);
+            throw new BusinessException("Error: Invalid ID.");
+        }
+        if (Objects.isNull(tryOnEffect.getStyleId())) {
+            log.error("Id 为：{} 的tryOnEffects记录,没有style_id", tryOnEffectsId);
+            throw new BusinessException("Cannot recreate outfit from past data.");
+        }
+
+        // 2. 组装参数
+        Style style = baseMapper.selectById(tryOnEffect.getStyleId());
+        if (Objects.nonNull(style)) {
+            OutfitRequest outfitRequest = outfitRequestMapper.selectById(style.getOutfitRequestId());
+            if (Objects.isNull(outfitRequest)){
+                log.error("找不到Id 为：{} 的OutfitRequest记录", style.getOutfitRequestId());
+                throw new BusinessException("Cannot recreate outfit from past data.");
+            }
+            SessionRecord sessionRecord = sessionRecordMapper.selectById(outfitRequest.getSessionRecordId());
+            if (Objects.isNull(sessionRecord)){
+                log.error("找不到Id 为：{} 的SessionRecord记录", outfitRequest.getSessionRecordId());
+                throw new BusinessException("Cannot recreate outfit from past data.");
+            }
+            RequestOutfitDTO requestOutfitDTO = getRequestOutfitDTO(outfitRequest, sessionRecord);
+
+            return requestOutfit(requestOutfitDTO);
+        } else {
+            log.error("找不到Id 为：{} 的Style记录", tryOnEffect.getStyleId());
+            throw new BusinessException("Cannot recreate outfit from past data.");
+        }
+    }
+
+    @NotNull
+    private static RequestOutfitDTO getRequestOutfitDTO(OutfitRequest outfitRequest, SessionRecord sessionRecord) {
+        RequestOutfitDTO requestOutfitDTO = new RequestOutfitDTO();
+        requestOutfitDTO.setCustomerId(outfitRequest.getCustomerId());
+        requestOutfitDTO.setCheckInId(outfitRequest.getVisitRecordId());
+        requestOutfitDTO.setStylist(outfitRequest.getStylist());
+        requestOutfitDTO.setGender(outfitRequest.getGender());
+        requestOutfitDTO.setNum(1);
+        requestOutfitDTO.setSessionId(sessionRecord.getSessionId());
+        requestOutfitDTO.setSummary(sessionRecord.getRequestSummary());
+        requestOutfitDTO.setOccasion(sessionRecord.getOccasionsList());
+        return requestOutfitDTO;
+    }
+
 }

src/main/java/com/aida/lanecarford/vo/CustomerVO.java

@@ -16,12 +16,15 @@ public class CustomerVO {
     /**
      * 顾客姓名
      */
-    @Schema(description = "顾客姓名", example = "张三", required = true)
+    @Schema(description = "顾客姓名", example = "张三")
     private String name;
 
     /**
      * 顾客邮箱
      */
-    @Schema(description = "顾客邮箱地址", example = "zhangsan@example.com", required = true)
+    @Schema(description = "顾客邮箱地址", example = "zhangsan@example.com")
     private String email;
+
+    @Schema(description = "顾客vipId", example = "1")
+    private String vipId;
 }