refactor(ai): 优化代码结构和更新时间处理

Administrator

yesterday 8130bed63a505cf718481c0d95d3fcf7b037db96

 src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java

@@ -66,6 +66,8 @@
            throw new FebsException("产品AI陪练不存在");
        }

        String companyId = aiProductRoleLink.getCompanyId();

        Date nowTime = new Date();
        LambdaQueryWrapper<AiMemberTalk> query = Wrappers.lambdaQuery(AiMemberTalk.class);
        query.eq(AiMemberTalk::getMemberId,memberUuid);
@@ -73,7 +75,7 @@
        query.last("limit 1");
        AiMemberTalk aiMemberTalk = aiMemberTalkService.getByQuery(query);
        if (ObjectUtil.isNull(aiMemberTalk)){
            aiMemberTalk = aiMemberTalkService.add(memberUuid,productId,nowTime);
            aiMemberTalk = aiMemberTalkService.add(memberUuid,companyId,productId,nowTime);
        }

        ApiMemberTalkStreamVo apiMemberTalkVo = new ApiMemberTalkStreamVo();
@@ -81,7 +83,7 @@
        AiProductQuestion question = aiProductQuestionService.createQuestion(productId);
        if (ObjectUtil.isNotNull(question)){
            title = question.getTitle();
            AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, title, nowTime);
            AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, title, nowTime);
            apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
            aiMemberTalkService.updateTimeUpdate(nowTime,aiMemberTalk.getId());
        }
@@ -128,7 +130,10 @@
        if (ObjectUtil.isNull(aiMemberTalk)){
            throw new FebsException("对话不存在");
        }
        aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),2,content,new Date());
        this.updateMemberTalkUpdateTime(aiMemberTalk.getId(), new Date());

        String companyId = aiMemberTalk.getCompanyId();
        aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),companyId,2,content,new Date());
        return new FebsResponse().success();
    }

@@ -274,13 +279,17 @@
        if (ObjectUtil.isNull(aiMemberTalk)){
            throw new FebsException("对话不存在");
        }

        this.updateMemberTalkUpdateTime(aiMemberTalk.getId(), new Date());

        String companyId = aiMemberTalk.getCompanyId();
        Integer type = dto.getType();
        String memberTalkItemId;
        String contentByCode = AiTalkOutputEnum.HIGH_LIGHT.getCodeByType(type);
        if(StrUtil.isEmpty(dto.getMemberTalkItemId())){
            HashMap<String, String> stringStringHashMap = new HashMap<>();
            stringStringHashMap.put(contentByCode,content);
            AiMemberTalkItem add = aiMemberTalkItemService.add(memberUuid, memberTalkId, 3, JSONUtil.toJsonStr(stringStringHashMap), new Date());
            AiMemberTalkItem add = aiMemberTalkItemService.add(memberUuid, memberTalkId,companyId, 3, JSONUtil.toJsonStr(stringStringHashMap), new Date());
            memberTalkItemId = add.getId();
        }else{
            memberTalkItemId = dto.getMemberTalkItemId();
@@ -302,4 +311,13 @@
        return new FebsResponse().success().data(stringStringHashMap);
    }

    @Override
    public void updateMemberTalkUpdateTime(String memberTalkId, Date updateTime) {
        aiMemberTalkMapper.update(null,
                Wrappers.lambdaUpdate(AiMemberTalk.class)
                .set(AiMemberTalk::getUpdatedTime,updateTime)
                .eq(AiMemberTalk::getId,memberTalkId)
                );
    }

}