Added Rotation Management

core/src/main/java/com/flipkart/gjex/core/config/RotationManagementConfig.java

@@ -0,0 +1,24 @@
+package com.flipkart.gjex.core.config;
+
+import com.flipkart.gjex.core.web.RotationManagementResource;
+import org.glassfish.jersey.server.ResourceConfig;
+
+import javax.inject.Inject;
+import javax.inject.Named;
+import javax.inject.Singleton;
+
+/**
+ * RotationManagementConfig for registering RotationManagementResource.
+ * @author ajay.jalgaonkar
+ *
+ */
+
+@Singleton
+@Named("RotationManagementConfig")
+public class RotationManagementConfig extends ResourceConfig  {
+
+  @Inject
+  public RotationManagementConfig (RotationManagementResource rotationManagementResource) {
+		register(rotationManagementResource);
+  }
+}

core/src/main/java/com/flipkart/gjex/core/task/RotationManagementBasedHealthCheck.java

@@ -0,0 +1,54 @@
+package com.flipkart.gjex.core.task;
+
+import com.flipkart.gjex.core.logging.Logging;
+import io.dropwizard.metrics5.health.HealthCheck;
+
+import javax.inject.Singleton;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+/**
+ * Rotation management based health check for the app
+ * @author ajay.jalgaonkar
+ */
+
+@Singleton
+public class RotationManagementBasedHealthCheck extends HealthCheck implements Logging {
+  private static final String BIR = "bir";
+  private static final String OOR = "oor";
+
+  private AtomicBoolean state;
+
+  public RotationManagementBasedHealthCheck() {
+    state = new AtomicBoolean(true);
+  }
+
+  @Override
+  protected Result check() {
+    if (isBir()) {
+      info("Returning healthy status.");
+      return Result.healthy("Server is " + getStatus());
+    } else {
+      info("Returning unhealthy status.");
+      return Result.unhealthy("Server is " + getStatus());
+    }
+  }
+
+  public String getStatus() {
+    return isBir() ? BIR : OOR;
+  }
+
+  public String makeOor() {
+    state.set(false);
+    return OOR;
+  }
+
+  public String makeBir() {
+    state.set(true);
+    return BIR;
+  }
+
+  public boolean isBir() {
+    return state.get();
+  }
+
+}

core/src/main/java/com/flipkart/gjex/core/web/RotationManagementResource.java

@@ -0,0 +1,52 @@
+package com.flipkart.gjex.core.web;
+
+import com.flipkart.gjex.core.task.RotationManagementBasedHealthCheck;
+
+import javax.inject.Inject;
+import javax.inject.Named;
+import javax.inject.Singleton;
+import javax.servlet.ServletContext;
+import javax.ws.rs.GET;
+import javax.ws.rs.Path;
+import javax.ws.rs.Produces;
+import javax.ws.rs.core.Context;
+import javax.ws.rs.core.MediaType;
+import javax.ws.rs.core.Response;
+
+/**
+ * Resource for rotation management
+ * @author ajay.jalgaonkar
+ */
+
+@Singleton
+@Path("/")
+@Named
+public class RotationManagementResource {
+
+  @Context
+  private ServletContext servletContext;
+
+  @Inject
+  public RotationManagementResource(RotationManagementBasedHealthCheck rotationManagementBasedHealthCheck) {
+    this.rotationManagementBasedHealthCheck = rotationManagementBasedHealthCheck;
+  }
+
+  private RotationManagementBasedHealthCheck rotationManagementBasedHealthCheck;
+
+  @GET
+  @Path("/oor")
+  @Produces(MediaType.APPLICATION_JSON)
+  public Response oor() {
+    String response = this.rotationManagementBasedHealthCheck.makeOor();
+    return Response.status(Response.Status.OK).entity(response).build();
+  }
+
+  @GET
+  @Produces(MediaType.APPLICATION_JSON)
+  @Path("/bir")
+  public Response bir() {
+    String response = this.rotationManagementBasedHealthCheck.makeBir();
+    return Response.status(Response.Status.OK).entity(response).build();
+  }
+
+}

examples/src/main/java/com/flipkart/gjex/examples/helloworld/guice/HelloWorldModule.java

@@ -15,26 +15,22 @@
  */
 package com.flipkart.gjex.examples.helloworld.guice;
 
-import org.glassfish.jersey.server.ResourceConfig;
-
-import io.dropwizard.metrics5.health.HealthCheck;
 import com.flipkart.gjex.core.filter.Filter;
+import com.flipkart.gjex.core.task.RotationManagementBasedHealthCheck;
 import com.flipkart.gjex.core.tracing.TracingSampler;
 import com.flipkart.gjex.examples.helloworld.filter.AuthFilter;
 import com.flipkart.gjex.examples.helloworld.filter.LoggingFilter;
-import com.flipkart.gjex.examples.helloworld.healthcheck.AllIsWellHealthCheck;
 import com.flipkart.gjex.examples.helloworld.service.GreeterService;
 import com.flipkart.gjex.examples.helloworld.tracing.AllWhitelistTracingSampler;
 import com.flipkart.gjex.examples.helloworld.web.HelloWorldResourceConfig;
 import com.google.inject.AbstractModule;
 import com.google.inject.name.Names;
