api clientでwatchに対応しました

appthrust · Mar 13, 2024 · 0c99fe9 · 0c99fe9
1 parent 3680360
commit 0c99fe9
Show file tree

Hide file tree

Showing 2 changed files with 216 additions and 114 deletions.
diff --git a/examples/generate-all-k8s-client/k8s-client/client.ts b/examples/generate-all-k8s-client/k8s-client/client.ts
@@ -103,13 +103,20 @@ type HttpHeaderOptions = {
   headers?: Record<string, string> | undefined
 }
 
+export type WatchEventType = 'ADDED' | 'Modified' | 'Deleted' | 'BOOKMARK'
+export type WatchExtraOptions<T> = {
+  watchEventHandler: (e: {
+    type: WatchEventType
+    object: T
+  }) => MaybePromise<unknown>
+}
 export type Options = RetryOptions & HttpHeaderOptions
 
 export async function apiClient<Response>(
   arguments_: QueryArgumentsSpec,
-  extraOptions?: Options
+  extraOptions: Options & WatchExtraOptions<Response>
 ): Promise<Response> {
-  const maxRetries = extraOptions?.maxRetries ?? 3
+  const maxRetries = extraOptions.maxRetries ?? 3
 
   const defaultRetryCondition: RetryConditionFunction = ({ ...object }) => {
     const { res, attempt, error } = object
@@ -227,7 +234,12 @@ export async function apiClient<Response>(
       const isJsonResponse = contentType?.includes('application/json') ?? false
 
       if (isSuccess && isJsonResponse) {
-        if ('watch' in params && params.watch && response.body) {
+        if (
+          'watch' in params &&
+          params.watch &&
+          response.body &&
+          'watchEventHandler' in extraOptions
+        ) {
           const reader = response.body.getReader()
           const textDecoder = new TextDecoder()
           let buffer = ''
@@ -242,8 +254,7 @@ export async function apiClient<Response>(
               const line = buffer.slice(0, newlineIndex)
               buffer = buffer.slice(newlineIndex + 1)
 
-              console.log('----------------')
-              console.log(JSON.parse(line))
+              await extraOptions.watchEventHandler(JSON.parse(line))
             }
           }