结组作业4

今天继续写后端的代码，包括controller和mapper和service的代码

首先是和团队成员一起写的调用chat接口的controller的代码

package org.test.tongyuzhe.Controller;


import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.hunyuan.v20230901.HunyuanClient;
import com.tencentcloudapi.hunyuan.v20230901.models.*;
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.RestController;
import org.test.tongyuzhe.Pojo.ChatRequest;
import org.test.tongyuzhe.Pojo.Result;
import org.test.tongyuzhe.utils.JwtUtil;

import java.util.ArrayList;
import java.util.Map;

@RestController
@RequestMapping("/api")
public class ChatController {


    @PostMapping("/chat")
    public Result<String> processChatRequest(@RequestBody ChatRequest chatRequest) {
        System.out.println(chatRequest.toString());
        //验证token
        try {
            Map<String, Object> claim = JwtUtil.parseToken(chatRequest.getToken());


            String userId = claim.get("id").toString();

            ChatProRequest req = new ChatProRequest();
            ArrayList<Message> messages1 = new ArrayList<Message>();
            Message message1 = new Message();

            message1.setRole("system");
            message1.setContent("你是一个用于辅导儿童心理的对话专家，你的名字叫小e" +
                    "我要求你在与儿童对话的过程中，" +
                    "辅导儿童心理，同时引导他们树立正确的价值观；在与家长对话的时候，" +
                    "能够根据与儿童的对话内容分析儿童的心理健康状况，并给出家长建议");

            messages1.add(message1);
            messages1.addAll(chatRequest.getMessages());
            Message[] messages = messages1.toArray(new Message[0]);

            req.setMessages(messages);
            ChatProResponse response = sendPostRequest(req);

            System.out.println(response);

            Choice[] choices = response.getChoices();
            Message message = choices[0].getMessage();

            String choice = message.getContent();

            for (int i = 0; i < chatRequest.getMessages().size(); i++) {
                choice+=chatRequest.getMessages();
            }

            // System.out.println(choice);
            return Result.success(choice);
        } catch (Exception e) {
            //未登录
            Result.error("未登录");
            throw new RuntimeException(e);
        }



    }
    public static ChatProResponse sendPostRequest(ChatProRequest req) {
        try{
            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
            Credential cred = new Credential("AKIDMF9rRpGgzAYpdP2VnwxmNksmUyGKBop4", "MX6Ws6EFA0idP7LguBblmaDdrulQtGUd");
            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("hunyuan.tencentcloudapi.com");
            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);
            // 实例化要请求产品的client对象,clientProfile是可选的
            HunyuanClient client = new HunyuanClient(cred, "", clientProfile);
            // 实例化一个请求对象,每个接口都会对应一个request对象

            req.setTopP(0F);
            req.setTemperature(0F);
            req.setStream(false);
            // 返回的resp是一个ChatProResponse的实例，与请求对象对应
            ChatProResponse resp = client.ChatPro(req);
            return resp;

        } catch (TencentCloudSDKException e) {
            System.out.println(e.toString());
        }
        return null;
    }


}

然后mapper层主要是sql语句的书写，比较简单

package org.test.tongyuzhe.Mapper;

import org.apache.ibatis.annotations.Insert;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Select;
import org.test.tongyuzhe.Pojo.Chat;

import java.util.List;

@Mapper
public interface ChatMapper {
    @Select("select * from chat where userId=#{userId}")
    List<Chat> getMessage(Integer userId);
    @Insert("insert into chat(userId, request, response, time) VALUES (#{userId},#{request},#{response},#{time})")
    void insertMessage(Chat chat);
}