vovk 3.0.0-draft.281 → 3.0.0-draft.282
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
|
@@ -22,6 +22,7 @@ const defaultStreamHandler = async (response, controller) => {
|
|
|
22
22
|
const reader = response.body.getReader();
|
|
23
23
|
// if streaming is too rapid, we need to make sure that the loop is stopped
|
|
24
24
|
let canceled = false;
|
|
25
|
+
const subscribers = new Set();
|
|
25
26
|
async function* asyncIterator() {
|
|
26
27
|
let prepend = '';
|
|
27
28
|
while (true) {
|
|
@@ -52,6 +53,10 @@ const defaultStreamHandler = async (response, controller) => {
|
|
|
52
53
|
break;
|
|
53
54
|
}
|
|
54
55
|
if (data) {
|
|
56
|
+
subscribers.forEach((cb) => {
|
|
57
|
+
if (!canceled)
|
|
58
|
+
cb(data);
|
|
59
|
+
});
|
|
55
60
|
if ('isError' in data && 'reason' in data) {
|
|
56
61
|
const upcomingError = data.reason;
|
|
57
62
|
await reader.cancel();
|
|
@@ -72,6 +77,14 @@ const defaultStreamHandler = async (response, controller) => {
|
|
|
72
77
|
[Symbol.asyncIterator]: asyncIterator,
|
|
73
78
|
[Symbol.dispose]: () => controller.abort(),
|
|
74
79
|
[Symbol.asyncDispose]: () => controller.abort(),
|
|
80
|
+
onIterate: (cb) => {
|
|
81
|
+
if (canceled)
|
|
82
|
+
return () => { };
|
|
83
|
+
subscribers.add(cb);
|
|
84
|
+
return () => {
|
|
85
|
+
subscribers.delete(cb);
|
|
86
|
+
};
|
|
87
|
+
},
|
|
75
88
|
abort: () => {
|
|
76
89
|
canceled = true;
|
|
77
90
|
return controller.abort();
|
package/cjs/client/types.d.ts
CHANGED
|
@@ -30,6 +30,7 @@ export type VovkStreamAsyncIterable<T> = {
|
|
|
30
30
|
[Symbol.dispose](): Promise<void> | void;
|
|
31
31
|
[Symbol.asyncDispose](): Promise<void> | void;
|
|
32
32
|
[Symbol.asyncIterator](): AsyncIterator<T>;
|
|
33
|
+
onIterate: (cb: (data: T) => void) => () => void;
|
|
33
34
|
abort: () => Promise<void> | void;
|
|
34
35
|
};
|
|
35
36
|
type StaticMethodReturn<T extends ControllerStaticMethod> = ReturnType<T> extends NextResponse<infer U> | Promise<NextResponse<infer U>> ? U : ReturnType<T> extends Response | Promise<Response> ? Awaited<ReturnType<T>> : ReturnType<T>;
|
|
@@ -22,6 +22,7 @@ const defaultStreamHandler = async (response, controller) => {
|
|
|
22
22
|
const reader = response.body.getReader();
|
|
23
23
|
// if streaming is too rapid, we need to make sure that the loop is stopped
|
|
24
24
|
let canceled = false;
|
|
25
|
+
const subscribers = new Set();
|
|
25
26
|
async function* asyncIterator() {
|
|
26
27
|
let prepend = '';
|
|
27
28
|
while (true) {
|
|
@@ -52,6 +53,10 @@ const defaultStreamHandler = async (response, controller) => {
|
|
|
52
53
|
break;
|
|
53
54
|
}
|
|
54
55
|
if (data) {
|
|
56
|
+
subscribers.forEach((cb) => {
|
|
57
|
+
if (!canceled)
|
|
58
|
+
cb(data);
|
|
59
|
+
});
|
|
55
60
|
if ('isError' in data && 'reason' in data) {
|
|
56
61
|
const upcomingError = data.reason;
|
|
57
62
|
await reader.cancel();
|
|
@@ -72,6 +77,14 @@ const defaultStreamHandler = async (response, controller) => {
|
|
|
72
77
|
[Symbol.asyncIterator]: asyncIterator,
|
|
73
78
|
[Symbol.dispose]: () => controller.abort(),
|
|
74
79
|
[Symbol.asyncDispose]: () => controller.abort(),
|
|
80
|
+
onIterate: (cb) => {
|
|
81
|
+
if (canceled)
|
|
82
|
+
return () => { };
|
|
83
|
+
subscribers.add(cb);
|
|
84
|
+
return () => {
|
|
85
|
+
subscribers.delete(cb);
|
|
86
|
+
};
|
|
87
|
+
},
|
|
75
88
|
abort: () => {
|
|
76
89
|
canceled = true;
|
|
77
90
|
return controller.abort();
|
package/mjs/client/types.d.ts
CHANGED
|
@@ -30,6 +30,7 @@ export type VovkStreamAsyncIterable<T> = {
|
|
|
30
30
|
[Symbol.dispose](): Promise<void> | void;
|
|
31
31
|
[Symbol.asyncDispose](): Promise<void> | void;
|
|
32
32
|
[Symbol.asyncIterator](): AsyncIterator<T>;
|
|
33
|
+
onIterate: (cb: (data: T) => void) => () => void;
|
|
33
34
|
abort: () => Promise<void> | void;
|
|
34
35
|
};
|
|
35
36
|
type StaticMethodReturn<T extends ControllerStaticMethod> = ReturnType<T> extends NextResponse<infer U> | Promise<NextResponse<infer U>> ? U : ReturnType<T> extends Response | Promise<Response> ? Awaited<ReturnType<T>> : ReturnType<T>;
|