7x31/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/Yidadaa/ChatGPT-Next-Web.git synced 2025-08-08 14:59:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

using stream_fetch in App

Browse Source
This commit is contained in:
lloydzhou
2024-09-29 20:32:36 +08:00
parent dd77ad5d74
commit 9e6ee50fa6
2 changed files with 52 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

110
app/utils/stream.ts
View File

@@ -12,9 +12,55 @@ type ResponseEvent = {
};
};
type StreamResponse = {
request_id: number;
status: number;
status_text: string;
headers: Record<string, string>;
};
export function fetch(url: string, options?: RequestInit): Promise<any> {
if (window.__TAURI__) {
const { signal, method = "GET", headers = {}, body = [] } = options || {};
let unlisten: Function | undefined;
let request_id = 0;
const ts = new TransformStream();
const writer = ts.writable.getWriter();
const close = () => {
unlisten && unlisten();
writer.ready.then(() => {
try {
writer.releaseLock();
} catch (e) {
console.error(e);
}
ts.writable.close();
});
};
if (signal) {
signal.addEventListener("abort", () => close());
}
// @ts-ignore 2. listen response multi times, and write to Response.body
window.__TAURI__.event
.listen("stream-response", (e: ResponseEvent) => {
const { request_id: rid, chunk, status } = e?.payload || {};
if (request_id != rid) {
return;
}
if (chunk) {
writer &&
writer.ready.then(() => {
writer && writer.write(new Uint8Array(chunk));
});
} else if (status === 0) {
// end of body
close();
}
})
.then((u: Function) => (unlisten = u));
return window.__TAURI__
.invoke("stream_fetch", {
method,
@@ -26,61 +72,11 @@ export function fetch(url: string, options?: RequestInit): Promise<any> {
? Array.from(new TextEncoder().encode(body))
: [],
})
.then(
(res: {
request_id: number;
status: number;
status_text: string;
headers: Record<string, string>;
}) => {
const { request_id, status, status_text: statusText, headers } = res;
console.log("send request_id", request_id, status, statusText);
let unlisten: Function | undefined;
const ts = new TransformStream();
const writer = ts.writable.getWriter();
const close = () => {
unlisten && unlisten();
writer.ready.then(() => {
try {
writer.releaseLock();
} catch (e) {
console.error(e);
}
ts.writable.close();
});
};
const response = new Response(ts.readable, {
status,
statusText,
headers,
});
if (signal) {
signal.addEventListener("abort", () => close());
}
// @ts-ignore 2. listen response multi times, and write to Response.body
window.__TAURI__.event
.listen("stream-response", (e: ResponseEvent) => {
const { id, payload } = e;
const { request_id: rid, chunk, status } = payload;
if (request_id != rid) {
return;
}
if (chunk) {
writer &&
writer.ready.then(() => {
writer && writer.write(new Uint8Array(chunk));
});
} else if (status === 0) {
// end of body
close();
}
})
.then((u: Function) => (unlisten = u));
return response;
},
)
.then((res: StreamResponse) => {
request_id = res.request_id;
const { status, status_text: statusText, headers } = res;
return new Response(ts.readable, { status, statusText, headers });
})
.catch((e) => {
console.error("stream error", e);
throw e;
@@ -88,7 +84,3 @@ export function fetch(url: string, options?: RequestInit): Promise<any> {
}
return window.fetch(url, options);
}
if (undefined !== window) {
window.tauriFetch = fetch;
}