aidlab/lanecarford_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
xupei
2025-12-23 11:38:48 +08:00
parent 5e6e9a8787
commit 2972ab1dca
13 changed files with 274 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/main/java/com/aida/lanecarford/controller/CustomerController.java
View File

@@ -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) {
return ApiResponse.success(customerService.customerCheckIn(vipId));
public ApiResponse<CustomerCheckInVO> customerCheckIn(@RequestParam String nickname) {
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));
}
}

11
src/main/java/com/aida/lanecarford/controller/StyleController.java
View File

@@ -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));
}
}

4
src/main/java/com/aida/lanecarford/dto/OutfitCallbackDTO.java
View File

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

9
src/main/java/com/aida/lanecarford/dto/RequestOutfitDTO.java
View File

@@ -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;
}

6
src/main/java/com/aida/lanecarford/entity/Customer.java
View File

@@ -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
*/

5
src/main/java/com/aida/lanecarford/entity/OutfitRequest.java
View File

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

51
src/main/java/com/aida/lanecarford/entity/SessionRecord.java Normal file
View File

@@ -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);
}
}
}

9
src/main/java/com/aida/lanecarford/mapper/SessionRecordMapper.java Normal file
View File

@@ -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> {
}

2
src/main/java/com/aida/lanecarford/service/CustomerService.java
View File

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

2
src/main/java/com/aida/lanecarford/service/StyleService.java
View File

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

62
src/main/java/com/aida/lanecarford/service/impl/CustomerServiceImpl.java
View File

@@ -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) {
if (StringUtil.isNullOrEmpty(vipId)) {
throw new BusinessException("Please enter a VIP ID.");
public CustomerCheckInVO customerCheckIn(String nickname) {
// sales ID即当前的用户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.setVipId(vipId);
customer.setCreatedTime(LocalDateTime.now());
save(customer);
// customer = new Customer();
// customer.setVipId(vipId);
// customer.setCreatedTime(LocalDateTime.now());
//
// 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;
}
}

114
src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java
View File

@@ -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(),
stylistPathEnum.getName(), requestOutfitDTO.getGender(), requestOutfitDTO.getSessionId());
Map<String, Object> params = setRequestOutfitParams(requestOutfitDTO, stylistPathEnum);
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("num_outfits", num);
params.put("stylist_path", stylistPath);
params.put("user_id", requestOutfitDTO.getCustomerId().toString());
params.put("num_outfits", requestOutfitDTO.getNum());
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;
}
}

7
src/main/java/com/aida/lanecarford/vo/CustomerVO.java
View File

@@ -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;
}