1. import java.util.concurrent.*;
  2. import java.util.function.Supplier;
  4. public class ApiPerformanceTest {
  6. /**
  7. * Measures the performance of an API endpoint with a timeout.
  8. *
  9. * @param endpoint The supplier that provides the API endpoint.
  10. * @param timeoutMs The timeout in milliseconds.
  11. * @return The result of the API call, or null if an error occurred or timeout.
  12. */
  13. public static Object measureApiPerformance(Supplier<Object> endpoint, long timeoutMs) {
  14. ExecutorService executor = Executors.newSingleThreadExecutor();
  15. Future<Object> future = executor.submit(() -> {
  16. try {
  17. return endpoint.get(); // Execute the API endpoint
  18. } catch (Exception e) {
  19. // Handle exceptions (e.g., network errors, API errors)
  20. System.err.println("Error during API call: " + e.getMessage());
  21. return null; // Or throw, depending on desired behavior
  22. }
  23. });
  25. try {
  26. return future.get(timeoutMs, TimeUnit.MILLISECONDS); // Wait for the result with a timeout
  27. } catch (TimeoutException e) {
  28. System.err.println("API call timed out after " + timeoutMs + "ms.");
  29. return null; // Indicate timeout
  30. } finally {
  31. executor.shutdown(); // Shut down the executor
  32. }
  33. }
  36. public static void main(String[] args) {
  37. // Example usage:
  38. Supplier<String> myApiEndpoint = () -> {
  39. try {
  40. // Simulate an API call with a delay
  41. Thread.sleep(500);
  42. return "API Response";
  43. } catch (InterruptedException e) {
  44. Thread.currentThread().interrupt();
  45. return null;
  46. }
  47. };
  49. Object result = measureApiPerformance(myApiEndpoint, 1000);
  51. if (result != null) {
  52. System.out.println("API Response: " + result);
  53. } else {
  54. System.out.println("API call failed or timed out.");
  55. }
  56. }
  57. }

Add your comment