Timeout in Hono

[vivekpatel5996]

Hello all,

I want to keep middleware in Hono for general timeout. Let's say if any API takes more than two minutes I want to throw error to the client with message something like timeout occurred. I am using Hono for cloudflare worker.
How Can I achieve this ?

Your help will be much appreciated.

Thanks

[EdamAme-x]

How about passing the next function and the setTimeout function to the Promise function to reject and resolve?

[vivekpatel5996]

Hi @EdamAme-x Thanks for your reply. Can you please add code for this?

[hlobil]

import { Hono, MiddlewareHandler } from 'hono';
import { HTTPException } from 'hono/http-exception';


class TimeoutException extends HTTPException {
  constructor(message: string = 'Timeout occurred') {
    super(408, { message });
  }
}

const timeout = (prom: Promise<any>, time: number | undefined, exception: any) => {
  let timer: string | number | NodeJS.Timeout | undefined;
  return Promise.race([
    prom,
    new Promise((_r, rej) => (timer = setTimeout(rej, time, exception))),
  ]).finally(() => clearTimeout(timer));
};

// Timeout Middleware
function timeoutMiddleware(ms: number): MiddlewareHandler {
  return async (c, next) => {
    return await timeout(next(), ms, new TimeoutException());
  };
}

const app = new Hono();

// Apply the middleware
app.use('*', timeoutMiddleware(5_000)); // 30 seconds

// Define your routes
app.get('/', async (c) => {
  return c.json({ message: 'Hello World' });
});

app.get('/timeout', async (c) => {
  await new Promise((resolve) => {
    setTimeout(resolve, 31_000); // 31 seconds
  });
  return c.json({ message: 'Hello World' });
});

app.get('/error', async (c) => {
  throw new Error('Something went wrong');
});

app.onError((error, c) => {
  if (error instanceof TimeoutException) {
    return c.json({ message: error.message }, error.status); // 408 Request Timeout
  }
  return c.json({ message: error.message }, 500); // 500 Internal Server Error
});

import { serve } from '@hono/node-server';
serve(app).listen(3000);

// Start the application
// addEventListener('fetch', (event) => {
//   event.respondWith(app.fetch(event.request));
// });