-
+import io.dropwizard.metrics5.health.HealthCheck;
 import io.grpc.BindableService;
 import io.grpc.ManagedChannel;
 import io.grpc.ManagedChannelBuilder;
 import io.grpc.examples.helloworld.GreeterGrpc;
-
-
+import org.glassfish.jersey.server.ResourceConfig;
 
 /**
  * Guice module for wiring sample Service to GJEX runtime
@@ -56,7 +52,7 @@ protected void configure() {
 		bind(Filter.class).annotatedWith(Names.named("LoggingFilter")).to(LoggingFilter.class);
 		bind(Filter.class).annotatedWith(Names.named("AuthFilter")).to(AuthFilter.class);
 		bind(TracingSampler.class).to(AllWhitelistTracingSampler.class);
-		bind(HealthCheck.class).to(AllIsWellHealthCheck.class);
+		bind(HealthCheck.class).to(RotationManagementBasedHealthCheck.class);
 		bind(ResourceConfig.class).annotatedWith(Names.named("HelloWorldResourceConfig")).to(HelloWorldResourceConfig.class);
 	}
 

examples/src/main/java/com/flipkart/gjex/examples/helloworld/web/HelloWorldResourceConfig.java

@@ -15,12 +15,14 @@
  */
 package com.flipkart.gjex.examples.helloworld.web;
 
+import com.flipkart.gjex.core.config.RotationManagementConfig;
+import com.flipkart.gjex.core.web.RotationManagementResource;
+import org.glassfish.jersey.server.ResourceConfig;
+
 import javax.inject.Inject;
 import javax.inject.Named;
 import javax.inject.Singleton;
 
-import org.glassfish.jersey.server.ResourceConfig;
-
 /**
  * ResourceConfig example for registering all custom web resources for a GJEX application. 
  * @author regu.b
@@ -31,8 +33,8 @@
 public class HelloWorldResourceConfig extends ResourceConfig {
 
 	@Inject
-	public HelloWorldResourceConfig (HelloWorldResource1 helloWorldresource1, 
-			HelloWorldResource2 helloWorldresource2) {
+	public HelloWorldResourceConfig (HelloWorldResource1 helloWorldresource1,
+																	 HelloWorldResource2 helloWorldresource2) {
 		register(helloWorldresource1);
 		register(helloWorldresource2);
 	}

guice/src/main/java/com/flipkart/gjex/guice/module/DashboardModule.java

@@ -25,6 +25,7 @@
 import javax.inject.Named;
 import javax.inject.Singleton;
 
+import com.flipkart.gjex.core.web.RotationManagementResource;
 import org.eclipse.jetty.server.Server;
 import org.eclipse.jetty.server.ServerConnector;
 import org.eclipse.jetty.servlet.DefaultServlet;
@@ -139,13 +140,21 @@ Server getDashboardJettyServer(@Named("Dashboard.service.port") int port,
 	@Singleton
 	Server getAPIJettyServer(@Named("APIVanillaJettyServer")Server server,
 							@Named("ApiServletContext") ServletContextHandler context,
-							@Named("HealthCheckResourceConfig")ResourceConfig healthcheckResourceConfig,
+							@Named("HealthCheckResourceConfig") ResourceConfig healthCheckResourceConfig,
+							@Named("RotationManagementResourceConfig") ResourceConfig rotationManagementResourceConfig,
 							@Named("TracingResourceConfig")ResourceConfig tracingResourceConfig,
 							@Named("TracingSamplerHolder")TracingSamplerHolder tracingSamplerHolder,
 							@Named("JSONMarshallingProvider")JacksonJaxbJsonProvider provider) throws URISyntaxException, UnknownHostException {
-		healthcheckResourceConfig.register(provider);
-		ServletHolder healthcheckServlet = new ServletHolder(new ServletContainer(healthcheckResourceConfig));
-		context.addServlet(healthcheckServlet, "/healthcheck"); // registering Health Check servlet under the /healthcheck path
+		healthCheckResourceConfig.register(provider);
+		ServletHolder healthCheckServlet =
+				new ServletHolder(new ServletContainer(healthCheckResourceConfig));
+		context.addServlet(healthCheckServlet, "/healthcheck"); // registering Health Check servlet under the /healthcheck path
+
+		rotationManagementResourceConfig.register(provider);
+		ServletHolder rotationManagementServlet =
+				new ServletHolder(new ServletContainer(rotationManagementResourceConfig));
+		context.addServlet(rotationManagementServlet, "/rotation/*"); // registering Rotation
+		// Management servlet under the /rotation path
 
 		tracingResourceConfig.register(provider);
 		ServletHolder tracingServlet = new ServletHolder(new ServletContainer(tracingResourceConfig));
@@ -209,6 +218,16 @@ ResourceConfig getAPIResourceConfig(HealthCheckResource healthCheckResource) {
 		return resourceConfig;
 	}
 
+	@Named("RotationManagementResourceConfig")
+	@Singleton
+	@Provides
+	ResourceConfig getRotationManagementResourceConfig(RotationManagementResource rotationManagementResource) {
+		ResourceConfig resourceConfig = new ResourceConfig();
+		resourceConfig.register(rotationManagementResource);
+		resourceConfig.setApplicationName(Constants.GJEX_CORE_APPLICATION);
+		return resourceConfig;
+	}
+
 	@Named("TracingResourceConfig")
 	@Singleton
 	@Provides