Package com.google.genai.gaos.utils

Class AsyncHooks

java.lang.Object

com.google.genai.gaos.utils.AsyncHooks

All Implemented Interfaces:

AsyncHook.AfterError, AsyncHook.AfterSuccess, AsyncHook.BeforeRequest

public class AsyncHooks
extends Object
implements AsyncHook.BeforeRequest, AsyncHook.AfterSuccess, AsyncHook.AfterError

Async hook registry for runtime request/response processing.

Example usage:

 asyncHooks.registerBeforeRequest((context, request) ->
     CompletableFuture.completedFuture(
         request.toBuilder()
             .header("transaction-id", UUID.randomUUID().toString())
             .build()));
 

Field Summary

Fields inherited from interface com.google.genai.gaos.utils.AsyncHook.AfterError

DEFAULT

Fields inherited from interface com.google.genai.gaos.utils.AsyncHook.AfterSuccess

DEFAULT

Fields inherited from interface com.google.genai.gaos.utils.AsyncHook.BeforeRequest

DEFAULT

Constructor Summary

Constructors

Constructor	Description
AsyncHooks()

Method Summary

Modifier and Type	Method	Description
CompletableFuture<HttpResponse<InputStream>>	afterError(Hook.AfterErrorContext context, HttpResponse<InputStream> response, Throwable error)	Either returns an HttpResponse or throws an Exception.
CompletableFuture<HttpResponse<InputStream>>	afterSuccess(Hook.AfterSuccessContext context, HttpResponse<InputStream> response)	Transforms the given response before response processing occurs.
CompletableFuture<HttpRequest>	beforeRequest(Hook.BeforeRequestContext context, HttpRequest request)	Transforms the given HttpRequest before sending.
AsyncHooks	registerAfterError(AsyncHook.AfterError afterError)	Registers an async after-error hook.
AsyncHooks	registerAfterSuccess(AsyncHook.AfterSuccess afterSuccess)	Registers an async after-success hook.
AsyncHooks	registerBeforeRequest(AsyncHook.BeforeRequest beforeRequest)	Registers an async before-request hook.

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

AsyncHooks

public AsyncHooks()

Method Details

registerBeforeRequest

public AsyncHooks registerBeforeRequest(AsyncHook.BeforeRequest beforeRequest)

Registers an async before-request hook. Hooks are chained in registration order.

Parameters:

beforeRequest - async hook returning CompletableFuture<HttpRequest>

Returns:

this

registerAfterSuccess

public AsyncHooks registerAfterSuccess(AsyncHook.AfterSuccess afterSuccess)

Registers an async after-success hook. Hooks are chained in registration order.

Parameters:

afterSuccess - async hook returning CompletableFuture<HttpResponse>

Returns:

this

registerAfterError

public AsyncHooks registerAfterError(AsyncHook.AfterError afterError)

Registers an async after-error hook. Hooks are chained in registration order.

Parameters:

afterError - async hook for error handling

Returns:

this

beforeRequest

public CompletableFuture<HttpRequest> beforeRequest(Hook.BeforeRequestContext context,
 HttpRequest request)

Description copied from interface: AsyncHook.BeforeRequest

Transforms the given HttpRequest before sending.

Note that HttpRequest is immutable. To modify the request use HttpRequest.toBuilder() which copies the request into a builder.

Specified by:

beforeRequest in interface AsyncHook.BeforeRequest

Parameters:

context - context for the hook call

request - request to be transformed

Returns:

transformed request

afterSuccess

public CompletableFuture<HttpResponse<InputStream>> afterSuccess(Hook.AfterSuccessContext context,
 HttpResponse<InputStream> response)

Description copied from interface: AsyncHook.AfterSuccess

Transforms the given response before response processing occurs.

Specified by:

afterSuccess in interface AsyncHook.AfterSuccess

Parameters:

context - context for the hook call

response - response to be transformed

Returns:

transformed response

afterError

public CompletableFuture<HttpResponse<InputStream>> afterError(Hook.AfterErrorContext context,
 HttpResponse<InputStream> response,
 Throwable error)

Description copied from interface: AsyncHook.AfterError

Either returns an HttpResponse or throws an Exception. Must be passed either a response or an error (both can't be absent).

Specified by:

afterError in interface AsyncHook.AfterError

Parameters:

context - context for the error

response - response information if available.

error - the optional exception. If response present then the error is for-info only, it was the last error in the chain of AfterError hook calls leading to this one

Returns:

HTTP response if method decides that an exception is not to be thrown