当Spring Boot遇见豆包大模型:一场流式响应的"魔法吟唱"仪式
一、前言:关于流式响应的奇妙比喻
想象一下你正在火锅店点单,如果服务员必须等所有菜品都备齐才一次性端上来,你可能会饿得把菜单都啃了。而流式响应就像贴心的服务员,毛肚刚切好就立刻端上桌,肥牛卷还在空中飞着就送到你面前——这就是我们今天要施展的"异步上菜大法"!
注:完整代码见底部
二、Spring Boot魔法阵搭建
2.1 召唤SSE精灵
@PostMapping(value = "/ask", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public SseEmitter ask() {
SseEmitter emitter = new SseEmitter(60_000L); // 设置60秒超时结界
//...后续魔法吟唱
}
这段咒语相当于在Spring Boot的魔法阵中召唤了一个SSE(Server-Sent Events)精灵,它将负责持续不断地把豆包大模型的智慧结晶传送给前端。
2.2 构建跨次元传送门
@CrossOrigin(origins = "*") // 允许所有位面的访问
这行代码就像在霍格沃茨的墙上开了个任意门,让前端、移动端、甚至隔壁王大爷的智能拐杖都能连接我们的服务。
三、豆包大模型召唤仪式
3.1 拼装召唤咒语
ArkService arkService = ArkService.builder()
.apiKey("你的魔法密钥") // 请自行替换成阿拉霍洞开咒
.build();
这里我们正在拼装哈利·波特级别的魔法道具,apiKey
就是你的魔杖认证,记得千万不要像把WiFi密码写成纸条贴在电梯里一样暴露它!
3.2 构建魔法对话卷轴
List<ChatMessage> chatMessages = new ArrayList<>();
chatMessages.add(ChatMessage.builder()
.role(ChatMessageRole.USER) // 麻瓜用户身份
.content("求Java策略模式代码") // 你的灵魂拷问
.build());
这相当于在羊皮纸上书写你的问题,就像给猫头鹰系上求助信。注意提问要像调制福灵剂一样精准,才能得到理想的回答。
四、流式响应的炼金术
4.1 启动魔法反应堆
new Thread(() -> {
// 这里开始炼制长生不老药...哦不,处理响应
}).start();
我们开启了多线程炼丹炉(误),确保主线程不会像被石化的赫敏一样僵住。
4.2 实时传输咒语
arkService.streamChatCompletion(...).forEach(j -> {
emitter.send(SseEmitter.event()
.data(JSON.toJSONString(aiChatDTO))); // 把知识碎片装进漂流瓶
});
这就像用魔法把一本厚书拆成一页页的羊皮纸,通过飞路网持续传送。前端可以像收快递一样实时展示每个字的到来。
五、魔法的收尾工作
5.1 关闭魔法阵
finally {
arkService.shutdownExecutor(); // 收拾魔法实验室
}
就像优秀的巫师总会清理遗忘咒的施法痕迹,这里确保我们的魔法资源不会像韦斯莱家的飞车一样失控乱跑。
5.2 异常处理小精灵
emitter.completeWithError(ex); // 把锅甩给异常对象
当魔药课发生爆炸时,我们要优雅地把错误信息封装成记忆球,而不是让整个霍格沃茨城堡崩溃。
六、实战效果展示
当你在控制台看到这样的输出:
"定义策略接口->创建具体策略类->配置上下文..."
说明你的魔法已经生效!前端会像收到邓布利多军的秘密信号一样,逐字显示出策略模式的代码实现。
postman效果图
七、写给麻瓜程序员的注意事项
- 密钥保护:把你的apiKey当成隐形衣,千万别学洛哈特教授到处炫耀
- 超时设置:60秒足够熬制一锅标准的缓和剂,但复杂问题可能需要更久
- 流量控制:别让你的魔法管道像韦斯莱笑话商店的烟花一样喷涌过度
八、结语:你已学会的魔法
现在你已经掌握了:
- 搭建SSE双向镜通信
- 流式响应炼金术
- 大模型召唤仪式
- 异常处理的黑魔法防御术
快去创建你自己的"魔法AI助手"吧!记住:好的代码应该像福灵剂一样令人愉悦,而不是像呕吐味的比比多味豆让人措手不及。愿代码之神永远眷顾你的IDE!
附
package com.ideal.jason.controller;
import com.alibaba.fastjson.JSON;
import com.volcengine.ark.runtime.model.completion.chat.ChatCompletionRequest;
import com.volcengine.ark.runtime.model.completion.chat.ChatMessage;
import com.volcengine.ark.runtime.model.completion.chat.ChatMessageRole;
import com.volcengine.ark.runtime.service.ArkService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
/**
* @author: jason
* @Date: 14 2月 2025
*/
@CrossOrigin(origins = "*")
@Slf4j
@RestController
@RequestMapping("/api/ai")
public class AIController {
@PostMapping(value = "/ask", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public SseEmitter ask() {
SseEmitter emitter = new SseEmitter(60_000L); // 超时时间 60 秒
String apiKey = "ccde85e1-6ae4-1234-abe1-423164b9f965";//替换成自己的 API Key
String content = "java策略模式代码样例提供";//替换成自己的 API Key
String model = "doubao-1.5-pro-32k-250115";//替换成自己的 模型id
// 创建ArkService实例
ArkService arkService = ArkService.builder().apiKey(apiKey).build();
// 模拟流式数据
new Thread(() -> {
try {
// 初始化消息列表
List<ChatMessage> chatMessages = new ArrayList<>();
// 创建用户消息
ChatMessage userMessage = ChatMessage.builder()
.role(ChatMessageRole.USER) // 设置消息角色为用户
.content(content) // //替换成自己的 提问内容
.build();
// 将用户消息添加到消息列表
chatMessages.add(userMessage);
// 创建聊天完成请求
ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder()
.model(model)// 需要替换为Model ID
.messages(chatMessages) // 设置消息列表
.stream(true)//以流式返回
.build();
// 发送聊天完成请求并打印响应
// 获取响应并打印每个选择的消息内容
StringBuilder sb = new StringBuilder();
arkService.streamChatCompletion(chatCompletionRequest).blockingIterable().forEach(j -> j.getChoices()
.forEach(choice -> {
//System.out.print(choice.getMessage().getContent());
AIChatDTO aiChatDTO =
AIChatDTO.builder().content(choice.getMessage().getContent().toString())
.build();
try {
emitter.send(SseEmitter.event()
//.id(choice.getMessage().getToolCallId())
.data(JSON.toJSONString(aiChatDTO)));
sb.append(aiChatDTO.getContent());
} catch (IOException e) {
log.error("ask IOException");
}
}
));
System.out.println(sb);
emitter.complete();
} catch (Exception ex) {
emitter.completeWithError(ex);
} finally {
// 关闭服务执行器
arkService.shutdownExecutor();
}
}).start();
return emitter;
}
}
maven依赖
<!-- 豆包 -->
<dependency>
<groupId>com.volcengine</groupId>
<artifactId>volcengine-java-sdk-ark-runtime</artifactId>
<version>LATEST</version>
</dependency>