add endpoint annotation

CDCgov · Jan 8, 2024 · be2209f · be2209f
1 parent 215afe5
commit be2209f
Showing 1 changed file with 22 additions and 20 deletions.
diff --git a/...main/java/gov/cdc/usds/simplereport/api/heathcheck/BackendAndDatabaseHealthIndicator.java b/...main/java/gov/cdc/usds/simplereport/api/heathcheck/BackendAndDatabaseHealthIndicator.java
@@ -6,36 +6,38 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.hibernate.exception.JDBCConnectionException;
+import org.springframework.boot.actuate.endpoint.annotation.Endpoint;
 import org.springframework.boot.actuate.health.Health;
 import org.springframework.boot.actuate.health.HealthIndicator;
 import org.springframework.stereotype.Component;
 
 @Component("backend-and-db-smoke-test")
+@Endpoint
 @Slf4j
 @RequiredArgsConstructor
 public class BackendAndDatabaseHealthIndicator implements HealthIndicator {
-  private final FeatureFlagRepository _ffRepo;
-  private final OktaRepository _oktaRepo;
-  public static final String ACTIVE_LITERAL = "ACTIVE";
+    private final FeatureFlagRepository _ffRepo;
+    private final OktaRepository _oktaRepo;
+    public static final String ACTIVE_LITERAL = "ACTIVE";
 
-  @Override
-  public Health health() {
-    try {
-      _ffRepo.findAll();
-      String oktaStatus = _oktaRepo.getApplicationStatusForHealthCheck();
+    @Override
+    public Health health() {
+        try {
+            _ffRepo.findAll();
+            String oktaStatus = _oktaRepo.getApplicationStatusForHealthCheck();
 
-      if (!ACTIVE_LITERAL.equals(oktaStatus)) {
-        log.info("Okta status didn't return ACTIVE, instead returned " + oktaStatus);
-        return Health.down().build();
-      }
+            if (!ACTIVE_LITERAL.equals(oktaStatus)) {
+                log.info("Okta status didn't return ACTIVE, instead returned " + oktaStatus);
+                return Health.down().build();
+            }
 
-      return Health.up().build();
-    } catch (JDBCConnectionException e) {
-      return Health.down().build();
-      // Okta API call errored
-    } catch (ApiException e) {
-      log.info(e.getMessage());
-      return Health.down().build();
+            return Health.up().build();
+        } catch (JDBCConnectionException e) {
+            return Health.down().build();
+            // Okta API call errored
+        } catch (ApiException e) {
+            log.info(e.getMessage());
+            return Health.down().build();
+        }
     }
-  }
 }