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