tnb/file/file-service/src/test/java/ChatTest.java

import cn.reghao.jutil.jdk.converter.DateTimeConverter;
import cn.reghao.jutil.jdk.serializer.JsonConverter;
import cn.reghao.jutil.jdk.string.SnowFlake;
import cn.reghao.tnb.file.app.FileApplication;
import cn.reghao.tnb.file.app.zchat.db.mapper.ChatDialogueMapper;
import cn.reghao.tnb.file.app.zchat.db.mapper.ChatMessageMapper;
import cn.reghao.tnb.file.app.zchat.db.mapper.ChatTransferMapper;
import cn.reghao.tnb.file.app.zchat.model.constant.ChatMsgType;
import cn.reghao.tnb.file.app.zchat.model.po.ChatDialogue;
import cn.reghao.tnb.file.app.zchat.model.po.ChatMessage;
import cn.reghao.tnb.file.app.zchat.model.po.ChatTransfer;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.ActiveProfiles;

import java.io.File;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

/**
 * @author reghao
 * @date 2026-06-08 13:29:28
 */
@Slf4j
@ActiveProfiles("dev")
@SpringBootTest(classes = FileApplication.class)
public class ChatTest {
    @Test
    public void chatTest() {
        List<ChatMessage> list = new ArrayList<>();
        JsonArray jsonArray = JsonConverter.jsonFileToObject(new File("/home/reghao/Downloads/chat.json"), JsonArray.class);
        for (JsonElement jsonElement : jsonArray) {
            try {
                JsonObject jsonObject = jsonElement.getAsJsonObject();
                long senderId = jsonObject.get("senderId").getAsLong();
                String content = jsonObject.get("content").getAsString();
                String createAt = jsonObject.get("createAt").getAsString();
                LocalDateTime createAt1 = DateTimeConverter.localDateTime2(createAt);

                long messageId = snowFlake.nextId();
                ChatMessage chatMessage = new ChatMessage(messageId, chatId, senderId);
                chatMessage.setMsgType(ChatMsgType.text.getCode());
                chatMessage.setContent(content);
                chatMessage.setCreateAt(createAt1);
                chatMessage.setSenderId(senderId);
                list.add(chatMessage);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        //chatMessageTest();
        chatMessageMapper.saveAll(list);
    }

    @Autowired
    ChatMessageMapper chatMessageMapper;
    @Autowired
    ChatTransferMapper chatTransferMapper;

    SnowFlake snowFlake = new SnowFlake(1L, 1L);
    long chatId = 65422888961L;
    int chatType = 1;
    private void chatMessageTest() {
        long senderId = 10001L;

        long messageId = snowFlake.nextId();
        ChatMessage chatMessage = new ChatMessage(messageId, chatId, senderId);
        chatMessage.setContent("");
        chatMessage.setMsgType(ChatMsgType.chatTransfer.getCode());
        ChatTransfer chatTransfer = new ChatTransfer(messageId, 1.02);

        chatMessageMapper.save(chatMessage);
        chatTransferMapper.save(chatTransfer);
    }

    @Autowired
    ChatDialogueMapper chatDialogueMapper;
    private void chatDialogueTest() {
        long senderId = 10001L;
        long receiverId = 10002L;

        ChatDialogue chatDialogue1 = new ChatDialogue(chatId, chatType, senderId, receiverId);
        ChatDialogue chatDialogue2 = new ChatDialogue(chatId, chatType, receiverId, senderId);
        chatDialogueMapper.saveAll(List.of(chatDialogue1, chatDialogue2));
    }
}