Java以流的形式返回前端

r囧r小猫 2024-03-23 09:57 9阅读 0赞

**前言：为了实现像ChatGPT一样的效果：文字进行逐个显示，后端返回的时候需要以流的形式**。

**目录**

一、字符串流

二、文件流

--------------------

#### 一、字符串流 ####

@PostMapping("returnStream")
        public void returnStream(HttpServletResponse response) throws IOException {
            String message = "我是一段等待已流形式返回的文字";
            // 以流的形式返回
            ServletOutputStream out = null;
            ByteArrayOutputStream baos = null;
            try {
                InputStream inStream = new ByteArrayInputStream(message.getBytes());
                byte[] buffer = new byte[1024];
                int len;
                baos = new ByteArrayOutputStream();
                while ((len = inStream.read(buffer)) != -1) {
                    baos.write(buffer, 0, len);
                }
                out = response.getOutputStream();
                out.write(baos.toByteArray());
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                Objects.requireNonNull(baos).flush();
                baos.close();
                Objects.requireNonNull(out).flush();
                out.close();
            }
    
        }

![d95dedf0033248ed806c9505f026cf39.png][]

#### 二、文件流 ####

ServletOutputStream out = null;
    		ByteArrayOutputStream baos = null;
    		try {
    			File file=new File(filename);
    			InputStream inStream=new FileInputStream(file);
    			byte[] buffer = new byte[1024];
    			int len;
    			baos = new ByteArrayOutputStream();
    			while ((len = inStream.read(buffer)) != -1) {
    				baos.write(buffer, 0, len);
    			}
    			out = response.getOutputStream();
    			out.write(baos.toByteArray());
    		} catch (Exception e) {
    			e.printStackTrace();
    		} finally {
    			baos.flush();
    			baos.close();
    			out.flush();
    			out.close();
    		}

[d95dedf0033248ed806c9505f026cf39.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/23/814ab41030f2468d93544f6e6e009c4a.png